E5BotForSQLite/model/sqlite.go

24 lines
363 B
Go
Raw Normal View History

2021-06-14 21:37:58 +08:00
package model
import (
2021-06-15 09:02:17 +08:00
"go.uber.org/zap"
2021-06-14 21:37:58 +08:00
"gorm.io/driver/sqlite"
"gorm.io/gorm"
"time"
)
var DB *gorm.DB
2021-06-15 09:02:17 +08:00
func InitDB() {
2021-06-14 21:37:58 +08:00
var err error
DB, err = gorm.Open(sqlite.Open("data.db"), &gorm.Config{
NowFunc: func() time.Time {
2021-06-15 09:02:17 +08:00
return time.Now()
2021-06-14 21:37:58 +08:00
},
})
if err != nil {
2021-06-15 09:02:17 +08:00
zap.S().Errorw("failed to open db", "error", err)
2021-06-14 21:37:58 +08:00
}
DB.AutoMigrate(&Client{})
}