plusone
/
plusone-commons
1
0
Fork 1
Code Issues 1 Pull Requests Releases Activity

补充缺失的校验

dev
ZhouXY108 2024-09-03 15:34:52 +08:00
parent 7a744c8953
commit 79aebe4fcc
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/main/java/xyz/zhouxy/plusone/commons/collection/AbstractMapWrapper.java
View File

@ -123,15 +123,31 @@ public abstract class AbstractMapWrapper<K, V, T extends AbstractMapWrapper<K, V
} }
public final V putIfAbsent(K key, V value) { public final V putIfAbsent(K key, V value) {
if (this.keyChecker != null) {
this.keyChecker.accept(key);
}
if (this.valueChecker != null) {
this.valueChecker.accept(value);
}
return this.map.putIfAbsent(key, value); return this.map.putIfAbsent(key, value);
} }
public final V computeIfAbsent(K key, Function<? super K, ? extends V> mappingFunction) { public final V computeIfAbsent(K key, Function<? super K, ? extends V> mappingFunction) {
if (this.keyChecker != null) {
this.keyChecker.accept(key);
}
Function<? super K, ? extends V> func = (K k) -> {
V value = mappingFunction.apply(k);
if (this.valueChecker != null) {
this.valueChecker.accept(value);
}
return value;
};
if (this.map instanceof ConcurrentHashMap) { if (this.map instanceof ConcurrentHashMap) {
return ConcurrentHashMapTools.computeIfAbsent( return ConcurrentHashMapTools.computeIfAbsent(
(ConcurrentHashMap<K, V>) this.map, key, mappingFunction); (ConcurrentHashMap<K, V>) this.map, key, func);
} else { } else {
return this.map.computeIfAbsent(key, mappingFunction); return this.map.computeIfAbsent(key, func);
} }
} }
@ -139,7 +155,7 @@ public abstract class AbstractMapWrapper<K, V, T extends AbstractMapWrapper<K, V
return this.map; return this.map;
} }
public final Map<K, V> exportUnmodifiableMapMap() { public final Map<K, V> exportUnmodifiableMap() {
return Collections.unmodifiableMap(this.map); return Collections.unmodifiableMap(this.map);
} }