gsh-digital-services/seaweedFS
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improve api

Browse Source
This commit is contained in:
chrislu
2024-04-28 12:42:58 -07:00
parent 1a2332c257
commit 0da1b7b0f9
2 changed files with 7 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
weed/mq/schema/schema_builder.go
View File

@@ -39,16 +39,6 @@ func (rtb *RecordTypeBuilder) SetField(name string, scalarType *schema_pb.Type)
return rtb return rtb
} }
func (rtb *RecordTypeBuilder) SetLongField(name string) *RecordTypeBuilder {
return rtb.SetField(name, TypeLong)
}
func (rtb *RecordTypeBuilder) SetFloatField(name string) *RecordTypeBuilder {
return rtb.SetField(name, TypeFloat)
}
func (rtb *RecordTypeBuilder) SetStringField(name string) *RecordTypeBuilder {
return rtb.SetField(name, TypeString)
}
func (rtb *RecordTypeBuilder) SetRecordField(name string, recordTypeBuilder *RecordTypeBuilder) *RecordTypeBuilder { func (rtb *RecordTypeBuilder) SetRecordField(name string, recordTypeBuilder *RecordTypeBuilder) *RecordTypeBuilder {
rtb.recordType.Fields = append(rtb.recordType.Fields, &schema_pb.Field{ rtb.recordType.Fields = append(rtb.recordType.Fields, &schema_pb.Field{
Name: name, Name: name,
@@ -57,14 +47,6 @@ func (rtb *RecordTypeBuilder) SetRecordField(name string, recordTypeBuilder *Rec
return rtb return rtb
} }
func (rtb *RecordTypeBuilder) SetListField(name string, elementType *schema_pb.Type) *RecordTypeBuilder {
rtb.recordType.Fields = append(rtb.recordType.Fields, &schema_pb.Field{
Name: name,
Type: &schema_pb.Type{Kind: &schema_pb.Type_ListType{ListType: &schema_pb.ListType{ElementType: elementType}}},
})
return rtb
}
func ListOf(elementType *schema_pb.Type) *schema_pb.Type { func ListOf(elementType *schema_pb.Type) *schema_pb.Type {
return &schema_pb.Type{Kind: &schema_pb.Type_ListType{ListType: &schema_pb.ListType{ElementType: elementType}}} return &schema_pb.Type{Kind: &schema_pb.Type_ListType{ListType: &schema_pb.ListType{ElementType: elementType}}}
} }

14
weed/mq/schema/write_parquet_test.go
View File

@@ -13,15 +13,15 @@ import (
func TestWriteReadParquet(t *testing.T) { func TestWriteReadParquet(t *testing.T) {
// create a schema_pb.RecordType // create a schema_pb.RecordType
recordType := NewRecordTypeBuilder(). recordType := NewRecordTypeBuilder().
SetLongField("ID"). SetField("ID", TypeLong).
SetLongField("CreatedAt"). SetField("CreatedAt", TypeLong).
SetRecordField("Person", NewRecordTypeBuilder(). SetRecordField("Person", NewRecordTypeBuilder().
SetStringField("zName"). SetField("zName", TypeString).
SetListField("emails", TypeString)). SetField("emails", ListOf(TypeString))).
SetStringField("Company"). SetField("Company", TypeString).
SetRecordField("Address", NewRecordTypeBuilder(). SetRecordField("Address", NewRecordTypeBuilder().
SetStringField("Street"). SetField("Street", TypeString).
SetStringField("City")).Build() SetField("City", TypeString)).Build()
fmt.Printf("RecordType: %v\n", recordType) fmt.Printf("RecordType: %v\n", recordType)
// create a parquet schema // create a parquet schema