fix: expand multitenancy support

Author: Xunnamius

package-lock.json

@@ -96,6 +96,7 @@
     "type-fest": "^4.41.0"
   },
   "devDependencies": {
+    "@-xun/api-strategy": "^1.3.8",
     "@-xun/fs": "^2.0.0",
     "@-xun/mongo-test": "^3.0.4",
     "@-xun/symbiote": "^4.11.3",

package.json

@@ -1,17 +1,23 @@
 import { LogTag } from '@-xun/cli/logging';
-import { getDb } from '@-xun/mongo-schema';
+import { getClient, getDb, setSchemaConfig } from '@-xun/mongo-schema';
 import { runWithMongoSchemaMultitenancy } from '@-xun/mongo-schema/multitenant';
 
-import { oneSecondInMs, TargetProblem, Task } from 'universe:constant.ts';
+import {
+  oneSecondInMs,
+  TargetProblem,
+  targetProblemBackends,
+  Task
+} from 'universe:constant.ts';
+
 import { ErrorMessage } from 'universe:error.ts';
 import { skipListrTask, waitForListr2OutputReady } from 'universe:util.ts';
 
 import type { GlobalExecutionContext } from 'universe:configure.ts';
 import type { ActualTargetProblem } from 'universe:constant.ts';
 import type { TaskRunnerContext } from 'universe:util.ts';
 
-const fullPrettyName = 'ban hammer';
-const taskType = Task.BanHammer;
+export const fullPrettyName = 'ban hammer';
+export const taskType = Task.BanHammer;
 
 export default async function task(
   taskName: string,
@@ -26,7 +32,9 @@ export default async function task(
 
   await waitForListr2OutputReady(debug);
 
-  await runWithMongoSchemaMultitenancy(target, async () => {
+  await runWithMongoSchemaMultitenancy(`${target}-${taskType}`, async () => {
+    let backend;
+
     switch (target) {
       case TargetProblem.ElectionsCpl:
       case TargetProblem.ElectionsIrv:
@@ -41,8 +49,13 @@ export default async function task(
         return;
       }
 
-      case TargetProblem.Drive:
+      case TargetProblem.Drive: {
+        backend = await targetProblemBackends.drive;
+        break;
+      }
+
       case TargetProblem.Qoverflow: {
+        backend = await targetProblemBackends.qoverflow;
         break;
       }
 
@@ -51,8 +64,16 @@ export default async function task(
       }
     }
 
-    const bannedCount = await unleashBanHammer();
+    // ? Prewarm shared memory
+    await getClient({
+      MONGODB_URI: getConfig(`${target}.mongodbUri`, 'string')
+    });
 
+    setSchemaConfig(backend.db.getSchemaConfig());
+
+    const [bannedCount, cursor] = await unleashBanHammer();
+
+    await cursor.close();
     taskLog([LogTag.IF_NOT_QUIETED], '~%O clients are currently banned', bannedCount);
   });
 
@@ -273,11 +294,9 @@ export default async function task(
     debug('aggregation pipeline: %O', pipeline);
 
     const cursor = db.collection('request-log').aggregate(pipeline);
+    await cursor.toArray();
 
-    await cursor.next();
-    await cursor.close();
-
-    return db.collection('limited-log').estimatedDocumentCount();
+    return [db.collection('limited-log').estimatedDocumentCount(), cursor] as const;
   }
 
   function getConfigFor(target: ActualTargetProblem) {

src/tasks/ban-hammer.ts

@@ -17,8 +17,8 @@ import type { GlobalExecutionContext } from 'universe:configure.ts';
 import type { ActualTargetProblem } from 'universe:constant.ts';
 import type { TaskRunnerContext } from 'universe:util.ts';
 
-const fullPrettyName = 'prune data';
-const taskType = Task.PruneData;
+export const fullPrettyName = 'prune data';
+export const taskType = Task.PruneData;
 
 const maxAllowedBytes = parseBytes('450mb')!;
 
@@ -68,7 +68,7 @@ export default async function task(
 
   await waitForListr2OutputReady(debug);
 
-  await runWithMongoSchemaMultitenancy(target, async () => {
+  await runWithMongoSchemaMultitenancy(`${target}-${taskType}`, async () => {
     switch (target) {
       case TargetProblem.ElectionsCpl:
       case TargetProblem.ElectionsIrv: