demo merge into project

pkg/common/db/mysql.go

@@ -182,6 +182,16 @@ func initMysqlDB() {
 		panic(err.Error())
 	}
 
+	sqlTable = "CREATE TABLE IF NOT EXISTS `register` (" +
+		" `account` varchar(255) NOT NULL," +
+		" `password` varchar(255) NOT NULL," +
+		" PRIMARY KEY (`account`)  USING BTREE" +
+		" ) ENGINE=InnoDB DEFAULT CHARSET=latin1 ROW_FORMAT=DYNAMIC;"
+	err = db.Exec(sqlTable).Error
+	if err != nil {
+		panic(err.Error())
+	}
+
 }
 
 func (m *mysqlDB) DefaultGormDB() (*gorm.DB, error) {

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

@@ -0,0 +1,64 @@
+package im_mysql_model
+
+import (
+	"Open_IM/pkg/common/db"
+	_ "github.com/jinzhu/gorm"
+)
+
+type GetRegisterParams struct {
+	Account string `json:"account"`
+}
+type Register struct {
+	Account  string `gorm:"column:account"`
+	Password string `gorm:"column:password"`
+}
+
+func GetRegister(params *GetRegisterParams) (Register, error, int64) {
+	var r Register
+	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
+	if err != nil {
+		return r, err, 0
+	}
+	result := dbConn.
+		Where(&Register{Account: params.Account}).
+		Find(&r)
+	return r, result.Error, result.RowsAffected
+}
+
+type SetPasswordParams struct {
+	Account  string `json:"account"`
+	Password string `json:"password"`
+}
+
+func SetPassword(params *SetPasswordParams) (Register, error) {
+	r := Register{
+		Account:  params.Account,
+		Password: params.Password,
+	}
+	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
+	if err != nil {
+		return r, err
+	}
+
+	result := dbConn.Create(&r)
+
+	return r, result.Error
+}
+
+func Login(params *Register) int64 {
+	var r Register
+	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
+	if err != nil {
+		return 3
+	}
+	result := dbConn.
+		Where(&Register{Account: params.Account}).
+		Find(&r)
+	if result.Error != nil && result.RowsAffected == 0 {
+		return 1
+	}
+	if r.Password != params.Password {
+		return 2
+	}
+	return 0
+}

pkg/common/db/redisModel.go

@@ -88,11 +88,6 @@ func (d *DataBases) GetTokenMapByUidPid(userID, platformID string) (map[string]i
 }
 func (d *DataBases) SetTokenMapByUidPid(userID string, platformID int32, m map[string]int) error {
 	key := uidPidToken + userID + ":" + constant.PlatformIDToName(platformID)
-	//var a []interface{}
-	//for k, v := range m {
-	//	a = append(a, k)
-	//	a = append(a, v)
-	//}
 	_, err := d.Exec("hmset", key, redis.Args{}.Add().AddFlat(m)...)
 	return err
 }

pkg/common/db/redisModel_test.go

@@ -8,8 +8,8 @@ import (
 
 func Test_SetTokenMapByUidPid(t *testing.T) {
 	m := make(map[string]int, 0)
-	m["哈哈"] = 1
-	m["heihei"] = 2
+	m["test1"] = 1
+	m["test2"] = 2
 	m["2332"] = 4
 	_ = DB.SetTokenMapByUidPid("1234", 2, m)