Browse Source

chore: improves types

e22m4u 2 years ago
parent
commit
dec52fc99f
1 changed files with 8 additions and 1 deletions
  1. 8 1
      src/schema.d.ts

+ 8 - 1
src/schema.d.ts

@@ -1,7 +1,10 @@
+import {ModelId} from './types';
+import {ModelData} from './types';
 import {Service} from '@e22m4u/service';
 import {Repository} from './repository';
 import {ModelDefinition} from './definition';
 import {DatasourceDefinition} from './definition';
+import {DEFAULT_PRIMARY_KEY_PROPERTY_NAME} from './definition';
 
 /**
  * Schema.
@@ -26,5 +29,9 @@ export declare class Schema extends Service {
    *
    * @param modelName
    */
-  getRepository(modelName: string): Repository;
+  getRepository<
+    Data extends ModelData = ModelData,
+    IdType extends ModelId = ModelId,
+    IdName extends string = DEFAULT_PRIMARY_KEY_PROPERTY_NAME,
+  >(modelName: string): Repository<Data, IdType, IdName>;
 }