Merge pull request #134 from jamebal/jmal

使用MongoDB URL里的数据库名称

Author: Kerwin1202

service/src/storage/mongo.ts

@@ -6,12 +6,14 @@ import type { ChatOptions, Config, UsageResponse } from './model'
 dotenv.config()
 
 const url = process.env.MONGODB_URL
+const parsedUrl = new URL(url)
+const dbName = (parsedUrl.pathname && parsedUrl.pathname !== '/') ? parsedUrl.pathname.substring(1) : 'chatgpt'
 const client = new MongoClient(url)
-const chatCol = client.db('chatgpt').collection('chat')
-const roomCol = client.db('chatgpt').collection('chat_room')
-const userCol = client.db('chatgpt').collection('user')
-const configCol = client.db('chatgpt').collection('config')
-const usageCol = client.db('chatgpt').collection('chat_usage')
+const chatCol = client.db(dbName).collection('chat')
+const roomCol = client.db(dbName).collection('chat_room')
+const userCol = client.db(dbName).collection('user')
+const configCol = client.db(dbName).collection('config')
+const usageCol = client.db(dbName).collection('chat_usage')
 
 /**
  * 插入聊天信息