Update last updated index to use multimap.

FINTLabs · Jun 11, 2020 · 73b4e4e · 73b4e4e
1 parent 283a522
commit 73b4e4e
Showing 1 changed file with 14 additions and 18 deletions.
diff --git a/src/main/java/no/fint/cache/FintCache.java b/src/main/java/no/fint/cache/FintCache.java
@@ -1,6 +1,7 @@
 package no.fint.cache;
 
 import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMultimap;
 import lombok.Getter;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
@@ -25,13 +26,13 @@ public class FintCache<T extends Serializable> implements Cache<T>, Serializable
     private CacheMetaData cacheMetaData;
     private List<CacheObject<T>> cacheObjects;
     private Map<Integer, Index> index;
-    private NavigableMap<Long, Index> timestampIndex;
+    private ImmutableMultimap<Long, Integer> lastUpdatedIndex;
 
     public FintCache() {
         cacheMetaData = new CacheMetaData();
         cacheObjects = Collections.emptyList();
         index = Collections.emptyMap();
-        timestampIndex = Collections.emptyNavigableMap();
+        lastUpdatedIndex = ImmutableMultimap.of();
     }
 
     @Override
@@ -98,8 +99,10 @@ public void addCache(List<CacheObject<T>> objects) {
 
     @Override
     public void flush() {
-        flushMetaData();
+        cacheMetaData = new CacheMetaData();
         cacheObjects = Collections.emptyList();
+        index = Collections.emptyMap();
+        lastUpdatedIndex = ImmutableMultimap.of();
     }
 
     @Override
@@ -113,12 +116,12 @@ public List<T> getSourceList() {
 
     @Override
     public Stream<CacheObject<T>> streamSince(long timestamp) {
-        return timestampIndex
-                .tailMap(timestamp, false)
-                .values()
+        return lastUpdatedIndex
+                .entries()
                 .stream()
-                .flatMapToInt(Index::stream)
-                .mapToObj(cacheObjects::get);
+                .filter(e -> e.getKey() > timestamp)
+                .map(Map.Entry::getValue)
+                .map(cacheObjects::get);
         //return cacheObjects.stream().filter(cacheObject -> (cacheObject.getLastUpdated() > timestamp));
     }
 
@@ -137,7 +140,7 @@ public List<?> getSourceListSince(long timestamp) {
     @SneakyThrows
     private void updateMetaData() {
         Map<Integer, Index> newIndex = new HashMap<>();
-        NavigableMap<Long, Index> newTimestampIndex = new TreeMap<>();
+        final ImmutableMultimap.Builder<Long, Integer> lastUpdatedBuilder = ImmutableMultimap.builder();
         cacheMetaData.setCacheCount(cacheObjects.size());
         cacheMetaData.setLastUpdated(System.currentTimeMillis());
         MessageDigest digest = MessageDigest.getInstance("SHA-1");
@@ -147,12 +150,12 @@ private void updateMetaData() {
             CacheObject<T> it = iterator.next();
             digest.update(it.rawChecksum());
             IntStream.of(it.getHashCodes()).forEach(key -> newIndex.compute(key, createIndex(i)));
-            newTimestampIndex.compute(it.getLastUpdated(), createIndex(i));
+            lastUpdatedBuilder.put(it.getLastUpdated(), i);
         }
         cacheMetaData.setChecksum(digest.digest());
         cacheMetaData.setSize(cacheObjects.parallelStream().mapToLong(CacheObject::getSize).sum());
         index = newIndex;
-        timestampIndex = newTimestampIndex;
+        lastUpdatedIndex = lastUpdatedBuilder.build();
     }
 
     private static <K> BiFunction<K, Index, Index> createIndex(int i) {
@@ -172,13 +175,6 @@ private Map<String, CacheObject<T>> getCacheMap(List<CacheObject<T>> list) {
         return list.parallelStream().collect(Collectors.toMap(CacheObject::getChecksum, Function.identity(), (a, b) -> b));
     }
 
-    private void flushMetaData() {
-        cacheMetaData.setCacheCount(0);
-        cacheMetaData.setLastUpdated(0);
-        cacheMetaData.setChecksum(null);
-        cacheMetaData.setSize(0L);
-    }
-
     @Override
     public long getLastUpdated() {
         return cacheMetaData.getLastUpdated();