68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package test
|
||
|
||
import (
|
||
model "catface/app/model"
|
||
"fmt"
|
||
"reflect"
|
||
"strings"
|
||
"testing"
|
||
"unicode"
|
||
)
|
||
|
||
// 获取处理后的 json 标签值
|
||
func getProcessedJSONTag(field reflect.StructField) string {
|
||
jsonTag := field.Tag.Get("json")
|
||
if jsonTag != "" {
|
||
// 移除 omitempty
|
||
jsonTag = strings.Replace(jsonTag, ",omitempty", "", -1)
|
||
jsonTag = strings.Replace(jsonTag, "omitempty", "", -1)
|
||
}
|
||
return jsonTag
|
||
}
|
||
|
||
func Test(tt *testing.T) {
|
||
animal := model.Animal{}
|
||
|
||
// 获取 Animal 类型的反射类型
|
||
t := reflect.TypeOf(animal)
|
||
|
||
// 创建一个空的 map，用于存储字段名
|
||
fieldMap := make(map[string]bool)
|
||
|
||
// 遍历结构体的所有字段
|
||
fmt.Println("All fields:", t.NumField())
|
||
for i := 0; i < t.NumField(); i++ {
|
||
field := t.Field(i)
|
||
fieldName := field.Name
|
||
|
||
// 获取 json 标签中的值
|
||
jsonTag := getProcessedJSONTag(field)
|
||
if jsonTag != "" {
|
||
// 如果有 json 标签，则使用标签中的值
|
||
fieldMap[jsonTag] = true
|
||
} else {
|
||
// 如果没有 json 标签，则转换字段名为下划线命名
|
||
convertedFieldName := camelToSnake(fieldName)
|
||
fieldMap[convertedFieldName] = true
|
||
}
|
||
}
|
||
|
||
// 打印 map
|
||
fmt.Println("Fields as keys:")
|
||
for key := range fieldMap {
|
||
fmt.Println(key)
|
||
}
|
||
}
|
||
|
||
// 将驼峰命名转换为下划线命名
|
||
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()
|
||
}
|