demo merge into project

pkg/common/config/config.go

@@ -155,6 +155,23 @@ type config struct {
 		CallbackUrl     string `yaml:"callbackUrl"`
 		CallBackTimeOut int    `yaml:"callbackTimeOut"`
 	}
+	Demo struct {
+		Port         []int `yaml:"openImDemoPort"`
+		AliSMSVerify struct {
+			AccessKeyID                  string `yaml:"accessKeyId"`
+			AccessKeySecret              string `yaml:"accessKeySecret"`
+			SignName                     string `yaml:"signName"`
+			VerificationCodeTemplateCode string `yaml:"verificationCodeTemplateCode"`
+		}
+		SuperCode string `yaml:"superCode"`
+		Mail      struct {
+			Title                   string `yaml:"title"`
+			SenderMail              string `yaml:"senderMail"`
+			SenderAuthorizationCode string `yaml:"senderAuthorizationCode"`
+			SmtpAddr                string `yaml:"smtpAddr"`
+			SmtpPort                int    `yaml:"smtpPort"`
+		}
+	}
 }
 
 func init() {

pkg/common/constant/error.go

@@ -60,6 +60,17 @@ var (
 	TokenUnknown     = errors.New("couldn't handle this token")
 )
 
+const (
+	NoError          = 0
+	FormattingError  = 10001
+	DatabaseError    = 10002
+	LogicalError     = 10003
+	ServerError      = 10004
+	HttpError        = 10005
+	IoErrot          = 10006
+	IntentionalError = 10007
+)
+
 func (e *ErrInfo) Error() string {
 	return e.ErrMsg
 }

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)