mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-28 22:45:26 +08:00
48 lines
978 B
Go
48 lines
978 B
Go
package reader
|
|
|
|
/*
|
|
|
|
#cgo CFLAGS: -I../core/include
|
|
|
|
#cgo LDFLAGS: -L../core/lib -lmilvus_dog_segment -Wl,-rpath=../core/lib
|
|
|
|
#include "partition_c.h"
|
|
|
|
*/
|
|
import "C"
|
|
import (
|
|
"errors"
|
|
)
|
|
|
|
type Collection struct {
|
|
CollectionPtr *C.Collection
|
|
CollectionName string
|
|
Partitions []*Partition
|
|
}
|
|
|
|
func (c *Collection) NewPartition(partitionName string) *Partition {
|
|
cName := C.CString(partitionName)
|
|
partitionPtr := C.NewPartition(c.CollectionPtr, cName)
|
|
|
|
var newPartition = &Partition{PartitionPtr: partitionPtr, PartitionName: partitionName}
|
|
c.Partitions = append(c.Partitions, newPartition)
|
|
return newPartition
|
|
}
|
|
|
|
func (c *Collection) DeletePartition(partition *Partition) {
|
|
cPtr := partition.PartitionPtr
|
|
C.DeletePartition(cPtr)
|
|
|
|
// TODO: remove from c.Partitions
|
|
}
|
|
|
|
func (c *Collection) GetSegments() ([]*Segment, error) {
|
|
segments, status := C.GetSegments(c.CollectionPtr)
|
|
|
|
if status != 0 {
|
|
return nil, errors.New("get segments failed")
|
|
}
|
|
|
|
return segments, nil
|
|
}
|