119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package helper
|
||
|
||
import (
|
||
"crypto/md5"
|
||
"fmt"
|
||
"github.com/golang/protobuf/proto"
|
||
"net/url"
|
||
"os"
|
||
"reflect"
|
||
"runtime"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
func Log(name string, msg ...interface{}) {
|
||
_, file, line, _ := runtime.Caller(1)
|
||
timeLayout := "2006-01-01 03:04:05" //转化所需模板
|
||
var datetime = time.Unix(time.Now().Unix(), 0).Format(timeLayout)
|
||
fmt.Println(name, msg, file, line, datetime)
|
||
}
|
||
|
||
func LogErr(err error) {
|
||
if err != nil {
|
||
Log("err", err.Error())
|
||
}
|
||
}
|
||
|
||
func Path(path string) (string, error) {
|
||
if _, err := os.Stat(path); os.IsNotExist(err) {
|
||
if err := os.MkdirAll(path, 0777); err != nil {
|
||
return "", err
|
||
}
|
||
}
|
||
return path, nil
|
||
}
|
||
|
||
func FileExists(filePath string) bool {
|
||
_, err := os.Stat(filePath)
|
||
return err == nil || os.IsExist(err)
|
||
}
|
||
|
||
// TrackTime 运行时间 defer TrackTime()()
|
||
func TrackTime() func() {
|
||
pre := time.Now()
|
||
return func() {
|
||
elapsed := time.Since(pre)
|
||
fmt.Println("elapsed:", elapsed)
|
||
}
|
||
}
|
||
|
||
// Ter 三目运算 Ter(true, 1, 2)
|
||
func Ter[T any](cond bool, a, b T) T {
|
||
if cond {
|
||
return a
|
||
}
|
||
return b
|
||
}
|
||
|
||
// get参转结构体
|
||
func ParseQueryStringToStruct(query string, target interface{}) error {
|
||
// 解析查询字符串
|
||
values, err := url.ParseQuery(query)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 获取目标结构体的反射值
|
||
rv := reflect.ValueOf(target).Elem()
|
||
|
||
// 遍历查询字符串中的每个键值对
|
||
for key, values := range values {
|
||
// 假设每个查询参数只传递一个值（如果有多个值，这里只取第一个）
|
||
value := values[0]
|
||
|
||
// 查找目标结构体中对应的字段
|
||
field := rv.FieldByName(strings.Title(key)) // 假设字段名与查询参数名相同，但首字母大写
|
||
if !field.IsValid() || !field.CanSet() {
|
||
// 如果找不到对应的字段，或者字段不可设置，则跳过
|
||
continue
|
||
}
|
||
|
||
// 根据字段类型设置值
|
||
switch field.Kind() {
|
||
case reflect.String:
|
||
field.SetString(value)
|
||
case reflect.Int:
|
||
intValue, err := strconv.Atoi(value)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
field.SetInt(int64(intValue))
|
||
// 可以根据需要添加更多类型处理
|
||
default:
|
||
// 不支持的类型，可以选择跳过或返回错误
|
||
continue
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// GenerateMD5Hash 生成 MD5 的函数，接收一个实现了 proto.Message 接口的结构体
|
||
func GenerateMD5Hash(msg proto.Message) (string, error) {
|
||
// 使用 proto.Marshal 序列化结构体，proto.Marshal 接受任何实现 proto.Message 接口的结构体
|
||
data, err := proto.Marshal(msg)
|
||
if err != nil {
|
||
return "", fmt.Errorf("failed to serialize message: %v", err)
|
||
}
|
||
|
||
// 计算 MD5
|
||
hash := md5.New()
|
||
hash.Write(data)
|
||
hashBytes := hash.Sum(nil)
|
||
|
||
// 返回 MD5 值的十六进制字符串
|
||
return fmt.Sprintf("%x", hashBytes), nil
|
||
}
|