sms_tool

sms_tool/sms_client.go

@@ -0,0 +1,136 @@
+package sms_tool
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
+	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"
+)
+
+type SmsClient struct {
+	SmsConfig
+	Cache  ICacheAdapter
+	client *dysmsapi20170525.Client
+}
+
+type SmsConfig struct {
+	AccessKeyId     string `json:"accessKeyId"`
+	AccessKeySecret string `json:"accessKeySecret"`
+	Endpoint        string `json:"endpoint"`
+}
+
+func NewSmsClient(smsConfig *SmsConfig) (smsClient SmsClient, err error) {
+	config := &openapi.Config{}
+	if smsConfig.AccessKeyId == "" {
+		return SmsClient{}, errors.New("AccessKeyId 不能为空")
+	}
+	if smsConfig.AccessKeySecret == "" {
+		return SmsClient{}, errors.New("AccessKeySecret 不能为空")
+	}
+	if smsConfig.Endpoint == "" {
+		smsConfig.Endpoint = "dysmsapi.aliyuncs.com"
+	}
+	config.AccessKeyId = tea.String(smsConfig.AccessKeyId)
+	config.AccessKeySecret = tea.String(smsConfig.AccessKeySecret)
+	// Endpoint 请参考 https://api.aliyun.com/product/Dysmsapi
+	config.Endpoint = tea.String(smsConfig.Endpoint)
+
+	smsClient.client = &dysmsapi20170525.Client{}
+	smsClient.client, err = dysmsapi20170525.NewClient(config)
+	return smsClient, err
+}
+
+// SendSms 发送短信
+func (c *SmsClient) SendSms(params *SendSmsParams) (res *dysmsapi20170525.SendSmsResponse, err error) {
+	smsParams := &dysmsapi20170525.SendSmsRequest{
+		OutId:                tea.String(params.OutId),
+		OwnerId:              tea.Int64(params.OwnerId),
+		PhoneNumbers:         tea.String(params.PhoneNumbers),
+		ResourceOwnerAccount: tea.String(params.ResourceOwnerAccount),
+		ResourceOwnerId:      tea.Int64(params.ResourceOwnerId),
+		SignName:             tea.String(params.SignName),
+		SmsUpExtendCode:      tea.String(params.SmsUpExtendCode),
+		TemplateCode:         tea.String(params.TemplateCode),
+		TemplateParam:        tea.String(params.TemplateParam),
+	}
+	runtime := &util.RuntimeOptions{}
+	return c.client.SendSmsWithOptions(smsParams, runtime)
+}
+
+// GetCode 获取验证码
+func (c *SmsClient) GetCode(ctx context.Context, key string) (code string, err error) {
+	if c.Cache == nil {
+		return "", errors.New("缓存不能为空")
+	}
+	value, err := c.Cache.Get(ctx, key)
+	if err != nil {
+		return "", err
+	}
+	return value.(string), nil
+}
+
+// SaveCode 保存验证码
+func (c *SmsClient) SaveCode(ctx context.Context, key string, code string, expire time.Duration, frequency time.Duration) (err error) {
+	if c.Cache == nil {
+		return errors.New("缓存不能为空")
+	}
+	frequencyKey := fmt.Sprintf(CodeFrequencyKey, key)
+	if frequency != 0 {
+		// 判断验证码是否频繁
+		frequencyValue, err := c.Cache.Get(ctx, frequencyKey)
+		if err != nil {
+			if errors.Is(err, redis.Nil) {
+				// 缓存中没有数据
+				fmt.Printf("缓存中没有数据")
+			} else {
+				return err
+			}
+		}
+		if frequencyValue != nil && frequencyValue.(string) == CodeFrequencyValue {
+			return errors.New("code发送频繁，请稍后再试")
+		}
+	}
+	//保存code
+	err = c.Cache.Set(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
+}
+
+// DeleteCode 删除验证码
+func (c *SmsClient) DeleteCode(ctx context.Context, key string) (err error) {
+	if c.Cache == nil {
+		return errors.New("缓存不能为空")
+	}
+	err = c.Cache.Del(ctx, key)
+	return err
+}
+
+// VerifyCode 校验验证码
+func (c *SmsClient) VerifyCode(ctx context.Context, key string, verifyCode string) (ok bool, err error) {
+	//获取验证码
+	code, err := c.GetCode(ctx, key)
+	if err != nil {
+		return false, err
+	}
+	//校验验证码
+	if code != verifyCode {
+		return false, errors.New("code不一致")
+	}
+	//删除验证码
+	_ = c.DeleteCode(ctx, key)
+	return true, nil
+}