forked from imbytecat/fullstack-starter
refactor(db): 暴露 closeDB() 函数以支持连接池清理
This commit is contained in:
+11
-6
@@ -10,11 +10,16 @@ export const createDB = () =>
|
|||||||
|
|
||||||
export type DB = ReturnType<typeof createDB>
|
export type DB = ReturnType<typeof createDB>
|
||||||
|
|
||||||
export const getDB = (() => {
|
let _db: DB | null = null
|
||||||
let db: DB | null = null
|
|
||||||
|
|
||||||
return (): DB => {
|
export const getDB = (): DB => {
|
||||||
db ??= createDB()
|
_db ??= createDB()
|
||||||
return db
|
return _db
|
||||||
|
}
|
||||||
|
|
||||||
|
export const closeDB = async () => {
|
||||||
|
if (_db) {
|
||||||
|
await _db.$client.end()
|
||||||
|
_db = null
|
||||||
|
}
|
||||||
}
|
}
|
||||||
})()
|
|
||||||
|
|||||||
Reference in New Issue
Block a user