envio 2.26.0-rc.0 → 2.26.0-rc.2

package/src/Persistence.res.js

@@ -2,6 +2,8 @@
 'use strict';
 
 var Js_exn = require("rescript/lib/js/js_exn.js");
+var Logging = require("./Logging.res.js");
+var Prometheus = require("./Prometheus.res.js");
 var EntityHistory = require("./db/EntityHistory.res.js");
 var ErrorHandling = require("./ErrorHandling.res.js");
 var Caml_exceptions = require("rescript/lib/js/caml_exceptions.js");
@@ -22,7 +24,7 @@ var entityHistoryActionEnumConfig = {
   default: "SET"
 };
 
-function make(userEntities, dcRegistryEntityConfig, allEnums, staticTables, storage, cacheStorage, onStorageInitialize) {
+function make(userEntities, dcRegistryEntityConfig, allEnums, staticTables, storage) {
   var allEntities = userEntities.concat([dcRegistryEntityConfig]);
   var allEnums$1 = allEnums.concat([entityHistoryActionEnumConfig]);
   return {
@@ -31,14 +33,21 @@ function make(userEntities, dcRegistryEntityConfig, allEnums, staticTables, stor
           allEntities: allEntities,
           allEnums: allEnums$1,
           storageStatus: "Unknown",
-          storage: storage,
-          onStorageInitialize: onStorageInitialize,
-          cacheStorage: cacheStorage
+          storage: storage
         };
 }
 
-async function init(persistence, skipIsInitializedCheckOpt, resetOpt) {
-  var skipIsInitializedCheck = skipIsInitializedCheckOpt !== undefined ? skipIsInitializedCheckOpt : false;
+async function loadInitialCache(persistence, withUpload) {
+  var effectCacheRecords = await persistence.storage.restoreEffectCache(withUpload);
+  var cache = {};
+  effectCacheRecords.forEach(function (record) {
+        Prometheus.EffectCacheCount.set(record.count, record.effectName);
+        cache[record.effectName] = record;
+      });
+  return cache;
+}
+
+async function init(persistence, resetOpt) {
   var reset = resetOpt !== undefined ? resetOpt : false;
   try {
     var promise = persistence.storageStatus;
@@ -64,20 +73,26 @@ async function init(persistence, skipIsInitializedCheckOpt, resetOpt) {
       TAG: "Initializing",
       _0: promise$1
     };
-    if (!(reset || skipIsInitializedCheck) && await persistence.storage.isInitialized()) {
+    if (reset || !await persistence.storage.isInitialized()) {
+      Logging.info("Initializing the indexer storage...");
+      await persistence.storage.initialize(persistence.allEntities, persistence.staticTables, persistence.allEnums);
+      Logging.info("The indexer storage is ready. Uploading cache...");
       persistence.storageStatus = {
         TAG: "Ready",
-        cleanRun: false
+        cleanRun: true,
+        cache: await loadInitialCache(persistence, true)
       };
     } else {
-      await persistence.storage.initialize(persistence.allEntities, persistence.staticTables, persistence.allEnums, reset || !skipIsInitializedCheck);
-      persistence.storageStatus = {
-        TAG: "Ready",
-        cleanRun: true
-      };
-      var onStorageInitialize = persistence.onStorageInitialize;
-      if (onStorageInitialize !== undefined) {
-        await onStorageInitialize();
+      var match = persistence.storageStatus;
+      var tmp;
+      tmp = typeof match !== "object" || match.TAG !== "Initializing" ? false : true;
+      if (tmp) {
+        Logging.info("The indexer storage is ready.");
+        persistence.storageStatus = {
+          TAG: "Ready",
+          cleanRun: false,
+          cache: await loadInitialCache(persistence, false)
+        };
       }
       
     }
@@ -98,9 +113,39 @@ function getInitializedStorageOrThrow(persistence) {
   }
 }
 
+async function setEffectCacheOrThrow(persistence, effect, items) {
+  var match = persistence.storageStatus;
+  if (typeof match !== "object") {
+    return Js_exn.raiseError("Failed to access the indexer storage. The Persistence layer is not initialized.");
+  }
+  if (match.TAG === "Initializing") {
+    return Js_exn.raiseError("Failed to access the indexer storage. The Persistence layer is not initialized.");
+  }
+  var cache = match.cache;
+  var storage = persistence.storage;
+  var effectName = effect.name;
+  var c = cache[effectName];
+  var effectCacheRecord;
+  if (c !== undefined) {
+    effectCacheRecord = c;
+  } else {
+    var c$1 = {
+      effectName: effectName,
+      count: 0
+    };
+    cache[effectName] = c$1;
+    effectCacheRecord = c$1;
+  }
+  var initialize = effectCacheRecord.count === 0;
+  await storage.setEffectCacheOrThrow(effect, items, initialize);
+  effectCacheRecord.count = effectCacheRecord.count + items.length | 0;
+  return Prometheus.EffectCacheCount.set(effectCacheRecord.count, effectName);
+}
+
 exports.StorageError = StorageError;
 exports.entityHistoryActionEnumConfig = entityHistoryActionEnumConfig;
 exports.make = make;
 exports.init = init;
 exports.getInitializedStorageOrThrow = getInitializedStorageOrThrow;
-/* EntityHistory Not a pure module */
+exports.setEffectCacheOrThrow = setEffectCacheOrThrow;
+/* Logging Not a pure module */