add set client init

pkg/common/db/model_struct.go

@@ -339,3 +339,11 @@ type RegisterAddFriend struct {
 func (RegisterAddFriend) TableName() string {
 	return "register_add_friend"
 }
+
+type ClientInitConfig struct {
+	DiscoverPageURL string `gorm:"column:discover_page_url;size:64" json:"version"`
+}
+
+func (ClientInitConfig) TableName() string {
+	return "client_init_config"
+}

pkg/common/db/mysql.go

@@ -78,7 +78,8 @@ func initMysqlDB() {
 		&GroupMember{},
 		&GroupRequest{},
 		&User{},
-		&Black{}, &ChatLog{}, &Register{}, &Conversation{}, &AppVersion{}, &Department{}, &BlackList{}, &IpLimit{}, &UserIpLimit{}, &Invitation{}, &RegisterAddFriend{})
+		&Black{}, &ChatLog{}, &Register{}, &Conversation{}, &AppVersion{}, &Department{}, &BlackList{}, &IpLimit{}, &UserIpLimit{}, &Invitation{}, &RegisterAddFriend{},
+		&ClientInitConfig{})
 	db.Set("gorm:table_options", "CHARSET=utf8")
 	db.Set("gorm:table_options", "collation=utf8_unicode_ci")
 
@@ -163,6 +164,12 @@ func initMysqlDB() {
 		fmt.Println("CreateTable Invitation")
 		db.Migrator().CreateTable(&Invitation{})
 	}
+
+	if !db.Migrator().HasTable(&ClientInitConfig{}) {
+		fmt.Println("CreateTable ClientInitConfig")
+		db.Migrator().CreateTable(&ClientInitConfig{})
+	}
+
 	DB.MysqlDB.db = db
 	return
 }

pkg/common/db/mysql_model/im_mysql_model/client_config.go

@@ -0,0 +1,22 @@
+package im_mysql_model
+
+import "Open_IM/pkg/common/db"
+
+func SetClientInitConfig(m map[string]interface{}) error {
+	result := db.DB.MysqlDB.DefaultGormDB().Model(&db.ClientInitConfig{}).Updates(m)
+	if result.Error != nil {
+		return result.Error
+	}
+	if result.RowsAffected == 1 {
+		err := db.DB.MysqlDB.DefaultGormDB().Model(&db.ClientInitConfig{}).Create(m).Error
+		return err
+	}
+
+	return nil
+}
+
+func GetClientInitConfig() (db.ClientInitConfig, error) {
+	var config db.ClientInitConfig
+	err := db.DB.MysqlDB.DefaultGormDB().Model((&db.ClientInitConfig{})).First(&config).Error
+	return config, err
+}