refacotr: support default config for orm.client

Author: elrrrrrrr

core/orm-decorator/src/decorator/Model.ts

@@ -3,6 +3,7 @@ import { ModelInfoUtil } from '../util/ModelInfoUtil';
 
 export interface ModelParams {
   tableName?: string;
+  dataSource?: string;
 }
 
 export const MODEL_PROTO_IMPL_TYPE = 'MODEL_PROTO';
@@ -18,6 +19,9 @@ export function Model(param?: ModelParams) {
     if (param?.tableName) {
       ModelInfoUtil.setTableName(param.tableName, clazz);
     }
+    if (param?.dataSource) {
+      ModelInfoUtil.setDataSource(param.dataSource, clazz);
+    }
     func(clazz);
   };
 }

plugin/orm/lib/LeoricRegister.ts

@@ -17,22 +17,35 @@ export class LeoricRegister extends Base {
     this.realmMap = new Map();
   }
 
-  getOrCreateRealm(datasource: string | undefined): any {
+  getConfig(datasource?: string) {
     let config: OrmConfig | undefined;
     if (!datasource) {
       config = this.dataSourceManager.getDefaultConfig();
     } else {
       config = this.dataSourceManager.getConfig(datasource);
     }
-    if (!config) {
-      throw new Error(`not found datasource for ${datasource}`);
+    return config;
+  }
+
+  getRealm(config: OrmConfig | undefined): Realm | undefined {
+    if (!config?.database) {
+      return undefined;
     }
-    let realm = this.realmMap.get(config.database);
-    if (realm) {
-      return realm;
+    const realm = this.realmMap.get(config.database);
+    return realm;
+  }
+
+  getOrCreateRealm(datasource: string | undefined): any {
+    const config = this.getConfig(datasource);
+    let realm: Realm | undefined;
+    if (config) {
+      realm = this.getRealm(config);
+      if (realm) {
+        return realm;
+      }
     }
     realm = new (Realm as any)({ ...config });
-    this.realmMap.set(config.database, realm);
+    this.realmMap.set(config!.database, realm);
     return realm;
   }
 

plugin/orm/lib/SingletonORM.ts

@@ -15,11 +15,16 @@ export class Orm {
 
   // default dataSource
   get client(): Realm {
-    return this.getClient(undefined);
+    const defaultConfig = this.leoricRegister.getConfig();
+    return this.leoricRegister.getRealm(defaultConfig)!;
   }
 
-  getClient(dataSource: string | undefined): Realm {
-    return this.leoricRegister.getOrCreateRealm(dataSource);
+  getClient(datasource: string): Realm {
+    const config = this.leoricRegister.getConfig(datasource);
+    if (!config) {
+      throw new Error(`not found ${datasource} datasource`);
+    }
+    return this.leoricRegister.getOrCreateRealm(config.database)!;
   }
 
 }