feat: Adapt the Gorm code base

Author: ZPascal

db/client.go

@@ -13,16 +13,16 @@ type Client interface {
 	Create(value interface{}) (int64, error)
 	Delete(value interface{}, where ...interface{}) (int64, error)
 	Save(value interface{}) (int64, error)
-	Update(attrs ...interface{}) (int64, error)
+	Update(column string, value interface{}) (int64, error)
 	First(out interface{}, where ...interface{}) error
 	Find(out interface{}, where ...interface{}) error
 	AutoMigrate(values ...interface{}) error
 	Begin() Client
 	Rollback() error
 	Commit() error
 	HasTable(value interface{}) bool
-	AddUniqueIndex(indexName string, columns ...string) error
-	RemoveIndex(indexName string) error
+	AddUniqueIndex(indexName string, columns interface{}) error
+	RemoveIndex(indexName string, columns interface{}) error
 	Model(value interface{}) Client
 	Exec(query string, args ...interface{}) int64
 	Rows(tableName string) (*sql.Rows, error)
@@ -42,16 +42,12 @@ func (c *gormClient) DropColumn(name string) error {
 func (c *gormClient) Close() error {
 	return c.Close()
 }
-func (c *gormClient) AddUniqueIndex(indexName string, columns ...string) error {
-	//TODO Add the interface abstraction layer
-	var value interface{}
-	return c.db.Migrator().CreateIndex(value, indexName)
+func (c *gormClient) AddUniqueIndex(indexName string, columns interface{}) error {
+	return c.db.Migrator().CreateIndex(columns, indexName)
 }
 
-func (c *gormClient) RemoveIndex(indexName string) error {
-	//TODO Add the interface abstraction layer
-	var value interface{}
-	return c.db.Migrator().DropIndex(value, indexName)
+func (c *gormClient) RemoveIndex(indexName string, columns interface{}) error {
+	return c.db.Migrator().DropIndex(columns, indexName)
 }
 
 func (c *gormClient) Model(value interface{}) Client {
@@ -80,9 +76,8 @@ func (c *gormClient) Save(value interface{}) (int64, error) {
 	return newDb.RowsAffected, newDb.Error
 }
 
-func (c *gormClient) Update(attrs ...interface{}) (int64, error) {
-	//TODO Adapt the functionality
-	newDb := c.db.Update(attrs...)
+func (c *gormClient) Update(column string, value interface{}) (int64, error) {
+	newDb := c.db.Update(column, value)
 	return newDb.RowsAffected, newDb.Error
 }
 

db/fakes/fake_client.go

@@ -18,5 +18,9 @@ func (v *V2UpdateRgMigration) Version() int {
 }
 
 func (v *V2UpdateRgMigration) Run(sqlDB *db.SqlDB) error {
-	return sqlDB.Client.Model(&models.RouterGroup{}).AddUniqueIndex("idx_rg_name", "name")
+	type routerGroup struct {
+		models.Model
+		Name string `gorm:"index:idx_rg_name,unique" json:"name"`
+	}
+	return sqlDB.Client.AddUniqueIndex("idx_rg_name", &routerGroup{})
 }

migration/V2_update_rg_migration.go

@@ -18,10 +18,11 @@ func (v *V4AddRgUniqIdxTCPRoute) Version() int {
 }
 
 func (v *V4AddRgUniqIdxTCPRoute) Run(sqlDB *db.SqlDB) error {
-	err := sqlDB.Client.Model(&models.TcpRouteMapping{}).RemoveIndex("idx_tcp_route")
+	err := sqlDB.Client.RemoveIndex("idx_tcp_route", &models.TcpRouteMapping{})
 	if err != nil {
 		return err
 	}
-	err = sqlDB.Client.Model(&models.TcpRouteMapping{}).AddUniqueIndex("idx_tcp_route", "router_group_guid", "host_port", "host_ip", "external_port")
+
+	err = sqlDB.Client.AddUniqueIndex("idx_tcp_route", &models.TcpRouteMapping{})
 	return err
 }

migration/V4_add_rg_uniq_idx_tcp_route_migration.go

@@ -18,7 +18,7 @@ func (v *V5SniHostnameMigration) Version() int {
 }
 
 func (v *V5SniHostnameMigration) Run(sqlDB *db.SqlDB) error {
-	_, err := sqlDB.Client.Model(&models.TcpRouteMapping{}).RemoveIndex("idx_tcp_route")
+	err := sqlDB.Client.RemoveIndex("idx_tcp_route", &models.TcpRouteMapping{})
 	if err != nil {
 		return err
 	}