refactor: add strict partial loading support to repository methods an…

…d em.populate
mikro-orm · Apr 10, 2023 · d87545a · d87545a · edorgeville · Apr 11, 2023
1 parent 8a9f8e8
commit d87545a
Show file tree

Hide file tree

Showing 4 changed files with 139 additions and 117 deletions.
diff --git a/packages/core/src/EntityManager.ts b/packages/core/src/EntityManager.ts
@@ -1380,19 +1380,20 @@ export class EntityManager<D extends IDatabaseDriver = IDatabaseDriver> {
   async populate<
     Entity extends object,
     Hint extends string = never,
-  >(entities: Entity | Entity[], populate: AutoPath<Entity, Hint>[] | boolean, options: EntityLoaderOptions<Entity, Hint> = {}): Promise<Loaded<Entity, Hint>[]> {
+    Fields extends string = never,
+  >(entities: Entity | Entity[], populate: AutoPath<Entity, Hint>[] | boolean, options: EntityLoaderOptions<Entity, Hint, Fields> = {}): Promise<Loaded<Entity, Hint, Fields>[]> {
     entities = Utils.asArray(entities);
 
     if (entities.length === 0) {
-      return entities as Loaded<Entity, Hint>[];
+      return entities as Loaded<Entity, Hint, Fields>[];
     }
 
     const em = this.getContext();
     const entityName = (entities[0] as Dictionary).constructor.name;
     const preparedPopulate = em.preparePopulate<Entity>(entityName, { populate: populate as true });
     await em.entityLoader.populate(entityName, entities, preparedPopulate, options);
 
-    return entities as Loaded<Entity, Hint>[];
+    return entities as Loaded<Entity, Hint, Fields>[];
   }
 
   /**