milvus/internal/reader/partition.go

package reader

/*

#cgo CFLAGS: -I${SRCDIR}/../core/output/include

#cgo LDFLAGS: -L${SRCDIR}/../core/output/lib -lmilvus_dog_segment -Wl,-rpath=${SRCDIR}/../core/output/lib

#include "collection_c.h"
#include "partition_c.h"
#include "segment_c.h"

*/
import "C"

type Partition struct {
	PartitionPtr  C.CPartition
	PartitionName string
	Segments      []*Segment
}

func (p *Partition) NewSegment(segmentId int64) *Segment {
	/*
		CSegmentBase
		NewSegment(CPartition partition, unsigned long segment_id);
	*/
	segmentPtr := C.NewSegment(p.PartitionPtr, C.ulong(segmentId))

	var newSegment = &Segment{SegmentPtr: segmentPtr, SegmentId: segmentId}
	p.Segments = append(p.Segments, newSegment)
	return newSegment
}

func (p *Partition) DeleteSegment(segment *Segment) {
	/*
		void
		DeleteSegment(CSegmentBase segment);
	*/
	cPtr := segment.SegmentPtr
	C.DeleteSegment(cPtr)

	// TODO: remove from p.Segments
}