fix code

internal/demo/register/ip_limit.go

@@ -44,17 +44,14 @@ func QueryIPRegister(c *gin.Context) {
 	resp.UserIDList = userIDList
 	ipLimit, err := imdb.QueryIPLimits(req.IP)
 	if err != nil {
-		if gorm.IsRecordNotFoundError(err) {
-			resp.Status = 0
-		} else {
-			log.NewError(req.OperationID, "QueryIPLimits failed", req.IP, err.Error())
-			c.JSON(http.StatusInternalServerError, gin.H{"errCode": constant.ErrDB.ErrCode, "errMsg": "QueryIPLimits error!"})
-			return
-		}
+		log.NewError(req.OperationID, "QueryIPLimits failed", req.IP, err.Error())
 	} else {
-		if ipLimit.Ip != "" {
-			resp.Status = 1
+		if ipLimit != nil {
+			if ipLimit.Ip != "" {
+				resp.Status = 1
+			}
 		}
+
 	}
 	log.NewInfo(req.OperationID, utils.GetSelfFuncName(), "resp:", resp)
 	c.JSON(http.StatusOK, gin.H{"errCode": 0, "errMsg": "", "data": resp})

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).Find(&ip).Error
+	err := db.DB.MysqlDB.DefaultGormDB().Model(&db.IpLimit{}).Where("ip=?", ip).First(&ip).Error
 	return &ipLimit, err
 }