fix code

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

@@ -12,13 +12,14 @@ func GetRegister(account, areaCode, userID string) (*db.Register, error) {
 		userID, "", account, account, areaCode).Take(&r).Error
 }
 
-func SetPassword(account, password, ex, userID, areaCode string) error {
+func SetPassword(account, password, ex, userID, areaCode, ip string) error {
 	r := db.Register{
-		Account:  account,
-		Password: password,
-		Ex:       ex,
-		UserID:   userID,
-		AreaCode: areaCode,
+		Account:    account,
+		Password:   password,
+		Ex:         ex,
+		UserID:     userID,
+		RegisterIP: ip,
+		AreaCode:   areaCode,
 	}
 	return db.DB.MysqlDB.DefaultGormDB().Table("registers").Create(&r).Error
 }

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

@@ -34,7 +34,7 @@ func IsLimitUserLoginIp(userID string, LoginIp string) (bool, error) {
 
 func QueryIPLimits(ip string) (*db.IpLimit, error) {
 	var ipLimit db.IpLimit
-	err := db.DB.MysqlDB.DefaultGormDB().Model(&db.IpLimit{}).Where("ip=?", ip).First(&ip).Error
+	err := db.DB.MysqlDB.DefaultGormDB().Model(&db.IpLimit{}).Where("ip=?", ip).Find(&ip).Error
 	return &ipLimit, err
 }