feat(core): Update query service decorator to have correct generics

doug-martin · Sep 17, 2020 · 74dc618 · 74dc618
1 parent 6b629d9
commit 74dc618
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/core/src/assemblers/assembler.ts b/packages/core/src/assemblers/assembler.ts
@@ -102,5 +102,5 @@ export function getAssembler<DTO, Entity>(
 export function getAssemblerClasses<DTO, Entity>(
   AssemblerClass: Class<Assembler<DTO, Entity>>,
 ): MetaValue<AssemblerClasses<DTO, Entity>> {
-  return assemblerReflector.get(AssemblerClass as Class<Assembler<unknown, unknown>>);
+  return assemblerReflector.get(AssemblerClass);
 }
diff --git a/packages/core/src/services/query.service.ts b/packages/core/src/services/query.service.ts
@@ -263,8 +263,8 @@ export interface QueryService<DTO, C = DeepPartial<DTO>, U = DeepPartial<DTO>> {
  * @param DTOClass - the DTO class that the QueryService is used for.
  */
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
-export function QueryService<DTO>(DTOClass: Class<DTO>) {
-  return <Cls extends Class<QueryService<DTO, unknown, unknown>>>(cls: Cls): Cls | void => {
+export function QueryService<DTO, C = DeepPartial<DTO>, U = DeepPartial<DTO>>(DTOClass: Class<DTO>) {
+  return <Cls extends Class<QueryService<DTO, C, U>>>(cls: Cls): Cls | void => {
     return Injectable()(cls);
   };
 }