add cold.Index

vinceanalytics · Feb 9, 2024 · 5f06c00 · 5f06c00
1 parent 0c5e37d
commit 5f06c00
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/segment/cold/index.go b/segment/cold/index.go
@@ -15,12 +15,20 @@ var skipFields = map[string]struct{}{
 	camel.Case(v1.Filters_Duration.String()): {},
 }
 
-func New(r arrow.Record) (segment.Segment, error) {
+type Index struct {
+	segment.Segment
+	DiskSize uint64
+}
+
+func New(r arrow.Record) (*Index, error) {
 	document := NewRecord(r, func(r arrow.Record, i int) bool {
 		_, ok := skipFields[r.ColumnName(i)]
 		return !ok
 	})
 	defer document.Release()
-	seg, _, err := ice.New([]segment.Document{document}, func(s string, i int) float32 { return 0 })
-	return seg, err
+	seg, size, err := ice.New([]segment.Document{document}, func(s string, i int) float32 { return 0 })
+	if err != nil {
+		return nil, err
+	}
+	return &Index{Segment: seg, DiskSize: size}, nil
 }