Java实现LRU缓存方案?

Java实现LRU缓存方法?

/**
* 可以使用双向链表和哈希表实现一个LRU缓存
* 1、使用双向链表实现的队列,队列的最大容量是缓存的大小。在使用的过程中,把最近使用的页面移动到队列头,最近没有使用的放到队尾
* 2、使用一个哈希表,把页号作为键,把缓存在队列中的节点的的地址作为值
* 当引用一个页面时,这个所需的页面在内存中,需要把这个页对应的节点移动到队列的前面,并且更新在哈希表中结点的地址
* 当所需的页面不在内存时,我们把它存储到内存中
* 当队列满时,那么就从队列尾部移除一个结点,并将新结点添加到队列的前面
*/

package com.lru;

import java.util.HashMap;

/**
 * @ClassName LRU
 * @Description 最近最少使用
 * @Author Administrator
 * @Date 2019/5/30 20:05
 * @Version 1.0
 **/
public class LRU<K,V> {
    /**
     * 可以使用双向链表和哈希表实现一个LRU缓存
     * 1、使用双向链表实现的队列,队列的最大容量是缓存的大小。在使用的过程中,把最近使用的页面移动到队列头,最近没有使用的放到队尾
     * 2、使用一个哈希表,把页号作为键,把缓存在队列中的节点的的地址作为值
     * 当引用一个页面时,这个所需的页面在内存中,需要把这个页对应的节点移动到队列的前面,并且更新在哈希表中结点的地址
     * 当所需的页面不在内存时,我们把它存储到内存中
     * 当队列满时,那么就从队列尾部移除一个结点,并将新结点添加到队列的前面
     */

    private final int MAX_CACHE_SIZE;
    private Entry first;
    private Entry last;

    private HashMap<K, Entry<K, V>> hashMap;

    /**
     *
     * @param cacheSize 初始的缓存的大小
     */
    public LRU(int cacheSize) {
        this.MAX_CACHE_SIZE = cacheSize;
        hashMap = new HashMap<K, Entry<K, V>>();
    }

    public void put(K key, V value) {
        Entry entry = getEntry(key);
        if(entry == null) {
            if(hashMap.size() >= MAX_CACHE_SIZE) {
                hashMap.remove(last.key);
                removeLast();
            }
            entry = new Entry();
            entry.key = key;
        }
        entry.value = value;
        moveToFirst(entry);
        hashMap.put(key, entry);
    }


    /**
     * 获取key键对应的值
     * @param key
     * @return
     */
    public V get(K key) {
        Entry<K, V> entry = getEntry(key);
        if(entry == null) {
            return null;
        }
        // 将当前entry移动到开头
        moveToFirst(entry);
        return entry.value;
    }

    /**
     * 将最近访问的entry移动到开头
     * @param entry
     */
    private void moveToFirst(Entry entry) {
        /**
         * 1、如果是第一个元素,则不需要移动
         */
        if(entry == first) {
            return;
        }

        /**
         * 2、如果entry节点的前面的节点存在,则改变前面的节点的指向
         */
        if(entry.pre != null) {
            entry.pre.next = entry.next;
        }

        /**
         * 3、如果entry节点的后面存在,则改变后面节点的前向指向
         */
        if(entry.next != null) {
            entry.next.pre = entry.pre;
        }

        /**
         * 4、如果当前结点是最后一个结点,那么该变last指向
         */
        if(entry == last) {
            last = last.pre;
        }

        /**
         * 5、
         */
        if(first == null || last == null) {
            first = last = entry;
            return ;
        }

        /**
         * 移动entry到队头
         */
        entry.next = first;
        first.pre = entry;
        entry.pre = null;
    }

    /**
     * 删除
     */
    private void removeLast() {
        if(last != null) {
            last = last.pre;
            if(last == null) {
                first = null;
            } else {
                last.next = null;
            }

        }
    }

    private Entry<K, V> getEntry(K key) {
        return hashMap.get(key);
    }

    /**
     * 存储键值的节点
     * @param <K>
     * @param <V>
     */
    class Entry<K, V> {
        public Entry pre;
        public Entry next;
        public K key;
        public V value;
    }
}

  

测试主类

package com.lru;

/**
 * @ClassName Main
 * @Description LRU缓存测试主类
 * @Author Administrator
 * @Date 2019/5/30 20:29
 * @Version 1.0
 **/
public class Main {

    public static void main(String[] args) {
        LRU<Integer, String> lru = new LRU<>(5);
        lru.put(1, "String1");
        lru.put(2, "String2");
        lru.put(3, "String3");
        lru.put(4, "String4");
        lru.put(5, "String5");
        lru.put(6, "String6");
        System.out.println(lru.get(1));
    }
}

  

猜你喜欢

转载自www.cnblogs.com/wylwyl/p/10951943.html