SCHISANDRA/schisandra-cloud-album
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚡ login performance optimization

Browse Source
This commit is contained in:
landaiqing
2024-08-25 00:06:00 +08:00
parent 974a96b6e0
commit 6e47e72514
8 changed files with 149 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
api/captcha_api/captcha_api.go
View File

@@ -30,32 +30,37 @@ func (CaptchaAPI) GenerateRotateCaptcha(c *gin.Context) {
captchaData, err := global.RotateCaptcha.Generate()
if err != nil {
global.LOG.Fatalln(err)
result.FailWithNull(c)
return
}
blockData := captchaData.GetData()
if blockData == nil {
result.FailWithNull(c)
return
}
var masterImageBase64, thumbImageBase64 string
masterImageBase64 = captchaData.GetMasterImage().ToBase64()
thumbImageBase64 = captchaData.GetThumbImage().ToBase64()
masterImageBase64 := captchaData.GetMasterImage().ToBase64()
thumbImageBase64 := captchaData.GetThumbImage().ToBase64()
dotsByte, err := json.Marshal(blockData)
if err != nil {
global.LOG.Fatalln(err)
result.FailWithNull(c)
return
}
key := helper.StringToMD5(string(dotsByte))
err = redis.Set(constant.UserLoginCaptchaRedisKey+key, dotsByte, time.Minute).Err()
if err != nil {
global.LOG.Fatalln(err)
result.FailWithNull(c)
return
}
bt := map[string]interface{}{
result.OkWithData(map[string]interface{}{
"key": key,
"image": masterImageBase64,
"thumb": thumbImageBase64,
}
result.OkWithData(bt, c)
}, c)
}
// CheckRotateData 验证旋转验证码
@@ -67,30 +72,36 @@ func (CaptchaAPI) GenerateRotateCaptcha(c *gin.Context) {
// @Success 200 {string} json
// @Router /api/captcha/rotate/check [post]
func (CaptchaAPI) CheckRotateData(c *gin.Context) {
rotateRequest := dto.RotateCaptchaRequest{}
err := c.ShouldBindJSON(&rotateRequest)
angle := rotateRequest.Angle
key := rotateRequest.Key
if err != nil {
var rotateRequest dto.RotateCaptchaRequest
if err := c.ShouldBindJSON(&rotateRequest); err != nil {
result.FailWithNull(c)
return
}
cacheDataByte, err := redis.Get(constant.UserLoginCaptchaRedisKey + key).Bytes()
if len(cacheDataByte) == 0 || err != nil {
cacheDataByte, err := redis.Get(constant.UserLoginCaptchaRedisKey + rotateRequest.Key).Bytes()
if err != nil || len(cacheDataByte) == 0 {
result.FailWithCodeAndMessage(1011, ginI18n.MustGetMessage(c, "CaptchaExpired"), c)
return
}
var dct *rotate.Block
if err := json.Unmarshal(cacheDataByte, &dct); err != nil {
result.FailWithNull(c)
return
}
sAngle, _ := strconv.ParseFloat(fmt.Sprintf("%v", angle), 64)
sAngle, err := strconv.ParseFloat(fmt.Sprintf("%v", rotateRequest.Angle), 64)
if err != nil {
result.FailWithNull(c)
return
}
chkRet := rotate.CheckAngle(int64(sAngle), int64(dct.Angle), 2)
if chkRet {
result.OkWithMessage("success", c)
return
}
result.FailWithMessage("fail", c)
}