milvus/reader/segment.go

70 lines
1.6 KiB
Go
Raw Normal View History

package reader
/*
#cgo CFLAGS: -I../core/include
#cgo LDFLAGS: -L../core/lib -lmilvus_dog_segment -Wl,-rpath=../core/lib
#include "segment_c.h"
*/
import "C"
import (
"github.com/czs007/suvlim/pulsar/schema"
)
const SegmentLifetime = 20000
type Segment struct {
SegmentPtr *C.SegmentBase
SegmentId uint64
SegmentCloseTime uint64
}
func (s *Segment) GetRowCount() int64 {
// TODO: C type to go type
return C.GetRowCount(s)
}
func (s *Segment) GetStatus() int {
// TODO: C type to go type
return C.GetStatus(s)
}
func (s *Segment) GetMaxTimestamp() uint64 {
// TODO: C type to go type
return C.GetMaxTimestamp(s)
}
func (s *Segment) GetMinTimestamp() uint64 {
// TODO: C type to go type
return C.GetMinTimestamp(s)
}
func (s *Segment) GetDeletedCount() uint64 {
// TODO: C type to go type
return C.GetDeletedCount(s)
}
func (s *Segment) Close() {
// TODO: C type to go type
C.CloseSegment(s)
}
////////////////////////////////////////////////////////////////////////////
func SegmentInsert(segment *Segment, collectionName string, partitionTag string, entityIds *[]int64, timestamps *[]uint64, dataChunk [][]*schema.FieldValue) ResultEntityIds {
// TODO: wrap cgo
return ResultEntityIds{}
}
func SegmentDelete(segment *Segment, collectionName string, entityIds *[]int64, timestamps *[]uint64) ResultEntityIds {
// TODO: wrap cgo
return ResultEntityIds{}
}
func SegmentSearch(segment *Segment, collectionName string, queryString string, timestamps *[]uint64, vectorRecord *[]schema.VectorRecord) ResultEntityIds {
// TODO: wrap cgo
return ResultEntityIds{}
}