aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorchrislu <chris.lu@gmail.com>2024-04-28 12:40:00 -0700
committerchrislu <chris.lu@gmail.com>2024-04-28 12:40:00 -0700
commit1a2332c257e761c9788e73d8c8af7cc9d7484b52 (patch)
tree08cb4bc126ea3212946a56ee90587ba7271d7298
parent5ce8f505466b0c4bbdc467a1a3dac933d5a4b129 (diff)
downloadseaweedfs-1a2332c257e761c9788e73d8c8af7cc9d7484b52.tar.xz
seaweedfs-1a2332c257e761c9788e73d8c8af7cc9d7484b52.zip
rename api
-rw-r--r--weed/mq/schema/schema_builder.go14
-rw-r--r--weed/mq/schema/struct_to_schema_test.go4
-rw-r--r--weed/mq/schema/to_parquet_levels_test.go14
3 files changed, 10 insertions, 22 deletions
diff --git a/weed/mq/schema/schema_builder.go b/weed/mq/schema/schema_builder.go
index 224c79309..0bd324fb3 100644
--- a/weed/mq/schema/schema_builder.go
+++ b/weed/mq/schema/schema_builder.go
@@ -39,24 +39,12 @@ func (rtb *RecordTypeBuilder) SetField(name string, scalarType *schema_pb.Type)
return rtb
}
-func (rtb *RecordTypeBuilder) SetBoolField(name string) *RecordTypeBuilder {
- return rtb.SetField(name, TypeBoolean)
-}
-func (rtb *RecordTypeBuilder) SetIntegerField(name string) *RecordTypeBuilder {
- return rtb.SetField(name, TypeInteger)
-}
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) SetDoubleField(name string) *RecordTypeBuilder {
- return rtb.SetField(name, TypeDouble)
-}
-func (rtb *RecordTypeBuilder) SetBytesField(name string) *RecordTypeBuilder {
- return rtb.SetField(name, TypeBytes)
-}
func (rtb *RecordTypeBuilder) SetStringField(name string) *RecordTypeBuilder {
return rtb.SetField(name, TypeString)
}
@@ -77,6 +65,6 @@ func (rtb *RecordTypeBuilder) SetListField(name string, elementType *schema_pb.T
return rtb
}
-func List(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}}}
}
diff --git a/weed/mq/schema/struct_to_schema_test.go b/weed/mq/schema/struct_to_schema_test.go
index 6a3d5687f..84fb48326 100644
--- a/weed/mq/schema/struct_to_schema_test.go
+++ b/weed/mq/schema/struct_to_schema_test.go
@@ -44,7 +44,7 @@ func TestStructToSchema(t *testing.T) {
}{},
},
want: NewRecordTypeBuilder().
- SetField("Field1", List(TypeInteger)).
+ SetField("Field1", ListOf(TypeInteger)).
SetField("Field2", TypeString).
Build(),
},
@@ -97,7 +97,7 @@ func TestStructToSchema(t *testing.T) {
SetField("Field1", TypeInteger).
SetRecordField("Field2", NewRecordTypeBuilder().
SetField("Field3", TypeString).
- SetField("Field4", List(TypeInteger)).
+ SetField("Field4", ListOf(TypeInteger)).
SetRecordField("Field5", NewRecordTypeBuilder().
SetField("Field6", TypeString).
SetField("Field7", TypeBytes),
diff --git a/weed/mq/schema/to_parquet_levels_test.go b/weed/mq/schema/to_parquet_levels_test.go
index e9976212e..b7ed4dde1 100644
--- a/weed/mq/schema/to_parquet_levels_test.go
+++ b/weed/mq/schema/to_parquet_levels_test.go
@@ -19,15 +19,15 @@ func TestToParquetLevels(t *testing.T) {
name: "nested type",
args: args{
NewRecordTypeBuilder().
- SetLongField("ID").
- SetLongField("CreatedAt").
+ SetField("ID", TypeLong).
+ SetField("CreatedAt", TypeLong).
SetRecordField("Person", NewRecordTypeBuilder().
- SetStringField("zName").
- SetListField("emails", TypeString)).
- SetStringField("Company").
+ SetField("zName", TypeString).
+ SetField("emails", ListOf(TypeString))).
+ SetField("Company", TypeString).
SetRecordField("Address", NewRecordTypeBuilder().
- SetStringField("Street").
- SetStringField("City")).Build(),
+ SetField("Street", TypeString).
+ SetField("City", TypeString)).Build(),
},
want: &ParquetLevels{
startColumnIndex: 0,