315 lines
7.2 KiB
Go
315 lines
7.2 KiB
Go
|
package utils
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"crypto/md5"
|
|||
|
|
"crypto/rand"
|
|||
|
|
"encoding/hex"
|
|||
|
|
"fmt"
|
|||
|
|
"math"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"golang.org/x/crypto/bcrypt"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// HashPassword 密码加密
|
|||
|
|
func HashPassword(password string) (string, error) {
|
|||
|
|
bytes, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
|
|||
|
|
return string(bytes), err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CheckPassword 验证密码
|
|||
|
|
func CheckPassword(password, hash string) bool {
|
|||
|
|
err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
|
|||
|
|
return err == nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MD5 MD5加密
|
|||
|
|
func MD5(str string) string {
|
|||
|
|
h := md5.New()
|
|||
|
|
h.Write([]byte(str))
|
|||
|
|
return hex.EncodeToString(h.Sum(nil))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateRandomString 生成随机字符串
|
|||
|
|
func GenerateRandomString(length int) string {
|
|||
|
|
const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
|||
|
|
b := make([]byte, length)
|
|||
|
|
for i := range b {
|
|||
|
|
randomByte := make([]byte, 1)
|
|||
|
|
rand.Read(randomByte)
|
|||
|
|
b[i] = charset[randomByte[0]%byte(len(charset))]
|
|||
|
|
}
|
|||
|
|
return string(b)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateOrderNo 生成订单号
|
|||
|
|
func GenerateOrderNo() string {
|
|||
|
|
now := time.Now()
|
|||
|
|
timestamp := now.Format("20060102150405")
|
|||
|
|
random := GenerateRandomString(6)
|
|||
|
|
return timestamp + strings.ToUpper(random)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateWechatOutTradeNo 生成微信支付商户订单号
|
|||
|
|
// 格式：WX + 时间戳(精确到毫秒) + 8位随机字符串
|
|||
|
|
// 确保每次支付请求都有唯一的订单号，避免重复支付问题
|
|||
|
|
func GenerateWechatOutTradeNo() string {
|
|||
|
|
now := time.Now()
|
|||
|
|
// 使用纳秒时间戳确保唯一性
|
|||
|
|
timestamp := now.Format("20060102150405") + fmt.Sprintf("%03d", now.Nanosecond()/1000000)
|
|||
|
|
random := GenerateRandomString(8)
|
|||
|
|
return "WX" + timestamp + strings.ToUpper(random)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateRefundNo 生成退款单号
|
|||
|
|
func GenerateRefundNo() string {
|
|||
|
|
now := time.Now()
|
|||
|
|
timestamp := now.Format("20060102150405")
|
|||
|
|
random := GenerateRandomString(6)
|
|||
|
|
return "RF" + timestamp + strings.ToUpper(random)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateWechatOutRefundNo 生成微信退款单号
|
|||
|
|
// 格式：WXR + 时间戳(精确到毫秒) + 8位随机字符串
|
|||
|
|
func GenerateWechatOutRefundNo() string {
|
|||
|
|
now := time.Now()
|
|||
|
|
// 使用纳秒时间戳确保唯一性
|
|||
|
|
timestamp := now.Format("20060102150405") + fmt.Sprintf("%03d", now.Nanosecond()/1000000)
|
|||
|
|
random := GenerateRandomString(8)
|
|||
|
|
return "WXR" + timestamp + strings.ToUpper(random)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// StringToUint 字符串转uint
|
|||
|
|
func StringToUint(str string) uint {
|
|||
|
|
if str == "" {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
num, err := strconv.ParseUint(str, 10, 32)
|
|||
|
|
if err != nil {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
return uint(num)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// StringToInt 字符串转int
|
|||
|
|
func StringToInt(str string) int {
|
|||
|
|
if str == "" {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
num, err := strconv.Atoi(str)
|
|||
|
|
if err != nil {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
return num
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// StringToFloat64 字符串转float64
|
|||
|
|
func StringToFloat64(str string) float64 {
|
|||
|
|
if str == "" {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
num, err := strconv.ParseFloat(str, 64)
|
|||
|
|
if err != nil {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
return num
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UintToString uint转字符串
|
|||
|
|
func UintToString(num uint) string {
|
|||
|
|
return strconv.FormatUint(uint64(num), 10)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IntToString 整数转字符串
|
|||
|
|
func IntToString(num int) string {
|
|||
|
|
return strconv.Itoa(num)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// FloatToString 浮点数转字符串
|
|||
|
|
func FloatToString(num float64) string {
|
|||
|
|
return fmt.Sprintf("%.0f", num)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BoolToInt bool转int
|
|||
|
|
func BoolToInt(b bool) int {
|
|||
|
|
if b {
|
|||
|
|
return 1
|
|||
|
|
}
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// FormatPrice 格式化价格（分转元）
|
|||
|
|
func FormatPrice(price uint) string {
|
|||
|
|
yuan := float64(price) / 100
|
|||
|
|
return fmt.Sprintf("%.2f", yuan)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ParsePrice 解析价格（元转分）
|
|||
|
|
func ParsePrice(priceStr string) uint {
|
|||
|
|
price, err := strconv.ParseFloat(priceStr, 64)
|
|||
|
|
if err != nil {
|
|||
|
|
return 0
|
|||
|
|
}
|
|||
|
|
return uint(math.Round(price * 100))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// InArray 检查元素是否在数组中
|
|||
|
|
func InArray(needle interface{}, haystack []interface{}) bool {
|
|||
|
|
for _, item := range haystack {
|
|||
|
|
if item == needle {
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// InStringArray 检查字符串是否在字符串数组中
|
|||
|
|
func InStringArray(needle string, haystack []string) bool {
|
|||
|
|
for _, item := range haystack {
|
|||
|
|
if item == needle {
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// InUintArray 检查uint是否在uint数组中
|
|||
|
|
func InUintArray(needle uint, haystack []uint) bool {
|
|||
|
|
for _, item := range haystack {
|
|||
|
|
if item == needle {
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RemoveDuplicateStrings 去除字符串数组重复元素
|
|||
|
|
func RemoveDuplicateStrings(slice []string) []string {
|
|||
|
|
keys := make(map[string]bool)
|
|||
|
|
result := []string{}
|
|||
|
|
for _, item := range slice {
|
|||
|
|
if !keys[item] {
|
|||
|
|
keys[item] = true
|
|||
|
|
result = append(result, item)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RemoveDuplicateUints 去除uint数组重复元素
|
|||
|
|
func RemoveDuplicateUints(slice []uint) []uint {
|
|||
|
|
keys := make(map[uint]bool)
|
|||
|
|
result := []uint{}
|
|||
|
|
for _, item := range slice {
|
|||
|
|
if !keys[item] {
|
|||
|
|
keys[item] = true
|
|||
|
|
result = append(result, item)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetClientIP 获取客户端IP
|
|||
|
|
func GetClientIP(remoteAddr, xForwardedFor, xRealIP string) string {
|
|||
|
|
if xRealIP != "" {
|
|||
|
|
return xRealIP
|
|||
|
|
}
|
|||
|
|
if xForwardedFor != "" {
|
|||
|
|
ips := strings.Split(xForwardedFor, ",")
|
|||
|
|
if len(ips) > 0 {
|
|||
|
|
return strings.TrimSpace(ips[0])
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if remoteAddr != "" {
|
|||
|
|
ip := strings.Split(remoteAddr, ":")[0]
|
|||
|
|
return ip
|
|||
|
|
}
|
|||
|
|
return "unknown"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Pagination 分页计算
|
|||
|
|
type Pagination struct {
|
|||
|
|
Page int `json:"page"`
|
|||
|
|
PageSize int `json:"page_size"`
|
|||
|
|
Total int64 `json:"total"`
|
|||
|
|
Offset int `json:"offset"`
|
|||
|
|
Limit int `json:"limit"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewPagination 创建分页对象
|
|||
|
|
func NewPagination(page, pageSize int) *Pagination {
|
|||
|
|
if page <= 0 {
|
|||
|
|
page = 1
|
|||
|
|
}
|
|||
|
|
if pageSize <= 0 {
|
|||
|
|
pageSize = 10
|
|||
|
|
}
|
|||
|
|
if pageSize > 100 {
|
|||
|
|
pageSize = 100
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
offset := (page - 1) * pageSize
|
|||
|
|
return &Pagination{
|
|||
|
|
Page: page,
|
|||
|
|
PageSize: pageSize,
|
|||
|
|
Offset: offset,
|
|||
|
|
Limit: pageSize,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsValidEmail 验证邮箱格式
|
|||
|
|
func IsValidEmail(email string) bool {
|
|||
|
|
// 简单的邮箱格式验证
|
|||
|
|
return strings.Contains(email, "@") && strings.Contains(email, ".")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsValidPhone 验证手机号格式
|
|||
|
|
func IsValidPhone(phone string) bool {
|
|||
|
|
// 简单的手机号格式验证（中国大陆）
|
|||
|
|
if len(phone) != 11 {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
return strings.HasPrefix(phone, "1")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TruncateString 截断字符串
|
|||
|
|
func TruncateString(str string, length int) string {
|
|||
|
|
if len(str) <= length {
|
|||
|
|
return str
|
|||
|
|
}
|
|||
|
|
return str[:length] + "..."
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetTimeRange 获取时间范围
|
|||
|
|
func GetTimeRange(rangeType string) (time.Time, time.Time) {
|
|||
|
|
now := time.Now()
|
|||
|
|
var start, end time.Time
|
|||
|
|
|
|||
|
|
switch rangeType {
|
|||
|
|
case "today":
|
|||
|
|
start = time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
|
|||
|
|
end = start.Add(24 * time.Hour).Add(-time.Nanosecond)
|
|||
|
|
case "yesterday":
|
|||
|
|
yesterday := now.AddDate(0, 0, -1)
|
|||
|
|
start = time.Date(yesterday.Year(), yesterday.Month(), yesterday.Day(), 0, 0, 0, 0, yesterday.Location())
|
|||
|
|
end = start.Add(24 * time.Hour).Add(-time.Nanosecond)
|
|||
|
|
case "week":
|
|||
|
|
weekday := int(now.Weekday())
|
|||
|
|
if weekday == 0 {
|
|||
|
|
weekday = 7
|
|||
|
|
}
|
|||
|
|
start = now.AddDate(0, 0, -weekday+1)
|
|||
|
|
start = time.Date(start.Year(), start.Month(), start.Day(), 0, 0, 0, 0, start.Location())
|
|||
|
|
end = start.AddDate(0, 0, 7).Add(-time.Nanosecond)
|
|||
|
|
case "month":
|
|||
|
|
start = time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location())
|
|||
|
|
end = start.AddDate(0, 1, 0).Add(-time.Nanosecond)
|
|||
|
|
default:
|
|||
|
|
start = now
|
|||
|
|
end = now
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return start, end
|
|||
|
|
}
|