Skip to content

Commit 1f9b9a8

Browse files
committed
use stream to filter null value on save
1 parent c5365b4 commit 1f9b9a8

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

agent/storage/src/main/java/me/hsgamer/topper/agent/storage/StorageAgent.java

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import java.util.concurrent.atomic.AtomicReference;
1414
import java.util.logging.Level;
1515
import java.util.logging.Logger;
16+
import java.util.stream.Collectors;
1617

1718
public class StorageAgent<K, V> implements Agent, DataEntryAgent<K, V>, Runnable {
1819
private final Logger logger;
@@ -53,15 +54,17 @@ private void save(boolean urgent) {
5354
return;
5455
}
5556

56-
Map<K, V> finalMap = new HashMap<>();
5757
Set<K> removeKeys = new HashSet<>();
58-
for (Map.Entry<K, V> entry : map.entrySet()) {
59-
if (entry.getValue() == null) {
60-
removeKeys.add(entry.getKey());
61-
} else {
62-
finalMap.put(entry.getKey(), entry.getValue());
63-
}
64-
}
58+
Map<K, V> finalMap = map.entrySet()
59+
.stream()
60+
.filter(entry -> {
61+
if (entry.getValue() == null) {
62+
removeKeys.add(entry.getKey());
63+
return false;
64+
}
65+
return true;
66+
})
67+
.collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
6568

6669
Runnable saveTask = () -> {
6770
try {

0 commit comments

Comments
 (0)