public/ssgf_utils
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 28 Wiki Activity

Merge branch 'feature/sms_tool'

Browse Source
This commit is contained in:
yuguojian 2025-06-18 17:41:21 +08:00
commit db27554374
3 changed files with 17 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
sms_tool/cache_model.go
View File

@ -9,4 +9,5 @@ type ICacheAdapter interface {
Get(ctx context.Context, key string) (value interface{}, err error)
Set(ctx context.Context, key string, value interface{}, expire time.Duration) (err error)
Del(ctx context.Context, key string) (err error)
SetNX(ctx context.Context, key string, value interface{}, expire time.Duration) (ok bool, err error)
}

30
sms_tool/sms_client.go
View File

@ -8,7 +8,6 @@ import (
dysmsapi20170525 "github.com/alibabacloud-go/dysmsapi-20170525/v5/client"
util "github.com/alibabacloud-go/tea-utils/v2/service"
"github.com/alibabacloud-go/tea/tea"
"github.com/redis/go-redis/v9"
"time"
)
@ -81,10 +80,7 @@ func (c *SmsClient) SaveCode(ctx context.Context, key string, code string, expir
}
//保存code
err = c.Cache.Set(ctx, key, code, expire)
if err != nil {
return err
}
return nil
}
// DeleteCode 删除验证码
@ -102,34 +98,18 @@ func (c *SmsClient) SaveCodeWithFrequency(ctx context.Context, key string, code
return errors.New("缓存不能为空")
}
frequencyKey := fmt.Sprintf(CodeFrequencyKey, key)
if frequency != 0 {
// 判断验证码是否频繁
frequencyValue, err := c.Cache.Get(ctx, frequencyKey)
if frequency <= 0 {
return errors.New("频率不能小于0")
}
ok, err := c.Cache.SetNX(ctx, frequencyKey, CodeFrequencyValue, frequency)
if err != nil {
if errors.Is(err, redis.Nil) {
// 缓存中没有数据
fmt.Printf("缓存中没有数据")
} else {
return err
}
}
if frequencyValue != nil && frequencyValue.(string) == CodeFrequencyValue {
if !ok {
return errors.New("code发送频繁，请稍后再试")
}
}
//保存code
err = c.SaveCode(ctx, key, code, expire)
if err != nil {
return err
}
if frequency != 0 {
// 保存验证码发送频率
err = c.Cache.Set(ctx, frequencyKey, CodeFrequencyValue, frequency)
if err != nil {
return err
}
}
return nil
}
// VerifyCode 校验验证码

8
sms_tool/sms_client_test.go
View File

@ -51,6 +51,14 @@ func (r *RedisCacheAdapter) Del(ctx context.Context, key string) error {
return r.client.Del(ctx, key).Err()
}
func (r *RedisCacheAdapter) SetNX(ctx context.Context, key string, value interface{}, expire time.Duration) (ok bool, err error) {
data, err := json.Marshal(value)
if err != nil {
return false, err
}
return r.client.SetNX(ctx, key, data, expire).Result()
}
var (
//SMS
accessKeyId = os.Getenv("SMS_ALIBABA_CLOUD_ACCESS_KEY_ID")