mirror of
https://github.com/lzh-1625/go_process_manager.git
synced 2025-10-04 15:42:44 +08:00
56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package repository
|
||
|
||
import (
|
||
"log"
|
||
"os"
|
||
"time"
|
||
|
||
"github.com/lzh-1625/go_process_manager/internal/app/model"
|
||
"github.com/lzh-1625/go_process_manager/internal/app/repository/query"
|
||
|
||
"github.com/glebarez/sqlite"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/logger"
|
||
)
|
||
|
||
var (
|
||
db *gorm.DB
|
||
defaultConfig = gorm.Session{PrepareStmt: true, SkipDefaultTransaction: true}
|
||
)
|
||
|
||
func InitDb() {
|
||
newLogger := logger.New(
|
||
log.New(os.Stdout, "\r\n", log.LstdFlags),
|
||
logger.Config{
|
||
SlowThreshold: time.Second,
|
||
LogLevel: logger.Silent,
|
||
IgnoreRecordNotFoundError: true,
|
||
ParameterizedQueries: true,
|
||
Colorful: true,
|
||
},
|
||
)
|
||
gdb, err := gorm.Open(sqlite.Open("data.db"), &gorm.Config{
|
||
Logger: newLogger,
|
||
})
|
||
if err != nil {
|
||
log.Panicf("sqlite数据库初始化失败！\n错误原因：%v", err)
|
||
}
|
||
sqlDB, err := gdb.DB()
|
||
if err != nil {
|
||
log.Panicf("sqlite数据库初始化失败！\n错误原因：%v", err)
|
||
}
|
||
sqlDB.SetConnMaxLifetime(time.Hour)
|
||
db = gdb.Session(&defaultConfig)
|
||
db = gdb.Session(&defaultConfig).Debug()
|
||
db.AutoMigrate(&model.Process{}, &model.User{}, &model.Permission{}, &model.Push{}, &model.Config{}, &model.ProcessLog{}, &model.Task{}, &model.WsShare{})
|
||
|
||
// g := gen.NewGenerator(gen.Config{
|
||
// OutPath: "internal/app/repository/query",
|
||
// Mode: gen.WithoutContext | gen.WithDefaultQuery | gen.WithQueryInterface, // generate mode
|
||
// })
|
||
// g.UseDB(db)
|
||
// g.ApplyBasic(&model.Process{}, &model.User{}, &model.Permission{}, &model.Push{}, &model.Config{}, &model.ProcessLog{}, &model.Task{}, &model.WsShare{})
|
||
// g.Execute()
|
||
query.SetDefault(db)
|
||
}
|