130 lines
3.1 KiB
Go
130 lines
3.1 KiB
Go
|
package sms
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"crypto/rand"
|
|||
|
|
"fmt"
|
|||
|
|
"math/big"
|
|||
|
|
|
|||
|
|
"github.com/aliyun/alibaba-cloud-sdk-go/services/dysmsapi"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
|
|||
|
|
"tyapi-server/internal/config"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Service 短信服务接口
|
|||
|
|
type Service interface {
|
|||
|
|
SendVerificationCode(ctx context.Context, phone string, code string) error
|
|||
|
|
GenerateCode(length int) string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AliSMSService 阿里云短信服务实现
|
|||
|
|
type AliSMSService struct {
|
|||
|
|
client *dysmsapi.Client
|
|||
|
|
config config.SMSConfig
|
|||
|
|
logger *zap.Logger
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewAliSMSService 创建阿里云短信服务
|
|||
|
|
func NewAliSMSService(cfg config.SMSConfig, logger *zap.Logger) (*AliSMSService, error) {
|
|||
|
|
client, err := dysmsapi.NewClientWithAccessKey("cn-hangzhou", cfg.AccessKeyID, cfg.AccessKeySecret)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("创建短信客户端失败: %w", err)
|
|||
|
|
}
|
|||
|
|
return &AliSMSService{
|
|||
|
|
client: client,
|
|||
|
|
config: cfg,
|
|||
|
|
logger: logger,
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendVerificationCode 发送验证码
|
|||
|
|
func (s *AliSMSService) SendVerificationCode(ctx context.Context, phone string, code string) error {
|
|||
|
|
request := dysmsapi.CreateSendSmsRequest()
|
|||
|
|
request.Scheme = "https"
|
|||
|
|
request.PhoneNumbers = phone
|
|||
|
|
request.SignName = s.config.SignName
|
|||
|
|
request.TemplateCode = s.config.TemplateCode
|
|||
|
|
request.TemplateParam = fmt.Sprintf(`{"code":"%s"}`, code)
|
|||
|
|
|
|||
|
|
response, err := s.client.SendSms(request)
|
|||
|
|
if err != nil {
|
|||
|
|
s.logger.Error("Failed to send SMS",
|
|||
|
|
zap.String("phone", phone),
|
|||
|
|
zap.Error(err))
|
|||
|
|
return fmt.Errorf("短信发送失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if response.Code != "OK" {
|
|||
|
|
s.logger.Error("SMS send failed",
|
|||
|
|
zap.String("phone", phone),
|
|||
|
|
zap.String("code", response.Code),
|
|||
|
|
zap.String("message", response.Message))
|
|||
|
|
return fmt.Errorf("短信发送失败: %s - %s", response.Code, response.Message)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
s.logger.Info("SMS sent successfully",
|
|||
|
|
zap.String("phone", phone),
|
|||
|
|
zap.String("bizId", response.BizId))
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateCode 生成验证码
|
|||
|
|
func (s *AliSMSService) GenerateCode(length int) string {
|
|||
|
|
if length <= 0 {
|
|||
|
|
length = 6
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 生成指定长度的数字验证码
|
|||
|
|
max := big.NewInt(int64(pow10(length)))
|
|||
|
|
n, _ := rand.Int(rand.Reader, max)
|
|||
|
|
|
|||
|
|
// 格式化为指定长度，不足时前面补0
|
|||
|
|
format := fmt.Sprintf("%%0%dd", length)
|
|||
|
|
return fmt.Sprintf(format, n.Int64())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// pow10 计算10的n次方
|
|||
|
|
func pow10(n int) int {
|
|||
|
|
result := 1
|
|||
|
|
for i := 0; i < n; i++ {
|
|||
|
|
result *= 10
|
|||
|
|
}
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MockSMSService 模拟短信服务（用于开发和测试）
|
|||
|
|
type MockSMSService struct {
|
|||
|
|
logger *zap.Logger
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewMockSMSService 创建模拟短信服务
|
|||
|
|
func NewMockSMSService(logger *zap.Logger) *MockSMSService {
|
|||
|
|
return &MockSMSService{
|
|||
|
|
logger: logger,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendVerificationCode 模拟发送验证码
|
|||
|
|
func (s *MockSMSService) SendVerificationCode(ctx context.Context, phone string, code string) error {
|
|||
|
|
s.logger.Info("Mock SMS sent",
|
|||
|
|
zap.String("phone", phone),
|
|||
|
|
zap.String("code", code))
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateCode 生成验证码
|
|||
|
|
func (s *MockSMSService) GenerateCode(length int) string {
|
|||
|
|
if length <= 0 {
|
|||
|
|
length = 6
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 开发环境使用固定验证码便于测试
|
|||
|
|
result := ""
|
|||
|
|
for i := 0; i < length; i++ {
|
|||
|
|
result += "1"
|
|||
|
|
}
|
|||
|
|
return result
|
|||
|
|
}
|