82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/bytedance/sonic"
|
|
)
|
|
|
|
// MarshalQueryParams 基于 struct 字段的 json tag 自动构建 URL 查询参数 map。
|
|
//
|
|
// 规则：
|
|
// - 字段必须有 json tag，tag 名（逗号前的部分）作为参数 key
|
|
// - 指针字段为 nil 时跳过（对应 optional 语义）；非 nil 时解引用处理
|
|
// - 基础类型（string / int / uint / float / bool）转为字符串
|
|
// - 复杂类型（struct / map / slice 等）自动 marshal 为 JSON 字符串
|
|
// - 无 json tag 或 tag 为 "-" 的字段被忽略
|
|
func MarshalQueryParams(v any) (map[string]string, error) {
|
|
rv := reflect.ValueOf(v)
|
|
if rv.Kind() == reflect.Ptr {
|
|
if rv.IsNil() {
|
|
return nil, nil
|
|
}
|
|
rv = rv.Elem()
|
|
}
|
|
if rv.Kind() != reflect.Struct {
|
|
return nil, fmt.Errorf("MarshalQueryParams: 期望 struct，实际 %s", rv.Kind())
|
|
}
|
|
|
|
rst := make(map[string]string)
|
|
rt := rv.Type()
|
|
for i := 0; i < rt.NumField(); i++ {
|
|
field := rt.Field(i)
|
|
tag := field.Tag.Get("json")
|
|
if tag == "" || tag == "-" {
|
|
continue
|
|
}
|
|
name, _, _ := strings.Cut(tag, ",")
|
|
|
|
fv := rv.Field(i)
|
|
if fv.Kind() == reflect.Ptr {
|
|
if fv.IsNil() {
|
|
continue
|
|
}
|
|
fv = fv.Elem()
|
|
}
|
|
|
|
val, err := queryValue(fv)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("MarshalQueryParams: 字段 %s: %w", field.Name, err)
|
|
}
|
|
rst[name] = val
|
|
}
|
|
return rst, nil
|
|
}
|
|
|
|
func queryValue(v reflect.Value) (string, error) {
|
|
switch v.Kind() {
|
|
case reflect.String:
|
|
return v.String(), nil
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
return strconv.FormatInt(v.Int(), 10), nil
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
return strconv.FormatUint(v.Uint(), 10), nil
|
|
case reflect.Float32, reflect.Float64:
|
|
return strconv.FormatFloat(v.Float(), 'f', -1, 64), nil
|
|
case reflect.Bool:
|
|
return strconv.FormatBool(v.Bool()), nil
|
|
default:
|
|
if !v.CanInterface() {
|
|
return "", fmt.Errorf("无法读取未导出字段")
|
|
}
|
|
b, err := sonic.Marshal(v.Interface())
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(b), nil
|
|
}
|
|
}
|