Allow solo runs with appIds and partnerIds

EdgeApp · Oct 10, 2023 · de645a6 · de645a6
1 parent cb5179b
commit de645a6
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 7 deletions.
diff --git a/config.sample.json b/config.sample.json
@@ -1,8 +1,8 @@
 {
   "couchDbFullpath": "http://admin:admin@localhost:5984",
   "httpPort": "8000",
-  "soloAppId": null,
-  "soloPluginId": null,
+  "soloAppIds": null,
+  "soloPartnerIds": null,
   "timeoutOverrideMins": null,
   "cacheLookbackMonths": null
 }
diff --git a/src/cacheEngine.ts b/src/cacheEngine.ts
@@ -111,13 +111,21 @@ export async function cacheEngine(): Promise<void> {
     const rawApps = await reportsApps.find(query)
     const apps = asApps(rawApps.docs)
     for (const app of apps) {
-      if (config.soloAppId != null && config.soloAppId !== app.appId) continue
-
+      if (
+        config.soloAppIds != null &&
+        config.soloAppIds.includes(app.appId) === false
+      ) {
+        continue
+      }
       const partnerIds = Object.keys(app.partnerIds)
 
       for (const partnerId of partnerIds) {
-        if (config.soloPartnerId != null && config.soloPartnerId !== partnerId)
+        if (
+          config.soloPartnerIds != null &&
+          config.soloPartnerIds.includes(partnerId) === false
+        ) {
           continue
+        }
         for (const timePeriod of TIME_PERIODS) {
           const data = await getAnalytic(
             start,

diff --git a/src/queryEngine.ts b/src/queryEngine.ts
@@ -160,15 +160,24 @@ export async function queryEngine(): Promise<void> {
     let remainingPartners: String[] = []
     // loop over every app
     for (const app of apps) {
-      if (config.soloAppId != null && config.soloAppId !== app.appId) continue
+      if (
+        config.soloAppIds != null &&
+        config.soloAppIds.includes(app.appId) === false
+      ) {
+        continue
+      }
       let partnerStatus: string[] = []
       // loop over every pluginId that app uses
       remainingPartners = Object.keys(app.partnerIds)
       for (const partnerId in app.partnerIds) {
         const pluginId = app.partnerIds[partnerId].pluginId ?? partnerId
 
-        if (config.soloPartnerId != null && config.soloPartnerId !== partnerId)
+        if (
+          config.soloPartnerIds != null &&
+          config.soloPartnerIds.includes(partnerId) === false
+        ) {
           continue
+        }
         remainingPartners.push(partnerId)
         promiseArray.push(
           runPlugin(app, partnerId, pluginId, dbProgress).finally(() => {