milvus/internal/reader/result.go

package reader

import (
	"context"
	masterPb "github.com/zilliztech/milvus-distributed/internal/proto/master"
	msgPb "github.com/zilliztech/milvus-distributed/internal/proto/message"
)

type ResultEntityIds []int64

type SearchResult struct {
	ResultIds       []int64
	ResultDistances []float32
}

func (node *QueryNode) PublishSearchResult(results *msgPb.QueryResult) msgPb.Status {
	var ctx = context.Background()

	node.messageClient.SendResult(ctx, *results, results.ProxyId)

	return msgPb.Status{ErrorCode: msgPb.ErrorCode_SUCCESS}
}

func (node *QueryNode) PublishFailedSearchResult() msgPb.Status {
	var results = msgPb.QueryResult{
		Status: &msgPb.Status{
			ErrorCode: 1,
			Reason:    "Search Failed",
		},
	}

	var ctx = context.Background()

	node.messageClient.SendResult(ctx, results, results.ProxyId)
	return msgPb.Status{ErrorCode: msgPb.ErrorCode_SUCCESS}
}

func (node *QueryNode) PublicStatistic(statisticData *[]masterPb.SegmentStat) msgPb.Status {
	var ctx = context.Background()

	node.messageClient.SendSegmentsStatistic(ctx, statisticData)

	return msgPb.Status{ErrorCode: msgPb.ErrorCode_SUCCESS}
}