package models import ( "basic.com/valib/logger.git" "github.com/jinzhu/gorm" _ "github.com/jinzhu/gorm/dialects/sqlite" "vamicro/config" ) var db *gorm.DB var err error // Init creates a connection to mysql database and // migrates any new models func Init() { db, err = gorm.Open(config.DBconf.Name, "../config/stack-service.db") if err != nil { logger.Debug("db open error ", err) } db.LogMode(true) //db.SetLogger(&DbLogger{}) db.AutoMigrate(&FileStack{}, &FileAnalysis{}, &FileAnalysisSetting{}) initSetting() } // GetDB ... func GetDB() *gorm.DB { return db } func CloseDB() { db.Close() } func initSetting() { db.Exec("INSERT INTO file_analysis_setting SELECT 1,'/opt/vasystem/files', 0, 0, 0 where not exists (select 1 from file_analysis_setting)") //添加默认的通道设置 }