demo modify

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

@@ -5,60 +5,24 @@ import (
 	_ "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
+func GetRegister(account string) (*db.Register, error) {
 	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
 	if err != nil {
-		return r, err, 0
+		return nil, err
 	}
-	result := dbConn.
-		Where(&Register{Account: params.Account}).
-		Find(&r)
-	return r, result.Error, result.RowsAffected
+	var r db.Register
+	return &r, dbConn.Table("registers").Where("account = ?",
+		account).Take(&r).Error
 }
-
-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,
+func SetPassword(account, password string) error {
+	r := db.Register{
+		Account:  account,
+		Password: password,
 	}
 	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
 	if err != nil {
-		return r, err
+		return err
 	}
+	return dbConn.Table("registers").Create(&r).Error
 
-	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
 }