@orion-js/dogs 4.6.5 → 4.6.6

package/dist/index.js

@@ -177,7 +177,8 @@ __runInitializers(_init, 1, JobsHistoryRepo);
 // src/repos/JobsRepo.ts
 import { generateId } from "@orion-js/helpers";
 import { logger } from "@orion-js/logger";
-import { MongoCollection as MongoCollection2, Repository as Repository2 } from "@orion-js/mongodb";
+import { MongoCollection as MongoCollection2, MongoDB as MongoDB2, Repository as Repository2 } from "@orion-js/mongodb";
+import { clean, validate } from "@orion-js/schema";
 
 // src/services/partitions.ts
 function validatePartitionsCount(nPartitions) {
@@ -617,13 +618,13 @@ var JobsRepo = class {
     if (jobs.length === 0) {
       return { scheduledCount: 0, skippedCount: 0, errors: [] };
     }
-    let scheduledCount = 0;
-    let skippedCount = 0;
     const errors = [];
+    const preparedJobs = [];
     for (let i = 0; i < jobs.length; i++) {
       const job = jobs[i];
       try {
-        await this.jobs.insertOne({
+        const record = await clean(JobRecordSchema, {
+          _id: this.jobs.generateId(),
           jobName: job.name,
           uniqueIdentifier: job.uniqueIdentifier,
           params: job.params,
@@ -632,16 +633,60 @@ var JobsRepo = class {
           type: "event",
           ...job.partition === void 0 ? {} : { partition: job.partition }
         });
-        scheduledCount++;
+        await validate(JobRecordSchema, record);
+        preparedJobs.push({ index: i, job, record });
       } catch (error) {
-        if (error.isValidationError && Object.values(error.validationErrors).includes("notUnique") && job.uniqueIdentifier) {
-          logger.info(`Job "${job.name}" with identifier "${job.uniqueIdentifier}" already exists`);
-          skippedCount++;
-        } else {
+        errors.push({
+          index: i,
+          error: error instanceof Error ? error : new Error(String(error)),
+          job
+        });
+      }
+    }
+    if (preparedJobs.length === 0) {
+      return { scheduledCount: 0, skippedCount: 0, errors };
+    }
+    const rawCollection = await this.jobs.getRawCollection();
+    await this.jobs.createIndexesPromise;
+    const operations = preparedJobs.map(
+      ({ record }) => record.uniqueIdentifier === void 0 ? { insertOne: { document: record } } : {
+        updateOne: {
+          filter: { uniqueIdentifier: record.uniqueIdentifier },
+          update: { $setOnInsert: record },
+          upsert: true
+        }
+      }
+    );
+    let scheduledCount = 0;
+    let skippedCount = 0;
+    try {
+      const result = await rawCollection.bulkWrite(operations, { ordered: false });
+      scheduledCount = result.insertedCount + result.upsertedCount;
+      skippedCount = result.matchedCount;
+    } catch (error) {
+      const writeErrors = error instanceof MongoDB2.MongoBulkWriteError ? Array.isArray(error.writeErrors) ? error.writeErrors : [error.writeErrors] : [];
+      if (!(error instanceof MongoDB2.MongoBulkWriteError) || writeErrors.length === 0) {
+        for (const preparedJob of preparedJobs) {
           errors.push({
-            index: i,
+            index: preparedJob.index,
             error: error instanceof Error ? error : new Error(String(error)),
-            job
+            job: preparedJob.job
+          });
+        }
+      } else {
+        scheduledCount = error.result.insertedCount + error.result.upsertedCount;
+        skippedCount = error.result.matchedCount;
+        for (const writeError of writeErrors) {
+          const preparedJob = preparedJobs[writeError.index];
+          if (!preparedJob) continue;
+          if (writeError.code === 11e3 && preparedJob.record.uniqueIdentifier !== void 0) {
+            skippedCount++;
+            continue;
+          }
+          errors.push({
+            index: preparedJob.index,
+            error: new Error(writeError.errmsg),
+            job: preparedJob.job
           });
         }
       }