mirror of https://github.com/rocboss/paopao-ce
Michael Li
1 year ago
commit
0552008b5b
@ -0,0 +1,34 @@
|
|||||||
|
// Copyright 2023 ROC. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package conf
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/rueian/rueidis"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
_redisClient rueidis.Client
|
||||||
|
_once sync.Once
|
||||||
|
)
|
||||||
|
|
||||||
|
func MustRedisClient() rueidis.Client {
|
||||||
|
_once.Do(func() {
|
||||||
|
client, err := rueidis.NewClient(rueidis.ClientOption{
|
||||||
|
InitAddress: redisSetting.InitAddress,
|
||||||
|
Username: redisSetting.Username,
|
||||||
|
Password: redisSetting.Password,
|
||||||
|
SelectDB: redisSetting.SelectDB,
|
||||||
|
ConnWriteTimeout: redisSetting.ConnWriteTimeout,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("create a redis client failed: %s", err)
|
||||||
|
}
|
||||||
|
_redisClient = client
|
||||||
|
})
|
||||||
|
return _redisClient
|
||||||
|
}
|
||||||
@ -0,0 +1,108 @@
|
|||||||
|
package cache
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
"unsafe"
|
||||||
|
|
||||||
|
"github.com/rocboss/paopao-ce/internal/core"
|
||||||
|
"github.com/rueian/rueidis"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
_ core.RedisCache = (*redisCache)(nil)
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
_pushToSearchJobKey = "JOB_PUSH_TO_SEARCH"
|
||||||
|
_countLoginErrKey = "PaoPaoUserLoginErr:"
|
||||||
|
_imgCaptchaKey = "PaoPaoCaptcha:"
|
||||||
|
_smsCaptchaKey = "PaoPaoSmsCaptcha:"
|
||||||
|
_countWhisperKey = "WhisperTimes:"
|
||||||
|
_rechargeStatusKey = "PaoPaoRecharge:"
|
||||||
|
)
|
||||||
|
|
||||||
|
type redisCache struct {
|
||||||
|
c rueidis.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) SetPushToSearchJob(ctx context.Context) error {
|
||||||
|
return r.c.Do(ctx, r.c.B().Set().
|
||||||
|
Key(_pushToSearchJobKey).Value("1").
|
||||||
|
Nx().ExSeconds(3600).
|
||||||
|
Build()).Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) DelPushToSearchJob(ctx context.Context) error {
|
||||||
|
return r.c.Do(ctx, r.c.B().Del().Key(_pushToSearchJobKey).Build()).Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) SetImgCaptcha(ctx context.Context, id string, value string) error {
|
||||||
|
return r.c.Do(ctx, r.c.B().Set().
|
||||||
|
Key(_imgCaptchaKey+id).Value(value).
|
||||||
|
ExSeconds(300).
|
||||||
|
Build()).Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) GetImgCaptcha(ctx context.Context, id string) (string, error) {
|
||||||
|
res, err := r.c.Do(ctx, r.c.B().Get().Key(_imgCaptchaKey+id).Build()).AsBytes()
|
||||||
|
return unsafe.String(&res[0], len(res)), err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) DelImgCaptcha(ctx context.Context, id string) error {
|
||||||
|
return r.c.Do(ctx, r.c.B().Del().Key(_imgCaptchaKey+id).Build()).Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) GetCountSmsCaptcha(ctx context.Context, phone string) (int64, error) {
|
||||||
|
return r.c.Do(ctx, r.c.B().Get().Key(_smsCaptchaKey+phone).Build()).AsInt64()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) IncrCountSmsCaptcha(ctx context.Context, phone string) (err error) {
|
||||||
|
if err = r.c.Do(ctx, r.c.B().Incr().Key(_smsCaptchaKey+phone).Build()).Error(); err == nil {
|
||||||
|
currentTime := time.Now()
|
||||||
|
endTime := time.Date(currentTime.Year(), currentTime.Month(), currentTime.Day(), 23, 59, 59, 0, currentTime.Location())
|
||||||
|
err = r.c.Do(ctx, r.c.B().Expire().Key(_smsCaptchaKey+phone).Seconds(int64(endTime.Sub(currentTime)/time.Second)).Build()).Error()
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) GetCountLoginErr(ctx context.Context, id int64) (int64, error) {
|
||||||
|
return r.c.Do(ctx, r.c.B().Get().Key(fmt.Sprintf("%s:%d", _countLoginErrKey, id)).Build()).AsInt64()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) DelCountLoginErr(ctx context.Context, id int64) error {
|
||||||
|
return r.c.Do(ctx, r.c.B().Del().Key(fmt.Sprintf("%s:%d", _countLoginErrKey, id)).Build()).Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) IncrCountLoginErr(ctx context.Context, id int64) error {
|
||||||
|
err := r.c.Do(ctx, r.c.B().Incr().Key(fmt.Sprintf("%s:%d", _countLoginErrKey, id)).Build()).Error()
|
||||||
|
if err == nil {
|
||||||
|
err = r.c.Do(ctx, r.c.B().Expire().Key(fmt.Sprintf("%s:%d", _countLoginErrKey, id)).Seconds(3600).Build()).Error()
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) GetCountWhisper(ctx context.Context, uid int64) (int64, error) {
|
||||||
|
return r.c.Do(ctx, r.c.B().Get().Key(fmt.Sprintf("%s:%d", _countWhisperKey, uid)).Build()).AsInt64()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) IncrCountWhisper(ctx context.Context, uid int64) (err error) {
|
||||||
|
key := fmt.Sprintf("%s:%d", _countWhisperKey, uid)
|
||||||
|
if err = r.c.Do(ctx, r.c.B().Incr().Key(key).Build()).Error(); err == nil {
|
||||||
|
currentTime := time.Now()
|
||||||
|
endTime := time.Date(currentTime.Year(), currentTime.Month(), currentTime.Day(), 23, 59, 59, 0, currentTime.Location())
|
||||||
|
err = r.c.Do(ctx, r.c.B().Expire().Key(key).Seconds(int64(endTime.Sub(currentTime)/time.Second)).Build()).Error()
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) SetRechargeStatus(ctx context.Context, tradeNo string) error {
|
||||||
|
return r.c.Do(ctx, r.c.B().Set().
|
||||||
|
Key(_rechargeStatusKey+tradeNo).Value("1").
|
||||||
|
Nx().ExSeconds(5).Build()).Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *redisCache) DelRechargeStatus(ctx context.Context, tradeNo string) error {
|
||||||
|
return r.c.Do(ctx, r.c.B().Del().Key(_rechargeStatusKey+tradeNo).Build()).Error()
|
||||||
|
}
|
||||||
Loading…
Reference in new issue