52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package payment
|
||
|
||
import (
|
||
"crypto/rand"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"strconv"
|
||
"sync/atomic"
|
||
"time"
|
||
)
|
||
|
||
const (
|
||
// 订单号前缀
|
||
AlipayPrefix = "ALI" // 支付宝订单前缀
|
||
WechatPrefix = "WX" // 微信支付订单前缀
|
||
ApplePrefix = "APP" // 苹果支付订单前缀
|
||
)
|
||
|
||
const (
|
||
RefundPrefix = "REF" // 退款订单前缀
|
||
)
|
||
|
||
// 全局原子计数器
|
||
var orderCounter uint32 = 0
|
||
|
||
// GenerateOrderNo 生成统一格式的订单号
|
||
// prefix: 订单前缀（ALI/WX/APP）
|
||
// 返回格式：前缀 + 时间戳(10位) + 计数器(6位) + 随机数(6位)
|
||
// 例如：ALI202403151234567890123456
|
||
func GenerateOrderNo(prefix string) string {
|
||
// 获取当前时间戳（秒级）
|
||
timestamp := time.Now().Unix()
|
||
timeStr := strconv.FormatInt(timestamp, 10)
|
||
|
||
// 原子递增计数器
|
||
counter := atomic.AddUint32(&orderCounter, 1)
|
||
|
||
// 生成4字节真随机数
|
||
randomBytes := make([]byte, 4)
|
||
_, err := rand.Read(randomBytes)
|
||
if err != nil {
|
||
// 如果随机数生成失败，回退到使用时间纳秒数据
|
||
randomBytes = []byte(strconv.FormatInt(time.Now().UnixNano()%1000000, 16))
|
||
}
|
||
randomHex := hex.EncodeToString(randomBytes)
|
||
|
||
// 组合所有部分: 前缀 + 时间戳 + 计数器 + 随机数
|
||
orderNo := fmt.Sprintf("%s_%s%06x%s", prefix, timeStr, counter%0xFFFFFF, randomHex[:6])
|
||
|
||
return orderNo
|
||
}
|