mirror of https://github.com/rocboss/paopao-ce
Michael Li
2 years ago
commit
dd9cac1b01
@ -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