52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
// Package utils -----------------------------
|
||
// @file : requestDataToProto.go
|
||
// @author : JJXu
|
||
// @contact : wavingbear@163.com
|
||
// @time : 2023/8/28 17:57
|
||
// -------------------------------------------
|
||
package utils
|
||
|
||
import (
|
||
"reflect"
|
||
"strings"
|
||
)
|
||
|
||
// http请求转proto请求
|
||
func RequestDataConvert(from interface{}, to interface{}) {
|
||
var proxyField = "Query"
|
||
fromValue := reflect.ValueOf(from)
|
||
toValue := reflect.ValueOf(to)
|
||
toType := reflect.TypeOf(to)
|
||
|
||
// 获取From结构体的字段信息
|
||
fromType := fromValue.Type().Elem()
|
||
for i := 0; i < fromType.NumField(); i++ {
|
||
// 获取字段名和字段值
|
||
fieldName := fromType.Field(i).Name
|
||
fieldValue := fromValue.Elem().FieldByName(fieldName)
|
||
if fieldName != proxyField {
|
||
_, exists := toType.Elem().FieldByName(fieldName)
|
||
if exists {
|
||
// 设置To结构体中相应字段的值
|
||
toValue.Elem().FieldByName(fieldName).Set(fieldValue)
|
||
}
|
||
}
|
||
}
|
||
queryField, exists := toType.Elem().FieldByName(proxyField)
|
||
if exists {
|
||
var queryFieldTypeName string
|
||
// 指针类型额外处理，拿到真实的数据类型
|
||
if queryField.Type.Kind() == reflect.Ptr {
|
||
queryFieldTypeName = queryField.Type.Elem().String()
|
||
} else {
|
||
queryFieldTypeName = queryField.Type.Kind().String()
|
||
}
|
||
//处理拿到的结构体类型如 utils.xxxx的类型，去掉utils.这部分
|
||
if strings.Contains(queryFieldTypeName, ".") {
|
||
queryFieldTypeName = strings.Split(queryFieldTypeName, ".")[1]
|
||
}
|
||
fromQueryValue := fromValue.Elem().FieldByName(queryFieldTypeName)
|
||
toValue.Elem().FieldByName(proxyField).Set(fromQueryValue.Addr())
|
||
}
|
||
}
|