diff --git a/controllers/user_controller.go b/controllers/user_controller.go index e0f1c97..6357c63 100644 --- a/controllers/user_controller.go +++ b/controllers/user_controller.go @@ -12,8 +12,23 @@ import ( //刷新验证码 POST func RefreshCode(c *gin.Context) { + var ( + err error + user = &struct { + Width int `json:"width" form:"width" binding:"required"` + Height int `json:"height" form:"height" binding:"required"` + }{} + ) + if err = c.ShouldBind(&user); err != nil { + c.JSON(http.StatusOK, &CallBack{ + Code: model.ErrCodeNormal, + Desc: err.Error(), + }) + globals.SugarLogger.Debugf("End API :%s error:%v:", c.Request.URL, err) + return + } globals.SugarLogger.Debugf("RefreshCode, ip: %v", c.ClientIP()) - utils.Captcha(c, 4) + utils.Captcha(c, user.Width, user.Height, 4) } type UserParam struct { diff --git a/utils/utils.go b/utils/utils.go index ae6aae7..d2da471 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -51,9 +51,8 @@ func RandStringBytesWithNumber(n int) string { return string(b) } -func Captcha(c *gin.Context, length ...int) { +func Captcha(c *gin.Context, w, h int, length ...int) { l := captcha.DefaultLen - w, h := 107, 36 if len(length) == 1 { l = length[0] }