eta_api/services/captcha_redis.go

package services

import (
	"eta_gn/eta_api/utils"
	"fmt"
	"time"
)

type CaptchaRedis struct{}

func (r CaptchaRedis) Set(id string, value string) (err error) {
	if utils.Rc == nil {
		err = fmt.Errorf("redis config err")
		return
	}
	key := utils.CaptchaCachePrefix + id
	b := utils.Rc.SetNX(key, value, time.Minute*5)
	if !b {
		err = fmt.Errorf("redis setnx err")
	}
	return
}

func (r CaptchaRedis) Get(id string, clear bool) (code string) {
	if utils.Rc == nil {
		return
	}
	key := utils.CaptchaCachePrefix + id
	val, err := utils.Rc.RedisString(key)
	if err != nil {
		return
	}
	code = val
	if clear {
		_ = utils.Rc.Delete(key)
	}
	return
}

func (r CaptchaRedis) Verify(id, answer string, clear bool) bool {
	v := r.Get(id, clear)
	return v == answer
}