releaseb add IdGzipCacheSize

Sammers21 · Jan 8, 2025 · 0e810fa · 0e810fa
1 parent e8e9a47
commit 0e810fa
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/src/io/github/sammers/pla/logic/CharUpdater.java b/src/io/github/sammers/pla/logic/CharUpdater.java
@@ -128,6 +128,7 @@ public Completable updateCharacters(String region,
                     merged.add(randomNotUpdatedIt.next());
                 }
             }
+            characterCache.calculateSizeMetrics();
             // merged into list of nickNames
             log.info("There is {} new chars, {} existing chars and {} random not updated chars in region {} that need to be updated. We have {} {} to do it",
                 newCharsSorted.size(), existingSorted.size(), randomNotUpdatedChars.size(), region, timeout, timeoutUnits.name());

diff --git a/src/io/github/sammers/pla/logic/CharacterCache.java b/src/io/github/sammers/pla/logic/CharacterCache.java
@@ -3,6 +3,7 @@
 import io.github.sammers.pla.blizzard.BracketType;
 import io.github.sammers.pla.blizzard.WowAPICharacter;
 import io.github.sammers.pla.db.Character;
+import io.prometheus.metrics.core.metrics.Gauge;
 
 import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
@@ -15,13 +16,24 @@ public class CharacterCache {
     public final Map<Integer, Set<Long>> alts;
     public final RealmStats realmStats;
 
+    public final Gauge idCacheSize = Gauge.builder()
+        .name("IdGzipCacheSize")
+        .help("How many characters are in the id cache")
+        .labelNames()
+        .register();
+
     public CharacterCache() {
         nameCache = new ConcurrentHashMap<>();
         idCache = new ConcurrentHashMap<>();
         alts = new ConcurrentHashMap<>();
         realmStats = new RealmStats();
     }
 
+    public void calculateSizeMetrics() {
+        long bytes = idCache.values().stream().mapToLong(b -> b.length).sum();
+        idCacheSize.set(bytes);
+    }
+
     public WowAPICharacter getByFullName(String name) {
         return WowAPICharacter.fromGzippedJson(nameCache.get(name));
     }
@@ -91,7 +103,7 @@ public Set<WowAPICharacter> findAltsInconsistenciesAndFix(WowAPICharacter charac
             WowAPICharacter changedAlts = ch.changeAlts(longs);
             return changedAlts;
         }).collect(Collectors.toSet());
-            return resm;
+        return resm;
     }
 
     public Collection<byte[]> values() {

diff --git a/src/io/github/sammers/pla/logic/Ladder.java b/src/io/github/sammers/pla/logic/Ladder.java
@@ -50,7 +50,6 @@ public class Ladder {
     public final CharUpdater charUpdater;
     private final DB db;
     private final BlizzardAPI blizzardAPI;
-
     private final AtomicBoolean charsLoaded = new AtomicBoolean(false);
 
     public Ladder(WebClient web, DB db, BlizzardAPI blizzardAPI, CharacterCache characterCache, Refs refs, Map<String, Cutoffs> regionCutoff) {
@@ -576,7 +575,10 @@ public Completable loadWowCharApiData(String region) {
                         emitter.onComplete();
                     });
                 }));
-        });
+        }).andThen(Completable.defer(() -> {
+            characterCache.calculateSizeMetrics();
+            return Completable.complete();
+        }));
     }
 
     private Completable loadLast(String bracket, String region) {