54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package model_handler
|
|
|
|
import (
|
|
"reflect"
|
|
"strings"
|
|
"unicode"
|
|
)
|
|
|
|
// 将驼峰命名转换为下划线命名
|
|
func camelToSnake(s string) string {
|
|
var result strings.Builder
|
|
for i, r := range s {
|
|
if i > 0 && unicode.IsUpper(r) {
|
|
result.WriteRune('_')
|
|
}
|
|
result.WriteRune(unicode.ToLower(r))
|
|
}
|
|
return result.String()
|
|
}
|
|
|
|
// 获取处理后的 json 标签值
|
|
func getProcessedJSONTag(field reflect.StructField) string {
|
|
jsonTag := field.Tag.Get("json")
|
|
if jsonTag != "" {
|
|
// INFO 移除 omitempty
|
|
jsonTag = strings.Replace(jsonTag, ",omitempty", "", -1)
|
|
jsonTag = strings.Replace(jsonTag, "omitempty", "", -1)
|
|
}
|
|
return jsonTag
|
|
}
|
|
|
|
func GetModelField(v interface{}) map[string]bool {
|
|
t := reflect.TypeOf(v) // TODO 特化处理掉 BaseModel 这样的继承字段
|
|
|
|
fieldMap := make(map[string]bool)
|
|
for i := 0; i < t.NumField(); i++ {
|
|
field := t.Field(i)
|
|
fieldName := field.Name
|
|
|
|
// 获取 json 标签中的值
|
|
jsonTag := getProcessedJSONTag(field)
|
|
if jsonTag != "" {
|
|
// 如果有 json 标签，则使用标签中的值
|
|
fieldMap[jsonTag] = false
|
|
} else {
|
|
// 如果没有 json 标签，则转换字段名为下划线命名
|
|
convertedFieldName := camelToSnake(fieldName)
|
|
fieldMap[convertedFieldName] = false
|
|
}
|
|
}
|
|
fieldMap["id"] = true // INFO default ID 默认都会返回。
|
|
return fieldMap
|
|
}
|