Commit 8a1eb84518538c58903f619d5dc16d9d2ef5d23a
1 parent
69fcd276
Exists in
master
and in
1 other branch
refactor: orm重构
Showing
13 changed files
with
240 additions
and
244 deletions
Show diff stats
dto/cardholder-record.go
| ... | ... | @@ -2,8 +2,8 @@ package dto |
| 2 | 2 | |
| 3 | 3 | import ( |
| 4 | 4 | "apigame/service/constd" |
| 5 | + "apigame/util/utdto" | |
| 5 | 6 | "apigame/util/util-lx/lxtime" |
| 6 | - "apigame/util/utildto" | |
| 7 | 7 | ) |
| 8 | 8 | |
| 9 | 9 | // CardHolderRecordBase 开卡包活动日志公共 |
| ... | ... | @@ -39,7 +39,7 @@ type CardHolderRecordOpen struct { |
| 39 | 39 | } |
| 40 | 40 | |
| 41 | 41 | func (d *CardHolderRecordOpen) TableName() string { |
| 42 | - return utildto.MYSQL_TABLE_TEMPLATE | |
| 42 | + return utdto.MYSQL_TABLE_TEMPLATE | |
| 43 | 43 | } |
| 44 | 44 | func (d *CardHolderRecordOpen) GetTableName(gameId string) string { |
| 45 | 45 | return constd.MYSQL_TABLE_S_CARDHOLDER_RECORD_OPEN + gameId |
| ... | ... | @@ -62,7 +62,7 @@ type CardHolderRecordRewardAlbum struct { |
| 62 | 62 | } |
| 63 | 63 | |
| 64 | 64 | func (d *CardHolderRecordRewardAlbum) TableName() string { |
| 65 | - return utildto.MYSQL_TABLE_TEMPLATE | |
| 65 | + return utdto.MYSQL_TABLE_TEMPLATE | |
| 66 | 66 | } |
| 67 | 67 | func (d *CardHolderRecordRewardAlbum) GetTableName(gameId string) string { |
| 68 | 68 | return constd.MYSQL_TABLE_S_CARDHOLDER_RECORD_REWARDALBUM + gameId |
| ... | ... | @@ -84,7 +84,7 @@ type CardHolderRecordRewardRound struct { |
| 84 | 84 | } |
| 85 | 85 | |
| 86 | 86 | func (d *CardHolderRecordRewardRound) TableName() string { |
| 87 | - return utildto.MYSQL_TABLE_TEMPLATE | |
| 87 | + return utdto.MYSQL_TABLE_TEMPLATE | |
| 88 | 88 | } |
| 89 | 89 | func (d *CardHolderRecordRewardRound) GetTableName(gameId string) string { |
| 90 | 90 | return constd.MYSQL_TABLE_S_CARDHOLDER_RECORD_REWARDROUND + gameId | ... | ... |
dto/cardholder.go
| ... | ... | @@ -2,9 +2,9 @@ package dto |
| 2 | 2 | |
| 3 | 3 | import ( |
| 4 | 4 | "apigame/service/constd" |
| 5 | + "apigame/util/utdto" | |
| 5 | 6 | "apigame/util/util-lx/lxalilog" |
| 6 | 7 | "apigame/util/util-lx/lxtime" |
| 7 | - "apigame/util/utildto" | |
| 8 | 8 | "encoding/json" |
| 9 | 9 | ) |
| 10 | 10 | |
| ... | ... | @@ -47,7 +47,7 @@ func NewCardHolderDataDetails() *CardHolderDataDetails { |
| 47 | 47 | } |
| 48 | 48 | |
| 49 | 49 | func (d *CardHolderData) TableName() string { |
| 50 | - return utildto.MYSQL_TABLE_TEMPLATE | |
| 50 | + return utdto.MYSQL_TABLE_TEMPLATE | |
| 51 | 51 | } |
| 52 | 52 | |
| 53 | 53 | func (d *CardHolderData) GetTableName(gameId string) string { | ... | ... |
dto/dto.go
| 1 | 1 | package dto |
| 2 | 2 | |
| 3 | 3 | import ( |
| 4 | - "apigame/util/utildto" | |
| 4 | + "apigame/util/utdto" | |
| 5 | + "apigame/util/util-lx/lxalilog" | |
| 5 | 6 | "gorm.io/gorm" |
| 7 | + "strings" | |
| 6 | 8 | ) |
| 7 | 9 | |
| 8 | -func InitTable(db *gorm.DB, tb ITableData, gameId string) { | |
| 10 | +func InitTable(db *gorm.DB, gameId string, tb IDtoData) { | |
| 9 | 11 | tableName := tb.GetTableName(gameId) |
| 10 | - utildto.InitTable(db, tb, tableName) | |
| 12 | + utdto.InitTable(db, tb, tableName) | |
| 13 | +} | |
| 14 | + | |
| 15 | +func Insert(db *gorm.DB, gameId string, data IDtoData) { | |
| 16 | + utdto.Insert(db, data, data.GetTableName(gameId)) | |
| 17 | +} | |
| 18 | + | |
| 19 | +func Update(db *gorm.DB, gameId string, data IDtoData) { | |
| 20 | + utdto.Update(db, data, data.GetTableName(gameId)) | |
| 21 | +} | |
| 22 | + | |
| 23 | +func Save(db *gorm.DB, gameId string, data IDtoData) { | |
| 24 | + utdto.Save(db, data, data.GetTableName(gameId)) | |
| 25 | +} | |
| 26 | + | |
| 27 | +func First(db *gorm.DB, gameId string, data IDtoData) (has bool, err error) { | |
| 28 | + result := utdto.First(db, data, data.GetTableName(gameId)) | |
| 29 | + err = result.Error | |
| 30 | + if err == nil { | |
| 31 | + has = true | |
| 32 | + } else { | |
| 33 | + has = false | |
| 34 | + if strings.Contains(err.Error(), "record not found") { | |
| 35 | + err = nil | |
| 36 | + } else { | |
| 37 | + lxalilog.Errors(err, gameId) | |
| 38 | + return | |
| 39 | + } | |
| 40 | + } | |
| 41 | + return | |
| 11 | 42 | } | ... | ... |
dto/init-dto.go
| ... | ... | @@ -2,7 +2,7 @@ package dto |
| 2 | 2 | |
| 3 | 3 | import ( |
| 4 | 4 | "apigame/service/constd" |
| 5 | - "apigame/util/utildto" | |
| 5 | + "apigame/util/utdto" | |
| 6 | 6 | "fmt" |
| 7 | 7 | "github.com/astaxie/beego" |
| 8 | 8 | "gorm.io/gorm" |
| ... | ... | @@ -16,12 +16,12 @@ var ( |
| 16 | 16 | func InitDto() { |
| 17 | 17 | |
| 18 | 18 | // ConnectMySQL |
| 19 | - if db, err := utildto.ConnectMySQL(beego.AppConfig.String("mysql::mergecommon")); err == nil { | |
| 19 | + if db, err := utdto.ConnectMySQL(beego.AppConfig.String("mysql::mergecommon")); err == nil { | |
| 20 | 20 | DbCommon = db |
| 21 | 21 | } else { |
| 22 | 22 | fmt.Println(" InitMysqls Error::" + err.Error()) |
| 23 | 23 | } |
| 24 | - if db, err := utildto.ConnectMySQL(beego.AppConfig.String("mysql::" + constd.MYSQL_MERGECONFIG)); err == nil { | |
| 24 | + if db, err := utdto.ConnectMySQL(beego.AppConfig.String("mysql::" + constd.MYSQL_MERGECONFIG)); err == nil { | |
| 25 | 25 | DbConfig = db |
| 26 | 26 | } else { |
| 27 | 27 | |
| ... | ... | @@ -30,10 +30,10 @@ func InitDto() { |
| 30 | 30 | // create table |
| 31 | 31 | // 卡牌卡包 |
| 32 | 32 | for _, gameId := range constd.GameListCardHolder { |
| 33 | - InitTable(DbCommon, new(CardHolderData), gameId) | |
| 34 | - InitTable(DbCommon, new(CardHolderRecordOpen), gameId) | |
| 35 | - InitTable(DbCommon, new(CardHolderRecordRewardAlbum), gameId) | |
| 36 | - InitTable(DbCommon, new(CardHolderRecordRewardRound), gameId) | |
| 33 | + InitTable(DbCommon, gameId, new(CardHolderData)) | |
| 34 | + InitTable(DbCommon, gameId, new(CardHolderRecordOpen)) | |
| 35 | + InitTable(DbCommon, gameId, new(CardHolderRecordRewardAlbum)) | |
| 36 | + InitTable(DbCommon, gameId, new(CardHolderRecordRewardRound)) | |
| 37 | 37 | } |
| 38 | 38 | |
| 39 | 39 | } | ... | ... |
dto/interface.go
| 1 | 1 | package dto |
| 2 | 2 | |
| 3 | -// ITableData 数据库表 | |
| 4 | -type ITableData interface { | |
| 5 | - // GetTableName 表名 后缀是 gameId | |
| 6 | - GetTableName(gameId string) string | |
| 7 | -} | |
| 8 | - | |
| 9 | -type IRecordData interface { | |
| 10 | - // TableName 表名模板 | |
| 11 | - TableName() string | |
| 12 | - // GetTableName 表名 后缀是 gameId | |
| 13 | - GetTableName(gameId string) string | |
| 14 | -} | |
| 15 | - | |
| 16 | -// IPlayerData 玩家数据表 | |
| 17 | -type IPlayerData interface { | |
| 3 | +// IDtoData dto数据 | |
| 4 | +type IDtoData interface { | |
| 18 | 5 | // TableName 表名模板 |
| 19 | 6 | TableName() string |
| 20 | 7 | // GetTableName 表名 后缀是 gameId |
| 21 | 8 | GetTableName(gameId string) string |
| 22 | - // Init 新建 | |
| 23 | - Init(uid int64) | |
| 24 | 9 | } | ... | ... |
dto/player-data.go
| ... | ... | @@ -1,37 +0,0 @@ |
| 1 | -package dto | |
| 2 | - | |
| 3 | -import ( | |
| 4 | - "apigame/util/util-lx/lxalilog" | |
| 5 | - "apigame/util/utildto" | |
| 6 | - "strings" | |
| 7 | -) | |
| 8 | - | |
| 9 | -func LoadPlayerData(gameId string, uid int64, data IPlayerData) (has bool, err error) { | |
| 10 | - result := utildto.First(DbCommon, data, data.GetTableName(gameId)) | |
| 11 | - err = result.Error | |
| 12 | - if err == nil { | |
| 13 | - has = true | |
| 14 | - } else { | |
| 15 | - has = false | |
| 16 | - if strings.Contains(err.Error(), "record not found") { | |
| 17 | - err = nil | |
| 18 | - } else { | |
| 19 | - lxalilog.Errors(err, gameId, uid) | |
| 20 | - return | |
| 21 | - } | |
| 22 | - } | |
| 23 | - // 档案不存在 新建 | |
| 24 | - if !has { | |
| 25 | - data.Init(uid) | |
| 26 | - CreatePlayerData(gameId, data) | |
| 27 | - } | |
| 28 | - return | |
| 29 | -} | |
| 30 | - | |
| 31 | -func CreatePlayerData(gameId string, data IPlayerData) { | |
| 32 | - utildto.Insert(DbCommon, data, data.GetTableName(gameId)) | |
| 33 | -} | |
| 34 | - | |
| 35 | -func SavePlayerData(gameId string, data IPlayerData) { | |
| 36 | - utildto.Save(DbCommon, data, data.GetTableName(gameId)) | |
| 37 | -} |
dto/record.go
| 1 | 1 | package dto |
| 2 | 2 | |
| 3 | -import "apigame/util/utildto" | |
| 3 | +import "apigame/util/utdto" | |
| 4 | 4 | |
| 5 | -func SaveRecord(gameId string, data IRecordData) { | |
| 6 | - utildto.Insert(DbCommon, data, data.GetTableName(gameId)) | |
| 5 | +func SaveRecord(gameId string, data IDtoData) { | |
| 6 | + utdto.Insert(DbCommon, data, data.GetTableName(gameId)) | |
| 7 | 7 | } | ... | ... |
service/cardholder/logic.go
| ... | ... | @@ -14,13 +14,13 @@ func SaveData(gameId string, d *dto.CardHolderData) { |
| 14 | 14 | d.UpdateTime = lxtime.NowUninx() |
| 15 | 15 | d.Encode() |
| 16 | 16 | |
| 17 | - dto.SavePlayerData(gameId, d) | |
| 17 | + dto.Update(dto.DbCommon, gameId, d) | |
| 18 | 18 | } |
| 19 | 19 | |
| 20 | 20 | func _LoadData(gameId string, uid int64) (d *dto.CardHolderData) { |
| 21 | 21 | d = dto.NewCardHolderData(uid) |
| 22 | 22 | |
| 23 | - has, err := dto.LoadPlayerData(gameId, uid, d) | |
| 23 | + has, err := dto.First(dto.DbCommon, gameId, d) | |
| 24 | 24 | if err != nil { |
| 25 | 25 | lxalilog.Errors(err, "cardholder._LoadData", gameId, uid) |
| 26 | 26 | return |
| ... | ... | @@ -28,6 +28,9 @@ func _LoadData(gameId string, uid int64) (d *dto.CardHolderData) { |
| 28 | 28 | // 解析 |
| 29 | 29 | if has { |
| 30 | 30 | d.Decode() |
| 31 | + } else { | |
| 32 | + d.Init(uid) | |
| 33 | + dto.Insert(dto.DbCommon, gameId, d) | |
| 31 | 34 | } |
| 32 | 35 | return |
| 33 | 36 | } | ... | ... |
| ... | ... | @@ -0,0 +1,37 @@ |
| 1 | +package utdto | |
| 2 | + | |
| 3 | +import ( | |
| 4 | + "gorm.io/gorm" | |
| 5 | + "strings" | |
| 6 | +) | |
| 7 | + | |
| 8 | +// Insert 插入 | |
| 9 | +func Insert(db *gorm.DB, value any, tableName string) *gorm.DB { | |
| 10 | + stmt := db.Session(&gorm.Session{DryRun: true}).Create(value).Statement | |
| 11 | + stmtSQL := stmt.SQL.String() | |
| 12 | + sql := strings.Replace(stmtSQL, MYSQL_TABLE_TEMPLATE, tableName, -1) | |
| 13 | + return db.Exec(sql, stmt.Vars...) | |
| 14 | +} | |
| 15 | + | |
| 16 | +// Update 更新 | |
| 17 | +func Update(db *gorm.DB, value any, tableName string) *gorm.DB { | |
| 18 | + stmt := db.Session(&gorm.Session{DryRun: true}).Updates(value).Statement | |
| 19 | + stmtSQL := stmt.SQL.String() | |
| 20 | + sql := strings.Replace(stmtSQL, MYSQL_TABLE_TEMPLATE, tableName, -1) | |
| 21 | + return db.Exec(sql, stmt.Vars...) | |
| 22 | +} | |
| 23 | + | |
| 24 | +// Save 保存 | |
| 25 | +func Save(db *gorm.DB, value any, tableName string) *gorm.DB { | |
| 26 | + stmt := db.Session(&gorm.Session{DryRun: true}).Save(value).Statement | |
| 27 | + stmtSQL := stmt.SQL.String() | |
| 28 | + sql := strings.Replace(stmtSQL, MYSQL_TABLE_TEMPLATE, tableName, -1) | |
| 29 | + return db.Exec(sql, stmt.Vars...) | |
| 30 | +} | |
| 31 | + | |
| 32 | +func First(db *gorm.DB, value any, tableName string) *gorm.DB { | |
| 33 | + stmt := db.Session(&gorm.Session{DryRun: true}).First(value).Statement | |
| 34 | + stmtSQL := stmt.SQL.String() | |
| 35 | + sql := strings.Replace(stmtSQL, MYSQL_TABLE_TEMPLATE, tableName, -1) | |
| 36 | + return db.Raw(sql, stmt.Vars...).Scan(value) | |
| 37 | +} | ... | ... |
| ... | ... | @@ -0,0 +1,12 @@ |
| 1 | +package utdto | |
| 2 | + | |
| 3 | +import ( | |
| 4 | + "gorm.io/driver/mysql" | |
| 5 | + "gorm.io/gorm" | |
| 6 | +) | |
| 7 | + | |
| 8 | +const MYSQL_TABLE_TEMPLATE = "all_table_template" // 表名模板 | |
| 9 | + | |
| 10 | +func ConnectMySQL(dsn string) (db *gorm.DB, err error) { | |
| 11 | + return gorm.Open(mysql.Open(dsn), &gorm.Config{}) | |
| 12 | +} | ... | ... |
| ... | ... | @@ -0,0 +1,132 @@ |
| 1 | +package utdto | |
| 2 | + | |
| 3 | +import ( | |
| 4 | + "fmt" | |
| 5 | + "gorm.io/driver/mysql" | |
| 6 | + "gorm.io/gorm" | |
| 7 | + "gorm.io/gorm/clause" | |
| 8 | + "gorm.io/gorm/migrator" | |
| 9 | + "strings" | |
| 10 | +) | |
| 11 | + | |
| 12 | +func InitTable(db *gorm.DB, value any, tableName string) { | |
| 13 | + m := db.Migrator().(mysql.Migrator).Migrator | |
| 14 | + queryTx, execTx := m.GetQueryAndExecTx() | |
| 15 | + if HasTable(queryTx.Migrator().(mysql.Migrator).Migrator, value, tableName) { | |
| 16 | + fmt.Println(tableName + " exist, skip!") | |
| 17 | + } else { | |
| 18 | + err := CreateTable(execTx, value, tableName) | |
| 19 | + if err != nil { | |
| 20 | + fmt.Println("err", err) | |
| 21 | + } | |
| 22 | + } | |
| 23 | +} | |
| 24 | + | |
| 25 | +func HasTable(m migrator.Migrator, value interface{}, tableName string) bool { | |
| 26 | + var count int64 | |
| 27 | + | |
| 28 | + m.RunWithValue(value, func(stmt *gorm.Statement) error { | |
| 29 | + currentDatabase := m.DB.Migrator().CurrentDatabase() | |
| 30 | + return m.DB.Raw("SELECT count(*) FROM information_schema.tables WHERE table_schema = ? AND table_name = ? AND table_type = ?", currentDatabase, tableName, "BASE TABLE").Row().Scan(&count) | |
| 31 | + }) | |
| 32 | + | |
| 33 | + return count > 0 | |
| 34 | +} | |
| 35 | + | |
| 36 | +func CreateTable(execTx *gorm.DB, value any, tableName string) error { | |
| 37 | + m := execTx.Migrator().(mysql.Migrator).Migrator | |
| 38 | + tx := m.DB.Session(&gorm.Session{}) | |
| 39 | + | |
| 40 | + if err := m.RunWithValue(value, func(stmt *gorm.Statement) (err error) { | |
| 41 | + var ( | |
| 42 | + createTableSQL = "CREATE TABLE ? (" | |
| 43 | + values = []interface{}{clause.Table{Name: tableName}} | |
| 44 | + hasPrimaryKeyInDataType bool | |
| 45 | + ) | |
| 46 | + | |
| 47 | + for _, dbName := range stmt.Schema.DBNames { | |
| 48 | + field := stmt.Schema.FieldsByDBName[dbName] | |
| 49 | + if !field.IgnoreMigration { | |
| 50 | + createTableSQL += "? ?" | |
| 51 | + hasPrimaryKeyInDataType = hasPrimaryKeyInDataType || strings.Contains(strings.ToUpper(m.DataTypeOf(field)), "PRIMARY KEY") | |
| 52 | + values = append(values, clause.Column{Name: dbName}, m.DB.Migrator().FullDataTypeOf(field)) | |
| 53 | + createTableSQL += "," | |
| 54 | + } | |
| 55 | + } | |
| 56 | + | |
| 57 | + if !hasPrimaryKeyInDataType && len(stmt.Schema.PrimaryFields) > 0 { | |
| 58 | + createTableSQL += "PRIMARY KEY ?," | |
| 59 | + primaryKeys := make([]interface{}, 0, len(stmt.Schema.PrimaryFields)) | |
| 60 | + for _, field := range stmt.Schema.PrimaryFields { | |
| 61 | + primaryKeys = append(primaryKeys, clause.Column{Name: field.DBName}) | |
| 62 | + } | |
| 63 | + | |
| 64 | + values = append(values, primaryKeys) | |
| 65 | + } | |
| 66 | + | |
| 67 | + for _, idx := range stmt.Schema.ParseIndexes() { | |
| 68 | + if m.CreateIndexAfterCreateTable { | |
| 69 | + defer func(value interface{}, name string) { | |
| 70 | + if err == nil { | |
| 71 | + err = tx.Migrator().CreateIndex(value, name) | |
| 72 | + } | |
| 73 | + }(value, idx.Name) | |
| 74 | + } else { | |
| 75 | + if idx.Class != "" { | |
| 76 | + createTableSQL += idx.Class + " " | |
| 77 | + } | |
| 78 | + createTableSQL += "INDEX ? ?" | |
| 79 | + | |
| 80 | + if idx.Comment != "" { | |
| 81 | + createTableSQL += fmt.Sprintf(" COMMENT '%s'", idx.Comment) | |
| 82 | + } | |
| 83 | + | |
| 84 | + if idx.Option != "" { | |
| 85 | + createTableSQL += " " + idx.Option | |
| 86 | + } | |
| 87 | + | |
| 88 | + createTableSQL += "," | |
| 89 | + values = append(values, clause.Column{Name: idx.Name}, tx.Migrator().(migrator.BuildIndexOptionsInterface).BuildIndexOptions(idx.Fields, stmt)) | |
| 90 | + } | |
| 91 | + } | |
| 92 | + | |
| 93 | + if !m.DB.DisableForeignKeyConstraintWhenMigrating && !m.DB.IgnoreRelationshipsWhenMigrating { | |
| 94 | + for _, rel := range stmt.Schema.Relationships.Relations { | |
| 95 | + if rel.Field.IgnoreMigration { | |
| 96 | + continue | |
| 97 | + } | |
| 98 | + if constraint := rel.ParseConstraint(); constraint != nil { | |
| 99 | + if constraint.Schema == stmt.Schema { | |
| 100 | + sql, vars := constraint.Build() | |
| 101 | + createTableSQL += sql + "," | |
| 102 | + values = append(values, vars...) | |
| 103 | + } | |
| 104 | + } | |
| 105 | + } | |
| 106 | + } | |
| 107 | + | |
| 108 | + for _, uni := range stmt.Schema.ParseUniqueConstraints() { | |
| 109 | + createTableSQL += "CONSTRAINT ? UNIQUE (?)," | |
| 110 | + values = append(values, clause.Column{Name: uni.Name}, clause.Expr{SQL: stmt.Quote(uni.Field.DBName)}) | |
| 111 | + } | |
| 112 | + | |
| 113 | + for _, chk := range stmt.Schema.ParseCheckConstraints() { | |
| 114 | + createTableSQL += "CONSTRAINT ? CHECK (?)," | |
| 115 | + values = append(values, clause.Column{Name: chk.Name}, clause.Expr{SQL: chk.Constraint}) | |
| 116 | + } | |
| 117 | + | |
| 118 | + createTableSQL = strings.TrimSuffix(createTableSQL, ",") | |
| 119 | + | |
| 120 | + createTableSQL += ")" | |
| 121 | + | |
| 122 | + if tableOption, ok := m.DB.Get("gorm:table_options"); ok { | |
| 123 | + createTableSQL += fmt.Sprint(tableOption) | |
| 124 | + } | |
| 125 | + | |
| 126 | + err = tx.Exec(createTableSQL, values...).Error | |
| 127 | + return err | |
| 128 | + }); err != nil { | |
| 129 | + return err | |
| 130 | + } | |
| 131 | + return nil | |
| 132 | +} | ... | ... |
util/utildto/mysql.go
| ... | ... | @@ -1,169 +0,0 @@ |
| 1 | -package utildto | |
| 2 | - | |
| 3 | -import ( | |
| 4 | - "fmt" | |
| 5 | - "gorm.io/driver/mysql" | |
| 6 | - "gorm.io/gorm" | |
| 7 | - "gorm.io/gorm/clause" | |
| 8 | - "gorm.io/gorm/migrator" | |
| 9 | - "strings" | |
| 10 | -) | |
| 11 | - | |
| 12 | -const MYSQL_TABLE_TEMPLATE = "all_table_template" // 表名模板 | |
| 13 | - | |
| 14 | -func ConnectMySQL(dsn string) (db *gorm.DB, err error) { | |
| 15 | - return gorm.Open(mysql.Open(dsn), &gorm.Config{}) | |
| 16 | -} | |
| 17 | - | |
| 18 | -func InitTable(db *gorm.DB, value any, tableName string) { | |
| 19 | - m := db.Migrator().(mysql.Migrator).Migrator | |
| 20 | - queryTx, execTx := m.GetQueryAndExecTx() | |
| 21 | - if HasTable(queryTx.Migrator().(mysql.Migrator).Migrator, value, tableName) { | |
| 22 | - fmt.Println(tableName + " exist, skip!") | |
| 23 | - } else { | |
| 24 | - err := CreateTable(execTx, value, tableName) | |
| 25 | - if err != nil { | |
| 26 | - fmt.Println("err", err) | |
| 27 | - } | |
| 28 | - } | |
| 29 | -} | |
| 30 | - | |
| 31 | -func HasTable(m migrator.Migrator, value interface{}, tableName string) bool { | |
| 32 | - var count int64 | |
| 33 | - | |
| 34 | - m.RunWithValue(value, func(stmt *gorm.Statement) error { | |
| 35 | - currentDatabase := m.DB.Migrator().CurrentDatabase() | |
| 36 | - return m.DB.Raw("SELECT count(*) FROM information_schema.tables WHERE table_schema = ? AND table_name = ? AND table_type = ?", currentDatabase, tableName, "BASE TABLE").Row().Scan(&count) | |
| 37 | - }) | |
| 38 | - | |
| 39 | - return count > 0 | |
| 40 | -} | |
| 41 | - | |
| 42 | -func CreateTable(execTx *gorm.DB, value any, tableName string) error { | |
| 43 | - m := execTx.Migrator().(mysql.Migrator).Migrator | |
| 44 | - tx := m.DB.Session(&gorm.Session{}) | |
| 45 | - | |
| 46 | - if err := m.RunWithValue(value, func(stmt *gorm.Statement) (err error) { | |
| 47 | - var ( | |
| 48 | - createTableSQL = "CREATE TABLE ? (" | |
| 49 | - values = []interface{}{clause.Table{Name: tableName}} | |
| 50 | - hasPrimaryKeyInDataType bool | |
| 51 | - ) | |
| 52 | - | |
| 53 | - for _, dbName := range stmt.Schema.DBNames { | |
| 54 | - field := stmt.Schema.FieldsByDBName[dbName] | |
| 55 | - if !field.IgnoreMigration { | |
| 56 | - createTableSQL += "? ?" | |
| 57 | - hasPrimaryKeyInDataType = hasPrimaryKeyInDataType || strings.Contains(strings.ToUpper(m.DataTypeOf(field)), "PRIMARY KEY") | |
| 58 | - values = append(values, clause.Column{Name: dbName}, m.DB.Migrator().FullDataTypeOf(field)) | |
| 59 | - createTableSQL += "," | |
| 60 | - } | |
| 61 | - } | |
| 62 | - | |
| 63 | - if !hasPrimaryKeyInDataType && len(stmt.Schema.PrimaryFields) > 0 { | |
| 64 | - createTableSQL += "PRIMARY KEY ?," | |
| 65 | - primaryKeys := make([]interface{}, 0, len(stmt.Schema.PrimaryFields)) | |
| 66 | - for _, field := range stmt.Schema.PrimaryFields { | |
| 67 | - primaryKeys = append(primaryKeys, clause.Column{Name: field.DBName}) | |
| 68 | - } | |
| 69 | - | |
| 70 | - values = append(values, primaryKeys) | |
| 71 | - } | |
| 72 | - | |
| 73 | - for _, idx := range stmt.Schema.ParseIndexes() { | |
| 74 | - if m.CreateIndexAfterCreateTable { | |
| 75 | - defer func(value interface{}, name string) { | |
| 76 | - if err == nil { | |
| 77 | - err = tx.Migrator().CreateIndex(value, name) | |
| 78 | - } | |
| 79 | - }(value, idx.Name) | |
| 80 | - } else { | |
| 81 | - if idx.Class != "" { | |
| 82 | - createTableSQL += idx.Class + " " | |
| 83 | - } | |
| 84 | - createTableSQL += "INDEX ? ?" | |
| 85 | - | |
| 86 | - if idx.Comment != "" { | |
| 87 | - createTableSQL += fmt.Sprintf(" COMMENT '%s'", idx.Comment) | |
| 88 | - } | |
| 89 | - | |
| 90 | - if idx.Option != "" { | |
| 91 | - createTableSQL += " " + idx.Option | |
| 92 | - } | |
| 93 | - | |
| 94 | - createTableSQL += "," | |
| 95 | - values = append(values, clause.Column{Name: idx.Name}, tx.Migrator().(migrator.BuildIndexOptionsInterface).BuildIndexOptions(idx.Fields, stmt)) | |
| 96 | - } | |
| 97 | - } | |
| 98 | - | |
| 99 | - if !m.DB.DisableForeignKeyConstraintWhenMigrating && !m.DB.IgnoreRelationshipsWhenMigrating { | |
| 100 | - for _, rel := range stmt.Schema.Relationships.Relations { | |
| 101 | - if rel.Field.IgnoreMigration { | |
| 102 | - continue | |
| 103 | - } | |
| 104 | - if constraint := rel.ParseConstraint(); constraint != nil { | |
| 105 | - if constraint.Schema == stmt.Schema { | |
| 106 | - sql, vars := constraint.Build() | |
| 107 | - createTableSQL += sql + "," | |
| 108 | - values = append(values, vars...) | |
| 109 | - } | |
| 110 | - } | |
| 111 | - } | |
| 112 | - } | |
| 113 | - | |
| 114 | - for _, uni := range stmt.Schema.ParseUniqueConstraints() { | |
| 115 | - createTableSQL += "CONSTRAINT ? UNIQUE (?)," | |
| 116 | - values = append(values, clause.Column{Name: uni.Name}, clause.Expr{SQL: stmt.Quote(uni.Field.DBName)}) | |
| 117 | - } | |
| 118 | - | |
| 119 | - for _, chk := range stmt.Schema.ParseCheckConstraints() { | |
| 120 | - createTableSQL += "CONSTRAINT ? CHECK (?)," | |
| 121 | - values = append(values, clause.Column{Name: chk.Name}, clause.Expr{SQL: chk.Constraint}) | |
| 122 | - } | |
| 123 | - | |
| 124 | - createTableSQL = strings.TrimSuffix(createTableSQL, ",") | |
| 125 | - | |
| 126 | - createTableSQL += ")" | |
| 127 | - | |
| 128 | - if tableOption, ok := m.DB.Get("gorm:table_options"); ok { | |
| 129 | - createTableSQL += fmt.Sprint(tableOption) | |
| 130 | - } | |
| 131 | - | |
| 132 | - err = tx.Exec(createTableSQL, values...).Error | |
| 133 | - return err | |
| 134 | - }); err != nil { | |
| 135 | - return err | |
| 136 | - } | |
| 137 | - return nil | |
| 138 | -} | |
| 139 | - | |
| 140 | -// Insert 插入 | |
| 141 | -func Insert(db *gorm.DB, value any, tableName string) *gorm.DB { | |
| 142 | - stmt := db.Session(&gorm.Session{DryRun: true}).Create(value).Statement | |
| 143 | - stmtSQL := stmt.SQL.String() | |
| 144 | - sql := strings.Replace(stmtSQL, MYSQL_TABLE_TEMPLATE, tableName, -1) | |
| 145 | - return db.Exec(sql, stmt.Vars...) | |
| 146 | -} | |
| 147 | - | |
| 148 | -// Update 更新 | |
| 149 | -func Update(db *gorm.DB, value any, tableName string) *gorm.DB { | |
| 150 | - stmt := db.Session(&gorm.Session{DryRun: true}).Updates(value).Statement | |
| 151 | - stmtSQL := stmt.SQL.String() | |
| 152 | - sql := strings.Replace(stmtSQL, MYSQL_TABLE_TEMPLATE, tableName, -1) | |
| 153 | - return db.Exec(sql, stmt.Vars...) | |
| 154 | -} | |
| 155 | - | |
| 156 | -// Save 保存 | |
| 157 | -func Save(db *gorm.DB, value any, tableName string) *gorm.DB { | |
| 158 | - stmt := db.Session(&gorm.Session{DryRun: true}).Save(value).Statement | |
| 159 | - stmtSQL := stmt.SQL.String() | |
| 160 | - sql := strings.Replace(stmtSQL, MYSQL_TABLE_TEMPLATE, tableName, -1) | |
| 161 | - return db.Exec(sql, stmt.Vars...) | |
| 162 | -} | |
| 163 | - | |
| 164 | -func First(db *gorm.DB, value any, tableName string) *gorm.DB { | |
| 165 | - stmt := db.Session(&gorm.Session{DryRun: true}).First(value).Statement | |
| 166 | - stmtSQL := stmt.SQL.String() | |
| 167 | - sql := strings.Replace(stmtSQL, MYSQL_TABLE_TEMPLATE, tableName, -1) | |
| 168 | - return db.Raw(sql, stmt.Vars...).Scan(value) | |
| 169 | -} |