github_mirrors/hotgo
1
0
Fork 0
mirror of https://github.com/bufanyun/hotgo.git synced 2025-08-26 16:46:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(captcha): 优化验证代码

Browse Source
This commit is contained in:
mh-swift
2024-07-29 13:54:06 +08:00
parent 804d5d5e59
commit 2e322e2606
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
server/internal/library/captcha/captcha.go
View File

@@ -7,12 +7,16 @@ package captcha
import (
"context"
"image/color"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/text/gstr"
"github.com/mojocn/base64Captcha"
"image/color"
)
// store 验证码存储方式
var store = base64Captcha.DefaultMemStore
// Generate 生成验证码
func Generate(ctx context.Context) (id string, base64 string) {
// 字符
@@ -47,7 +51,7 @@ func Generate(ctx context.Context) (id string, base64 string) {
Fonts: []string{"chromohv.ttf"},
}
c := base64Captcha.NewCaptcha(driver.ConvertFonts(), base64Captcha.DefaultMemStore)
c := base64Captcha.NewCaptcha(driver.ConvertFonts(), store)
id, base64, _, err := c.Generate()
if err != nil {
g.Log().Errorf(ctx, "captcha.Generate err:%+v", err)
@@ -60,6 +64,5 @@ func Verify(id, answer string) bool {
if id == "" || answer == "" {
return false
}
c := base64Captcha.NewCaptcha(new(base64Captcha.DriverString), base64Captcha.DefaultMemStore)
return c.Verify(id, gstr.ToLower(answer), true)
return store.Verify(id, gstr.ToLower(answer), true)
}