當初始化專案的時候，之前需要手動匯入資料庫sql檔案，現在可以直接在命令列引數里輸入go-fly install ,就可以把資料庫檔案導進去了
實作方式就是使用gorm執行sql檔案里面的每行sql陳述句
讀入sql檔案后把字串按照分號分隔，轉成[]string ，然后再回圈執行
實體代碼是（自行更改檔案目錄）：
func install(){ sqlFile:=config.Dir+"go-fly.sql" isExit,_:=tools.IsFileExist(config.MysqlConf) dataExit,_:=tools.IsFileExist(sqlFile) if !isExit||!dataExit{ fmt.Println("config/mysql.json 資料庫組態檔或者資料庫檔案go-fly.sql不存在") os.Exit(1) } sqls,_:=ioutil.ReadFile(sqlFile) sqlArr:=strings.Split(string(sqls),";") for _,sql:=range sqlArr{ if sql==""{ continue } models.Execute(sql) } }
go-fly代碼中資料庫操作類
package models import ( "fmt" "github.com/jinzhu/gorm" "github.com/taoshihan1991/imaptool/config" "time" ) var DB *gorm.DB type Model struct { ID uint `gorm:"primary_key" json:"id"` CreatedAt time.Time `json:"created_at"` UpdatedAt time.Time `json:"updated_at"` DeletedAt *time.Time `sql:"index" json:"deleted_at"` } func init(){ mysql:=config.CreateMysql() dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&parseTime=True&loc=Local", mysql.Username, mysql.Password, mysql.Server, mysql.Port, mysql.Database) DB,_=gorm.Open("mysql",dsn) DB.SingularTable(true) DB.LogMode(true) DB.DB().SetMaxIdleConns(10) DB.DB().SetMaxOpenConns(100) } func Execute(sql string){ DB.Exec(sql) } func CloseDB() { defer DB.Close() }
簡化的邏輯就是gorm open以后得到的實體執行一下Exec方法，執行sql陳述句
執行完成以后的效果：
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/341.html
標籤：Go
上一篇：[Go]GO語言實戰-GO-FLY在線客服cobra庫命令列引數決議
下一篇：分布式系統選主場景分析及實作