feat(plugin-clickhouse): update lifecycle

artusjs · Oct 8, 2024 · f869f2b · f869f2b
1 parent abdea30
commit f869f2b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/packages/plugins/clickhouse/src/client.ts b/packages/plugins/clickhouse/src/client.ts
@@ -12,7 +12,7 @@ export type ClickHouseConfig = NodeClickHouseClientConfigOptions & {
   id: InjectEnum.Client,
   scope: ScopeEnum.SINGLETON,
 })
-export class ClickHouseClient {
+export default class ClickHouseClient {
   _config: any;
 
   private clickhouseClient: NodeClickHouseClient;
@@ -34,4 +34,4 @@ export class ClickHouseClient {
   }
 }
 
-export default ClickHouseClient;
+export { ClickHouseClient };
diff --git a/packages/plugins/clickhouse/src/lifecycle.ts b/packages/plugins/clickhouse/src/lifecycle.ts
@@ -7,6 +7,7 @@ import {
   LifecycleHook,
 } from '@artus/core';
 import { ClickHouseConfig, ClickHouseClient } from './client';
+import { InjectEnum } from './constants';
 
 @LifecycleHookUnit()
 export default class PluginLifecycle implements ApplicationLifecycle {
@@ -26,7 +27,7 @@ export default class PluginLifecycle implements ApplicationLifecycle {
     }
 
     this.logger.info('[clickhouse] staring clickhouse with url: %s', config.url);
-    const clickhouse = this.app.container.get('ARTUSX_CLICKHOUSE') as ClickHouseClient;
+    const clickhouse = this.app.container.get(InjectEnum.Client) as ClickHouseClient;
     await clickhouse.init(config);
   }
 }