47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
package captcha
|
||
|
||
import (
|
||
"context"
|
||
"picgo/corelib"
|
||
"time"
|
||
)
|
||
|
||
// Store 存储验证码接口
|
||
type Store interface {
|
||
// SetCode cid 当前id
|
||
// oid 旧id
|
||
// code 验证码
|
||
// expired 过期时间（毫秒）
|
||
// return nil/error
|
||
SetCode(cid string, oid string, code string, expired int)
|
||
// GetCode cid 当前id
|
||
// return code
|
||
GetCode(cid string) string
|
||
}
|
||
|
||
// SetCode 验证码保存到redis
|
||
func SetCode(cid string, code string, expired int64) (err error) {
|
||
ctx := context.Background()
|
||
cid = genRedisKey(cid)
|
||
if err = corelib.RdbClient.Set(ctx, cid, code, time.Duration(expired)*time.Second).Err(); err != nil {
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
// GetCode 从redis中查询验证码，查询之后删除
|
||
func GetCode(cid string) (code string, err error) {
|
||
ctx := context.Background()
|
||
cid = genRedisKey(cid)
|
||
if code, err = corelib.RdbClient.Get(ctx, cid).Result(); err != nil {
|
||
return
|
||
}
|
||
//rdb.Del(ctx, cid)
|
||
return
|
||
}
|
||
|
||
// captcha缓存key拼接
|
||
func genRedisKey(id string) (res string) {
|
||
return corelib.CaptchaKey + id
|
||
}
|