first-commit

node_modules/search-insights/lib/utils/objectQueryTracker.ts

@@ -0,0 +1,74 @@
+import { LocalStorage } from "./localStorage";
+
+interface ObjectQueryMap {
+  [indexAndObjectId: string]: [queryId: string, timestamp: number];
+}
+
+const STORE = "AlgoliaObjectQueryCache";
+const LIMIT = 5000; // 1 entry is typically no more than 100 bytes, so this is ~500kB worth of data - most browsers allow at least 5MB per origin
+const FREE = 1000;
+
+function getCache(): ObjectQueryMap {
+  return LocalStorage.get(STORE) ?? {};
+}
+function setCache(objectQueryMap: ObjectQueryMap): void {
+  LocalStorage.set(STORE, limited(objectQueryMap));
+}
+
+function limited(objectQueryMap: ObjectQueryMap): ObjectQueryMap {
+  return Object.keys(objectQueryMap).length > LIMIT
+    ? purgeOldest(objectQueryMap)
+    : objectQueryMap;
+}
+
+function purgeOldest(objectQueryMap: ObjectQueryMap): ObjectQueryMap {
+  const sorted = Object.entries(objectQueryMap).sort(
+    ([, [, aTimestamp]], [, [, bTimestamp]]) => bTimestamp - aTimestamp
+  );
+
+  const newObjectQueryMap = sorted
+    .slice(0, sorted.length - FREE - 1)
+    .reduce<ObjectQueryMap>(
+      (acc, [key, value]) => ({
+        ...acc,
+        [key]: value
+      }),
+      {}
+    );
+
+  return newObjectQueryMap;
+}
+
+function makeKey(index: string, objectID: string): string {
+  return `${index}_${objectID}`;
+}
+
+export function storeQueryForObject(
+  index: string,
+  objectID: string,
+  queryID: string
+): void {
+  const objectQueryMap = getCache();
+
+  objectQueryMap[makeKey(index, objectID)] = [queryID, Date.now()];
+  setCache(objectQueryMap);
+}
+
+export function getQueryForObject(
+  index: string,
+  objectID: string
+): [queryId: string, timestamp: number] | undefined {
+  return getCache()[makeKey(index, objectID)];
+}
+
+export function removeQueryForObjects(
+  index: string,
+  objectIDs: string[]
+): void {
+  const objectQueryMap = getCache();
+
+  objectIDs.forEach((objectID) => {
+    delete objectQueryMap[makeKey(index, objectID)];
+  });
+  setCache(objectQueryMap);
+}