milvus/reader/segment.go
bigsheeper 5a57b62f0c Refactor collection and partition, wrap cgo interfaces
Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
2020-08-31 15:54:44 +08:00

70 lines
1.5 KiB
Go

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 (
"suvlim/pulsar/schema"
)
const SegmentLifetime = 20000
type Segment struct {
SegmentPtr *C.SegmentBase
SegmentId int32
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{}
}