add disconnect hooks for closing everyting

src/index.ts

@@ -10,6 +10,7 @@ import {
   Caller,
   Provide,
   Inject,
+  LifecycleEvents,
 } from 'koishi-thirdeye';
 import { Connection, ConnectionOptions, createConnection } from 'typeorm';
 import { Context, Database } from 'koishi';
@@ -31,7 +32,10 @@ declare module 'koishi' {
 
 @Provide('typeorm', { immediate: true })
 @DefinePlugin({ name: 'typeorm', schema: TypeORMPluginConfig })
-export default class TypeORMPlugin extends BasePlugin<TypeORMPluginConfigLike> {
+export default class TypeORMPlugin
+  extends BasePlugin<TypeORMPluginConfigLike>
+  implements LifecycleEvents
+{
   private registryMap = new Map<string, ConnectionEntry>();
   private entityMap = new Map<TypeORMEntity, string>();
 
@@ -96,4 +100,11 @@ export default class TypeORMPlugin extends BasePlugin<TypeORMPluginConfigLike> {
     }
     return connection;
   }
+
+  async onDisconnect() {
+    const tokens = Array.from(this.registryMap.keys());
+    await Promise.all(tokens.map((token) => this.close(token)));
+    this.registryMap.clear();
+    this.entityMap.clear();
+  }
 }