Update index.js

mongoosejs · Mar 8, 2024 · 7fbffe5 · 7fbffe5
1 parent 0017585
commit 7fbffe5
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/index.js b/index.js
@@ -72,11 +72,14 @@ function getSchemaForDoc(schema, res) {
   }
 
   const discriminatorValue = res[schema.discriminatorMapping.key];
-  const childSchema = Object.entries(schema.discriminators).find(([modelName, childSchema]) => {
-    // Match against an explicit discriminator value if present
+  let childSchema = undefined;
+  for (const name of Object.keys(schema.discriminators)) {
     const matchValue = childSchema.discriminatorMapping.value || modelName;
-    return discriminatorValue === matchValue;
-  })[1];
+    if (matchValue === discriminatorValue) {
+      childSchema = schema.discriminators[name];
+      break;
+    }
+  }
   return childSchema;
 }