diff --git a/dameng_test.go b/dameng_test.go index 9222d72..a5b139a 100644 --- a/dameng_test.go +++ b/dameng_test.go @@ -99,6 +99,12 @@ func TestGormConnExample(t *testing.T) { } else { t.Logf("迁移表结构成功！") } + // 测试重新迁移 schema + if err = db.AutoMigrate(&Product{}); err != nil { + t.Errorf("迁移表结构失败：%v", err) + } else { + t.Logf("迁移表结构成功！") + } // Create data := Product{Code: "D42", Price: 100, Remark1: "VARCHAR", Remark2: "CLOB"} diff --git a/migrator_test.go b/migrator_test.go new file mode 100644 index 0000000..380a21c --- /dev/null +++ b/migrator_test.go @@ -0,0 +1,196 @@ +package dameng + +import ( + "encoding/json" + "reflect" + "testing" + "time" + + "gorm.io/gorm" +) + +func TestMigrator_AutoMigrate(t *testing.T) { + options := map[string]string{ + "schema": dmSchema, + "appName": "dm_TestMigrator", + "connectTimeout": "30000", + } + db, err := gorm.Open(New(Config{ + DriverName: DriverName, + DSN: BuildUrl(dmUsername, dmPassword, dmHost, dmPort, options), + VarcharSizeIsCharLength: true, + })) + if err != nil { + t.Fatalf("initialize db session based on dialector got error: %v", err) + } + + type args struct { + drop bool + models []interface{} + comments []string + } + tests := []struct { + name string + args args + wantErr bool + }{ + {name: "TestTableUser-create", args: args{models: []interface{}{TestTableUser{}}, comments: []string{"用户信息表"}}}, + {name: "TestTableUser-alter", args: args{models: []interface{}{TestTableUser{}}, comments: []string{"用户信息表"}, drop: true}}, + {name: "TestTableUserNoComments", args: args{models: []interface{}{TestTableUserNoComments{}}, comments: []string{"用户信息表"}}}, + {name: "TestTableUserAddColumn", args: args{models: []interface{}{TestTableUserAddColumn{}}, comments: []string{"用户信息表"}}}, + {name: "TestTableUserMigrateColumn", args: args{models: []interface{}{TestTableUserMigrateColumn{}}, comments: []string{"用户信息表"}, drop: true}}, + {name: "TestTableColumnTypeModel-create", args: args{models: []interface{}{testTableColumnTypeModel{}}, comments: []string{"测试字段类型表"}}}, + {name: "TestTableColumnTypeModel-alter", args: args{models: []interface{}{testTableColumnTypeModel{}}, comments: []string{"测试字段类型表"}, drop: true}}, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if len(tt.args.models) == 0 { + t.Fatal("models is nil") + } + migrator := db.Set("gorm:table_comments", tt.args.comments).Migrator() + + if tt.args.drop { + for _, model := range tt.args.models { + if !migrator.HasTable(model) { + continue + } + if err = migrator.DropTable(model); err != nil { + t.Fatalf("DropTable() got error = %v", err) + } + } + } + + if err = migrator.AutoMigrate(tt.args.models...); (err != nil) != tt.wantErr { + t.Errorf("AutoMigrate() error = %v, wantErr %v", err, tt.wantErr) + } else if err == nil { + t.Log("AutoMigrate() success!") + } + + if tt.name == "TestTableUserMigrateColumn" { + wantUser := TestTableUserMigrateColumn{ + TestTableUser: TestTableUser{ + UID: "U0", + Name: "someone", + Account: "guest", + Password: "MAkOvrJ8JV", + Email: "", + PhoneNumber: "+8618888888888", + Sex: "1", + UserType: 1, + Enabled: true, + Remark: "Ahmad", + }, + AddNewColumn: "AddNewColumnValue", + CommentSingleQuote: "CommentSingleQuoteValue", + } + + result := db.Create(&wantUser) + if err = result.Error; err != nil { + t.Fatal(err) + } + + var gotUser TestTableUserMigrateColumn + result.Where(&TestTableUser{UID: "U0"}).Find(&gotUser) + if err = result.Error; err != nil { + t.Fatal(err) + } + gotUserBytes, _ := json.Marshal(gotUser) + t.Logf("gotUser Result: %s", gotUserBytes) + if !reflect.DeepEqual(gotUser, wantUser) { + wantUserBytes, _ := json.Marshal(wantUser) + t.Errorf("wantUser Info: %s", wantUserBytes) + } + } + }) + } +} + +// TestTableUser 测试用户信息表模型 +type TestTableUser struct { + ID uint64 `gorm:"column:id;size:64;not null;autoIncrement:true;autoIncrementIncrement:1;primaryKey;comment:自增 ID" json:"id"` + UID string `gorm:"column:uid;type:varchar(50);comment:用户身份标识" json:"uid"` + Name string `gorm:"column:name;size:50;comment:用户姓名" json:"name"` + + Account string `gorm:"column:account;type:varchar(50);comment:登录账号" json:"account"` + Password string `gorm:"column:password;type:varchar(512);comment:登录密码（密文）" json:"password"` + + Email string `gorm:"column:email;type:varchar(128);comment:邮箱地址" json:"email"` + PhoneNumber string `gorm:"column:phone_number;type:varchar(15);comment:E.164" json:"phoneNumber"` + + Sex string `gorm:"column:sex;type:char(1);comment:性别" json:"sex"` + Birthday *time.Time `gorm:"column:birthday;->:false;<-:create;comment:生日" json:"birthday,omitempty"` + + UserType int `gorm:"column:user_type;size:8;comment:用户类型" json:"userType"` + + Enabled bool `gorm:"column:enabled;comment:是否可用" json:"enabled"` + Remark string `gorm:"column:remark;size:1024;comment:备注信息" json:"remark"` +} + +func (TestTableUser) TableName() string { + return "test_user" +} + +type TestTableUserNoComments struct { + ID uint64 `gorm:"column:id;size:64;not null;autoIncrement:true;autoIncrementIncrement:1;primaryKey" json:"id"` + UID string `gorm:"column:name;type:varchar(50)" json:"uid"` + Name string `gorm:"column:name;size:50" json:"name"` + + Account string `gorm:"column:account;type:varchar(50)" json:"account"` + Password string `gorm:"column:password;type:varchar(512)" json:"password"` + + Email string `gorm:"column:email;type:varchar(128)" json:"email"` + PhoneNumber string `gorm:"column:phone_number;type:varchar(15)" json:"phoneNumber"` + + Sex string `gorm:"column:sex;type:char(1)" json:"sex"` + Birthday time.Time `gorm:"column:birthday" json:"birthday"` + + UserType int `gorm:"column:user_type;size:8" json:"userType"` + + Enabled bool `gorm:"column:enabled" json:"enabled"` + Remark string `gorm:"column:remark;size:1024" json:"remark"` +} + +func (TestTableUserNoComments) TableName() string { + return "test_user" +} + +type TestTableUserAddColumn struct { + TestTableUser + + AddNewColumn string `gorm:"column:add_new_column;type:varchar(100);comment:添加新字段"` +} + +func (TestTableUserAddColumn) TableName() string { + return "test_user" +} + +type TestTableUserMigrateColumn struct { + TestTableUser + + AddNewColumn string `gorm:"column:add_new_column;type:varchar(100);comment:测试添加新字段"` + CommentSingleQuote string `gorm:"column:comment_single_quote;comment:注释中存在单引号'[']'"` +} + +func (TestTableUserMigrateColumn) TableName() string { + return "test_user" +} + +type testTableColumnTypeModel struct { + ID int64 `gorm:"column:id;size:64;not null;autoIncrement:true;autoIncrementIncrement:1;primaryKey"` + Name string `gorm:"column:name;size:50"` + Age uint8 `gorm:"column:age;size:8"` + + Avatar []byte `gorm:"column:avatar;"` + + Balance float64 `gorm:"column:balance;type:decimal(18, 2)"` + Remark string `gorm:"column:remark;size:-1"` + Enabled bool `gorm:"column:enabled;"` + + CreatedAt time.Time + UpdatedAt time.Time + DeletedAt gorm.DeletedAt +} + +func (t testTableColumnTypeModel) TableName() string { + return "test_table_column_type" +}