feat(mysql, mongo): support custom tables

koishijs · Mar 18, 2021 · 875f75b · 875f75b
1 parent 49c2c39
commit 875f75b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/packages/plugin-mongo/src/database.ts b/packages/plugin-mongo/src/database.ts
@@ -1,7 +1,11 @@
 import { MongoClient, Db, Collection } from 'mongodb'
-import { App, Channel, Database, Tables, TableType, User } from 'koishi-core'
+import { App, Channel, Database, User, Tables as KoishiTables } from 'koishi-core'
 import { URLSearchParams } from 'url'
 
+type TableType = keyof Tables
+
+export interface Tables extends KoishiTables {}
+
 export interface Config {
   username?: string
   password?: string

diff --git a/packages/plugin-mysql/src/database.ts b/packages/plugin-mysql/src/database.ts
@@ -1,5 +1,5 @@
 import { createPool, Pool, PoolConfig, escape as mysqlEscape, escapeId, format, OkPacket, TypeCast } from 'mysql'
-import { TableType, Tables, App, Database } from 'koishi-core'
+import { Tables as KoishiTables, App, Database } from 'koishi-core'
 import { Logger } from 'koishi-utils'
 import { types } from 'util'
 
@@ -9,6 +9,10 @@ declare module 'mysql' {
   }
 }
 
+type TableType = keyof Tables
+
+export interface Tables extends KoishiTables {}
+
 const logger = new Logger('mysql')
 
 export interface Config extends PoolConfig {}