struct to schema correctly

This commit is contained in:
chrislu
2024-04-28 12:23:54 -07:00
parent f37862bffe
commit 8a0e5b6d54
3 changed files with 152 additions and 20 deletions

View File

@@ -5,20 +5,13 @@ import (
"reflect"
)
func StructToSchema(instance any) *RecordTypeBuilder {
rtb := NewRecordTypeBuilder()
func StructToSchema(instance any) *schema_pb.RecordType {
myType := reflect.TypeOf(instance)
for i := 0; i < myType.NumField(); i++ {
field := myType.Field(i)
fieldType := field.Type
fieldName := field.Name
schemaField := reflectTypeToSchemaType(fieldType)
if schemaField == nil {
continue
}
rtb.setField(fieldName, schemaField)
if myType.Kind() != reflect.Struct {
return nil
}
return rtb
st := reflectTypeToSchemaType(myType)
return st.GetRecordType()
}
func reflectTypeToSchemaType(t reflect.Type) *schema_pb.Type {
@@ -50,6 +43,26 @@ func reflectTypeToSchemaType(t reflect.Type) *schema_pb.Type {
}
}
}
case reflect.Struct:
recordType := &schema_pb.RecordType{}
for i := 0; i < t.NumField(); i++ {
field := t.Field(i)
fieldType := field.Type
fieldName := field.Name
schemaField := reflectTypeToSchemaType(fieldType)
if schemaField == nil {
return nil
}
recordType.Fields = append(recordType.Fields, &schema_pb.Field{
Name: fieldName,
Type: schemaField,
})
}
return &schema_pb.Type{
Kind: &schema_pb.Type_RecordType{
RecordType: recordType,
},
}
}
return nil
}