mirror of
https://github.com/eryajf/chatgpt-dingtalk.git
synced 2025-10-05 16:16:56 +08:00
42 lines
826 B
Go
42 lines
826 B
Go
package db
|
|
|
|
import (
|
|
"github.com/eryajf/chatgpt-dingtalk/pkg/logger"
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// 全局数据库对象
|
|
var DB *gorm.DB
|
|
|
|
// 初始化数据库
|
|
func InitDB() {
|
|
DB = ConnSqlite()
|
|
|
|
dbAutoMigrate()
|
|
}
|
|
|
|
// 自动迁移表结构
|
|
func dbAutoMigrate() {
|
|
_ = DB.AutoMigrate(
|
|
Chat{},
|
|
)
|
|
}
|
|
|
|
func ConnSqlite() *gorm.DB {
|
|
db, err := gorm.Open(sqlite.Open("data/dingtalkbot.sqlite"), &gorm.Config{
|
|
// 禁用外键(指定外键时不会在mysql创建真实的外键约束)
|
|
DisableForeignKeyConstraintWhenMigrating: true,
|
|
})
|
|
if err != nil {
|
|
logger.Fatal("failed to connect sqlite3: %v", err)
|
|
}
|
|
dbObj, err := db.DB()
|
|
if err != nil {
|
|
logger.Fatal("failed to get sqlite3 obj: %v", err)
|
|
}
|
|
// 参见： https://github.com/glebarez/sqlite/issues/52
|
|
dbObj.SetMaxOpenConns(1)
|
|
return db
|
|
}
|