public class LinkedHashMap<K,V>
extends HashMap<K,V>
implements Map<K,V>
{
//...
}
从LinkedHashMap的类定义可以看出,LinkedHashMap继承了HashMap,它具有HashMap的所有性质。以下主要看其与HashMap不同的地方。
其经典应用场景是实现LRU缓存,文章最后会给出一个简单的实现。
/**
* 双向链表的头结点
*/
transient LinkedHashMap.Entry<K,V> head;
/**
* 双向链表的尾结点
*/
transient LinkedHashMap.Entry<K,V> tail;
/**
* 维护结点顺序,true:维护访问顺序;false:维护插入顺序
*/
final boolean accessOrder;
static class Entry<K,V> extends HashMap.Node<K,V> {
Entry<K,V> before, after;
Entry(int hash, K key, V value, Node<K,V> next) {
super(hash, key, value, next);
}
}
Entry继承了HashMap的Node,并新增两个属性before和after,用来维护LinkedHashMap中的结点顺序的视图(所有结点组成的双向链表)。
以下代卖来自HashMap,LinkedHashMap并未重写该方法,而是使用钩子的方式来实现结点顺序的维护,具体由三个方法:
newNode
HashMap中有实现,LinkedHashMap对其进行了重写
afterNodeAccess
HashMap中是一个空方法,LinkedHashMap对其进行了重写
afterNodeInsertion
HashMap中是一个空方法,LinkedHashMap对其进行了重写
/**
* 将k-v对放入table中
*/
public V put(K key, V value) {
return putVal(hash(key), key, value, false, true);
}
/**
* 将k-v对放入table中
*/
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
boolean evict) {
Node<K,V>[] tab; Node<K,V> p; int n, i;
// 如果table数组为null或者table数组长度为0,则调用resize方法初始化table数组,返回数组长度
if ((tab = table) == null || (n = tab.length) == 0)
n = (tab = resize()).length;
// 如果table在计算出索引(idx)位置的值为null,则将新的k-v对构建为Node节点放在table数组的这个位置
// 同时这里为p变量赋值了
if ((p = tab[i = (n - 1) & hash]) == null)
// 这里构造结点与HashMap不同
tab[i] = newNode(hash, key, value, null);
// 如果table在计算出索引(idx)位置的值不为null
else {
Node<K,V> e; K k;
// p的hash与当前k的hash相等,且p的key与当前k的值相等,则视为同一个K,后面会对e处理
if (p.hash == hash &&
((k = p.key) == key || (key != null && key.equals(k))))
e = p;
// 表示有hash冲突,无论链表还是红黑树都是为了解决hash冲突
// 如果p的类型是TreeNode,则执行TreeNode节点的插入,即将当前k-v插入到红黑树中
else if (p instanceof TreeNode)
e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
else {
// 构造一个Node结点,并将其插入到链表的尾部;并且,在需要的时候,将该链表转换为红黑树
for (int binCount = 0; ; ++binCount) {
// 如果p的next节点(e)为空,则使用当前k-v构建Node并将其作为p的next节点
if ((e = p.next) == null) {
// 这里构造结点与HashMap不同
p.next = newNode(hash, key, value, null);
// 当循环到第8次,binCount = 7,条件为真,对table进行扩容或者将当前idx处的链表转换为红黑树具体看treeifyBin方法
if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
treeifyBin(tab, hash);
// 到这里表示k-v是一个全新的节点,并且已经放入table中,循环结束
break;
}
// 如果p的next节点(e)不为null,且e节点的hash与当前k的hash相等、e节点的key与当前k相等,则视为空一格K,循环结束,后面会对e处理
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
break;
// 如果前面两个if都不成立,继续查看链表的下一个节点是否满足上面的两个if语句片,直到链表的结尾即p.next值为null
p = e;
}
}
// 条件为真,表示有相同的K进来了,根据onlyIfAbsent和当前已映射的v值是否为空来进行节点value的操作
if (e != null) { // existing mapping for key
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
e.value = value;
// LinkedHashMap有实现,用来维护节点的添加或者访问顺序
afterNodeAccess(e);
// 返回该key关联的旧值
return oldValue;
}
}
++modCount;
// 执行到这里说明在putVal执行之前当前key在table中不存在,size记录的是key的个数
// 当size > 数组长度*0.75时,对table数组进行扩容
if (++size > threshold)
resize();
// LinkedHashMap有实现,用来维护节点的添加或者访问顺序
afterNodeInsertion(evict);
return null;
}
下面看看以上三个方法的实现及作用
这里创建新的Entry结点,并将该几点放到双向链表的尾部。
// 普通结点
Node<K,V> newNode(int hash, K key, V value, Node<K,V> e) {
LinkedHashMap.Entry<K,V> p =
new LinkedHashMap.Entry<K,V>(hash, key, value, e);
linkNodeLast(p);
return p;
}
// 红黑树结点
TreeNode<K,V> newTreeNode(int hash, K key, V value, Node<K,V> next) {
TreeNode<K,V> p = new TreeNode<K,V>(hash, key, value, next);
linkNodeLast(p);
return p;
}
private void linkNodeLast(LinkedHashMap.Entry<K,V> p) {
LinkedHashMap.Entry<K,V> last = tail;
tail = p;
if (last == null)
head = p;
else {
p.before = last;
last.after = p;
}
}
如果accessOrder为true,并且该结点不是双向链表的尾结点,则将该结点移动到双向链表的尾部
void afterNodeAccess(Node<K,V> e) { // move node to last
LinkedHashMap.Entry<K,V> last;
if (accessOrder && (last = tail) != e) {
LinkedHashMap.Entry<K,V> p =
(LinkedHashMap.Entry<K,V>)e, b = p.before, a = p.after;
// 将p从双向链表中删除
p.after = null; // 断开p与其后继结点的链接
if (b == null) // p无前驱结点,则将双向链表的头结点设置为p的后继结点,因为p要作为尾结点
head = a;
else // p有前驱结点,则使p的前驱结点的后继结点直接指向p的后街结点,即断开p的前驱结点与p的链接
b.after = a;
if (a != null) // p有后继结点,则将p的后继结点的前驱结点指向p的前驱结点,即断开p的后继结点与p的链接。到这里就将p从双向链表中删除了。
a.before = b;
else // p无后继结点,使将p的前驱结点作为双向链表的尾结点
last = b;
// 将p添加到双向链表的尾部
if (last == null) //
head = p;
else {
p.before = last;
last.after = p;
}
// tail指向指向p结点
tail = p;
++modCount;
}
}
LinkedHashMap中removeEldestEntry始终返回false,实际这里也没做什么。
void afterNodeInsertion(boolean evict) { // possibly remove eldest
LinkedHashMap.Entry<K,V> first;
if (evict && (first = head) != null && removeEldestEntry(first)) { // LinkedHashMap中removeEldestEntry始终返回false
K key = first.key;
removeNode(hash(key), key, null, false, true);
}
}
LinkedHashMap未重写该方法。
LinkedHashMap重写了该方法。首先get方法调用了父类HashMap的getNode方法,如果查找到该结点,则调用afterNodeAccess方法进行顺序维护的操作
/**
* 调用父类HashMap的getNode方法,如果查找到该结点,则进行维护顺序的操作
*/
public V get(Object key) {
Node<K,V> e;
if ((e = getNode(hash(key), key)) == null)
return null;
if (accessOrder)
afterNodeAccess(e);
return e.value;
}
以下代码来自父类HashMap,LinkedHashMap没有重写该方法,同get操作一样,使用钩子的方式来实现结点顺序的维护:
/**
*
*/
public V remove(Object key) {
Node<K,V> e;
return (e = removeNode(hash(key), key, null, false, true)) == null ?
null : e.value;
}
/**
*
*/
final Node<K,V> removeNode(int hash, Object key, Object value,
boolean matchValue, boolean movable) {
Node<K,V>[] tab; Node<K,V> p; int n, index;
if ((tab = table) != null && (n = tab.length) > 0 &&
(p = tab[index = (n - 1) & hash]) != null) {
// 查找key对应的结点
Node<K,V> node = null, e; K k; V v;
if (p.hash == hash &&
((k = p.key) == key || (key != null && key.equals(k))))// node结点是p,即node结点是链表头结点或红黑树跟结点
node = p;
else if ((e = p.next) != null) { // 是红黑树或者链表
if (p instanceof TreeNode)
node = ((TreeNode<K,V>)p).getTreeNode(hash, key);
else {
// 从链表查找key对应的结点,且始终保证p是node的前一个结点
do {
if (e.hash == hash &&
((k = e.key) == key ||
(key != null && key.equals(k)))) {
node = e;
break;
}
p = e;
} while ((e = e.next) != null);
}
}
// 从红黑树中、或者从链表中将结点node删除
if (node != null && (!matchValue || (v = node.value) == value ||
(value != null && value.equals(v)))) {
// 从红黑树中删除node结点
if (node instanceof TreeNode)
((TreeNode<K,V>)node).removeTreeNode(this, tab, movable);
// node结点是链表头结点
else if (node == p)
tab[index] = node.next;
// p是node的前一个结点
else
p.next = node.next;
++modCount;
--size;
afterNodeRemoval(node);
return node;
}
}
return null;
}
下面来看看afterNodeRemoval方法
该方法将结点从双向链表中删除
void afterNodeRemoval(Node<K,V> e) { // unlink
LinkedHashMap.Entry<K,V> p =
(LinkedHashMap.Entry<K,V>)e, b = p.before, a = p.after;
p.before = p.after = null;
if (b == null)
head = a;
else
b.after = a;
if (a == null)
tail = b;
else
a.before = b;
}
那究竟结点顺序是从哪里提现出来的呢??
分别看LinkedKeySet、LinkedValues、LinkedEntrySet的forEach方法
public final void forEach(Consumer<? super K> action) {
if (action == null)
throw new NullPointerException();
int mc = modCount;
for (LinkedHashMap.Entry<K,V> e = head; e != null; e = e.after)
action.accept(e.key);
if (modCount != mc)
throw new ConcurrentModificationException();
}
public final void forEach(Consumer<? super V> action) {
if (action == null)
throw new NullPointerException();
int mc = modCount;
for (LinkedHashMap.Entry<K,V> e = head; e != null; e = e.after)
action.accept(e.value);
if (modCount != mc)
throw new ConcurrentModificationException();
}
public final void forEach(Consumer<? super Map.Entry<K,V>> action) {
if (action == null)
throw new NullPointerException();
int mc = modCount;
for (LinkedHashMap.Entry<K,V> e = head; e != null; e = e.after)
action.accept(e);
if (modCount != mc)
throw new ConcurrentModificationException();
}
还有LinkedHashMap的forEach方法
public void forEach(BiConsumer<? super K, ? super V> action) {
if (action == null)
throw new NullPointerException();
int mc = modCount;
for (LinkedHashMap.Entry<K,V> e = head; e != null; e = e.after)
action.accept(e.key, e.value);
if (modCount != mc)
throw new ConcurrentModificationException();
}
这些forEach方法都是从双向链表的头结点开始遍历,直到没有更多元素。
以及LinkedHashMap内部定义迭代器
abstract class LinkedHashIterator {
LinkedHashMap.Entry<K,V> next;
LinkedHashMap.Entry<K,V> current;
int expectedModCount;
// next指向双向链表的头结点
LinkedHashIterator() {
next = head;
expectedModCount = modCount;
current = null;
}
public final boolean hasNext() {
return next != null;
}
// 而获取nextNode时,获取的是当前结点的后继结点
final LinkedHashMap.Entry<K,V> nextNode() {
LinkedHashMap.Entry<K,V> e = next;
if (modCount != expectedModCount)
throw new ConcurrentModificationException();
if (e == null)
throw new NoSuchElementException();
current = e;
next = e.after;
return e;
}
public final void remove() {
Node<K,V> p = current;
if (p == null)
throw new IllegalStateException();
if (modCount != expectedModCount)
throw new ConcurrentModificationException();
current = null;
K key = p.key;
// 这里调用了LinkedHashMap的removeNode方法,也会处罚结点顺序维护的操作
removeNode(hash(key), key, null, false, false);
expectedModCount = modCount;
}
}
// 以下三个迭代器都调用了父类LinkedHashIterator的nextNode结点
final class LinkedKeyIterator extends LinkedHashIterator
implements Iterator<K> {
public final K next() { return nextNode().getKey(); }
}
final class LinkedValueIterator extends LinkedHashIterator
implements Iterator<V> {
public final V next() { return nextNode().value; }
}
final class LinkedEntryIterator extends LinkedHashIterator
implements Iterator<Map.Entry<K,V>> {
public final Map.Entry<K,V> next() { return nextNode(); }
}
迭代器也是按照双向链表的顺序来返回相应的值,直到没有更多结点。
从上可以看出,无论是forEach方法,还是迭代器的next方法,都是按照LinkedHashMap中维护的一个双向链表来进行输出的。
LinkedHashMap内部维护了一个双向链表来维护结点遍历时的输出顺序。
LinkedHashMap的典型应用是实现基于LRU算法的缓存,那么,该怎样实现一个固定大小的LRU缓存呢???
答案就在afterNodeInsertion方法内调用的removeEldestEntry方法中。removeEldestEntry方法的含义是”是否删除’最老’的结点,即双向链表的头部结点”。
实现如下:
public class LruCache<K, V> extends LinkedHashMap<K, V> {
/**
* key-value对的最大数量,当size()方法返回值大于这个值,则进行旧结点的删除
*/
private int maxKeyCount;
private static final int INITIAL_CAPACITY = 16;
private static final float LOAD_FACTOR = 0.75F;
/**
* 只有当accessOrder为true时,才具有LRU的特性;否则,只维护写入顺序
*/
private static final boolean ACCESS_ORDER = true;
public LruCache(Integer maxKeyCount) {
super(INITIAL_CAPACITY, LOAD_FACTOR, ACCESS_ORDER);
this.maxKeyCount = maxKeyCount;
}
@Override
protected boolean removeEldestEntry(Map.Entry<K, V> eldest) {
return super.size() > maxKeyCount;
}
}
测试程序如下:
@Test
public void test() {
int maxKeyCount = 16;
LruCache<String, String> cache = new LruCache<>(maxKeyCount);
for (int i = 0; i < maxKeyCount << 1; i++) {
String key = i + "";
String value = key;
String put = cache.put(key, value);// put为null
}
// 这里应该输出按 16 - 31 顺序的共16个key
log.info("keySet: {}", cache.keySet());
log.info("values: {}", cache.values());
log.info("entrySet: {}", cache.entrySet());
String s = cache.get("16");
log.info("lru cache get {} --->> {}", "16", s);
// 这里应该输出按 17 - 31, 16 顺序的共16个key
log.info("keySet: {}", cache.keySet());
log.info("values: {}", cache.values());
log.info("entrySet: {}", cache.entrySet());
String put = cache.put("17", "1717");
log.info("cache put: {} --- >> {}, {}", "17", "1717", put);
// 这里应该输出按 18 - 31, 16, 17 顺序的共16个key
log.info("keySet: {}", cache.keySet());
log.info("values: {}", cache.values());
log.info("entrySet: {}", cache.entrySet());
}