milvus/reader/result.go
bigsheeper b948a3438e Add api about insertion, deletion and search
Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
2020-09-07 17:01:46 +08:00

42 lines
1.0 KiB
Go

package reader
import (
"fmt"
msgpb "github.com/czs007/suvlim/pkg/message"
"strconv"
)
type ResultEntityIds []int64
type SearchResult struct {
ResultIds []int64
ResultDistances []float32
}
func getResultTopicByClientId(clientId int64) string {
// TODO: Result topic?
return "result-topic/partition-" + strconv.FormatInt(clientId, 10)
}
func publishResult(ids *ResultEntityIds, clientId int64) msgpb.Status {
// TODO: Pulsar publish
var resultTopic = getResultTopicByClientId(clientId)
fmt.Println(resultTopic)
return msgpb.Status{ErrorCode: msgpb.ErrorCode_SUCCESS}
}
func publishSearchResult(searchResults *[]*SearchResult, clientId int64) msgpb.Status {
// TODO: Pulsar publish
var resultTopic = getResultTopicByClientId(clientId)
fmt.Println(resultTopic)
return msgpb.Status{ErrorCode: msgpb.ErrorCode_SUCCESS}
}
func publicStatistic(statisticTopic string) msgpb.Status {
// TODO: get statistic info
// getStatisticInfo()
// var info = getStatisticInfo()
// TODO: Pulsar publish
return msgpb.Status{ErrorCode: msgpb.ErrorCode_SUCCESS}
}