mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
f02bd8c8f5
Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
package querynode
|
|
|
|
/*
|
|
#cgo CFLAGS: -I${SRCDIR}/../core/output/include
|
|
#cgo LDFLAGS: -L${SRCDIR}/../core/output/lib -lmilvus_segcore -Wl,-rpath=${SRCDIR}/../core/output/lib
|
|
|
|
#include "segcore/collection_c.h"
|
|
#include "segcore/segment_c.h"
|
|
#include "segcore/plan_c.h"
|
|
*/
|
|
import "C"
|
|
import (
|
|
"errors"
|
|
"strconv"
|
|
"unsafe"
|
|
)
|
|
|
|
type Plan struct {
|
|
cPlan C.CPlan
|
|
}
|
|
|
|
func createPlan(col Collection, dsl string) (*Plan, error) {
|
|
cDsl := C.CString(dsl)
|
|
var cPlan C.CPlan
|
|
status := C.CreatePlan(col.collectionPtr, cDsl, &cPlan)
|
|
|
|
errorCode := status.error_code
|
|
|
|
if errorCode != 0 {
|
|
errorMsg := C.GoString(status.error_msg)
|
|
defer C.free(unsafe.Pointer(status.error_msg))
|
|
return nil, errors.New("Create plan failed, C runtime error detected, error code = " + strconv.Itoa(int(errorCode)) + ", error msg = " + errorMsg)
|
|
}
|
|
|
|
var newPlan = &Plan{cPlan: cPlan}
|
|
return newPlan, nil
|
|
}
|
|
|
|
func (plan *Plan) getTopK() int64 {
|
|
topK := C.GetTopK(plan.cPlan)
|
|
return int64(topK)
|
|
}
|
|
|
|
func (plan *Plan) getMetricType() string {
|
|
cMetricType := C.GetMetricType(plan.cPlan)
|
|
defer C.free(unsafe.Pointer(cMetricType))
|
|
metricType := C.GoString(cMetricType)
|
|
return metricType
|
|
}
|
|
|
|
func (plan *Plan) delete() {
|
|
C.DeletePlan(plan.cPlan)
|
|
}
|
|
|
|
type PlaceholderGroup struct {
|
|
cPlaceholderGroup C.CPlaceholderGroup
|
|
}
|
|
|
|
func parserPlaceholderGroup(plan *Plan, placeHolderBlob []byte) (*PlaceholderGroup, error) {
|
|
var blobPtr = unsafe.Pointer(&placeHolderBlob[0])
|
|
blobSize := C.long(len(placeHolderBlob))
|
|
var cPlaceholderGroup C.CPlaceholderGroup
|
|
status := C.ParsePlaceholderGroup(plan.cPlan, blobPtr, blobSize, &cPlaceholderGroup)
|
|
|
|
errorCode := status.error_code
|
|
|
|
if errorCode != 0 {
|
|
errorMsg := C.GoString(status.error_msg)
|
|
defer C.free(unsafe.Pointer(status.error_msg))
|
|
return nil, errors.New("Parser placeholder group failed, C runtime error detected, error code = " + strconv.Itoa(int(errorCode)) + ", error msg = " + errorMsg)
|
|
}
|
|
|
|
var newPlaceholderGroup = &PlaceholderGroup{cPlaceholderGroup: cPlaceholderGroup}
|
|
return newPlaceholderGroup, nil
|
|
}
|
|
|
|
func (pg *PlaceholderGroup) getNumOfQuery() int64 {
|
|
numQueries := C.GetNumOfQueries(pg.cPlaceholderGroup)
|
|
return int64(numQueries)
|
|
}
|
|
|
|
func (pg *PlaceholderGroup) delete() {
|
|
C.DeletePlaceholderGroup(pg.cPlaceholderGroup)
|
|
}
|