admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-06 17:18:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

enhance: timestamptz support groupby (#45762)

Browse Source 
issue: https://github.com/milvus-io/milvus/issues/45761

Signed-off-by: zhenshan.cao <zhenshan.cao@zilliz.com>
This commit is contained in:
zhenshan.cao 2025-11-21 18:39:05 +08:00 committed by GitHub
parent e00ad1098f
commit bec6d1d1e1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/util/typeutil/field_data.go
View File

@ -19,7 +19,7 @@ func NewFieldDataBuilder(dt schemapb.DataType, fillZero bool, capacity int) (*Fi
switch dt {
case schemapb.DataType_Bool,
schemapb.DataType_Int8, schemapb.DataType_Int16, schemapb.DataType_Int32, schemapb.DataType_Int64,
schemapb.DataType_VarChar:
schemapb.DataType_Timestamptz, schemapb.DataType_VarChar:
return &FieldDataBuilder{
dt: dt,
data: make([]any, 0, capacity),
@ -111,6 +111,26 @@ func (b *FieldDataBuilder) Build() *schemapb.FieldData {
},
},
}
case schemapb.DataType_Timestamptz:
val := make([]int64, 0, len(b.valid))
validIdx := 0
for _, v := range b.valid {
if v {
val = append(val, b.data[validIdx].(int64))
validIdx++
} else if b.fillZero {
val = append(val, 0)
}
}
field.Field = &schemapb.FieldData_Scalars{
Scalars: &schemapb.ScalarField{
Data: &schemapb.ScalarField_TimestamptzData{
TimestamptzData: &schemapb.TimestamptzArray{
Data: val,
},
},
},
}
case schemapb.DataType_VarChar:
val := make([]string, 0, len(b.valid))
validIdx := 0