From 63645d248c765244488cd34dbc1bb6528ca6b7c7 Mon Sep 17 00:00:00 2001 From: zhangzengfei <zhangzengfei@smartai.com> Date: 星期二, 05 九月 2023 09:58:13 +0800 Subject: [PATCH] 修复编译 --- version-control/models/db.go | 128 +++++++++++++++++++++--------------------- 1 files changed, 64 insertions(+), 64 deletions(-) diff --git a/version-control/models/db.go b/version-control/models/db.go index 2fa3f2f..e2529e3 100644 --- a/version-control/models/db.go +++ b/version-control/models/db.go @@ -1,64 +1,64 @@ -package models - -import ( - "fmt" - "github.com/jinzhu/gorm" - _ "github.com/jinzhu/gorm/dialects/sqlite" - "os" - "path" - "path/filepath" - "vamicro/config" -) - -var db *gorm.DB - - -// Init creates a connection to mysql database and -// migrates any new models -func Init() { - var err error - //dbDir := getDbDir() - //dbPath := path.Join(dbDir, "version.db") - db, err = gorm.Open(config.DBconf.Name, "../config/version.db") - if err != nil { - fmt.Println("../config/version.db", config.DBconf.Name) - fmt.Println("Db Init failed, ", err) - return - } else { - fmt.Println("Db Init ok, file:", "../config/version.db") - } - - db.LogMode(true) - //db.SetLogger(&DbLogger{}) - //&Program{}, - db.AutoMigrate(&SysSetting{}, &AutoBackupConf{}) - InitSysSettingData() -} - -//GetDB ... -func GetDB() *gorm.DB { - return db -} - -func CloseDB() { - db.Close() -} - -func getExePath() string { - dir, err := filepath.Abs(filepath.Dir(os.Args[0])) - if nil != err { - return "./" - } - - return dir -} - -func getDbDir() string { - dbPath := getExePath() - dbPath = path.Join(dbPath, "db") - if _, err := os.Stat(dbPath); os.IsNotExist(err) { - _ = os.MkdirAll(dbPath, 0777) - } - - return dbPath -} +package models + +import ( + "fmt" + "github.com/jinzhu/gorm" + _ "github.com/jinzhu/gorm/dialects/sqlite" + "os" + "path" + "path/filepath" + "vamicro/config" +) + +var db *gorm.DB + + +// Init creates a connection to mysql database and +// migrates any new models +func Init() { + var err error + //dbDir := getDbDir() + //dbPath := path.Join(dbDir, "version.db") + db, err = gorm.Open(config.DBconf.Name, "../config/version.db") + if err != nil { + fmt.Println("../config/version.db", config.DBconf.Name) + fmt.Println("Db Init failed, ", err) + return + } else { + fmt.Println("Db Init ok, file:", "../config/version.db") + } + + db.LogMode(true) + //db.SetLogger(&DbLogger{}) + //&Program{}, + db.AutoMigrate(&SysSetting{}, &AutoBackupConf{}) + InitSysSettingData() +} + +//GetDB ... +func GetDB() *gorm.DB { + return db +} + +func CloseDB() { + db.Close() +} + +func getExePath() string { + dir, err := filepath.Abs(filepath.Dir(os.Args[0])) + if nil != err { + return "./" + } + + return dir +} + +func getDbDir() string { + dbPath := getExePath() + dbPath = path.Join(dbPath, "db") + if _, err := os.Stat(dbPath); os.IsNotExist(err) { + _ = os.MkdirAll(dbPath, 0777) + } + + return dbPath +} -- Gitblit v1.8.0