admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-28 22:45:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add aggregation for query result

Browse Source 
Signed-off-by: cai.zhang <cai.zhang@zilliz.com>
This commit is contained in:
cai.zhang 2020-09-08 14:16:03 +08:00 committed by yefu.chen
parent 41d638abb5
commit 815a6ca1cf
16 changed files with 36862 additions and 1189 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2480
pkg/message/message.pb.go
View File

File diff suppressed because it is too large Load Diff

110
pkg/metapb/metapb.pb.go
View File

@ -1,21 +1,13 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// source: metapb.proto
/*
Package metapb is a generated protocol buffer package.
It is generated from these files:
metapb.proto
It has these top-level messages:
Cluster
Peer
*/
package metapb
import proto "github.com/golang/protobuf/proto"
import fmt "fmt"
import math "math"
import (
fmt "fmt"
proto "github.com/golang/protobuf/proto"
math "math"
)
// Reference imports to suppress errors if they are not otherwise used.
var _ = proto.Marshal
@ -26,7 +18,7 @@ var _ = math.Inf
// is compatible with the proto package it is being compiled against.
// A compilation error at this line likely means your copy of the
// proto package needs to be updated.
const _ = proto.ProtoPackageIsVersion2 // please upgrade the proto package
const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package
type PeerRole int32
@ -43,6 +35,7 @@ var PeerRole_name = map[int32]string{
2: "Write",
3: "Proxyer",
}
var PeerRole_value = map[string]int32{
"Master": 0,
"Reader": 1,
@ -53,19 +46,45 @@ var PeerRole_value = map[string]int32{
func (x PeerRole) String() string {
return proto.EnumName(PeerRole_name, int32(x))
}
func (PeerRole) EnumDescriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
type Cluster struct {
Id uint64 `protobuf:"varint,1,opt,name=id" json:"id,omitempty"`
// max peer count for a region.
// pd will do the auto-balance if region peer count mismatches.
MaxPeerCount uint32 `protobuf:"varint,1024,opt,name=max_peer_count,json=maxPeerCount" json:"max_peer_count,omitempty"`
func (PeerRole) EnumDescriptor() ([]byte, []int) {
return fileDescriptor_77b4d575d5a68dda, []int{0}
}
func (m *Cluster) Reset() { *m = Cluster{} }
func (m *Cluster) String() string { return proto.CompactTextString(m) }
func (*Cluster) ProtoMessage() {}
func (*Cluster) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
type Cluster struct {
Id uint64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"`
// max peer count for a region.
// pd will do the auto-balance if region peer count mismatches.
MaxPeerCount uint32 `protobuf:"varint,1024,opt,name=max_peer_count,json=maxPeerCount,proto3" json:"max_peer_count,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *Cluster) Reset() { *m = Cluster{} }
func (m *Cluster) String() string { return proto.CompactTextString(m) }
func (*Cluster) ProtoMessage() {}
func (*Cluster) Descriptor() ([]byte, []int) {
return fileDescriptor_77b4d575d5a68dda, []int{0}
}
func (m *Cluster) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_Cluster.Unmarshal(m, b)
}
func (m *Cluster) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_Cluster.Marshal(b, m, deterministic)
}
func (m *Cluster) XXX_Merge(src proto.Message) {
xxx_messageInfo_Cluster.Merge(m, src)
}
func (m *Cluster) XXX_Size() int {
return xxx_messageInfo_Cluster.Size(m)
}
func (m *Cluster) XXX_DiscardUnknown() {
xxx_messageInfo_Cluster.DiscardUnknown(m)
}
var xxx_messageInfo_Cluster proto.InternalMessageInfo
func (m *Cluster) GetId() uint64 {
if m != nil {
@ -82,15 +101,38 @@ func (m *Cluster) GetMaxPeerCount() uint32 {
}
type Peer struct {
Id uint64 `protobuf:"varint,1,opt,name=id" json:"id,omitempty"`
PeerId uint64 `protobuf:"varint,2,opt,name=peer_id,json=peerId" json:"peer_id,omitempty"`
Role PeerRole `protobuf:"varint,3,opt,name=role,enum=metapb.PeerRole" json:"role,omitempty"`
Id uint64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"`
PeerId uint64 `protobuf:"varint,2,opt,name=peer_id,json=peerId,proto3" json:"peer_id,omitempty"`
Role PeerRole `protobuf:"varint,3,opt,name=role,proto3,enum=metapb.PeerRole" json:"role,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *Peer) Reset() { *m = Peer{} }
func (m *Peer) String() string { return proto.CompactTextString(m) }
func (*Peer) ProtoMessage() {}
func (*Peer) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} }
func (m *Peer) Reset() { *m = Peer{} }
func (m *Peer) String() string { return proto.CompactTextString(m) }
func (*Peer) ProtoMessage() {}
func (*Peer) Descriptor() ([]byte, []int) {
return fileDescriptor_77b4d575d5a68dda, []int{1}
}
func (m *Peer) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_Peer.Unmarshal(m, b)
}
func (m *Peer) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_Peer.Marshal(b, m, deterministic)
}
func (m *Peer) XXX_Merge(src proto.Message) {
xxx_messageInfo_Peer.Merge(m, src)
}
func (m *Peer) XXX_Size() int {
return xxx_messageInfo_Peer.Size(m)
}
func (m *Peer) XXX_DiscardUnknown() {
xxx_messageInfo_Peer.DiscardUnknown(m)
}
var xxx_messageInfo_Peer proto.InternalMessageInfo
func (m *Peer) GetId() uint64 {
if m != nil {
@ -114,14 +156,14 @@ func (m *Peer) GetRole() PeerRole {
}
func init() {
proto.RegisterEnum("metapb.PeerRole", PeerRole_name, PeerRole_value)
proto.RegisterType((*Cluster)(nil), "metapb.Cluster")
proto.RegisterType((*Peer)(nil), "metapb.Peer")
proto.RegisterEnum("metapb.PeerRole", PeerRole_name, PeerRole_value)
}
func init() { proto.RegisterFile("metapb.proto", fileDescriptor0) }
func init() { proto.RegisterFile("metapb.proto", fileDescriptor_77b4d575d5a68dda) }
var fileDescriptor0 = []byte{
var fileDescriptor_77b4d575d5a68dda = []byte{
// 210 bytes of a gzipped FileDescriptorProto
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x64, 0x8f, 0xb1, 0x4b, 0xc5, 0x30,
0x10, 0xc6, 0x4d, 0x5e, 0x4d, 0x9f, 0xe7, 0xb3, 0x84, 0x2c, 0x66, 0x2c, 0x0f, 0x85, 0xe2, 0xd0,

543
pkg/pdpb/pdpb.pb.go
View File

@ -1,35 +1,17 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// source: pdpb.proto
/*
Package pdpb is a generated protocol buffer package.
It is generated from these files:
pdpb.proto
It has these top-level messages:
Member
RequestHeader
ResponseHeader
Error
TsoRequest
Timestamp
TsoResponse
AllocIDRequest
AllocIDResponse
HeartbeatRequest
HeartbeatResponse
*/
package pdpb
import proto "github.com/golang/protobuf/proto"
import fmt "fmt"
import math "math"
import metapb "github.com/czs007/suvlim/pkg/metapb"
import (
context "golang.org/x/net/context"
context "context"
fmt "fmt"
metapb "github.com/czs007/suvlim/pkg/metapb"
proto "github.com/golang/protobuf/proto"
grpc "google.golang.org/grpc"
codes "google.golang.org/grpc/codes"
status "google.golang.org/grpc/status"
math "math"
)
// Reference imports to suppress errors if they are not otherwise used.
@ -41,7 +23,7 @@ var _ = math.Inf
// is compatible with the proto package it is being compiled against.
// A compilation error at this line likely means your copy of the
// proto package needs to be updated.
const _ = proto.ProtoPackageIsVersion2 // please upgrade the proto package
const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package
type ErrorType int32
@ -64,6 +46,7 @@ var ErrorType_name = map[int32]string{
5: "INCOMPATIBLE_VERSION",
6: "REGION_NOT_FOUND",
}
var ErrorType_value = map[string]int32{
"OK": 0,
"UNKNOWN": 1,
@ -77,21 +60,47 @@ var ErrorType_value = map[string]int32{
func (x ErrorType) String() string {
return proto.EnumName(ErrorType_name, int32(x))
}
func (ErrorType) EnumDescriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
func (ErrorType) EnumDescriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{0}
}
type Member struct {
// name is the name of the PD member.
Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// member_id is the unique id of the PD member.
MemberId uint64 `protobuf:"varint,2,opt,name=member_id,json=memberId" json:"member_id,omitempty"`
PeerUrls []string `protobuf:"bytes,3,rep,name=peer_urls,json=peerUrls" json:"peer_urls,omitempty"`
ClientUrls []string `protobuf:"bytes,4,rep,name=client_urls,json=clientUrls" json:"client_urls,omitempty"`
MemberId uint64 `protobuf:"varint,2,opt,name=member_id,json=memberId,proto3" json:"member_id,omitempty"`
PeerUrls []string `protobuf:"bytes,3,rep,name=peer_urls,json=peerUrls,proto3" json:"peer_urls,omitempty"`
ClientUrls []string `protobuf:"bytes,4,rep,name=client_urls,json=clientUrls,proto3" json:"client_urls,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *Member) Reset() { *m = Member{} }
func (m *Member) String() string { return proto.CompactTextString(m) }
func (*Member) ProtoMessage() {}
func (*Member) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
func (m *Member) Reset() { *m = Member{} }
func (m *Member) String() string { return proto.CompactTextString(m) }
func (*Member) ProtoMessage() {}
func (*Member) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{0}
}
func (m *Member) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_Member.Unmarshal(m, b)
}
func (m *Member) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_Member.Marshal(b, m, deterministic)
}
func (m *Member) XXX_Merge(src proto.Message) {
xxx_messageInfo_Member.Merge(m, src)
}
func (m *Member) XXX_Size() int {
return xxx_messageInfo_Member.Size(m)
}
func (m *Member) XXX_DiscardUnknown() {
xxx_messageInfo_Member.DiscardUnknown(m)
}
var xxx_messageInfo_Member proto.InternalMessageInfo
func (m *Member) GetName() string {
if m != nil {
@ -123,13 +132,36 @@ func (m *Member) GetClientUrls() []string {
type RequestHeader struct {
// cluster_id is the ID of the cluster which be sent to.
ClusterId uint64 `protobuf:"varint,1,opt,name=cluster_id,json=clusterId" json:"cluster_id,omitempty"`
ClusterId uint64 `protobuf:"varint,1,opt,name=cluster_id,json=clusterId,proto3" json:"cluster_id,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *RequestHeader) Reset() { *m = RequestHeader{} }
func (m *RequestHeader) String() string { return proto.CompactTextString(m) }
func (*RequestHeader) ProtoMessage() {}
func (*RequestHeader) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} }
func (m *RequestHeader) Reset() { *m = RequestHeader{} }
func (m *RequestHeader) String() string { return proto.CompactTextString(m) }
func (*RequestHeader) ProtoMessage() {}
func (*RequestHeader) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{1}
}
func (m *RequestHeader) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_RequestHeader.Unmarshal(m, b)
}
func (m *RequestHeader) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_RequestHeader.Marshal(b, m, deterministic)
}
func (m *RequestHeader) XXX_Merge(src proto.Message) {
xxx_messageInfo_RequestHeader.Merge(m, src)
}
func (m *RequestHeader) XXX_Size() int {
return xxx_messageInfo_RequestHeader.Size(m)
}
func (m *RequestHeader) XXX_DiscardUnknown() {
xxx_messageInfo_RequestHeader.DiscardUnknown(m)
}
var xxx_messageInfo_RequestHeader proto.InternalMessageInfo
func (m *RequestHeader) GetClusterId() uint64 {
if m != nil {
@ -140,14 +172,37 @@ func (m *RequestHeader) GetClusterId() uint64 {
type ResponseHeader struct {
// cluster_id is the ID of the cluster which sent the response.
ClusterId uint64 `protobuf:"varint,1,opt,name=cluster_id,json=clusterId" json:"cluster_id,omitempty"`
Error *Error `protobuf:"bytes,2,opt,name=error" json:"error,omitempty"`
ClusterId uint64 `protobuf:"varint,1,opt,name=cluster_id,json=clusterId,proto3" json:"cluster_id,omitempty"`
Error *Error `protobuf:"bytes,2,opt,name=error,proto3" json:"error,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *ResponseHeader) Reset() { *m = ResponseHeader{} }
func (m *ResponseHeader) String() string { return proto.CompactTextString(m) }
func (*ResponseHeader) ProtoMessage() {}
func (*ResponseHeader) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{2} }
func (m *ResponseHeader) Reset() { *m = ResponseHeader{} }
func (m *ResponseHeader) String() string { return proto.CompactTextString(m) }
func (*ResponseHeader) ProtoMessage() {}
func (*ResponseHeader) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{2}
}
func (m *ResponseHeader) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_ResponseHeader.Unmarshal(m, b)
}
func (m *ResponseHeader) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_ResponseHeader.Marshal(b, m, deterministic)
}
func (m *ResponseHeader) XXX_Merge(src proto.Message) {
xxx_messageInfo_ResponseHeader.Merge(m, src)
}
func (m *ResponseHeader) XXX_Size() int {
return xxx_messageInfo_ResponseHeader.Size(m)
}
func (m *ResponseHeader) XXX_DiscardUnknown() {
xxx_messageInfo_ResponseHeader.DiscardUnknown(m)
}
var xxx_messageInfo_ResponseHeader proto.InternalMessageInfo
func (m *ResponseHeader) GetClusterId() uint64 {
if m != nil {
@ -164,14 +219,37 @@ func (m *ResponseHeader) GetError() *Error {
}
type Error struct {
Type ErrorType `protobuf:"varint,1,opt,name=type,enum=pdpb.ErrorType" json:"type,omitempty"`
Message string `protobuf:"bytes,2,opt,name=message" json:"message,omitempty"`
Type ErrorType `protobuf:"varint,1,opt,name=type,proto3,enum=pdpb.ErrorType" json:"type,omitempty"`
Message string `protobuf:"bytes,2,opt,name=message,proto3" json:"message,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *Error) Reset() { *m = Error{} }
func (m *Error) String() string { return proto.CompactTextString(m) }
func (*Error) ProtoMessage() {}
func (*Error) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{3} }
func (m *Error) Reset() { *m = Error{} }
func (m *Error) String() string { return proto.CompactTextString(m) }
func (*Error) ProtoMessage() {}
func (*Error) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{3}
}
func (m *Error) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_Error.Unmarshal(m, b)
}
func (m *Error) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_Error.Marshal(b, m, deterministic)
}
func (m *Error) XXX_Merge(src proto.Message) {
xxx_messageInfo_Error.Merge(m, src)
}
func (m *Error) XXX_Size() int {
return xxx_messageInfo_Error.Size(m)
}
func (m *Error) XXX_DiscardUnknown() {
xxx_messageInfo_Error.DiscardUnknown(m)
}
var xxx_messageInfo_Error proto.InternalMessageInfo
func (m *Error) GetType() ErrorType {
if m != nil {
@ -188,14 +266,37 @@ func (m *Error) GetMessage() string {
}
type TsoRequest struct {
Header *RequestHeader `protobuf:"bytes,1,opt,name=header" json:"header,omitempty"`
Count uint32 `protobuf:"varint,2,opt,name=count" json:"count,omitempty"`
Header *RequestHeader `protobuf:"bytes,1,opt,name=header,proto3" json:"header,omitempty"`
Count uint32 `protobuf:"varint,2,opt,name=count,proto3" json:"count,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *TsoRequest) Reset() { *m = TsoRequest{} }
func (m *TsoRequest) String() string { return proto.CompactTextString(m) }
func (*TsoRequest) ProtoMessage() {}
func (*TsoRequest) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{4} }
func (m *TsoRequest) Reset() { *m = TsoRequest{} }
func (m *TsoRequest) String() string { return proto.CompactTextString(m) }
func (*TsoRequest) ProtoMessage() {}
func (*TsoRequest) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{4}
}
func (m *TsoRequest) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_TsoRequest.Unmarshal(m, b)
}
func (m *TsoRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_TsoRequest.Marshal(b, m, deterministic)
}
func (m *TsoRequest) XXX_Merge(src proto.Message) {
xxx_messageInfo_TsoRequest.Merge(m, src)
}
func (m *TsoRequest) XXX_Size() int {
return xxx_messageInfo_TsoRequest.Size(m)
}
func (m *TsoRequest) XXX_DiscardUnknown() {
xxx_messageInfo_TsoRequest.DiscardUnknown(m)
}
var xxx_messageInfo_TsoRequest proto.InternalMessageInfo
func (m *TsoRequest) GetHeader() *RequestHeader {
if m != nil {
@ -212,14 +313,37 @@ func (m *TsoRequest) GetCount() uint32 {
}
type Timestamp struct {
Physical int64 `protobuf:"varint,1,opt,name=physical" json:"physical,omitempty"`
Logical int64 `protobuf:"varint,2,opt,name=logical" json:"logical,omitempty"`
Physical int64 `protobuf:"varint,1,opt,name=physical,proto3" json:"physical,omitempty"`
Logical int64 `protobuf:"varint,2,opt,name=logical,proto3" json:"logical,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *Timestamp) Reset() { *m = Timestamp{} }
func (m *Timestamp) String() string { return proto.CompactTextString(m) }
func (*Timestamp) ProtoMessage() {}
func (*Timestamp) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{5} }
func (m *Timestamp) Reset() { *m = Timestamp{} }
func (m *Timestamp) String() string { return proto.CompactTextString(m) }
func (*Timestamp) ProtoMessage() {}
func (*Timestamp) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{5}
}
func (m *Timestamp) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_Timestamp.Unmarshal(m, b)
}
func (m *Timestamp) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_Timestamp.Marshal(b, m, deterministic)
}
func (m *Timestamp) XXX_Merge(src proto.Message) {
xxx_messageInfo_Timestamp.Merge(m, src)
}
func (m *Timestamp) XXX_Size() int {
return xxx_messageInfo_Timestamp.Size(m)
}
func (m *Timestamp) XXX_DiscardUnknown() {
xxx_messageInfo_Timestamp.DiscardUnknown(m)
}
var xxx_messageInfo_Timestamp proto.InternalMessageInfo
func (m *Timestamp) GetPhysical() int64 {
if m != nil {
@ -236,15 +360,38 @@ func (m *Timestamp) GetLogical() int64 {
}
type TsoResponse struct {
Header *ResponseHeader `protobuf:"bytes,1,opt,name=header" json:"header,omitempty"`
Count uint32 `protobuf:"varint,2,opt,name=count" json:"count,omitempty"`
Timestamp *Timestamp `protobuf:"bytes,3,opt,name=timestamp" json:"timestamp,omitempty"`
Header *ResponseHeader `protobuf:"bytes,1,opt,name=header,proto3" json:"header,omitempty"`
Count uint32 `protobuf:"varint,2,opt,name=count,proto3" json:"count,omitempty"`
Timestamp *Timestamp `protobuf:"bytes,3,opt,name=timestamp,proto3" json:"timestamp,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *TsoResponse) Reset() { *m = TsoResponse{} }
func (m *TsoResponse) String() string { return proto.CompactTextString(m) }
func (*TsoResponse) ProtoMessage() {}
func (*TsoResponse) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{6} }
func (m *TsoResponse) Reset() { *m = TsoResponse{} }
func (m *TsoResponse) String() string { return proto.CompactTextString(m) }
func (*TsoResponse) ProtoMessage() {}
func (*TsoResponse) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{6}
}
func (m *TsoResponse) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_TsoResponse.Unmarshal(m, b)
}
func (m *TsoResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_TsoResponse.Marshal(b, m, deterministic)
}
func (m *TsoResponse) XXX_Merge(src proto.Message) {
xxx_messageInfo_TsoResponse.Merge(m, src)
}
func (m *TsoResponse) XXX_Size() int {
return xxx_messageInfo_TsoResponse.Size(m)
}
func (m *TsoResponse) XXX_DiscardUnknown() {
xxx_messageInfo_TsoResponse.DiscardUnknown(m)
}
var xxx_messageInfo_TsoResponse proto.InternalMessageInfo
func (m *TsoResponse) GetHeader() *ResponseHeader {
if m != nil {
@ -268,13 +415,36 @@ func (m *TsoResponse) GetTimestamp() *Timestamp {
}
type AllocIDRequest struct {
Header *RequestHeader `protobuf:"bytes,1,opt,name=header" json:"header,omitempty"`
Header *RequestHeader `protobuf:"bytes,1,opt,name=header,proto3" json:"header,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *AllocIDRequest) Reset() { *m = AllocIDRequest{} }
func (m *AllocIDRequest) String() string { return proto.CompactTextString(m) }
func (*AllocIDRequest) ProtoMessage() {}
func (*AllocIDRequest) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{7} }
func (m *AllocIDRequest) Reset() { *m = AllocIDRequest{} }
func (m *AllocIDRequest) String() string { return proto.CompactTextString(m) }
func (*AllocIDRequest) ProtoMessage() {}
func (*AllocIDRequest) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{7}
}
func (m *AllocIDRequest) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_AllocIDRequest.Unmarshal(m, b)
}
func (m *AllocIDRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_AllocIDRequest.Marshal(b, m, deterministic)
}
func (m *AllocIDRequest) XXX_Merge(src proto.Message) {
xxx_messageInfo_AllocIDRequest.Merge(m, src)
}
func (m *AllocIDRequest) XXX_Size() int {
return xxx_messageInfo_AllocIDRequest.Size(m)
}
func (m *AllocIDRequest) XXX_DiscardUnknown() {
xxx_messageInfo_AllocIDRequest.DiscardUnknown(m)
}
var xxx_messageInfo_AllocIDRequest proto.InternalMessageInfo
func (m *AllocIDRequest) GetHeader() *RequestHeader {
if m != nil {
@ -284,14 +454,37 @@ func (m *AllocIDRequest) GetHeader() *RequestHeader {
}
type AllocIDResponse struct {
Header *ResponseHeader `protobuf:"bytes,1,opt,name=header" json:"header,omitempty"`
Id uint64 `protobuf:"varint,2,opt,name=id" json:"id,omitempty"`
Header *ResponseHeader `protobuf:"bytes,1,opt,name=header,proto3" json:"header,omitempty"`
Id uint64 `protobuf:"varint,2,opt,name=id,proto3" json:"id,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *AllocIDResponse) Reset() { *m = AllocIDResponse{} }
func (m *AllocIDResponse) String() string { return proto.CompactTextString(m) }
func (*AllocIDResponse) ProtoMessage() {}
func (*AllocIDResponse) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{8} }
func (m *AllocIDResponse) Reset() { *m = AllocIDResponse{} }
func (m *AllocIDResponse) String() string { return proto.CompactTextString(m) }
func (*AllocIDResponse) ProtoMessage() {}
func (*AllocIDResponse) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{8}
}
func (m *AllocIDResponse) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_AllocIDResponse.Unmarshal(m, b)
}
func (m *AllocIDResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_AllocIDResponse.Marshal(b, m, deterministic)
}
func (m *AllocIDResponse) XXX_Merge(src proto.Message) {
xxx_messageInfo_AllocIDResponse.Merge(m, src)
}
func (m *AllocIDResponse) XXX_Size() int {
return xxx_messageInfo_AllocIDResponse.Size(m)
}
func (m *AllocIDResponse) XXX_DiscardUnknown() {
xxx_messageInfo_AllocIDResponse.DiscardUnknown(m)
}
var xxx_messageInfo_AllocIDResponse proto.InternalMessageInfo
func (m *AllocIDResponse) GetHeader() *ResponseHeader {
if m != nil {
@ -308,14 +501,37 @@ func (m *AllocIDResponse) GetId() uint64 {
}
type HeartbeatRequest struct {
Header *RequestHeader `protobuf:"bytes,1,opt,name=header" json:"header,omitempty"`
Peer *metapb.Peer `protobuf:"bytes,2,opt,name=peer" json:"peer,omitempty"`
Header *RequestHeader `protobuf:"bytes,1,opt,name=header,proto3" json:"header,omitempty"`
Peer *metapb.Peer `protobuf:"bytes,2,opt,name=peer,proto3" json:"peer,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *HeartbeatRequest) Reset() { *m = HeartbeatRequest{} }
func (m *HeartbeatRequest) String() string { return proto.CompactTextString(m) }
func (*HeartbeatRequest) ProtoMessage() {}
func (*HeartbeatRequest) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{9} }
func (m *HeartbeatRequest) Reset() { *m = HeartbeatRequest{} }
func (m *HeartbeatRequest) String() string { return proto.CompactTextString(m) }
func (*HeartbeatRequest) ProtoMessage() {}
func (*HeartbeatRequest) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{9}
}
func (m *HeartbeatRequest) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_HeartbeatRequest.Unmarshal(m, b)
}
func (m *HeartbeatRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_HeartbeatRequest.Marshal(b, m, deterministic)
}
func (m *HeartbeatRequest) XXX_Merge(src proto.Message) {
xxx_messageInfo_HeartbeatRequest.Merge(m, src)
}
func (m *HeartbeatRequest) XXX_Size() int {
return xxx_messageInfo_HeartbeatRequest.Size(m)
}
func (m *HeartbeatRequest) XXX_DiscardUnknown() {
xxx_messageInfo_HeartbeatRequest.DiscardUnknown(m)
}
var xxx_messageInfo_HeartbeatRequest proto.InternalMessageInfo
func (m *HeartbeatRequest) GetHeader() *RequestHeader {
if m != nil {
@ -332,13 +548,36 @@ func (m *HeartbeatRequest) GetPeer() *metapb.Peer {
}
type HeartbeatResponse struct {
Header *ResponseHeader `protobuf:"bytes,1,opt,name=header" json:"header,omitempty"`
Header *ResponseHeader `protobuf:"bytes,1,opt,name=header,proto3" json:"header,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
}
func (m *HeartbeatResponse) Reset() { *m = HeartbeatResponse{} }
func (m *HeartbeatResponse) String() string { return proto.CompactTextString(m) }
func (*HeartbeatResponse) ProtoMessage() {}
func (*HeartbeatResponse) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{10} }
func (m *HeartbeatResponse) Reset() { *m = HeartbeatResponse{} }
func (m *HeartbeatResponse) String() string { return proto.CompactTextString(m) }
func (*HeartbeatResponse) ProtoMessage() {}
func (*HeartbeatResponse) Descriptor() ([]byte, []int) {
return fileDescriptor_78b27e6f04f44c6e, []int{10}
}
func (m *HeartbeatResponse) XXX_Unmarshal(b []byte) error {
return xxx_messageInfo_HeartbeatResponse.Unmarshal(m, b)
}
func (m *HeartbeatResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
return xxx_messageInfo_HeartbeatResponse.Marshal(b, m, deterministic)
}
func (m *HeartbeatResponse) XXX_Merge(src proto.Message) {
xxx_messageInfo_HeartbeatResponse.Merge(m, src)
}
func (m *HeartbeatResponse) XXX_Size() int {
return xxx_messageInfo_HeartbeatResponse.Size(m)
}
func (m *HeartbeatResponse) XXX_DiscardUnknown() {
xxx_messageInfo_HeartbeatResponse.DiscardUnknown(m)
}
var xxx_messageInfo_HeartbeatResponse proto.InternalMessageInfo
func (m *HeartbeatResponse) GetHeader() *ResponseHeader {
if m != nil {
@ -348,6 +587,7 @@ func (m *HeartbeatResponse) GetHeader() *ResponseHeader {
}
func init() {
proto.RegisterEnum("pdpb.ErrorType", ErrorType_name, ErrorType_value)
proto.RegisterType((*Member)(nil), "pdpb.Member")
proto.RegisterType((*RequestHeader)(nil), "pdpb.RequestHeader")
proto.RegisterType((*ResponseHeader)(nil), "pdpb.ResponseHeader")
@ -359,7 +599,50 @@ func init() {
proto.RegisterType((*AllocIDResponse)(nil), "pdpb.AllocIDResponse")
proto.RegisterType((*HeartbeatRequest)(nil), "pdpb.HeartbeatRequest")
proto.RegisterType((*HeartbeatResponse)(nil), "pdpb.HeartbeatResponse")
proto.RegisterEnum("pdpb.ErrorType", ErrorType_name, ErrorType_value)
}
func init() { proto.RegisterFile("pdpb.proto", fileDescriptor_78b27e6f04f44c6e) }
var fileDescriptor_78b27e6f04f44c6e = []byte{
// 603 bytes of a gzipped FileDescriptorProto
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x9c, 0x54, 0x4d, 0x6f, 0xd3, 0x40,
0x10, 0xad, 0x1d, 0x27, 0xad, 0x27, 0x6d, 0xea, 0x4e, 0x03, 0x44, 0x41, 0x88, 0xb0, 0x5c, 0x22,
0x3e, 0xa2, 0x2a, 0x5c, 0xb8, 0x70, 0x70, 0x88, 0x4b, 0xad, 0xb6, 0xde, 0x68, 0xe3, 0x82, 0x38,
0x45, 0x4e, 0xbc, 0x6a, 0x23, 0xd9, 0xb1, 0xf1, 0x3a, 0x42, 0xbd, 0xf1, 0x27, 0xf8, 0x27, 0xfc,
0x40, 0xe4, 0xf5, 0x07, 0x69, 0x90, 0x10, 0xca, 0xcd, 0xf3, 0xde, 0xec, 0x9b, 0x37, 0x1f, 0x09,
0x40, 0xec, 0xc7, 0xf3, 0x41, 0x9c, 0x44, 0x69, 0x84, 0x5a, 0xf6, 0xdd, 0x3d, 0x0c, 0x79, 0xea,
0x95, 0x18, 0xf9, 0x0e, 0x8d, 0x6b, 0x1e, 0xce, 0x79, 0x82, 0x08, 0xda, 0xca, 0x0b, 0x79, 0x47,
0xe9, 0x29, 0x7d, 0x9d, 0xc9, 0x6f, 0x7c, 0x0a, 0x7a, 0x28, 0xd9, 0xd9, 0xd2, 0xef, 0xa8, 0x3d,
0xa5, 0xaf, 0xb1, 0x83, 0x1c, 0xb0, 0xfd, 0x8c, 0x8c, 0x39, 0x4f, 0x66, 0xeb, 0x24, 0x10, 0x9d,
0x5a, 0xaf, 0xd6, 0xd7, 0xd9, 0x41, 0x06, 0xdc, 0x24, 0x81, 0xc0, 0xe7, 0xd0, 0x5c, 0x04, 0x4b,
0xbe, 0x4a, 0x73, 0x5a, 0x93, 0x34, 0xe4, 0x50, 0x96, 0x40, 0x06, 0x70, 0xc4, 0xf8, 0xb7, 0x35,
0x17, 0xe9, 0x05, 0xf7, 0x7c, 0x9e, 0xe0, 0x33, 0x80, 0x45, 0xb0, 0x16, 0x69, 0x5e, 0x4c, 0x91,
0xc5, 0xf4, 0x02, 0xb1, 0x7d, 0xc2, 0xa0, 0xc5, 0xb8, 0x88, 0xa3, 0x95, 0xe0, 0xff, 0xf5, 0x00,
0x5f, 0x40, 0x9d, 0x27, 0x49, 0x94, 0x48, 0xdf, 0xcd, 0x61, 0x73, 0x20, 0x27, 0x61, 0x65, 0x10,
0xcb, 0x19, 0x72, 0x0e, 0x75, 0x19, 0xe3, 0x4b, 0xd0, 0xd2, 0xfb, 0x38, 0xef, 0xbd, 0x35, 0x3c,
0xde, 0x48, 0x75, 0xef, 0x63, 0xce, 0x24, 0x89, 0x1d, 0xd8, 0x0f, 0xb9, 0x10, 0xde, 0x2d, 0x97,
0x92, 0x3a, 0x2b, 0x43, 0x42, 0x01, 0x5c, 0x11, 0x15, 0xed, 0xe0, 0x6b, 0x68, 0xdc, 0x49, 0x87,
0x52, 0xae, 0x39, 0x3c, 0xcd, 0xe5, 0x1e, 0x74, 0xcb, 0x8a, 0x14, 0x6c, 0x43, 0x7d, 0x11, 0xad,
0x57, 0xa9, 0x94, 0x3c, 0x62, 0x79, 0x40, 0x4c, 0xd0, 0xdd, 0x65, 0xc8, 0x45, 0xea, 0x85, 0x31,
0x76, 0xe1, 0x20, 0xbe, 0xbb, 0x17, 0xcb, 0x85, 0x17, 0x48, 0xc5, 0x1a, 0xab, 0xe2, 0xcc, 0x53,
0x10, 0xdd, 0x4a, 0x4a, 0x95, 0x54, 0x19, 0x92, 0x1f, 0x0a, 0x34, 0xa5, 0xa9, 0x7c, 0x66, 0xf8,
0x66, 0xcb, 0x55, 0xbb, 0x74, 0xb5, 0x39, 0xd3, 0x7f, 0xdb, 0xc2, 0xb7, 0xa0, 0xa7, 0xa5, 0xad,
0x4e, 0x4d, 0xca, 0x14, 0xb3, 0xaa, 0xdc, 0xb2, 0x3f, 0x19, 0xe4, 0x03, 0xb4, 0xcc, 0x20, 0x88,
0x16, 0xf6, 0x78, 0x97, 0xd1, 0x10, 0x0a, 0xc7, 0xd5, 0xf3, 0x9d, 0x9a, 0x68, 0x81, 0x5a, 0x9d,
0xad, 0xba, 0xf4, 0x89, 0x07, 0xc6, 0x05, 0xf7, 0x92, 0x74, 0xce, 0xbd, 0x74, 0xa7, 0x65, 0xf5,
0x40, 0xcb, 0x0e, 0xbc, 0xb8, 0xa8, 0xc3, 0x41, 0xf1, 0x4b, 0x9a, 0x70, 0x9e, 0x30, 0xc9, 0x10,
0x13, 0x4e, 0x36, 0x4a, 0xec, 0xe2, 0xfa, 0xd5, 0x4f, 0x05, 0xf4, 0xea, 0xf4, 0xb0, 0x01, 0x2a,
0xbd, 0x34, 0xf6, 0xb0, 0x09, 0xfb, 0x37, 0xce, 0xa5, 0x43, 0xbf, 0x38, 0x86, 0x82, 0x6d, 0x30,
0x1c, 0xea, 0xce, 0x46, 0x94, 0xba, 0x53, 0x97, 0x99, 0x93, 0x89, 0x35, 0x36, 0x54, 0x3c, 0x85,
0xe3, 0xa9, 0x4b, 0x99, 0x35, 0x73, 0xe9, 0xf5, 0x68, 0xea, 0x52, 0xc7, 0x32, 0x6a, 0xd8, 0x81,
0xb6, 0x79, 0xc5, 0x2c, 0x73, 0xfc, 0xf5, 0x61, 0xba, 0x96, 0x31, 0xb6, 0xf3, 0x91, 0x5e, 0x4f,
0x4c, 0xd7, 0x1e, 0x5d, 0x59, 0xb3, 0xcf, 0x16, 0x9b, 0xda, 0xd4, 0x31, 0xea, 0x99, 0x3c, 0xb3,
0x3e, 0xd9, 0xd4, 0x99, 0x65, 0x55, 0xce, 0xe9, 0x8d, 0x33, 0x36, 0x1a, 0xc3, 0x5f, 0x0a, 0xa8,
0x93, 0x31, 0x9e, 0x41, 0xcd, 0x15, 0x11, 0x1a, 0xc5, 0xde, 0xab, 0xb3, 0xef, 0x9e, 0x6c, 0x20,
0x79, 0x63, 0x64, 0xaf, 0xaf, 0x9c, 0x29, 0xf8, 0x1e, 0xf6, 0x8b, 0x3d, 0x62, 0xd1, 0xf9, 0xc3,
0xab, 0xe8, 0x3e, 0xda, 0x42, 0xcb, 0xd7, 0x38, 0x02, 0xbd, 0x9a, 0x26, 0x3e, 0xce, 0xb3, 0xb6,
0x37, 0xd8, 0x7d, 0xf2, 0x17, 0xbe, 0x59, 0x7d, 0xde, 0x90, 0xff, 0x73, 0xef, 0x7e, 0x07, 0x00,
0x00, 0xff, 0xff, 0x34, 0x73, 0xe9, 0xc4, 0x09, 0x05, 0x00, 0x00,
}
// Reference imports to suppress errors if they are not otherwise used.
@ -370,8 +653,9 @@ var _ grpc.ClientConn
// is compatible with the grpc package it is being compiled against.
const _ = grpc.SupportPackageIsVersion4
// Client API for PD service
// PDClient is the client API for PD service.
//
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://godoc.org/google.golang.org/grpc#ClientConn.NewStream.
type PDClient interface {
Tso(ctx context.Context, opts ...grpc.CallOption) (PD_TsoClient, error)
AllocID(ctx context.Context, in *AllocIDRequest, opts ...grpc.CallOption) (*AllocIDResponse, error)
@ -387,7 +671,7 @@ func NewPDClient(cc *grpc.ClientConn) PDClient {
}
func (c *pDClient) Tso(ctx context.Context, opts ...grpc.CallOption) (PD_TsoClient, error) {
stream, err := grpc.NewClientStream(ctx, &_PD_serviceDesc.Streams[0], c.cc, "/pdpb.PD/Tso", opts...)
stream, err := c.cc.NewStream(ctx, &_PD_serviceDesc.Streams[0], "/pdpb.PD/Tso", opts...)
if err != nil {
return nil, err
}
@ -419,7 +703,7 @@ func (x *pDTsoClient) Recv() (*TsoResponse, error) {
func (c *pDClient) AllocID(ctx context.Context, in *AllocIDRequest, opts ...grpc.CallOption) (*AllocIDResponse, error) {
out := new(AllocIDResponse)
err := grpc.Invoke(ctx, "/pdpb.PD/AllocID", in, out, c.cc, opts...)
err := c.cc.Invoke(ctx, "/pdpb.PD/AllocID", in, out, opts...)
if err != nil {
return nil, err
}
@ -427,7 +711,7 @@ func (c *pDClient) AllocID(ctx context.Context, in *AllocIDRequest, opts ...grpc
}
func (c *pDClient) Heartbeat(ctx context.Context, opts ...grpc.CallOption) (PD_HeartbeatClient, error) {
stream, err := grpc.NewClientStream(ctx, &_PD_serviceDesc.Streams[1], c.cc, "/pdpb.PD/Heartbeat", opts...)
stream, err := c.cc.NewStream(ctx, &_PD_serviceDesc.Streams[1], "/pdpb.PD/Heartbeat", opts...)
if err != nil {
return nil, err
}
@ -457,14 +741,27 @@ func (x *pDHeartbeatClient) Recv() (*HeartbeatResponse, error) {
return m, nil
}
// Server API for PD service
// PDServer is the server API for PD service.
type PDServer interface {
Tso(PD_TsoServer) error
AllocID(context.Context, *AllocIDRequest) (*AllocIDResponse, error)
Heartbeat(PD_HeartbeatServer) error
}
// UnimplementedPDServer can be embedded to have forward compatible implementations.
type UnimplementedPDServer struct {
}
func (*UnimplementedPDServer) Tso(srv PD_TsoServer) error {
return status.Errorf(codes.Unimplemented, "method Tso not implemented")
}
func (*UnimplementedPDServer) AllocID(ctx context.Context, req *AllocIDRequest) (*AllocIDResponse, error) {
return nil, status.Errorf(codes.Unimplemented, "method AllocID not implemented")
}
func (*UnimplementedPDServer) Heartbeat(srv PD_HeartbeatServer) error {
return status.Errorf(codes.Unimplemented, "method Heartbeat not implemented")
}
func RegisterPDServer(s *grpc.Server, srv PDServer) {
s.RegisterService(&_PD_serviceDesc, srv)
}
@ -564,47 +861,3 @@ var _PD_serviceDesc = grpc.ServiceDesc{
},
Metadata: "pdpb.proto",
}
func init() { proto.RegisterFile("pdpb.proto", fileDescriptor0) }
var fileDescriptor0 = []byte{
// 603 bytes of a gzipped FileDescriptorProto
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x9c, 0x54, 0x4d, 0x6f, 0xd3, 0x40,
0x10, 0xad, 0x1d, 0x27, 0xad, 0x27, 0x6d, 0xea, 0x4e, 0x03, 0x44, 0x41, 0x88, 0xb0, 0x5c, 0x22,
0x3e, 0xa2, 0x2a, 0x5c, 0xb8, 0x70, 0x70, 0x88, 0x4b, 0xad, 0xb6, 0xde, 0x68, 0xe3, 0x82, 0x38,
0x45, 0x4e, 0xbc, 0x6a, 0x23, 0xd9, 0xb1, 0xf1, 0x3a, 0x42, 0xbd, 0xf1, 0x27, 0xf8, 0x27, 0xfc,
0x40, 0xe4, 0xf5, 0x07, 0x69, 0x90, 0x10, 0xca, 0xcd, 0xf3, 0xde, 0xec, 0x9b, 0x37, 0x1f, 0x09,
0x40, 0xec, 0xc7, 0xf3, 0x41, 0x9c, 0x44, 0x69, 0x84, 0x5a, 0xf6, 0xdd, 0x3d, 0x0c, 0x79, 0xea,
0x95, 0x18, 0xf9, 0x0e, 0x8d, 0x6b, 0x1e, 0xce, 0x79, 0x82, 0x08, 0xda, 0xca, 0x0b, 0x79, 0x47,
0xe9, 0x29, 0x7d, 0x9d, 0xc9, 0x6f, 0x7c, 0x0a, 0x7a, 0x28, 0xd9, 0xd9, 0xd2, 0xef, 0xa8, 0x3d,
0xa5, 0xaf, 0xb1, 0x83, 0x1c, 0xb0, 0xfd, 0x8c, 0x8c, 0x39, 0x4f, 0x66, 0xeb, 0x24, 0x10, 0x9d,
0x5a, 0xaf, 0xd6, 0xd7, 0xd9, 0x41, 0x06, 0xdc, 0x24, 0x81, 0xc0, 0xe7, 0xd0, 0x5c, 0x04, 0x4b,
0xbe, 0x4a, 0x73, 0x5a, 0x93, 0x34, 0xe4, 0x50, 0x96, 0x40, 0x06, 0x70, 0xc4, 0xf8, 0xb7, 0x35,
0x17, 0xe9, 0x05, 0xf7, 0x7c, 0x9e, 0xe0, 0x33, 0x80, 0x45, 0xb0, 0x16, 0x69, 0x5e, 0x4c, 0x91,
0xc5, 0xf4, 0x02, 0xb1, 0x7d, 0xc2, 0xa0, 0xc5, 0xb8, 0x88, 0xa3, 0x95, 0xe0, 0xff, 0xf5, 0x00,
0x5f, 0x40, 0x9d, 0x27, 0x49, 0x94, 0x48, 0xdf, 0xcd, 0x61, 0x73, 0x20, 0x27, 0x61, 0x65, 0x10,
0xcb, 0x19, 0x72, 0x0e, 0x75, 0x19, 0xe3, 0x4b, 0xd0, 0xd2, 0xfb, 0x38, 0xef, 0xbd, 0x35, 0x3c,
0xde, 0x48, 0x75, 0xef, 0x63, 0xce, 0x24, 0x89, 0x1d, 0xd8, 0x0f, 0xb9, 0x10, 0xde, 0x2d, 0x97,
0x92, 0x3a, 0x2b, 0x43, 0x42, 0x01, 0x5c, 0x11, 0x15, 0xed, 0xe0, 0x6b, 0x68, 0xdc, 0x49, 0x87,
0x52, 0xae, 0x39, 0x3c, 0xcd, 0xe5, 0x1e, 0x74, 0xcb, 0x8a, 0x14, 0x6c, 0x43, 0x7d, 0x11, 0xad,
0x57, 0xa9, 0x94, 0x3c, 0x62, 0x79, 0x40, 0x4c, 0xd0, 0xdd, 0x65, 0xc8, 0x45, 0xea, 0x85, 0x31,
0x76, 0xe1, 0x20, 0xbe, 0xbb, 0x17, 0xcb, 0x85, 0x17, 0x48, 0xc5, 0x1a, 0xab, 0xe2, 0xcc, 0x53,
0x10, 0xdd, 0x4a, 0x4a, 0x95, 0x54, 0x19, 0x92, 0x1f, 0x0a, 0x34, 0xa5, 0xa9, 0x7c, 0x66, 0xf8,
0x66, 0xcb, 0x55, 0xbb, 0x74, 0xb5, 0x39, 0xd3, 0x7f, 0xdb, 0xc2, 0xb7, 0xa0, 0xa7, 0xa5, 0xad,
0x4e, 0x4d, 0xca, 0x14, 0xb3, 0xaa, 0xdc, 0xb2, 0x3f, 0x19, 0xe4, 0x03, 0xb4, 0xcc, 0x20, 0x88,
0x16, 0xf6, 0x78, 0x97, 0xd1, 0x10, 0x0a, 0xc7, 0xd5, 0xf3, 0x9d, 0x9a, 0x68, 0x81, 0x5a, 0x9d,
0xad, 0xba, 0xf4, 0x89, 0x07, 0xc6, 0x05, 0xf7, 0x92, 0x74, 0xce, 0xbd, 0x74, 0xa7, 0x65, 0xf5,
0x40, 0xcb, 0x0e, 0xbc, 0xb8, 0xa8, 0xc3, 0x41, 0xf1, 0x4b, 0x9a, 0x70, 0x9e, 0x30, 0xc9, 0x10,
0x13, 0x4e, 0x36, 0x4a, 0xec, 0xe2, 0xfa, 0xd5, 0x4f, 0x05, 0xf4, 0xea, 0xf4, 0xb0, 0x01, 0x2a,
0xbd, 0x34, 0xf6, 0xb0, 0x09, 0xfb, 0x37, 0xce, 0xa5, 0x43, 0xbf, 0x38, 0x86, 0x82, 0x6d, 0x30,
0x1c, 0xea, 0xce, 0x46, 0x94, 0xba, 0x53, 0x97, 0x99, 0x93, 0x89, 0x35, 0x36, 0x54, 0x3c, 0x85,
0xe3, 0xa9, 0x4b, 0x99, 0x35, 0x73, 0xe9, 0xf5, 0x68, 0xea, 0x52, 0xc7, 0x32, 0x6a, 0xd8, 0x81,
0xb6, 0x79, 0xc5, 0x2c, 0x73, 0xfc, 0xf5, 0x61, 0xba, 0x96, 0x31, 0xb6, 0xf3, 0x91, 0x5e, 0x4f,
0x4c, 0xd7, 0x1e, 0x5d, 0x59, 0xb3, 0xcf, 0x16, 0x9b, 0xda, 0xd4, 0x31, 0xea, 0x99, 0x3c, 0xb3,
0x3e, 0xd9, 0xd4, 0x99, 0x65, 0x55, 0xce, 0xe9, 0x8d, 0x33, 0x36, 0x1a, 0xc3, 0x5f, 0x0a, 0xa8,
0x93, 0x31, 0x9e, 0x41, 0xcd, 0x15, 0x11, 0x1a, 0xc5, 0xde, 0xab, 0xb3, 0xef, 0x9e, 0x6c, 0x20,
0x79, 0x63, 0x64, 0xaf, 0xaf, 0x9c, 0x29, 0xf8, 0x1e, 0xf6, 0x8b, 0x3d, 0x62, 0xd1, 0xf9, 0xc3,
0xab, 0xe8, 0x3e, 0xda, 0x42, 0xcb, 0xd7, 0x38, 0x02, 0xbd, 0x9a, 0x26, 0x3e, 0xce, 0xb3, 0xb6,
0x37, 0xd8, 0x7d, 0xf2, 0x17, 0xbe, 0x59, 0x7d, 0xde, 0x90, 0xff, 0x73, 0xef, 0x7e, 0x07, 0x00,
0x00, 0xff, 0xff, 0x34, 0x73, 0xe9, 0xc4, 0x09, 0x05, 0x00, 0x00,
}

2
proto/message.proto
View File

@ -208,6 +208,8 @@ message QueryResult {
repeated float scores = 4;
repeated float distances = 5;
repeated KeyValuePair extra_params = 6;
int64 query_id = 7;
int64 client_id = 8;
}
/**

1093
proxy/src/grpc/gen-milvus/milvus.grpc.pb.cc Normal file
View File

File diff suppressed because it is too large Load Diff

4559
proxy/src/grpc/gen-milvus/milvus.grpc.pb.h Normal file
View File

File diff suppressed because it is too large Load Diff

16962
proxy/src/grpc/gen-milvus/milvus.pb.cc Normal file
View File

File diff suppressed because it is too large Load Diff

11538
proxy/src/grpc/gen-milvus/milvus.pb.h Normal file
View File

File diff suppressed because it is too large Load Diff

485
proxy/src/grpc/message.pb.cc
View File

@ -1089,6 +1089,8 @@ const ::PROTOBUF_NAMESPACE_ID::uint32 TableStruct_message_2eproto::offsets[] PRO
PROTOBUF_FIELD_OFFSET(::milvus::grpc::QueryResult, scores_),
PROTOBUF_FIELD_OFFSET(::milvus::grpc::QueryResult, distances_),
PROTOBUF_FIELD_OFFSET(::milvus::grpc::QueryResult, extra_params_),
PROTOBUF_FIELD_OFFSET(::milvus::grpc::QueryResult, query_id_),
PROTOBUF_FIELD_OFFSET(::milvus::grpc::QueryResult, client_id_),
~0u, // no _has_bits_
PROTOBUF_FIELD_OFFSET(::milvus::grpc::StringReply, _internal_metadata_),
~0u, // no _extensions_
@ -1326,32 +1328,32 @@ static const ::PROTOBUF_NAMESPACE_ID::internal::MigrationSchema schemas[] PROTOB
{ 132, -1, sizeof(::milvus::grpc::SearchInSegmentParam)},
{ 139, -1, sizeof(::milvus::grpc::Entities)},
{ 148, -1, sizeof(::milvus::grpc::QueryResult)},
{ 159, -1, sizeof(::milvus::grpc::StringReply)},
{ 166, -1, sizeof(::milvus::grpc::BoolReply)},
{ 173, -1, sizeof(::milvus::grpc::CollectionRowCount)},
{ 180, -1, sizeof(::milvus::grpc::Command)},
{ 186, -1, sizeof(::milvus::grpc::IndexParam)},
{ 196, -1, sizeof(::milvus::grpc::FlushParam)},
{ 202, -1, sizeof(::milvus::grpc::CompactParam)},
{ 209, -1, sizeof(::milvus::grpc::DeleteByIDParam)},
{ 216, -1, sizeof(::milvus::grpc::CollectionInfo)},
{ 223, -1, sizeof(::milvus::grpc::GetEntityIDsParam)},
{ 230, -1, sizeof(::milvus::grpc::EntityIdentity)},
{ 238, -1, sizeof(::milvus::grpc::VectorFieldParam)},
{ 244, -1, sizeof(::milvus::grpc::FieldType)},
{ 252, -1, sizeof(::milvus::grpc::FieldParam)},
{ 262, -1, sizeof(::milvus::grpc::VectorFieldRecord)},
{ 268, -1, sizeof(::milvus::grpc::TermQuery)},
{ 279, -1, sizeof(::milvus::grpc::CompareExpr)},
{ 286, -1, sizeof(::milvus::grpc::RangeQuery)},
{ 295, -1, sizeof(::milvus::grpc::VectorQuery)},
{ 305, -1, sizeof(::milvus::grpc::BooleanQuery)},
{ 312, -1, sizeof(::milvus::grpc::GeneralQuery)},
{ 322, -1, sizeof(::milvus::grpc::SearchParamPB)},
{ 331, -1, sizeof(::milvus::grpc::InsertOrDeleteMsg)},
{ 346, -1, sizeof(::milvus::grpc::SearchMsg)},
{ 358, -1, sizeof(::milvus::grpc::TimeSyncMsg)},
{ 366, -1, sizeof(::milvus::grpc::Key2SegMsg)},
{ 161, -1, sizeof(::milvus::grpc::StringReply)},
{ 168, -1, sizeof(::milvus::grpc::BoolReply)},
{ 175, -1, sizeof(::milvus::grpc::CollectionRowCount)},
{ 182, -1, sizeof(::milvus::grpc::Command)},
{ 188, -1, sizeof(::milvus::grpc::IndexParam)},
{ 198, -1, sizeof(::milvus::grpc::FlushParam)},
{ 204, -1, sizeof(::milvus::grpc::CompactParam)},
{ 211, -1, sizeof(::milvus::grpc::DeleteByIDParam)},
{ 218, -1, sizeof(::milvus::grpc::CollectionInfo)},
{ 225, -1, sizeof(::milvus::grpc::GetEntityIDsParam)},
{ 232, -1, sizeof(::milvus::grpc::EntityIdentity)},
{ 240, -1, sizeof(::milvus::grpc::VectorFieldParam)},
{ 246, -1, sizeof(::milvus::grpc::FieldType)},
{ 254, -1, sizeof(::milvus::grpc::FieldParam)},
{ 264, -1, sizeof(::milvus::grpc::VectorFieldRecord)},
{ 270, -1, sizeof(::milvus::grpc::TermQuery)},
{ 281, -1, sizeof(::milvus::grpc::CompareExpr)},
{ 288, -1, sizeof(::milvus::grpc::RangeQuery)},
{ 297, -1, sizeof(::milvus::grpc::VectorQuery)},
{ 307, -1, sizeof(::milvus::grpc::BooleanQuery)},
{ 314, -1, sizeof(::milvus::grpc::GeneralQuery)},
{ 324, -1, sizeof(::milvus::grpc::SearchParamPB)},
{ 333, -1, sizeof(::milvus::grpc::InsertOrDeleteMsg)},
{ 348, -1, sizeof(::milvus::grpc::SearchMsg)},
{ 360, -1, sizeof(::milvus::grpc::TimeSyncMsg)},
{ 368, -1, sizeof(::milvus::grpc::Key2SegMsg)},
};
static ::PROTOBUF_NAMESPACE_ID::Message const * const file_default_instances[] = {
@ -1449,150 +1451,151 @@ const char descriptor_table_protodef_message_2eproto[] PROTOBUF_SECTION_VARIABLE
"Param\"x\n\010Entities\022#\n\006status\030\001 \001(\0132\023.milv"
"us.grpc.Status\022\013\n\003ids\030\002 \003(\003\022\021\n\tvalid_row"
"\030\003 \003(\010\022\'\n\trows_data\030\004 \003(\0132\024.milvus.grpc."
"RowData\"\300\001\n\013QueryResult\022#\n\006status\030\001 \001(\0132"
"RowData\"\345\001\n\013QueryResult\022#\n\006status\030\001 \001(\0132"
"\023.milvus.grpc.Status\022\'\n\010entities\030\002 \001(\0132\025"
".milvus.grpc.Entities\022\017\n\007row_num\030\003 \001(\003\022\016"
"\n\006scores\030\004 \003(\002\022\021\n\tdistances\030\005 \003(\002\022/\n\014ext"
"ra_params\030\006 \003(\0132\031.milvus.grpc.KeyValuePa"
"ir\"H\n\013StringReply\022#\n\006status\030\001 \001(\0132\023.milv"
"us.grpc.Status\022\024\n\014string_reply\030\002 \001(\t\"D\n\t"
"BoolReply\022#\n\006status\030\001 \001(\0132\023.milvus.grpc."
"Status\022\022\n\nbool_reply\030\002 \001(\010\"W\n\022Collection"
"RowCount\022#\n\006status\030\001 \001(\0132\023.milvus.grpc.S"
"tatus\022\034\n\024collection_row_count\030\002 \001(\003\"\026\n\007C"
"ommand\022\013\n\003cmd\030\001 \001(\t\"\243\001\n\nIndexParam\022#\n\006st"
"atus\030\001 \001(\0132\023.milvus.grpc.Status\022\027\n\017colle"
"ction_name\030\002 \001(\t\022\022\n\nfield_name\030\003 \001(\t\022\022\n\n"
"index_name\030\004 \001(\t\022/\n\014extra_params\030\005 \003(\0132\031"
".milvus.grpc.KeyValuePair\"+\n\nFlushParam\022"
"\035\n\025collection_name_array\030\001 \003(\t\":\n\014Compac"
"tParam\022\027\n\017collection_name\030\001 \001(\t\022\021\n\tthres"
"hold\030\002 \001(\001\"<\n\017DeleteByIDParam\022\027\n\017collect"
"ion_name\030\001 \001(\t\022\020\n\010id_array\030\002 \003(\003\"H\n\016Coll"
"ectionInfo\022#\n\006status\030\001 \001(\0132\023.milvus.grpc"
".Status\022\021\n\tjson_info\030\002 \001(\t\"@\n\021GetEntityI"
"DsParam\022\027\n\017collection_name\030\001 \001(\t\022\022\n\nsegm"
"ent_id\030\002 \001(\003\"P\n\016EntityIdentity\022\027\n\017collec"
"tion_name\030\001 \001(\t\022\020\n\010id_array\030\002 \003(\003\022\023\n\013fie"
"ld_names\030\003 \003(\t\"%\n\020VectorFieldParam\022\021\n\tdi"
"mension\030\001 \001(\003\"w\n\tFieldType\022*\n\tdata_type\030"
"\001 \001(\0162\025.milvus.grpc.DataTypeH\000\0225\n\014vector"
"_param\030\002 \001(\0132\035.milvus.grpc.VectorFieldPa"
"ramH\000B\007\n\005value\"\255\001\n\nFieldParam\022\n\n\002id\030\001 \001("
"\004\022\014\n\004name\030\002 \001(\t\022#\n\004type\030\003 \001(\0162\025.milvus.g"
"rpc.DataType\022/\n\014index_params\030\004 \003(\0132\031.mil"
"vus.grpc.KeyValuePair\022/\n\014extra_params\030\005 "
"\003(\0132\031.milvus.grpc.KeyValuePair\"@\n\021Vector"
"FieldRecord\022+\n\005value\030\001 \003(\0132\034.milvus.grpc"
".VectorRowRecord\"\233\001\n\tTermQuery\022\022\n\nfield_"
"name\030\001 \001(\t\022\021\n\tint_value\030\002 \003(\003\022\024\n\014double_"
"value\030\003 \003(\001\022\021\n\tvalue_num\030\004 \001(\003\022\r\n\005boost\030"
"\005 \001(\002\022/\n\014extra_params\030\006 \003(\0132\031.milvus.grp"
"c.KeyValuePair\"N\n\013CompareExpr\022.\n\010operato"
"r\030\001 \001(\0162\034.milvus.grpc.CompareOperator\022\017\n"
"\007operand\030\002 \001(\t\"\213\001\n\nRangeQuery\022\022\n\nfield_n"
"ame\030\001 \001(\t\022)\n\007operand\030\002 \003(\0132\030.milvus.grpc"
".CompareExpr\022\r\n\005boost\030\003 \001(\002\022/\n\014extra_par"
"ams\030\004 \003(\0132\031.milvus.grpc.KeyValuePair\"\244\001\n"
"\013VectorQuery\022\022\n\nfield_name\030\001 \001(\t\022\023\n\013quer"
"y_boost\030\002 \001(\002\022-\n\007records\030\003 \003(\0132\034.milvus."
"grpc.VectorRowRecord\022\014\n\004topk\030\004 \001(\003\022/\n\014ex"
"tra_params\030\005 \003(\0132\031.milvus.grpc.KeyValueP"
"air\"c\n\014BooleanQuery\022!\n\005occur\030\001 \001(\0162\022.mil"
"vus.grpc.Occur\0220\n\rgeneral_query\030\002 \003(\0132\031."
"milvus.grpc.GeneralQuery\"\333\001\n\014GeneralQuer"
"y\0222\n\rboolean_query\030\001 \001(\0132\031.milvus.grpc.B"
"ooleanQueryH\000\022,\n\nterm_query\030\002 \001(\0132\026.milv"
"us.grpc.TermQueryH\000\022.\n\013range_query\030\003 \001(\013"
"2\027.milvus.grpc.RangeQueryH\000\0220\n\014vector_qu"
"ery\030\004 \001(\0132\030.milvus.grpc.VectorQueryH\000B\007\n"
"\005query\"\250\001\n\rSearchParamPB\022\027\n\017collection_n"
"ame\030\001 \001(\t\022\033\n\023partition_tag_array\030\002 \003(\t\0220"
"\n\rgeneral_query\030\003 \001(\0132\031.milvus.grpc.Gene"
"ralQuery\022/\n\014extra_params\030\004 \003(\0132\031.milvus."
"grpc.KeyValuePair\"\231\002\n\021InsertOrDeleteMsg\022"
"\027\n\017collection_name\030\001 \001(\t\022\'\n\trows_data\030\002 "
"\001(\0132\024.milvus.grpc.RowData\022\013\n\003uid\030\003 \001(\003\022\025"
"\n\rpartition_tag\030\004 \001(\t\022\021\n\ttimestamp\030\005 \001(\003"
"\022\022\n\nsegment_id\030\006 \001(\003\022\022\n\nchannel_id\030\007 \001(\003"
"\022\037\n\002op\030\010 \001(\0162\023.milvus.grpc.OpType\022\021\n\tcli"
"ent_id\030\t \001(\003\022/\n\014extra_params\030\n \003(\0132\031.mil"
"vus.grpc.KeyValuePair\"\316\001\n\tSearchMsg\022\027\n\017c"
"ollection_name\030\001 \001(\t\022-\n\007records\030\002 \001(\0132\034."
"milvus.grpc.VectorRowRecord\022\025\n\rpartition"
"_tag\030\003 \001(\t\022\013\n\003uid\030\004 \001(\003\022\021\n\ttimestamp\030\005 \001"
"(\003\022\021\n\tclient_id\030\006 \001(\003\022/\n\014extra_params\030\007 "
"\003(\0132\031.milvus.grpc.KeyValuePair\"[\n\013TimeSy"
"ncMsg\022\017\n\007peer_Id\030\001 \001(\003\022\021\n\tTimestamp\030\002 \001("
"\003\022(\n\tsync_type\030\003 \001(\0162\025.milvus.grpc.SyncT"
"ype\"-\n\nKey2SegMsg\022\013\n\003uid\030\001 \001(\003\022\022\n\nsegmen"
"t_id\030\002 \003(\003*\242\004\n\tErrorCode\022\013\n\007SUCCESS\020\000\022\024\n"
"\020UNEXPECTED_ERROR\020\001\022\022\n\016CONNECT_FAILED\020\002\022"
"\025\n\021PERMISSION_DENIED\020\003\022\031\n\025COLLECTION_NOT"
"_EXISTS\020\004\022\024\n\020ILLEGAL_ARGUMENT\020\005\022\025\n\021ILLEG"
"AL_DIMENSION\020\007\022\026\n\022ILLEGAL_INDEX_TYPE\020\010\022\033"
"\n\027ILLEGAL_COLLECTION_NAME\020\t\022\020\n\014ILLEGAL_T"
"OPK\020\n\022\025\n\021ILLEGAL_ROWRECORD\020\013\022\025\n\021ILLEGAL_"
"VECTOR_ID\020\014\022\031\n\025ILLEGAL_SEARCH_RESULT\020\r\022\022"
"\n\016FILE_NOT_FOUND\020\016\022\017\n\013META_FAILED\020\017\022\020\n\014C"
"ACHE_FAILED\020\020\022\030\n\024CANNOT_CREATE_FOLDER\020\021\022"
"\026\n\022CANNOT_CREATE_FILE\020\022\022\030\n\024CANNOT_DELETE"
"_FOLDER\020\023\022\026\n\022CANNOT_DELETE_FILE\020\024\022\025\n\021BUI"
"LD_INDEX_ERROR\020\025\022\021\n\rILLEGAL_NLIST\020\026\022\027\n\023I"
"LLEGAL_METRIC_TYPE\020\027\022\021\n\rOUT_OF_MEMORY\020\030*"
"\221\001\n\010DataType\022\010\n\004NONE\020\000\022\010\n\004BOOL\020\001\022\010\n\004INT8"
"\020\002\022\t\n\005INT16\020\003\022\t\n\005INT32\020\004\022\t\n\005INT64\020\005\022\t\n\005F"
"LOAT\020\n\022\n\n\006DOUBLE\020\013\022\n\n\006STRING\020\024\022\021\n\rVECTOR"
"_BINARY\020d\022\020\n\014VECTOR_FLOAT\020e*C\n\017CompareOp"
"erator\022\006\n\002LT\020\000\022\007\n\003LTE\020\001\022\006\n\002EQ\020\002\022\006\n\002GT\020\003\022"
"\007\n\003GTE\020\004\022\006\n\002NE\020\005*8\n\005Occur\022\013\n\007INVALID\020\000\022\010"
"\n\004MUST\020\001\022\n\n\006SHOULD\020\002\022\014\n\010MUST_NOT\020\003* \n\006Op"
"Type\022\n\n\006INSERT\020\000\022\n\n\006DELETE\020\001*\037\n\010SyncType"
"\022\010\n\004READ\020\000\022\t\n\005WRITE\020\0012\333\r\n\rMilvusService\022"
"\?\n\020CreateCollection\022\024.milvus.grpc.Mappin"
"g\032\023.milvus.grpc.Status\"\000\022F\n\rHasCollectio"
"n\022\033.milvus.grpc.CollectionName\032\026.milvus."
"grpc.BoolReply\"\000\022I\n\022DescribeCollection\022\033"
".milvus.grpc.CollectionName\032\024.milvus.grp"
"c.Mapping\"\000\022Q\n\017CountCollection\022\033.milvus."
"grpc.CollectionName\032\037.milvus.grpc.Collec"
"tionRowCount\"\000\022J\n\017ShowCollections\022\024.milv"
"us.grpc.Command\032\037.milvus.grpc.Collection"
"NameList\"\000\022P\n\022ShowCollectionInfo\022\033.milvu"
"s.grpc.CollectionName\032\033.milvus.grpc.Coll"
"ectionInfo\"\000\022D\n\016DropCollection\022\033.milvus."
"grpc.CollectionName\032\023.milvus.grpc.Status"
"\"\000\022=\n\013CreateIndex\022\027.milvus.grpc.IndexPar"
"am\032\023.milvus.grpc.Status\"\000\022C\n\rDescribeInd"
"ex\022\027.milvus.grpc.IndexParam\032\027.milvus.grp"
"c.IndexParam\"\000\022;\n\tDropIndex\022\027.milvus.grp"
"c.IndexParam\032\023.milvus.grpc.Status\"\000\022E\n\017C"
"reatePartition\022\033.milvus.grpc.PartitionPa"
"ram\032\023.milvus.grpc.Status\"\000\022E\n\014HasPartiti"
"on\022\033.milvus.grpc.PartitionParam\032\026.milvus"
".grpc.BoolReply\"\000\022K\n\016ShowPartitions\022\033.mi"
"lvus.grpc.CollectionName\032\032.milvus.grpc.P"
"artitionList\"\000\022C\n\rDropPartition\022\033.milvus"
".grpc.PartitionParam\032\023.milvus.grpc.Statu"
"s\"\000\022<\n\006Insert\022\030.milvus.grpc.InsertParam\032"
"\026.milvus.grpc.EntityIds\"\000\022E\n\rGetEntityBy"
"ID\022\033.milvus.grpc.EntityIdentity\032\025.milvus"
".grpc.Entities\"\000\022H\n\014GetEntityIDs\022\036.milvu"
"s.grpc.GetEntityIDsParam\032\026.milvus.grpc.E"
"ntityIds\"\000\022>\n\006Search\022\030.milvus.grpc.Searc"
"hParam\032\030.milvus.grpc.QueryResult\"\000\022P\n\017Se"
"archInSegment\022!.milvus.grpc.SearchInSegm"
"entParam\032\030.milvus.grpc.QueryResult\"\000\0227\n\003"
"Cmd\022\024.milvus.grpc.Command\032\030.milvus.grpc."
"StringReply\"\000\022A\n\nDeleteByID\022\034.milvus.grp"
"c.DeleteByIDParam\032\023.milvus.grpc.Status\"\000"
"\022G\n\021PreloadCollection\022\033.milvus.grpc.Coll"
"ectionName\032\023.milvus.grpc.Status\"\000\0227\n\005Flu"
"sh\022\027.milvus.grpc.FlushParam\032\023.milvus.grp"
"c.Status\"\000\022;\n\007Compact\022\031.milvus.grpc.Comp"
"actParam\032\023.milvus.grpc.Status\"\000\022B\n\010Searc"
"hPB\022\032.milvus.grpc.SearchParamPB\032\030.milvus"
".grpc.QueryResult\"\000B\007Z\005msgpbb\006proto3"
"ir\022\020\n\010query_id\030\007 \001(\003\022\021\n\tclient_id\030\010 \001(\003\""
"H\n\013StringReply\022#\n\006status\030\001 \001(\0132\023.milvus."
"grpc.Status\022\024\n\014string_reply\030\002 \001(\t\"D\n\tBoo"
"lReply\022#\n\006status\030\001 \001(\0132\023.milvus.grpc.Sta"
"tus\022\022\n\nbool_reply\030\002 \001(\010\"W\n\022CollectionRow"
"Count\022#\n\006status\030\001 \001(\0132\023.milvus.grpc.Stat"
"us\022\034\n\024collection_row_count\030\002 \001(\003\"\026\n\007Comm"
"and\022\013\n\003cmd\030\001 \001(\t\"\243\001\n\nIndexParam\022#\n\006statu"
"s\030\001 \001(\0132\023.milvus.grpc.Status\022\027\n\017collecti"
"on_name\030\002 \001(\t\022\022\n\nfield_name\030\003 \001(\t\022\022\n\nind"
"ex_name\030\004 \001(\t\022/\n\014extra_params\030\005 \003(\0132\031.mi"
"lvus.grpc.KeyValuePair\"+\n\nFlushParam\022\035\n\025"
"collection_name_array\030\001 \003(\t\":\n\014CompactPa"
"ram\022\027\n\017collection_name\030\001 \001(\t\022\021\n\tthreshol"
"d\030\002 \001(\001\"<\n\017DeleteByIDParam\022\027\n\017collection"
"_name\030\001 \001(\t\022\020\n\010id_array\030\002 \003(\003\"H\n\016Collect"
"ionInfo\022#\n\006status\030\001 \001(\0132\023.milvus.grpc.St"
"atus\022\021\n\tjson_info\030\002 \001(\t\"@\n\021GetEntityIDsP"
"aram\022\027\n\017collection_name\030\001 \001(\t\022\022\n\nsegment"
"_id\030\002 \001(\003\"P\n\016EntityIdentity\022\027\n\017collectio"
"n_name\030\001 \001(\t\022\020\n\010id_array\030\002 \003(\003\022\023\n\013field_"
"names\030\003 \003(\t\"%\n\020VectorFieldParam\022\021\n\tdimen"
"sion\030\001 \001(\003\"w\n\tFieldType\022*\n\tdata_type\030\001 \001"
"(\0162\025.milvus.grpc.DataTypeH\000\0225\n\014vector_pa"
"ram\030\002 \001(\0132\035.milvus.grpc.VectorFieldParam"
"H\000B\007\n\005value\"\255\001\n\nFieldParam\022\n\n\002id\030\001 \001(\004\022\014"
"\n\004name\030\002 \001(\t\022#\n\004type\030\003 \001(\0162\025.milvus.grpc"
".DataType\022/\n\014index_params\030\004 \003(\0132\031.milvus"
".grpc.KeyValuePair\022/\n\014extra_params\030\005 \003(\013"
"2\031.milvus.grpc.KeyValuePair\"@\n\021VectorFie"
"ldRecord\022+\n\005value\030\001 \003(\0132\034.milvus.grpc.Ve"
"ctorRowRecord\"\233\001\n\tTermQuery\022\022\n\nfield_nam"
"e\030\001 \001(\t\022\021\n\tint_value\030\002 \003(\003\022\024\n\014double_val"
"ue\030\003 \003(\001\022\021\n\tvalue_num\030\004 \001(\003\022\r\n\005boost\030\005 \001"
"(\002\022/\n\014extra_params\030\006 \003(\0132\031.milvus.grpc.K"
"eyValuePair\"N\n\013CompareExpr\022.\n\010operator\030\001"
" \001(\0162\034.milvus.grpc.CompareOperator\022\017\n\007op"
"erand\030\002 \001(\t\"\213\001\n\nRangeQuery\022\022\n\nfield_name"
"\030\001 \001(\t\022)\n\007operand\030\002 \003(\0132\030.milvus.grpc.Co"
"mpareExpr\022\r\n\005boost\030\003 \001(\002\022/\n\014extra_params"
"\030\004 \003(\0132\031.milvus.grpc.KeyValuePair\"\244\001\n\013Ve"
"ctorQuery\022\022\n\nfield_name\030\001 \001(\t\022\023\n\013query_b"
"oost\030\002 \001(\002\022-\n\007records\030\003 \003(\0132\034.milvus.grp"
"c.VectorRowRecord\022\014\n\004topk\030\004 \001(\003\022/\n\014extra"
"_params\030\005 \003(\0132\031.milvus.grpc.KeyValuePair"
"\"c\n\014BooleanQuery\022!\n\005occur\030\001 \001(\0162\022.milvus"
".grpc.Occur\0220\n\rgeneral_query\030\002 \003(\0132\031.mil"
"vus.grpc.GeneralQuery\"\333\001\n\014GeneralQuery\0222"
"\n\rboolean_query\030\001 \001(\0132\031.milvus.grpc.Bool"
"eanQueryH\000\022,\n\nterm_query\030\002 \001(\0132\026.milvus."
"grpc.TermQueryH\000\022.\n\013range_query\030\003 \001(\0132\027."
"milvus.grpc.RangeQueryH\000\0220\n\014vector_query"
"\030\004 \001(\0132\030.milvus.grpc.VectorQueryH\000B\007\n\005qu"
"ery\"\250\001\n\rSearchParamPB\022\027\n\017collection_name"
"\030\001 \001(\t\022\033\n\023partition_tag_array\030\002 \003(\t\0220\n\rg"
"eneral_query\030\003 \001(\0132\031.milvus.grpc.General"
"Query\022/\n\014extra_params\030\004 \003(\0132\031.milvus.grp"
"c.KeyValuePair\"\231\002\n\021InsertOrDeleteMsg\022\027\n\017"
"collection_name\030\001 \001(\t\022\'\n\trows_data\030\002 \001(\013"
"2\024.milvus.grpc.RowData\022\013\n\003uid\030\003 \001(\003\022\025\n\rp"
"artition_tag\030\004 \001(\t\022\021\n\ttimestamp\030\005 \001(\004\022\022\n"
"\nsegment_id\030\006 \001(\003\022\022\n\nchannel_id\030\007 \001(\003\022\037\n"
"\002op\030\010 \001(\0162\023.milvus.grpc.OpType\022\021\n\tclient"
"_id\030\t \001(\003\022/\n\014extra_params\030\n \003(\0132\031.milvus"
".grpc.KeyValuePair\"\316\001\n\tSearchMsg\022\027\n\017coll"
"ection_name\030\001 \001(\t\022-\n\007records\030\002 \001(\0132\034.mil"
"vus.grpc.VectorRowRecord\022\025\n\rpartition_ta"
"g\030\003 \001(\t\022\013\n\003uid\030\004 \001(\003\022\021\n\ttimestamp\030\005 \001(\004\022"
"\021\n\tclient_id\030\006 \001(\003\022/\n\014extra_params\030\007 \003(\013"
"2\031.milvus.grpc.KeyValuePair\"[\n\013TimeSyncM"
"sg\022\017\n\007peer_Id\030\001 \001(\003\022\021\n\tTimestamp\030\002 \001(\004\022("
"\n\tsync_type\030\003 \001(\0162\025.milvus.grpc.SyncType"
"\"-\n\nKey2SegMsg\022\013\n\003uid\030\001 \001(\003\022\022\n\nsegment_i"
"d\030\002 \003(\003*\242\004\n\tErrorCode\022\013\n\007SUCCESS\020\000\022\024\n\020UN"
"EXPECTED_ERROR\020\001\022\022\n\016CONNECT_FAILED\020\002\022\025\n\021"
"PERMISSION_DENIED\020\003\022\031\n\025COLLECTION_NOT_EX"
"ISTS\020\004\022\024\n\020ILLEGAL_ARGUMENT\020\005\022\025\n\021ILLEGAL_"
"DIMENSION\020\007\022\026\n\022ILLEGAL_INDEX_TYPE\020\010\022\033\n\027I"
"LLEGAL_COLLECTION_NAME\020\t\022\020\n\014ILLEGAL_TOPK"
"\020\n\022\025\n\021ILLEGAL_ROWRECORD\020\013\022\025\n\021ILLEGAL_VEC"
"TOR_ID\020\014\022\031\n\025ILLEGAL_SEARCH_RESULT\020\r\022\022\n\016F"
"ILE_NOT_FOUND\020\016\022\017\n\013META_FAILED\020\017\022\020\n\014CACH"
"E_FAILED\020\020\022\030\n\024CANNOT_CREATE_FOLDER\020\021\022\026\n\022"
"CANNOT_CREATE_FILE\020\022\022\030\n\024CANNOT_DELETE_FO"
"LDER\020\023\022\026\n\022CANNOT_DELETE_FILE\020\024\022\025\n\021BUILD_"
"INDEX_ERROR\020\025\022\021\n\rILLEGAL_NLIST\020\026\022\027\n\023ILLE"
"GAL_METRIC_TYPE\020\027\022\021\n\rOUT_OF_MEMORY\020\030*\221\001\n"
"\010DataType\022\010\n\004NONE\020\000\022\010\n\004BOOL\020\001\022\010\n\004INT8\020\002\022"
"\t\n\005INT16\020\003\022\t\n\005INT32\020\004\022\t\n\005INT64\020\005\022\t\n\005FLOA"
"T\020\n\022\n\n\006DOUBLE\020\013\022\n\n\006STRING\020\024\022\021\n\rVECTOR_BI"
"NARY\020d\022\020\n\014VECTOR_FLOAT\020e*C\n\017CompareOpera"
"tor\022\006\n\002LT\020\000\022\007\n\003LTE\020\001\022\006\n\002EQ\020\002\022\006\n\002GT\020\003\022\007\n\003"
"GTE\020\004\022\006\n\002NE\020\005*8\n\005Occur\022\013\n\007INVALID\020\000\022\010\n\004M"
"UST\020\001\022\n\n\006SHOULD\020\002\022\014\n\010MUST_NOT\020\003* \n\006OpTyp"
"e\022\n\n\006INSERT\020\000\022\n\n\006DELETE\020\001*\037\n\010SyncType\022\010\n"
"\004READ\020\000\022\t\n\005WRITE\020\0012\333\r\n\rMilvusService\022\?\n\020"
"CreateCollection\022\024.milvus.grpc.Mapping\032\023"
".milvus.grpc.Status\"\000\022F\n\rHasCollection\022\033"
".milvus.grpc.CollectionName\032\026.milvus.grp"
"c.BoolReply\"\000\022I\n\022DescribeCollection\022\033.mi"
"lvus.grpc.CollectionName\032\024.milvus.grpc.M"
"apping\"\000\022Q\n\017CountCollection\022\033.milvus.grp"
"c.CollectionName\032\037.milvus.grpc.Collectio"
"nRowCount\"\000\022J\n\017ShowCollections\022\024.milvus."
"grpc.Command\032\037.milvus.grpc.CollectionNam"
"eList\"\000\022P\n\022ShowCollectionInfo\022\033.milvus.g"
"rpc.CollectionName\032\033.milvus.grpc.Collect"
"ionInfo\"\000\022D\n\016DropCollection\022\033.milvus.grp"
"c.CollectionName\032\023.milvus.grpc.Status\"\000\022"
"=\n\013CreateIndex\022\027.milvus.grpc.IndexParam\032"
"\023.milvus.grpc.Status\"\000\022C\n\rDescribeIndex\022"
"\027.milvus.grpc.IndexParam\032\027.milvus.grpc.I"
"ndexParam\"\000\022;\n\tDropIndex\022\027.milvus.grpc.I"
"ndexParam\032\023.milvus.grpc.Status\"\000\022E\n\017Crea"
"tePartition\022\033.milvus.grpc.PartitionParam"
"\032\023.milvus.grpc.Status\"\000\022E\n\014HasPartition\022"
"\033.milvus.grpc.PartitionParam\032\026.milvus.gr"
"pc.BoolReply\"\000\022K\n\016ShowPartitions\022\033.milvu"
"s.grpc.CollectionName\032\032.milvus.grpc.Part"
"itionList\"\000\022C\n\rDropPartition\022\033.milvus.gr"
"pc.PartitionParam\032\023.milvus.grpc.Status\"\000"
"\022<\n\006Insert\022\030.milvus.grpc.InsertParam\032\026.m"
"ilvus.grpc.EntityIds\"\000\022E\n\rGetEntityByID\022"
"\033.milvus.grpc.EntityIdentity\032\025.milvus.gr"
"pc.Entities\"\000\022H\n\014GetEntityIDs\022\036.milvus.g"
"rpc.GetEntityIDsParam\032\026.milvus.grpc.Enti"
"tyIds\"\000\022>\n\006Search\022\030.milvus.grpc.SearchPa"
"ram\032\030.milvus.grpc.QueryResult\"\000\022P\n\017Searc"
"hInSegment\022!.milvus.grpc.SearchInSegment"
"Param\032\030.milvus.grpc.QueryResult\"\000\0227\n\003Cmd"
"\022\024.milvus.grpc.Command\032\030.milvus.grpc.Str"
"ingReply\"\000\022A\n\nDeleteByID\022\034.milvus.grpc.D"
"eleteByIDParam\032\023.milvus.grpc.Status\"\000\022G\n"
"\021PreloadCollection\022\033.milvus.grpc.Collect"
"ionName\032\023.milvus.grpc.Status\"\000\0227\n\005Flush\022"
"\027.milvus.grpc.FlushParam\032\023.milvus.grpc.S"
"tatus\"\000\022;\n\007Compact\022\031.milvus.grpc.Compact"
"Param\032\023.milvus.grpc.Status\"\000\022B\n\010SearchPB"
"\022\032.milvus.grpc.SearchParamPB\032\030.milvus.gr"
"pc.QueryResult\"\000B\007Z\005msgpbb\006proto3"
;
static const ::PROTOBUF_NAMESPACE_ID::internal::DescriptorTable*const descriptor_table_message_2eproto_deps[1] = {
};
@ -1647,7 +1650,7 @@ static ::PROTOBUF_NAMESPACE_ID::internal::SCCInfoBase*const descriptor_table_mes
static ::PROTOBUF_NAMESPACE_ID::internal::once_flag descriptor_table_message_2eproto_once;
static bool descriptor_table_message_2eproto_initialized = false;
const ::PROTOBUF_NAMESPACE_ID::internal::DescriptorTable descriptor_table_message_2eproto = {
&descriptor_table_message_2eproto_initialized, descriptor_table_protodef_message_2eproto, "message.proto", 7516,
&descriptor_table_message_2eproto_initialized, descriptor_table_protodef_message_2eproto, "message.proto", 7553,
&descriptor_table_message_2eproto_once, descriptor_table_message_2eproto_sccs, descriptor_table_message_2eproto_deps, 46, 0,
schemas, file_default_instances, TableStruct_message_2eproto::offsets,
file_level_metadata_message_2eproto, 47, file_level_enum_descriptors_message_2eproto, file_level_service_descriptors_message_2eproto,
@ -8808,15 +8811,17 @@ QueryResult::QueryResult(const QueryResult& from)
} else {
entities_ = nullptr;
}
row_num_ = from.row_num_;
::memcpy(&row_num_, &from.row_num_,
static_cast<size_t>(reinterpret_cast<char*>(&client_id_) -
reinterpret_cast<char*>(&row_num_)) + sizeof(client_id_));
// @@protoc_insertion_point(copy_constructor:milvus.grpc.QueryResult)
}
void QueryResult::SharedCtor() {
::PROTOBUF_NAMESPACE_ID::internal::InitSCC(&scc_info_QueryResult_message_2eproto.base);
::memset(&status_, 0, static_cast<size_t>(
reinterpret_cast<char*>(&row_num_) -
reinterpret_cast<char*>(&status_)) + sizeof(row_num_));
reinterpret_cast<char*>(&client_id_) -
reinterpret_cast<char*>(&status_)) + sizeof(client_id_));
}
QueryResult::~QueryResult() {
@ -8855,7 +8860,9 @@ void QueryResult::Clear() {
delete entities_;
}
entities_ = nullptr;
row_num_ = PROTOBUF_LONGLONG(0);
::memset(&row_num_, 0, static_cast<size_t>(
reinterpret_cast<char*>(&client_id_) -
reinterpret_cast<char*>(&row_num_)) + sizeof(client_id_));
_internal_metadata_.Clear();
}
@ -8920,6 +8927,20 @@ const char* QueryResult::_InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID
} while (::PROTOBUF_NAMESPACE_ID::internal::UnalignedLoad<::PROTOBUF_NAMESPACE_ID::uint8>(ptr) == 50);
} else goto handle_unusual;
continue;
// int64 query_id = 7;
case 7:
if (PROTOBUF_PREDICT_TRUE(static_cast<::PROTOBUF_NAMESPACE_ID::uint8>(tag) == 56)) {
query_id_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint(&ptr);
CHK_(ptr);
} else goto handle_unusual;
continue;
// int64 client_id = 8;
case 8:
if (PROTOBUF_PREDICT_TRUE(static_cast<::PROTOBUF_NAMESPACE_ID::uint8>(tag) == 64)) {
client_id_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint(&ptr);
CHK_(ptr);
} else goto handle_unusual;
continue;
default: {
handle_unusual:
if ((tag & 7) == 4 || tag == 0) {
@ -9028,6 +9049,32 @@ bool QueryResult::MergePartialFromCodedStream(
break;
}
// int64 query_id = 7;
case 7: {
if (static_cast< ::PROTOBUF_NAMESPACE_ID::uint8>(tag) == (56 & 0xFF)) {
DO_((::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::ReadPrimitive<
::PROTOBUF_NAMESPACE_ID::int64, ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_INT64>(
input, &query_id_)));
} else {
goto handle_unusual;
}
break;
}
// int64 client_id = 8;
case 8: {
if (static_cast< ::PROTOBUF_NAMESPACE_ID::uint8>(tag) == (64 & 0xFF)) {
DO_((::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::ReadPrimitive<
::PROTOBUF_NAMESPACE_ID::int64, ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_INT64>(
input, &client_id_)));
} else {
goto handle_unusual;
}
break;
}
default: {
handle_unusual:
if (tag == 0) {
@ -9099,6 +9146,16 @@ void QueryResult::SerializeWithCachedSizes(
output);
}
// int64 query_id = 7;
if (this->query_id() != 0) {
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64(7, this->query_id(), output);
}
// int64 client_id = 8;
if (this->client_id() != 0) {
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64(8, this->client_id(), output);
}
if (_internal_metadata_.have_unknown_fields()) {
::PROTOBUF_NAMESPACE_ID::internal::WireFormat::SerializeUnknownFields(
_internal_metadata_.unknown_fields(), output);
@ -9165,6 +9222,16 @@ void QueryResult::SerializeWithCachedSizes(
6, this->extra_params(static_cast<int>(i)), target);
}
// int64 query_id = 7;
if (this->query_id() != 0) {
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64ToArray(7, this->query_id(), target);
}
// int64 client_id = 8;
if (this->client_id() != 0) {
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64ToArray(8, this->client_id(), target);
}
if (_internal_metadata_.have_unknown_fields()) {
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormat::SerializeUnknownFieldsToArray(
_internal_metadata_.unknown_fields(), target);
@ -9248,6 +9315,20 @@ size_t QueryResult::ByteSizeLong() const {
this->row_num());
}
// int64 query_id = 7;
if (this->query_id() != 0) {
total_size += 1 +
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::Int64Size(
this->query_id());
}
// int64 client_id = 8;
if (this->client_id() != 0) {
total_size += 1 +
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::Int64Size(
this->client_id());
}
int cached_size = ::PROTOBUF_NAMESPACE_ID::internal::ToCachedSize(total_size);
SetCachedSize(cached_size);
return total_size;
@ -9287,6 +9368,12 @@ void QueryResult::MergeFrom(const QueryResult& from) {
if (from.row_num() != 0) {
set_row_num(from.row_num());
}
if (from.query_id() != 0) {
set_query_id(from.query_id());
}
if (from.client_id() != 0) {
set_client_id(from.client_id());
}
}
void QueryResult::CopyFrom(const ::PROTOBUF_NAMESPACE_ID::Message& from) {
@ -9316,6 +9403,8 @@ void QueryResult::InternalSwap(QueryResult* other) {
swap(status_, other->status_);
swap(entities_, other->entities_);
swap(row_num_, other->row_num_);
swap(query_id_, other->query_id_);
swap(client_id_, other->client_id_);
}
::PROTOBUF_NAMESPACE_ID::Metadata QueryResult::GetMetadata() const {
@ -17511,7 +17600,7 @@ const char* InsertOrDeleteMsg::_InternalParse(const char* ptr, ::PROTOBUF_NAMESP
CHK_(ptr);
} else goto handle_unusual;
continue;
// int64 timestamp = 5;
// uint64 timestamp = 5;
case 5:
if (PROTOBUF_PREDICT_TRUE(static_cast<::PROTOBUF_NAMESPACE_ID::uint8>(tag) == 40)) {
timestamp_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint(&ptr);
@ -17643,12 +17732,12 @@ bool InsertOrDeleteMsg::MergePartialFromCodedStream(
break;
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
case 5: {
if (static_cast< ::PROTOBUF_NAMESPACE_ID::uint8>(tag) == (40 & 0xFF)) {
DO_((::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::ReadPrimitive<
::PROTOBUF_NAMESPACE_ID::int64, ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_INT64>(
::PROTOBUF_NAMESPACE_ID::uint64, ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_UINT64>(
input, &timestamp_)));
} else {
goto handle_unusual;
@ -17778,9 +17867,9 @@ void InsertOrDeleteMsg::SerializeWithCachedSizes(
4, this->partition_tag(), output);
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
if (this->timestamp() != 0) {
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64(5, this->timestamp(), output);
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteUInt64(5, this->timestamp(), output);
}
// int64 segment_id = 6;
@ -17860,9 +17949,9 @@ void InsertOrDeleteMsg::SerializeWithCachedSizes(
4, this->partition_tag(), target);
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
if (this->timestamp() != 0) {
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64ToArray(5, this->timestamp(), target);
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteUInt64ToArray(5, this->timestamp(), target);
}
// int64 segment_id = 6;
@ -17954,10 +18043,10 @@ size_t InsertOrDeleteMsg::ByteSizeLong() const {
this->uid());
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
if (this->timestamp() != 0) {
total_size += 1 +
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::Int64Size(
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::UInt64Size(
this->timestamp());
}
@ -18215,7 +18304,7 @@ const char* SearchMsg::_InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::
CHK_(ptr);
} else goto handle_unusual;
continue;
// int64 timestamp = 5;
// uint64 timestamp = 5;
case 5:
if (PROTOBUF_PREDICT_TRUE(static_cast<::PROTOBUF_NAMESPACE_ID::uint8>(tag) == 40)) {
timestamp_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint(&ptr);
@ -18325,12 +18414,12 @@ bool SearchMsg::MergePartialFromCodedStream(
break;
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
case 5: {
if (static_cast< ::PROTOBUF_NAMESPACE_ID::uint8>(tag) == (40 & 0xFF)) {
DO_((::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::ReadPrimitive<
::PROTOBUF_NAMESPACE_ID::int64, ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_INT64>(
::PROTOBUF_NAMESPACE_ID::uint64, ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_UINT64>(
input, &timestamp_)));
} else {
goto handle_unusual;
@ -18420,9 +18509,9 @@ void SearchMsg::SerializeWithCachedSizes(
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64(4, this->uid(), output);
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
if (this->timestamp() != 0) {
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64(5, this->timestamp(), output);
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteUInt64(5, this->timestamp(), output);
}
// int64 client_id = 6;
@ -18486,9 +18575,9 @@ void SearchMsg::SerializeWithCachedSizes(
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64ToArray(4, this->uid(), target);
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
if (this->timestamp() != 0) {
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64ToArray(5, this->timestamp(), target);
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteUInt64ToArray(5, this->timestamp(), target);
}
// int64 client_id = 6;
@ -18564,10 +18653,10 @@ size_t SearchMsg::ByteSizeLong() const {
this->uid());
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
if (this->timestamp() != 0) {
total_size += 1 +
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::Int64Size(
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::UInt64Size(
this->timestamp());
}
@ -18738,7 +18827,7 @@ const char* TimeSyncMsg::_InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID
CHK_(ptr);
} else goto handle_unusual;
continue;
// int64 Timestamp = 2;
// uint64 Timestamp = 2;
case 2:
if (PROTOBUF_PREDICT_TRUE(static_cast<::PROTOBUF_NAMESPACE_ID::uint8>(tag) == 16)) {
timestamp_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint(&ptr);
@ -18796,12 +18885,12 @@ bool TimeSyncMsg::MergePartialFromCodedStream(
break;
}
// int64 Timestamp = 2;
// uint64 Timestamp = 2;
case 2: {
if (static_cast< ::PROTOBUF_NAMESPACE_ID::uint8>(tag) == (16 & 0xFF)) {
DO_((::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::ReadPrimitive<
::PROTOBUF_NAMESPACE_ID::int64, ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_INT64>(
::PROTOBUF_NAMESPACE_ID::uint64, ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_UINT64>(
input, &timestamp_)));
} else {
goto handle_unusual;
@ -18855,9 +18944,9 @@ void TimeSyncMsg::SerializeWithCachedSizes(
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64(1, this->peer_id(), output);
}
// int64 Timestamp = 2;
// uint64 Timestamp = 2;
if (this->timestamp() != 0) {
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64(2, this->timestamp(), output);
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteUInt64(2, this->timestamp(), output);
}
// .milvus.grpc.SyncType sync_type = 3;
@ -18884,9 +18973,9 @@ void TimeSyncMsg::SerializeWithCachedSizes(
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64ToArray(1, this->peer_id(), target);
}
// int64 Timestamp = 2;
// uint64 Timestamp = 2;
if (this->timestamp() != 0) {
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteInt64ToArray(2, this->timestamp(), target);
target = ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::WriteUInt64ToArray(2, this->timestamp(), target);
}
// .milvus.grpc.SyncType sync_type = 3;
@ -18923,10 +19012,10 @@ size_t TimeSyncMsg::ByteSizeLong() const {
this->peer_id());
}
// int64 Timestamp = 2;
// uint64 Timestamp = 2;
if (this->timestamp() != 0) {
total_size += 1 +
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::Int64Size(
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::UInt64Size(
this->timestamp());
}

90
proxy/src/grpc/message.pb.h
View File

@ -3645,6 +3645,8 @@ class QueryResult :
kStatusFieldNumber = 1,
kEntitiesFieldNumber = 2,
kRowNumFieldNumber = 3,
kQueryIdFieldNumber = 7,
kClientIdFieldNumber = 8,
};
// repeated float scores = 4;
int scores_size() const;
@ -3700,6 +3702,16 @@ class QueryResult :
::PROTOBUF_NAMESPACE_ID::int64 row_num() const;
void set_row_num(::PROTOBUF_NAMESPACE_ID::int64 value);
// int64 query_id = 7;
void clear_query_id();
::PROTOBUF_NAMESPACE_ID::int64 query_id() const;
void set_query_id(::PROTOBUF_NAMESPACE_ID::int64 value);
// int64 client_id = 8;
void clear_client_id();
::PROTOBUF_NAMESPACE_ID::int64 client_id() const;
void set_client_id(::PROTOBUF_NAMESPACE_ID::int64 value);
// @@protoc_insertion_point(class_scope:milvus.grpc.QueryResult)
private:
class _Internal;
@ -3713,6 +3725,8 @@ class QueryResult :
::milvus::grpc::Status* status_;
::milvus::grpc::Entities* entities_;
::PROTOBUF_NAMESPACE_ID::int64 row_num_;
::PROTOBUF_NAMESPACE_ID::int64 query_id_;
::PROTOBUF_NAMESPACE_ID::int64 client_id_;
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
friend struct ::TableStruct_message_2eproto;
};
@ -7338,10 +7352,10 @@ class InsertOrDeleteMsg :
::PROTOBUF_NAMESPACE_ID::int64 uid() const;
void set_uid(::PROTOBUF_NAMESPACE_ID::int64 value);
// int64 timestamp = 5;
// uint64 timestamp = 5;
void clear_timestamp();
::PROTOBUF_NAMESPACE_ID::int64 timestamp() const;
void set_timestamp(::PROTOBUF_NAMESPACE_ID::int64 value);
::PROTOBUF_NAMESPACE_ID::uint64 timestamp() const;
void set_timestamp(::PROTOBUF_NAMESPACE_ID::uint64 value);
// int64 segment_id = 6;
void clear_segment_id();
@ -7373,7 +7387,7 @@ class InsertOrDeleteMsg :
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr partition_tag_;
::milvus::grpc::RowData* rows_data_;
::PROTOBUF_NAMESPACE_ID::int64 uid_;
::PROTOBUF_NAMESPACE_ID::int64 timestamp_;
::PROTOBUF_NAMESPACE_ID::uint64 timestamp_;
::PROTOBUF_NAMESPACE_ID::int64 segment_id_;
::PROTOBUF_NAMESPACE_ID::int64 channel_id_;
::PROTOBUF_NAMESPACE_ID::int64 client_id_;
@ -7550,10 +7564,10 @@ class SearchMsg :
::PROTOBUF_NAMESPACE_ID::int64 uid() const;
void set_uid(::PROTOBUF_NAMESPACE_ID::int64 value);
// int64 timestamp = 5;
// uint64 timestamp = 5;
void clear_timestamp();
::PROTOBUF_NAMESPACE_ID::int64 timestamp() const;
void set_timestamp(::PROTOBUF_NAMESPACE_ID::int64 value);
::PROTOBUF_NAMESPACE_ID::uint64 timestamp() const;
void set_timestamp(::PROTOBUF_NAMESPACE_ID::uint64 value);
// int64 client_id = 6;
void clear_client_id();
@ -7570,7 +7584,7 @@ class SearchMsg :
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr partition_tag_;
::milvus::grpc::VectorRowRecord* records_;
::PROTOBUF_NAMESPACE_ID::int64 uid_;
::PROTOBUF_NAMESPACE_ID::int64 timestamp_;
::PROTOBUF_NAMESPACE_ID::uint64 timestamp_;
::PROTOBUF_NAMESPACE_ID::int64 client_id_;
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
friend struct ::TableStruct_message_2eproto;
@ -7699,10 +7713,10 @@ class TimeSyncMsg :
::PROTOBUF_NAMESPACE_ID::int64 peer_id() const;
void set_peer_id(::PROTOBUF_NAMESPACE_ID::int64 value);
// int64 Timestamp = 2;
// uint64 Timestamp = 2;
void clear_timestamp();
::PROTOBUF_NAMESPACE_ID::int64 timestamp() const;
void set_timestamp(::PROTOBUF_NAMESPACE_ID::int64 value);
::PROTOBUF_NAMESPACE_ID::uint64 timestamp() const;
void set_timestamp(::PROTOBUF_NAMESPACE_ID::uint64 value);
// .milvus.grpc.SyncType sync_type = 3;
void clear_sync_type();
@ -7715,7 +7729,7 @@ class TimeSyncMsg :
::PROTOBUF_NAMESPACE_ID::internal::InternalMetadataWithArena _internal_metadata_;
::PROTOBUF_NAMESPACE_ID::int64 peer_id_;
::PROTOBUF_NAMESPACE_ID::int64 timestamp_;
::PROTOBUF_NAMESPACE_ID::uint64 timestamp_;
int sync_type_;
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
friend struct ::TableStruct_message_2eproto;
@ -10261,6 +10275,34 @@ QueryResult::extra_params() const {
return extra_params_;
}
// int64 query_id = 7;
inline void QueryResult::clear_query_id() {
query_id_ = PROTOBUF_LONGLONG(0);
}
inline ::PROTOBUF_NAMESPACE_ID::int64 QueryResult::query_id() const {
// @@protoc_insertion_point(field_get:milvus.grpc.QueryResult.query_id)
return query_id_;
}
inline void QueryResult::set_query_id(::PROTOBUF_NAMESPACE_ID::int64 value) {
query_id_ = value;
// @@protoc_insertion_point(field_set:milvus.grpc.QueryResult.query_id)
}
// int64 client_id = 8;
inline void QueryResult::clear_client_id() {
client_id_ = PROTOBUF_LONGLONG(0);
}
inline ::PROTOBUF_NAMESPACE_ID::int64 QueryResult::client_id() const {
// @@protoc_insertion_point(field_get:milvus.grpc.QueryResult.client_id)
return client_id_;
}
inline void QueryResult::set_client_id(::PROTOBUF_NAMESPACE_ID::int64 value) {
client_id_ = value;
// @@protoc_insertion_point(field_set:milvus.grpc.QueryResult.client_id)
}
// -------------------------------------------------------------------
// StringReply
@ -12735,15 +12777,15 @@ inline void InsertOrDeleteMsg::set_allocated_partition_tag(std::string* partitio
// @@protoc_insertion_point(field_set_allocated:milvus.grpc.InsertOrDeleteMsg.partition_tag)
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
inline void InsertOrDeleteMsg::clear_timestamp() {
timestamp_ = PROTOBUF_LONGLONG(0);
timestamp_ = PROTOBUF_ULONGLONG(0);
}
inline ::PROTOBUF_NAMESPACE_ID::int64 InsertOrDeleteMsg::timestamp() const {
inline ::PROTOBUF_NAMESPACE_ID::uint64 InsertOrDeleteMsg::timestamp() const {
// @@protoc_insertion_point(field_get:milvus.grpc.InsertOrDeleteMsg.timestamp)
return timestamp_;
}
inline void InsertOrDeleteMsg::set_timestamp(::PROTOBUF_NAMESPACE_ID::int64 value) {
inline void InsertOrDeleteMsg::set_timestamp(::PROTOBUF_NAMESPACE_ID::uint64 value) {
timestamp_ = value;
// @@protoc_insertion_point(field_set:milvus.grpc.InsertOrDeleteMsg.timestamp)
@ -13006,15 +13048,15 @@ inline void SearchMsg::set_uid(::PROTOBUF_NAMESPACE_ID::int64 value) {
// @@protoc_insertion_point(field_set:milvus.grpc.SearchMsg.uid)
}
// int64 timestamp = 5;
// uint64 timestamp = 5;
inline void SearchMsg::clear_timestamp() {
timestamp_ = PROTOBUF_LONGLONG(0);
timestamp_ = PROTOBUF_ULONGLONG(0);
}
inline ::PROTOBUF_NAMESPACE_ID::int64 SearchMsg::timestamp() const {
inline ::PROTOBUF_NAMESPACE_ID::uint64 SearchMsg::timestamp() const {
// @@protoc_insertion_point(field_get:milvus.grpc.SearchMsg.timestamp)
return timestamp_;
}
inline void SearchMsg::set_timestamp(::PROTOBUF_NAMESPACE_ID::int64 value) {
inline void SearchMsg::set_timestamp(::PROTOBUF_NAMESPACE_ID::uint64 value) {
timestamp_ = value;
// @@protoc_insertion_point(field_set:milvus.grpc.SearchMsg.timestamp)
@ -13082,15 +13124,15 @@ inline void TimeSyncMsg::set_peer_id(::PROTOBUF_NAMESPACE_ID::int64 value) {
// @@protoc_insertion_point(field_set:milvus.grpc.TimeSyncMsg.peer_Id)
}
// int64 Timestamp = 2;
// uint64 Timestamp = 2;
inline void TimeSyncMsg::clear_timestamp() {
timestamp_ = PROTOBUF_LONGLONG(0);
timestamp_ = PROTOBUF_ULONGLONG(0);
}
inline ::PROTOBUF_NAMESPACE_ID::int64 TimeSyncMsg::timestamp() const {
inline ::PROTOBUF_NAMESPACE_ID::uint64 TimeSyncMsg::timestamp() const {
// @@protoc_insertion_point(field_get:milvus.grpc.TimeSyncMsg.Timestamp)
return timestamp_;
}
inline void TimeSyncMsg::set_timestamp(::PROTOBUF_NAMESPACE_ID::int64 value) {
inline void TimeSyncMsg::set_timestamp(::PROTOBUF_NAMESPACE_ID::uint64 value) {
timestamp_ = value;
// @@protoc_insertion_point(field_set:milvus.grpc.TimeSyncMsg.Timestamp)

2
proxy/src/message_client/CMakeLists.txt
View File

@ -15,4 +15,4 @@ target_include_directories(message_client_cpp PUBLIC ${PROJECT_BINARY_DIR}/third
target_link_libraries(message_client_cpp utils pulsarStatic libprotobuf)
#install(TARGETS message_client_cpp
# DESTINATION lib)
# DESTINATION lib)

90
proxy/src/message_client/ClientV2.cpp
View File

@ -52,8 +52,94 @@ Status MsgClientV2::Init(const std::string &insert_delete,
return Status::OK();
}
void MsgClientV2::GetQueryResult(int64_t query_id) {
throw std::exception();
int64_t GetQueryNodeNum() {
return 2;
}
milvus::grpc::QueryResult Aggregation(std::vector<std::shared_ptr<grpc::QueryResult>> &results){
//TODO: QueryNode has only one
int64_t length = results.size();
std::vector<float> all_scores;
std::vector<float> all_distance;
std::vector<grpc::KeyValuePair> all_kv_pairs;
std::vector<int> index(length * results[0]->scores_size());
for (int n = 0; n < length * results[0]->scores_size(); ++n) {
index[n] = n;
}
for (int i = 0; i < length; i++){
for (int j = 0; j < results[i]->scores_size(); j++){
all_scores.push_back(results[i]->scores()[j]);
all_distance.push_back(results[i]->distances()[j]);
all_kv_pairs.push_back(results[i]->extra_params()[j]);
}
}
for (int k = 0; k < all_distance.size() - 1; ++k) {
for (int l = k + 1; l < all_distance.size(); ++l) {
if (all_distance[l] > all_distance[k]){
float distance_temp = all_distance[k];
all_distance[k] = all_distance[l];
all_distance[l] = distance_temp;
int index_temp = index[k];
index[k] = index[l];
index[l] = index_temp;
}
}
}
grpc::QueryResult result;
// result_prt->set_allocated_status(const_cast<milvus::grpc::Status*>(&results[0]->status()));
result.mutable_status()->CopyFrom(results[0]->status());
result.mutable_entities()->CopyFrom(results[0]->entities());
result.set_row_num(results[0]->row_num());
for (int m = 0; m < results[0]->scores_size(); ++m) {
result.add_scores(all_scores[index[m]]);
result.add_distances(all_distance[m]);
result.add_extra_params();
result.mutable_extra_params(m)->CopyFrom(all_kv_pairs[index[m]]);
}
result.set_query_id(results[0]->query_id());
result.set_client_id(results[0]->client_id());
return result;
}
milvus::grpc::QueryResult MsgClientV2::GetQueryResult(int64_t query_id) {
// Result result = MsgProducer::createProducer("result-partition-0");
auto client = std::make_shared<MsgClient>("pulsar://localhost:6650");
MsgConsumer consumer(client, "my_consumer");
consumer.subscribe("result");
std::vector<std::shared_ptr<grpc::QueryResult>> results;
int64_t query_node_num = GetQueryNodeNum();
std::map<int64_t, std::vector<std::shared_ptr<grpc::QueryResult>>> total_results;
while (true) {
auto received_result = total_results[query_id];
if (received_result.size() == query_node_num) {
break;
}
Message msg;
consumer.receive(msg);
grpc::QueryResult search_res_msg;
auto status = search_res_msg.ParseFromString(msg.getDataAsString());
if (status) {
auto message = std::make_shared<grpc::QueryResult>(search_res_msg);
total_results[message->query_id()].push_back(message);
consumer.acknowledge(msg);
}
}
return Aggregation(total_results[query_id]);
}
Status MsgClientV2::SendMutMessage(const milvus::grpc::InsertParam &request) {

2
proxy/src/message_client/ClientV2.h
View File

@ -26,7 +26,7 @@ class MsgClientV2 {
//
Status SendQueryMessage(const milvus::grpc::SearchParam &request);
void GetQueryResult(int64_t query_id);
static milvus::grpc::QueryResult GetQueryResult(int64_t query_id);
private:

2
proxy/unittest/CMakeLists.txt
View File

@ -35,4 +35,4 @@ set( UNITTEST_LIBS libboost_system.a
)
add_subdirectory(message_client)
add_subdirectory(config)
add_subdirectory(config)

8
proxy/unittest/message_client/CMakeLists.txt
View File

@ -1,17 +1,21 @@
enable_testing()
set( GRPC_SERVICE_FILES
${MILVUS_ENGINE_SRC}/grpc/message.grpc.pb.cc
${MILVUS_ENGINE_SRC}/grpc/message.pb.cc
${MILVUS_ENGINE_SRC}/grpc/message.pb.cc
${MILVUS_ENGINE_SRC}/utils/Status.cpp
)
set(unittest_srcs
unittest_entry.cpp
consumer_test.cpp producer_test.cpp)
# consumer_test.cpp producer_test.cpp
get_result_test.cpp)
add_executable(test_pulsar ${unittest_srcs} ${GRPC_SERVICE_FILES})
include_directories(${MILVUS_ENGINE_SRC}/utils)
target_include_directories(test_pulsar PUBLIC ${PROJECT_BINARY_DIR}/thirdparty/pulsar/pulsar-src/pulsar-client-cpp/include)
target_include_directories(test_pulsar PUBLIC ${PROJECT_BINARY_DIR}/thirdparty/avro/avro-build/include)

85
proxy/unittest/message_client/get_result_test.cpp Normal file
View File

@ -0,0 +1,85 @@
#include <gtest/gtest.h>
#include "message_client/ClientV2.h"
TEST(CLIENT_CPP, GetResult) {
auto client= std::make_shared<milvus::message_client::MsgClient>("pulsar://localhost:6650");
milvus::message_client::MsgProducer producer(client,"test");
producer.createProducer("result");
producer.send("result");
milvus::grpc::QueryResult fake_message;
milvus::grpc::QueryResult fake_message2;
milvus::grpc::Status fake_status;
fake_status.set_error_code(milvus::grpc::ErrorCode::SUCCESS);
std::string aaa = "hahaha";
fake_status.set_reason(aaa);
milvus::grpc::RowData fake_row_data;
fake_row_data.set_blob("fake_row_data");
milvus::grpc::Entities fake_entities;
// fake_entities.set_allocated_status(&fake_status);
fake_entities.mutable_status()->CopyFrom(fake_status);
for (int i = 0; i < 10; i++){
fake_entities.add_ids(i);
fake_entities.add_valid_row(true);
fake_entities.add_rows_data()->CopyFrom(fake_row_data);
}
int64_t fake_row_num = 10;
float fake_scores[10] = {1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9, 0.0};
float fake_distance[10] = {9.7, 9.6, 9.5, 9.8, 8.7, 8.8, 9.9, 8.8, 9.7, 8.9};
std::vector<milvus::grpc::KeyValuePair> fake_extra_params;
milvus::grpc::KeyValuePair keyValuePair;
for (int i = 0; i < 10; ++i) {
keyValuePair.set_key(std::to_string(i));
keyValuePair.set_value(std::to_string(i + 10));
fake_extra_params.push_back(keyValuePair);
}
int64_t fake_query_id = 10;
int64_t fake_client_id = 1;
fake_message.mutable_status()->CopyFrom(fake_status);
fake_message.mutable_entities()->CopyFrom(fake_entities);
fake_message.set_row_num(fake_row_num);
for (int i = 0; i < 10; i++) {
fake_message.add_scores(fake_scores[i]);
fake_message.add_distances(fake_distance[i]);
fake_message.add_extra_params()->CopyFrom(fake_extra_params[i]);
}
fake_message.set_query_id(fake_query_id);
fake_message.set_client_id(fake_client_id);
float fake_scores2[10] = {2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8, 9.9, 0.0, 1.1};
float fake_distance2[10] = {9.8, 8.6, 9.6, 9.7, 8.9, 8.8, 9.0, 9.8, 9.7, 8.9};
fake_message2.mutable_status()->CopyFrom(fake_status);
fake_message2.mutable_entities()->CopyFrom(fake_entities);
fake_message2.set_row_num(fake_row_num);
for (int j = 0; j < 10; ++j) {
fake_message2.add_scores(fake_scores2[j]);
fake_message2.add_distances(fake_distance2[j]);
fake_message2.add_extra_params()->CopyFrom(fake_extra_params[j]);
}
fake_message2.set_query_id(fake_query_id);
fake_message2.set_client_id(fake_client_id);
producer.send(fake_message.SerializeAsString());
producer.send(fake_message2.SerializeAsString());
producer.close();
client->close();
auto result = milvus::message_client::MsgClientV2::GetQueryResult(fake_query_id);
std::cout << result.client_id() << std::endl;
for (int k = 0; k < result.distances_size(); ++k) {
std::cout << result.distances(k) << "\t";
}
std::cout << "hahah" << std::endl;
}