mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-02 11:59:00 +08:00
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package reader
|
|
|
|
/*
|
|
|
|
#cgo CFLAGS: -I../core/include
|
|
|
|
#cgo LDFLAGS: -L../core/lib -lmilvus_dog_segment -Wl,-rpath=../core/lib
|
|
|
|
#include "collection_c.h"
|
|
#include "partition_c.h"
|
|
#include "segment_c.h"
|
|
|
|
*/
|
|
import "C"
|
|
|
|
type Collection struct {
|
|
CollectionPtr C.CCollection
|
|
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) {
|
|
// TODO: add get segments
|
|
//segments, status := C.GetSegments(c.CollectionPtr)
|
|
//
|
|
//if status != 0 {
|
|
// return nil, errors.New("get segments failed")
|
|
//}
|
|
//
|
|
//return segments, nil
|
|
return nil, nil
|
|
}
|