diff --git a/handler/func_model.go b/handler/func_model.go index 0acbdc5..616c9e6 100644 --- a/handler/func_model.go +++ b/handler/func_model.go @@ -53,15 +53,13 @@ func FindFuncModel(c *gin.Context) { return } c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": funcModels}) - } else if req.ModelID != 0 { + } else { funcModels, err3 := service.FindFuncModelByUserID(userID) if err3 != nil { c.JSON(http.StatusOK, gin.H{"error": err3.Error(), "code": proto.FuncModelSearchFailed, "message": "failed"}) return } c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": funcModels}) - } else { - c.JSON(http.StatusOK, gin.H{"error": "type error", "code": proto.FuncModelSearchFailed, "message": "failed"}) } } else { c.JSON(http.StatusOK, gin.H{"error": err.Error(), "code": proto.ParameterError, "message": "failed"})