mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
77 lines
2.2 KiB
Go
77 lines
2.2 KiB
Go
package indexcgowrapper
|
|
|
|
/*
|
|
|
|
#cgo CFLAGS: -I${SRCDIR}/../../core/output/include
|
|
|
|
#cgo darwin LDFLAGS: -L${SRCDIR}/../../core/output/lib -lmilvus_indexbuilder -lmilvus_common -Wl,-rpath,"${SRCDIR}/../../core/output/lib"
|
|
#cgo linux LDFLAGS: -L${SRCDIR}/../../core/output/lib -lmilvus_indexbuilder -lmilvus_common -Wl,-rpath=${SRCDIR}/../../core/output/lib
|
|
#cgo windows LDFLAGS: -L${SRCDIR}/../../core/output/lib -lmilvus_indexbuilder -lmilvus_common -Wl,-rpath=${SRCDIR}/../../core/output/lib
|
|
|
|
#include <stdlib.h> // free
|
|
#include "indexbuilder/index_c.h"
|
|
|
|
*/
|
|
import "C"
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"unsafe"
|
|
|
|
"github.com/milvus-io/milvus/internal/log"
|
|
"github.com/milvus-io/milvus/internal/proto/commonpb"
|
|
)
|
|
|
|
func GetBinarySetKeys(cBinarySet C.CBinarySet) ([]string, error) {
|
|
size := int(C.GetBinarySetSize(cBinarySet))
|
|
if size == 0 {
|
|
return nil, fmt.Errorf("BinarySet size is zero!")
|
|
}
|
|
datas := make([]unsafe.Pointer, size)
|
|
|
|
C.GetBinarySetKeys(cBinarySet, unsafe.Pointer(&datas[0]))
|
|
ret := make([]string, size)
|
|
for i := 0; i < size; i++ {
|
|
ret[i] = C.GoString((*C.char)(datas[i]))
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func GetBinarySetValue(cBinarySet C.CBinarySet, key string) ([]byte, error) {
|
|
cIndexKey := C.CString(key)
|
|
defer C.free(unsafe.Pointer(cIndexKey))
|
|
ret := C.GetBinarySetValueSize(cBinarySet, cIndexKey)
|
|
size := int(ret)
|
|
if size == 0 {
|
|
return nil, fmt.Errorf("GetBinarySetValueSize size is zero!")
|
|
}
|
|
value := make([]byte, size)
|
|
status := C.CopyBinarySetValue(unsafe.Pointer(&value[0]), cIndexKey, cBinarySet)
|
|
|
|
if err := HandleCStatus(&status, "CopyBinarySetValue failed"); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return value, nil
|
|
}
|
|
|
|
// HandleCStatus deal with the error returned from CGO
|
|
func HandleCStatus(status *C.CStatus, extraInfo string) error {
|
|
if status.error_code == 0 {
|
|
return nil
|
|
}
|
|
errorCode := status.error_code
|
|
errorName, ok := commonpb.ErrorCode_name[int32(errorCode)]
|
|
if !ok {
|
|
errorName = "UnknownError"
|
|
}
|
|
errorMsg := C.GoString(status.error_msg)
|
|
defer C.free(unsafe.Pointer(status.error_msg))
|
|
|
|
finalMsg := fmt.Sprintf("[%s] %s", errorName, errorMsg)
|
|
logMsg := fmt.Sprintf("%s, C Runtime Exception: %s\n", extraInfo, finalMsg)
|
|
log.Warn(logMsg)
|
|
return errors.New(finalMsg)
|
|
}
|