mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 08:20:25 +08:00
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package captcha
|
||
|
||
import (
|
||
"mayfly-go/pkg/biz"
|
||
"mayfly-go/pkg/rediscli"
|
||
"time"
|
||
|
||
"github.com/mojocn/base64Captcha"
|
||
)
|
||
|
||
var store base64Captcha.Store
|
||
var driver base64Captcha.Driver = base64Captcha.DefaultDriverDigit
|
||
|
||
// 生成验证码
|
||
func Generate() (string, string) {
|
||
if store == nil {
|
||
if rediscli.GetCli() != nil {
|
||
store = new(RedisStore)
|
||
} else {
|
||
store = base64Captcha.DefaultMemStore
|
||
}
|
||
}
|
||
|
||
c := base64Captcha.NewCaptcha(driver, store)
|
||
// 获取
|
||
id, b64s, err := c.Generate()
|
||
biz.ErrIsNilAppendErr(err, "获取验证码错误: %s")
|
||
return id, b64s
|
||
}
|
||
|
||
// 验证验证码
|
||
func Verify(id string, val string) bool {
|
||
if store == nil || id == "" || val == "" {
|
||
return false
|
||
}
|
||
// 同时清理掉这个图片
|
||
return store.Verify(id, val, true)
|
||
}
|
||
|
||
type RedisStore struct {
|
||
}
|
||
|
||
const CAPTCHA = "mayfly:captcha:"
|
||
|
||
// 实现设置captcha的方法
|
||
func (r RedisStore) Set(id string, value string) error {
|
||
//time.Minute*2：有效时间2分钟
|
||
rediscli.Set(CAPTCHA+id, value, time.Minute*2)
|
||
return nil
|
||
}
|
||
|
||
// 实现获取captcha的方法
|
||
func (r RedisStore) Get(id string, clear bool) string {
|
||
key := CAPTCHA + id
|
||
val, err := rediscli.Get(key)
|
||
if err != nil {
|
||
return ""
|
||
}
|
||
if clear {
|
||
//clear为true，验证通过，删除这个验证码
|
||
rediscli.Del(key)
|
||
}
|
||
return val
|
||
}
|
||
|
||
// 实现验证captcha的方法
|
||
func (r RedisStore) Verify(id, answer string, clear bool) bool {
|
||
return r.Get(id, clear) == answer
|
||
}
|