实现个人客户微信认证和短信验证功能
- 添加个人客户微信登录和手机验证码登录接口 - 实现个人客户设备、ICCID、手机号关联管理 - 添加短信发送服务(HTTP 客户端) - 添加微信认证服务(含 mock 实现) - 添加 JWT Token 生成和验证工具 - 创建数据库迁移脚本(personal_customer 关联表) - 修复测试文件中的路由注册参数错误 - 重构 scripts 目录结构(分离独立脚本到子目录) Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
160
pkg/sms/client.go
Normal file
160
pkg/sms/client.go
Normal file
@@ -0,0 +1,160 @@
|
||||
package sms
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/md5"
|
||||
"encoding/hex"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/bytedance/sonic"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
// Client 短信客户端
|
||||
type Client struct {
|
||||
gatewayURL string // 短信网关地址
|
||||
username string // 账号用户名
|
||||
password string // 账号密码
|
||||
signature string // 短信签名
|
||||
timeout time.Duration // 请求超时时间
|
||||
logger *zap.Logger // 日志记录器
|
||||
httpClient HTTPClient // HTTP 客户端接口
|
||||
}
|
||||
|
||||
// HTTPClient HTTP 客户端接口(便于测试)
|
||||
type HTTPClient interface {
|
||||
Post(ctx context.Context, url string, body []byte) ([]byte, error)
|
||||
}
|
||||
|
||||
// NewClient 创建短信客户端
|
||||
func NewClient(gatewayURL, username, password, signature string, timeout time.Duration, logger *zap.Logger, httpClient HTTPClient) *Client {
|
||||
return &Client{
|
||||
gatewayURL: gatewayURL,
|
||||
username: username,
|
||||
password: password,
|
||||
signature: signature,
|
||||
timeout: timeout,
|
||||
logger: logger,
|
||||
httpClient: httpClient,
|
||||
}
|
||||
}
|
||||
|
||||
// SendMessage 发送短信
|
||||
// content: 短信内容(不包含签名,签名会自动添加)
|
||||
// phones: 接收手机号列表
|
||||
func (c *Client) SendMessage(ctx context.Context, content string, phones []string) (*SendResponse, error) {
|
||||
// 生成时间戳(毫秒)
|
||||
timestamp := time.Now().UnixMilli()
|
||||
|
||||
// 计算签名
|
||||
sign := c.calculateSign(timestamp)
|
||||
|
||||
// 构造完整的短信内容(添加签名)
|
||||
fullContent := c.signature + content
|
||||
|
||||
// 构造请求
|
||||
req := &SendRequest{
|
||||
UserName: c.username,
|
||||
Content: fullContent,
|
||||
PhoneList: phones,
|
||||
Timestamp: timestamp,
|
||||
Sign: sign,
|
||||
}
|
||||
|
||||
// 序列化请求
|
||||
reqBody, err := sonic.Marshal(req)
|
||||
if err != nil {
|
||||
c.logger.Error("序列化短信请求失败",
|
||||
zap.Error(err),
|
||||
zap.Strings("phones", phones),
|
||||
)
|
||||
return nil, fmt.Errorf("序列化短信请求失败: %w", err)
|
||||
}
|
||||
|
||||
// 记录请求日志(脱敏处理)
|
||||
c.logger.Info("发送短信请求",
|
||||
zap.Strings("phones", phones),
|
||||
zap.String("content_preview", c.truncateContent(content)),
|
||||
zap.Int64("timestamp", timestamp),
|
||||
)
|
||||
|
||||
// 发送请求
|
||||
url := c.gatewayURL + "/api/sendMessageMass"
|
||||
|
||||
// 创建带超时的上下文
|
||||
reqCtx, cancel := context.WithTimeout(ctx, c.timeout)
|
||||
defer cancel()
|
||||
|
||||
respBody, err := c.httpClient.Post(reqCtx, url, reqBody)
|
||||
if err != nil {
|
||||
c.logger.Error("发送短信请求失败",
|
||||
zap.Error(err),
|
||||
zap.String("url", url),
|
||||
zap.Strings("phones", phones),
|
||||
)
|
||||
return nil, fmt.Errorf("发送短信请求失败: %w", err)
|
||||
}
|
||||
|
||||
// 解析响应
|
||||
var resp SendResponse
|
||||
if err := sonic.Unmarshal(respBody, &resp); err != nil {
|
||||
c.logger.Error("解析短信响应失败",
|
||||
zap.Error(err),
|
||||
zap.ByteString("response_body", respBody),
|
||||
)
|
||||
return nil, fmt.Errorf("解析短信响应失败: %w", err)
|
||||
}
|
||||
|
||||
// 记录响应日志
|
||||
if resp.Code == CodeSuccess {
|
||||
c.logger.Info("短信发送成功",
|
||||
zap.Int64("msg_id", resp.MsgID),
|
||||
zap.Int("sms_count", resp.SMSCount),
|
||||
zap.Strings("phones", phones),
|
||||
)
|
||||
} else {
|
||||
c.logger.Warn("短信发送失败",
|
||||
zap.Int("code", resp.Code),
|
||||
zap.String("message", resp.Message),
|
||||
zap.Strings("phones", phones),
|
||||
)
|
||||
}
|
||||
|
||||
// 检查响应状态
|
||||
if resp.Code != CodeSuccess {
|
||||
return &resp, NewSMSError(resp.Code, resp.Message)
|
||||
}
|
||||
|
||||
return &resp, nil
|
||||
}
|
||||
|
||||
// calculateSign 计算签名
|
||||
// 规则: MD5(userName + timestamp + MD5(password))
|
||||
func (c *Client) calculateSign(timestamp int64) string {
|
||||
// 第一步:计算 MD5(password)
|
||||
passwordMD5 := c.md5Hash(c.password)
|
||||
|
||||
// 第二步:组合字符串
|
||||
combined := fmt.Sprintf("%s%d%s", c.username, timestamp, passwordMD5)
|
||||
|
||||
// 第三步:计算最终签名
|
||||
return c.md5Hash(combined)
|
||||
}
|
||||
|
||||
// md5Hash 计算 MD5 哈希值(小写)
|
||||
func (c *Client) md5Hash(s string) string {
|
||||
h := md5.New()
|
||||
h.Write([]byte(s))
|
||||
return hex.EncodeToString(h.Sum(nil))
|
||||
}
|
||||
|
||||
// truncateContent 截断内容用于日志记录
|
||||
func (c *Client) truncateContent(content string) string {
|
||||
const maxLen = 50
|
||||
runes := []rune(content)
|
||||
if len(runes) > maxLen {
|
||||
return string(runes[:maxLen]) + "..."
|
||||
}
|
||||
return content
|
||||
}
|
||||
Reference in New Issue
Block a user