milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add execute logic for balancing sealed segments (#11343) 

Signed-off-by: xige-16 <xi.ge@zilliz.com>
pull/9945/head
xige-16 2021-11-06 15:22:56 +08:00 committed by GitHub
parent dcf4cecea4
commit 82c199227c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
7 changed files with 482 additions and 160 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/proto/query_coord.proto
View File

@ -323,6 +323,8 @@ message LoadBalanceRequest {
common.MsgBase base = 1;
repeated int64 source_nodeIDs = 2;
TriggerCondition balance_reason = 3;
repeated int64 dst_nodeIDs = 4;
repeated int64 sealed_segmentIDs = 5;
}
//---------------- common query proto -----------------

308
internal/proto/querypb/query_coord.pb.go
View File

@ -2109,6 +2109,8 @@ type LoadBalanceRequest struct {
Base *commonpb.MsgBase `protobuf:"bytes,1,opt,name=base,proto3" json:"base,omitempty"`
SourceNodeIDs []int64 `protobuf:"varint,2,rep,packed,name=source_nodeIDs,json=sourceNodeIDs,proto3" json:"source_nodeIDs,omitempty"`
BalanceReason TriggerCondition `protobuf:"varint,3,opt,name=balance_reason,json=balanceReason,proto3,enum=milvus.proto.query.TriggerCondition" json:"balance_reason,omitempty"`
DstNodeIDs []int64 `protobuf:"varint,4,rep,packed,name=dst_nodeIDs,json=dstNodeIDs,proto3" json:"dst_nodeIDs,omitempty"`
SealedSegmentIDs []int64 `protobuf:"varint,5,rep,packed,name=sealed_segmentIDs,json=sealedSegmentIDs,proto3" json:"sealed_segmentIDs,omitempty"`
XXX_NoUnkeyedLiteral struct{} `json:"-"`
XXX_unrecognized []byte `json:"-"`
XXX_sizecache int32 `json:"-"`
@ -2160,6 +2162,20 @@ func (m *LoadBalanceRequest) GetBalanceReason() TriggerCondition {
return TriggerCondition_handoff
}
func (m *LoadBalanceRequest) GetDstNodeIDs() []int64 {
if m != nil {
return m.DstNodeIDs
}
return nil
}
func (m *LoadBalanceRequest) GetSealedSegmentIDs() []int64 {
if m != nil {
return m.SealedSegmentIDs
}
return nil
}
//---------------- common query proto -----------------
type SegmentChangeInfo struct {
OnlineNodeID int64 `protobuf:"varint,1,opt,name=online_nodeID,json=onlineNodeID,proto3" json:"online_nodeID,omitempty"`
@ -2312,151 +2328,153 @@ func init() {
func init() { proto.RegisterFile("query_coord.proto", fileDescriptor_aab7cc9a69ed26e8) }
var fileDescriptor_aab7cc9a69ed26e8 = []byte{
// 2299 bytes of a gzipped FileDescriptorProto
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x19, 0x4b, 0x6f, 0x1c, 0x49,
0xd9, 0x3d, 0x0f, 0x7b, 0xe6, 0x9b, 0x57, 0xa7, 0x12, 0xcf, 0x4e, 0x86, 0x24, 0xeb, 0xed, 0x6c,
0x1e, 0xeb, 0x65, 0x9d, 0xac, 0xb3, 0x3c, 0x56, 0xb0, 0x87, 0x8d, 0x67, 0xe3, 0x9d, 0x25, 0x71,
0x4c, 0x3b, 0xbb, 0x88, 0x28, 0x52, 0xd3, 0x33, 0x5d, 0x1e, 0xb7, 0xd2, 0xdd, 0x35, 0xe9, 0xea,
0x89, 0xe3, 0x9c, 0x39, 0xc0, 0x01, 0xf1, 0x03, 0x40, 0x48, 0x48, 0x20, 0xb4, 0x07, 0x8e, 0x70,
0xe0, 0x94, 0x9f, 0xc0, 0x2f, 0x40, 0x42, 0x70, 0x87, 0x0b, 0x9c, 0x51, 0x57, 0x55, 0xf7, 0xf4,
0xa3, 0xc6, 0x1e, 0xdb, 0x84, 0x44, 0x2b, 0x6e, 0xdd, 0x5f, 0x7d, 0xf5, 0xbd, 0x1f, 0xf5, 0x55,
0xc1, 0x99, 0x27, 0x13, 0xec, 0x1f, 0x18, 0x43, 0x42, 0x7c, 0x6b, 0x6d, 0xec, 0x93, 0x80, 0x20,
0xe4, 0xda, 0xce, 0xd3, 0x09, 0xe5, 0x7f, 0x6b, 0x6c, 0xbd, 0x5b, 0x1f, 0x12, 0xd7, 0x25, 0x1e,
0x87, 0x75, 0xeb, 0x49, 0x8c, 0x6e, 0xd3, 0xf6, 0x02, 0xec, 0x7b, 0xa6, 0x13, 0xad, 0xd2, 0xe1,
0x1e, 0x76, 0x4d, 0xf1, 0xa7, 0x5a, 0x66, 0x60, 0x26, 0xe9, 0x6b, 0x3f, 0x56, 0xa0, 0xbd, 0xb3,
0x47, 0xf6, 0x37, 0x88, 0xe3, 0xe0, 0x61, 0x60, 0x13, 0x8f, 0xea, 0xf8, 0xc9, 0x04, 0xd3, 0x00,
0xdd, 0x84, 0xd2, 0xc0, 0xa4, 0xb8, 0xa3, 0xac, 0x28, 0xd7, 0x6b, 0xeb, 0x17, 0xd6, 0x52, 0x92,
0x08, 0x11, 0xee, 0xd1, 0xd1, 0x6d, 0x93, 0x62, 0x9d, 0x61, 0x22, 0x04, 0x25, 0x6b, 0xd0, 0xef,
0x75, 0x0a, 0x2b, 0xca, 0xf5, 0xa2, 0xce, 0xbe, 0xd1, 0xdb, 0xd0, 0x18, 0xc6, 0xb4, 0xfb, 0x3d,
0xda, 0x29, 0xae, 0x14, 0xaf, 0x17, 0xf5, 0x34, 0x50, 0xfb, 0x9d, 0x02, 0x6f, 0xe4, 0xc4, 0xa0,
0x63, 0xe2, 0x51, 0x8c, 0x6e, 0xc1, 0x22, 0x0d, 0xcc, 0x60, 0x42, 0x85, 0x24, 0x5f, 0x93, 0x4a,
0xb2, 0xc3, 0x50, 0x74, 0x81, 0x9a, 0x67, 0x5b, 0x90, 0xb0, 0x45, 0xef, 0xc3, 0x39, 0xdb, 0xbb,
0x87, 0x5d, 0xe2, 0x1f, 0x18, 0x63, 0xec, 0x0f, 0xb1, 0x17, 0x98, 0x23, 0x1c, 0xc9, 0x78, 0x36,
0x5a, 0xdb, 0x9e, 0x2e, 0x69, 0xbf, 0x55, 0x60, 0x39, 0x94, 0x74, 0xdb, 0xf4, 0x03, 0xfb, 0x25,
0xd8, 0x4b, 0x83, 0x7a, 0x52, 0xc6, 0x4e, 0x91, 0xad, 0xa5, 0x60, 0x21, 0xce, 0x38, 0x62, 0x1f,
0xea, 0x56, 0x62, 0xe2, 0xa6, 0x60, 0xda, 0x6f, 0x84, 0x63, 0x93, 0x72, 0x9e, 0xc6, 0xa0, 0x59,
0x9e, 0x85, 0x3c, 0xcf, 0x93, 0x98, 0xf3, 0x85, 0x02, 0xcb, 0x77, 0x89, 0x69, 0x4d, 0x1d, 0xff,
0xbf, 0x37, 0xe7, 0x47, 0xb0, 0xc8, 0xb3, 0xa4, 0x53, 0x62, 0xbc, 0xae, 0xa4, 0x79, 0x89, 0x0c,
0x9a, 0x4a, 0xb8, 0xc3, 0x00, 0xba, 0xd8, 0xa4, 0xfd, 0x52, 0x81, 0x8e, 0x8e, 0x1d, 0x6c, 0x52,
0xfc, 0x2a, 0xb5, 0x68, 0xc3, 0xa2, 0x47, 0x2c, 0xdc, 0xef, 0x31, 0x2d, 0x8a, 0xba, 0xf8, 0xd3,
0xfe, 0x2e, 0x2c, 0xfc, 0x9a, 0x07, 0x6c, 0xc2, 0x0b, 0xe5, 0x93, 0x78, 0xe1, 0xc5, 0xd4, 0x0b,
0xaf, 0xbb, 0xa6, 0x53, 0x4f, 0x95, 0x53, 0x9e, 0xfa, 0x21, 0x9c, 0xdf, 0xf0, 0xb1, 0x19, 0xe0,
0xef, 0x87, 0x65, 0x7e, 0x63, 0xcf, 0xf4, 0x3c, 0xec, 0x44, 0x2a, 0x64, 0x99, 0x2b, 0x12, 0xe6,
0x1d, 0x58, 0x1a, 0xfb, 0xe4, 0xd9, 0x41, 0x2c, 0x77, 0xf4, 0xab, 0xfd, 0x5a, 0x81, 0xae, 0x8c,
0xf6, 0x69, 0x2a, 0xc2, 0x35, 0x68, 0xf9, 0x5c, 0x38, 0x63, 0xc8, 0xe9, 0x31, 0xae, 0x55, 0xbd,
0x29, 0xc0, 0x82, 0x0b, 0xba, 0x02, 0x4d, 0x1f, 0xd3, 0x89, 0x33, 0xc5, 0x2b, 0x32, 0xbc, 0x06,
0x87, 0x0a, 0x34, 0xed, 0x4b, 0x05, 0xce, 0x6f, 0xe2, 0x20, 0xf6, 0x5e, 0xc8, 0x0e, 0xbf, 0xa6,
0xd5, 0xf5, 0x57, 0x0a, 0xb4, 0x32, 0x82, 0xa2, 0x15, 0xa8, 0x25, 0x70, 0x84, 0x83, 0x92, 0x20,
0xf4, 0x6d, 0x28, 0x87, 0xb6, 0xc3, 0x4c, 0xa4, 0xe6, 0xba, 0xb6, 0x96, 0x6f, 0xee, 0x6b, 0x69,
0xaa, 0x3a, 0xdf, 0x80, 0x6e, 0xc0, 0x59, 0x49, 0x65, 0x15, 0xe2, 0xa3, 0x7c, 0x61, 0xd5, 0x7e,
0xaf, 0x40, 0x57, 0x66, 0xcc, 0xd3, 0x38, 0xfc, 0x21, 0xb4, 0x63, 0x6d, 0x0c, 0x0b, 0xd3, 0xa1,
0x6f, 0x8f, 0x59, 0x9a, 0xb1, 0x66, 0x50, 0x5b, 0xbf, 0x7c, 0xb4, 0x3e, 0x54, 0x5f, 0x8e, 0x49,
0xf4, 0x12, 0x14, 0xb4, 0x9f, 0x29, 0xb0, 0xbc, 0x89, 0x83, 0x1d, 0x3c, 0x72, 0xb1, 0x17, 0xf4,
0xbd, 0x5d, 0x72, 0x72, 0xc7, 0x5f, 0x02, 0xa0, 0x82, 0x4e, 0xdc, 0xa8, 0x12, 0x90, 0x79, 0x82,
0x40, 0xfb, 0x67, 0x11, 0x6a, 0x09, 0x61, 0xd0, 0x05, 0xa8, 0xc6, 0x14, 0x84, 0x6b, 0xa7, 0x80,
0x1c, 0xc5, 0x82, 0x24, 0xac, 0x32, 0xe1, 0x51, 0xcc, 0x87, 0xc7, 0x8c, 0x0a, 0x8e, 0xce, 0x43,
0xc5, 0xc5, 0xae, 0x41, 0xed, 0xe7, 0x58, 0x54, 0x8c, 0x25, 0x17, 0xbb, 0x3b, 0xf6, 0x73, 0x1c,
0x2e, 0x79, 0x13, 0xd7, 0xf0, 0xc9, 0x3e, 0xed, 0x2c, 0xf2, 0x25, 0x6f, 0xe2, 0xea, 0x64, 0x9f,
0xa2, 0x8b, 0x00, 0xb6, 0x67, 0xe1, 0x67, 0x86, 0x67, 0xba, 0xb8, 0xb3, 0xc4, 0x32, 0xae, 0xca,
0x20, 0x5b, 0xa6, 0x8b, 0xc3, 0x5a, 0xc1, 0x7e, 0xfa, 0xbd, 0x4e, 0x85, 0x6f, 0x14, 0xbf, 0xa1,
0xaa, 0x22, 0x4f, 0xfb, 0xbd, 0x4e, 0x95, 0xef, 0x8b, 0x01, 0xe8, 0x13, 0x68, 0x08, 0xbd, 0x0d,
0x1e, 0xcb, 0xc0, 0x62, 0x79, 0x45, 0xe6, 0x7b, 0x61, 0x40, 0x1e, 0xc9, 0x75, 0x9a, 0xf8, 0x43,
0x57, 0xa1, 0x39, 0x24, 0xee, 0xd8, 0x64, 0xd6, 0xb9, 0xe3, 0x13, 0xb7, 0x53, 0x63, 0x7e, 0xca,
0x40, 0xd1, 0x4d, 0x38, 0x3b, 0x64, 0x75, 0xcb, 0xba, 0x7d, 0xb0, 0x11, 0x2f, 0x75, 0xea, 0x2b,
0xca, 0xf5, 0x8a, 0x2e, 0x5b, 0x42, 0xdf, 0x8a, 0x92, 0xac, 0xc1, 0x04, 0x7b, 0x4b, 0x1e, 0xd9,
0x49, 0xc9, 0x38, 0x3e, 0x3b, 0x0a, 0x67, 0x43, 0xf0, 0x34, 0xe9, 0xf2, 0x0d, 0x28, 0xdb, 0xde,
0x2e, 0x89, 0xb2, 0xe3, 0xcd, 0x43, 0x2c, 0xc4, 0x98, 0x71, 0x6c, 0xed, 0x8f, 0x45, 0x68, 0x7f,
0x6c, 0x59, 0xb2, 0x1e, 0x70, 0xfc, 0x54, 0x98, 0x86, 0x54, 0x21, 0x15, 0x52, 0xf3, 0xd4, 0xc1,
0x77, 0xe1, 0x4c, 0xa6, 0xbe, 0x8b, 0xc8, 0xac, 0xea, 0x6a, 0xba, 0xc2, 0xf7, 0x7b, 0xe8, 0x1d,
0x50, 0xd3, 0x35, 0x5e, 0x74, 0xb7, 0xaa, 0xde, 0x4a, 0x55, 0xf9, 0x7e, 0x0f, 0x7d, 0x13, 0xde,
0x18, 0x39, 0x64, 0x60, 0x3a, 0x06, 0xc5, 0xa6, 0x83, 0x2d, 0x63, 0x9a, 0x58, 0x8b, 0x2c, 0x06,
0x96, 0xf9, 0xf2, 0x0e, 0x5b, 0xdd, 0x89, 0x93, 0x6c, 0x33, 0x8c, 0x3c, 0xfc, 0xd8, 0x18, 0x13,
0xca, 0x32, 0x86, 0xc5, 0x74, 0x2d, 0x5b, 0x45, 0xe3, 0xf9, 0xe7, 0x1e, 0x1d, 0x6d, 0x0b, 0xcc,
0x30, 0xf6, 0xf0, 0xe3, 0xe8, 0x0f, 0x7d, 0x0e, 0x6d, 0xa9, 0x00, 0xb4, 0x53, 0x99, 0xcf, 0x53,
0xe7, 0x24, 0x02, 0x52, 0xed, 0xaf, 0x0a, 0x9c, 0xd7, 0xb1, 0x4b, 0x9e, 0xe2, 0xaf, 0xac, 0xef,
0xb4, 0xbf, 0x15, 0xa0, 0xfd, 0x03, 0x33, 0x18, 0xee, 0xf5, 0x5c, 0x01, 0xa4, 0xaf, 0x46, 0xc1,
0x4c, 0x35, 0x2d, 0xe5, 0xab, 0x69, 0x9c, 0x7e, 0x65, 0x99, 0x53, 0xc3, 0x41, 0x78, 0xed, 0x8b,
0x48, 0xdf, 0x69, 0xfa, 0x25, 0x8e, 0xa1, 0x8b, 0x27, 0x38, 0x86, 0xa2, 0x0d, 0x68, 0xe0, 0x67,
0x43, 0x67, 0x62, 0x61, 0x83, 0x73, 0x5f, 0x62, 0xdc, 0x2f, 0x49, 0xb8, 0x27, 0x23, 0xaa, 0x2e,
0x36, 0xf5, 0x59, 0x09, 0x78, 0xa1, 0xc0, 0x79, 0x6e, 0x65, 0xec, 0x04, 0xe6, 0xab, 0x35, 0x74,
0x6c, 0xc6, 0xd2, 0x71, 0xcc, 0xa8, 0xfd, 0xa4, 0x08, 0x2d, 0xa1, 0x60, 0x38, 0x7c, 0xcc, 0xd1,
0x43, 0x33, 0x1e, 0x2d, 0xe4, 0x3d, 0x3a, 0x8f, 0xb8, 0xd1, 0xa1, 0xaf, 0x94, 0x38, 0xf4, 0x5d,
0x04, 0xd8, 0x75, 0x26, 0x74, 0xcf, 0x08, 0x6c, 0x37, 0xea, 0xa0, 0x55, 0x06, 0x79, 0x60, 0xbb,
0x18, 0x7d, 0x0c, 0xf5, 0x81, 0xed, 0x39, 0x64, 0x64, 0x8c, 0xcd, 0x60, 0x8f, 0xb2, 0x22, 0x24,
0xf7, 0xd8, 0x1d, 0x1b, 0x3b, 0xd6, 0x6d, 0x86, 0xab, 0xd7, 0xf8, 0x9e, 0xed, 0x70, 0x0b, 0xba,
0x04, 0xb5, 0xb0, 0x0d, 0x93, 0x5d, 0xde, 0x89, 0x97, 0x38, 0x0b, 0x6f, 0xe2, 0xde, 0xdf, 0x65,
0xbd, 0xf8, 0xbb, 0x50, 0x0d, 0x9b, 0x02, 0x75, 0xc8, 0x28, 0x2a, 0x32, 0x47, 0xd1, 0x9f, 0x6e,
0x40, 0x1f, 0x41, 0xd5, 0x0a, 0x03, 0x81, 0xed, 0xae, 0xce, 0x74, 0x03, 0x0b, 0x96, 0xbb, 0x64,
0xc4, 0xdc, 0x30, 0xdd, 0xa1, 0xfd, 0xbb, 0x00, 0x67, 0x43, 0x1f, 0x44, 0x85, 0xea, 0xe4, 0x71,
0x74, 0x11, 0xc0, 0xa2, 0x81, 0x91, 0x8a, 0xa5, 0xaa, 0x45, 0x83, 0x2d, 0x1e, 0x4e, 0x1f, 0x46,
0xa1, 0x52, 0x9c, 0x7d, 0x1c, 0xcc, 0xc4, 0x44, 0x3e, 0xeb, 0x4e, 0x32, 0x82, 0xa3, 0xef, 0x41,
0xd3, 0x21, 0xa6, 0x65, 0x0c, 0x89, 0x67, 0xf1, 0xde, 0x50, 0x66, 0xcd, 0xff, 0x6d, 0x99, 0x08,
0x0f, 0x7c, 0x7b, 0x34, 0xc2, 0xfe, 0x46, 0x84, 0xab, 0x37, 0x1c, 0x76, 0x01, 0x21, 0x7e, 0xd1,
0x65, 0x68, 0x50, 0x32, 0xf1, 0x87, 0x38, 0x52, 0x94, 0x1f, 0xac, 0xea, 0x1c, 0xb8, 0x25, 0x4f,
0x9d, 0x25, 0xc9, 0x19, 0xf2, 0x2f, 0x0a, 0xb4, 0xc5, 0x48, 0x7a, 0x7a, 0xdb, 0xcf, 0xca, 0xe1,
0x28, 0xe0, 0x8b, 0x87, 0x4c, 0x39, 0xa5, 0x39, 0xa6, 0x9c, 0xb2, 0x64, 0x50, 0x4d, 0x1f, 0xa4,
0x17, 0xb3, 0x07, 0x69, 0xed, 0x01, 0x34, 0xe2, 0x3e, 0xc0, 0x32, 0xfc, 0x32, 0x34, 0xb8, 0x58,
0x46, 0x68, 0x52, 0x6c, 0x45, 0x53, 0x2a, 0x07, 0xde, 0x65, 0xb0, 0x90, 0x6a, 0xdc, 0x67, 0xf8,
0xe1, 0xa8, 0xaa, 0x27, 0x20, 0xda, 0x1f, 0x0a, 0xa0, 0x26, 0x3b, 0x28, 0xa3, 0x3c, 0xcf, 0xf8,
0x7b, 0x0d, 0x5a, 0xe2, 0x02, 0x35, 0x6e, 0x63, 0x62, 0x20, 0x7d, 0x92, 0x24, 0xd7, 0x43, 0x1f,
0x40, 0x9b, 0x23, 0xe6, 0xda, 0x1e, 0x1f, 0x4c, 0xcf, 0xb1, 0x55, 0x3d, 0x73, 0x6e, 0x99, 0x7d,
0x6c, 0x28, 0x9d, 0xe2, 0xd8, 0x90, 0x3f, 0xd6, 0x94, 0x4f, 0x76, 0xac, 0xd1, 0xfe, 0x5c, 0x84,
0xe6, 0x34, 0x43, 0xe6, 0xb6, 0xda, 0x3c, 0x17, 0x7b, 0x5b, 0xa0, 0x4e, 0x27, 0x3f, 0x76, 0x5a,
0x3e, 0x34, 0xc9, 0xb3, 0x33, 0x5f, 0x6b, 0x9c, 0x19, 0x95, 0xef, 0x40, 0x43, 0xd8, 0xdc, 0x48,
0x36, 0x97, 0xb7, 0x64, 0xc4, 0x52, 0x11, 0xa6, 0xd7, 0x13, 0xbd, 0x86, 0xa2, 0x0f, 0xa1, 0xca,
0xf2, 0x3e, 0x38, 0x18, 0x63, 0x91, 0xf2, 0x17, 0x64, 0x34, 0xc2, 0xc8, 0x7b, 0x70, 0x30, 0xc6,
0x7a, 0xc5, 0x11, 0x5f, 0xa7, 0xed, 0xf3, 0xb7, 0x60, 0xd9, 0xe7, 0xa9, 0x6d, 0x19, 0x29, 0xf3,
0x2d, 0x31, 0xf3, 0x9d, 0x8b, 0x16, 0xb7, 0x93, 0x66, 0x9c, 0x31, 0xc5, 0x57, 0x66, 0x4e, 0xf1,
0xbf, 0x28, 0x40, 0x3b, 0x94, 0xfd, 0xb6, 0xe9, 0x98, 0xde, 0x10, 0xcf, 0x3f, 0x90, 0xfe, 0x77,
0x9a, 0x69, 0xae, 0x12, 0x96, 0x24, 0x95, 0x30, 0xdd, 0x14, 0xca, 0xd9, 0xa6, 0xf0, 0x26, 0xd4,
0x04, 0x0d, 0x8b, 0x78, 0x98, 0x19, 0xbb, 0xa2, 0x03, 0x07, 0xf5, 0x88, 0xc7, 0x46, 0xd8, 0x70,
0x3f, 0x5b, 0x5d, 0x62, 0xab, 0x4b, 0x16, 0x0d, 0xd8, 0xd2, 0x45, 0x80, 0xa7, 0xa6, 0x63, 0x5b,
0x2c, 0x48, 0x98, 0x99, 0x2a, 0x7a, 0x95, 0x41, 0x42, 0x13, 0x68, 0x3f, 0x57, 0xa0, 0xfd, 0xa9,
0xe9, 0x59, 0x64, 0x77, 0xf7, 0xf4, 0xf5, 0x75, 0x03, 0xa2, 0x01, 0xb5, 0x7f, 0x9c, 0xa1, 0x2d,
0xb5, 0x49, 0xfb, 0x93, 0x02, 0x28, 0xe1, 0xaf, 0x93, 0x4b, 0x73, 0x05, 0x9a, 0x29, 0xcb, 0xc7,
0xef, 0x17, 0x49, 0xd3, 0xd3, 0xb0, 0xef, 0x0d, 0x38, 0x2b, 0xc3, 0xc7, 0x26, 0x25, 0x1e, 0x73,
0xe3, 0xdc, 0x7d, 0x6f, 0x10, 0x89, 0x19, 0x6e, 0xd5, 0xfe, 0xa5, 0xc0, 0x19, 0xa1, 0x5a, 0x98,
0x71, 0x23, 0x1c, 0x95, 0x74, 0xe2, 0x39, 0xb6, 0x17, 0xc7, 0x80, 0xa8, 0x21, 0x1c, 0x28, 0x9c,
0xfc, 0x29, 0xb4, 0x04, 0x52, 0x5c, 0x13, 0xe7, 0xb4, 0x5f, 0x93, 0xef, 0x8b, 0xab, 0xe1, 0x15,
0x68, 0x92, 0xdd, 0xdd, 0x24, 0x3f, 0x1e, 0x98, 0x0d, 0x01, 0x15, 0x0c, 0x3f, 0x03, 0x35, 0x42,
0x3b, 0x6e, 0x15, 0x6e, 0x89, 0x8d, 0xf1, 0xdc, 0xf6, 0x53, 0x05, 0x3a, 0xe9, 0x9a, 0x9c, 0x50,
0xff, 0xf8, 0xae, 0xfb, 0x4e, 0x7a, 0xec, 0xbf, 0x72, 0x88, 0x3c, 0x53, 0x3e, 0xe2, 0x1c, 0xb4,
0xfa, 0x1c, 0x9a, 0xe9, 0xe2, 0x89, 0xea, 0x50, 0xd9, 0x22, 0xc1, 0x27, 0xcf, 0x6c, 0x1a, 0xa8,
0x0b, 0xa8, 0x09, 0xb0, 0x45, 0x82, 0x6d, 0x1f, 0x53, 0xec, 0x05, 0xaa, 0x82, 0x00, 0x16, 0xef,
0x7b, 0x3d, 0x9b, 0x3e, 0x56, 0x0b, 0xe8, 0xac, 0xb8, 0x92, 0x34, 0x9d, 0xbe, 0xa8, 0x24, 0x6a,
0x31, 0xdc, 0x1e, 0xff, 0x95, 0x90, 0x0a, 0xf5, 0x18, 0x65, 0x73, 0xfb, 0x73, 0xb5, 0x8c, 0xaa,
0x50, 0xe6, 0x9f, 0x8b, 0xab, 0xf7, 0x41, 0xcd, 0x86, 0x08, 0xaa, 0xc1, 0xd2, 0x1e, 0xcf, 0x30,
0x75, 0x01, 0xb5, 0xa0, 0xe6, 0x4c, 0x83, 0x5b, 0x55, 0x42, 0xc0, 0xc8, 0x1f, 0x0f, 0x45, 0x98,
0xab, 0x85, 0x90, 0x5b, 0xe8, 0xb5, 0x1e, 0xd9, 0xf7, 0xd4, 0xe2, 0xea, 0x67, 0x50, 0x4f, 0xde,
0xb3, 0xa0, 0x0a, 0x94, 0xb6, 0x88, 0x87, 0xd5, 0x85, 0x90, 0xec, 0xa6, 0x4f, 0xf6, 0x6d, 0x6f,
0xc4, 0x75, 0xb8, 0xe3, 0x93, 0xe7, 0xd8, 0x53, 0x0b, 0xe1, 0x42, 0xd8, 0x5c, 0xc3, 0x85, 0x62,
0xb8, 0xc0, 0x3b, 0xad, 0x5a, 0x5a, 0x7d, 0x1f, 0x2a, 0x51, 0x11, 0x47, 0x67, 0xa0, 0x91, 0x7a,
0xd0, 0x50, 0x17, 0x10, 0xe2, 0x07, 0xc0, 0x69, 0xb9, 0x56, 0x95, 0xf5, 0x7f, 0x00, 0x00, 0x3f,
0x47, 0x10, 0xe2, 0x5b, 0x68, 0x0c, 0x68, 0x13, 0x07, 0x1b, 0xc4, 0x1d, 0x13, 0x2f, 0x12, 0x89,
0xa2, 0x9b, 0x33, 0xda, 0x6c, 0x1e, 0x55, 0x68, 0xd9, 0xbd, 0x3a, 0x63, 0x47, 0x06, 0x5d, 0x5b,
0x40, 0x2e, 0xe3, 0x18, 0xce, 0x18, 0x0f, 0xec, 0xe1, 0xe3, 0xe8, 0x36, 0xfc, 0x10, 0x8e, 0x19,
0xd4, 0x88, 0x63, 0xa6, 0xc7, 0x8a, 0x9f, 0x9d, 0xc0, 0xb7, 0xbd, 0x51, 0x74, 0x45, 0xa5, 0x2d,
0xa0, 0x27, 0x70, 0x6e, 0x13, 0x33, 0xee, 0x36, 0x0d, 0xec, 0x21, 0x8d, 0x18, 0xae, 0xcf, 0x66,
0x98, 0x43, 0x3e, 0x26, 0x4b, 0x07, 0x5a, 0x99, 0x57, 0x5b, 0xb4, 0x2a, 0x8d, 0x77, 0xe9, 0x0b,
0x73, 0xf7, 0xdd, 0xb9, 0x70, 0x63, 0x6e, 0x36, 0x34, 0xd3, 0x2f, 0x9a, 0xe8, 0x9d, 0x59, 0x04,
0x72, 0x4f, 0x40, 0xdd, 0xd5, 0x79, 0x50, 0x63, 0x56, 0x0f, 0xa1, 0x99, 0x7e, 0x33, 0x93, 0xb3,
0x92, 0xbe, 0xab, 0x75, 0x0f, 0xbb, 0x1d, 0xd4, 0x16, 0xd0, 0x8f, 0xe0, 0x4c, 0xee, 0xa1, 0x0a,
0x7d, 0x5d, 0x46, 0x7e, 0xd6, 0x7b, 0xd6, 0x51, 0x1c, 0x84, 0xf4, 0x53, 0x2b, 0xce, 0x96, 0x3e,
0xf7, 0x62, 0x39, 0xbf, 0xf4, 0x09, 0xf2, 0x87, 0x49, 0x7f, 0x6c, 0x0e, 0x13, 0x40, 0xf9, 0xa7,
0x2a, 0xf4, 0x9e, 0x8c, 0xc5, 0xcc, 0xe7, 0xb2, 0xee, 0xda, 0xbc, 0xe8, 0xb1, 0xcb, 0x27, 0x2c,
0x5b, 0xb3, 0x8f, 0x3a, 0x52, 0xb6, 0x33, 0x5f, 0xa9, 0xe4, 0x6c, 0x67, 0xbf, 0xc3, 0xf0, 0xa0,
0x4e, 0x5f, 0x3a, 0xcb, 0x7d, 0x25, 0x7d, 0x1b, 0x91, 0x07, 0xb5, 0xfc, 0x0e, 0x5b, 0x5b, 0x40,
0x06, 0xc0, 0x26, 0x0e, 0xee, 0xe1, 0xc0, 0xb7, 0x87, 0x14, 0x5d, 0x95, 0xa6, 0xf8, 0x14, 0x21,
0xe2, 0x71, 0xed, 0x48, 0xbc, 0x88, 0xc1, 0xfa, 0x97, 0x00, 0x55, 0x66, 0xdd, 0xb0, 0x4b, 0xff,
0xbf, 0xe0, 0xbe, 0x84, 0x82, 0xfb, 0x08, 0x5a, 0x99, 0xb7, 0x01, 0x79, 0xc1, 0x95, 0x3f, 0x20,
0x1c, 0x95, 0x79, 0x03, 0x40, 0xf9, 0x0b, 0x6c, 0x79, 0x0a, 0xcc, 0xbc, 0xe8, 0x3e, 0x8a, 0xc7,
0x23, 0x68, 0x65, 0x2e, 0x90, 0xe5, 0x1a, 0xc8, 0x6f, 0x99, 0xe7, 0xd0, 0x20, 0x7f, 0x71, 0x2a,
0xd7, 0x60, 0xe6, 0x05, 0xeb, 0x51, 0x3c, 0xbe, 0x80, 0x7a, 0xf2, 0x3a, 0x0d, 0x5d, 0x9b, 0x55,
0x5b, 0x33, 0x43, 0xc9, 0xab, 0xaf, 0xac, 0x2f, 0xbf, 0xf3, 0x3c, 0x82, 0x56, 0xe6, 0xc6, 0x4b,
0xee, 0x5d, 0xf9, 0xb5, 0xd8, 0x51, 0xd4, 0xbf, 0x42, 0xb5, 0xf2, 0xf6, 0x07, 0x0f, 0xd7, 0x47,
0x76, 0xb0, 0x37, 0x19, 0x84, 0x5a, 0xde, 0xe0, 0x98, 0xef, 0xd9, 0x44, 0x7c, 0xdd, 0x88, 0x8a,
0xc6, 0x0d, 0x46, 0xe9, 0x06, 0x93, 0x76, 0x3c, 0x18, 0x2c, 0xb2, 0xdf, 0x5b, 0xff, 0x09, 0x00,
0x00, 0xff, 0xff, 0x2e, 0xfd, 0xc9, 0x8e, 0x2a, 0x28, 0x00, 0x00,
// 2321 bytes of a gzipped FileDescriptorProto
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x19, 0xcb, 0x6e, 0x1c, 0xc7,
0x91, 0xb3, 0x0f, 0x72, 0xb7, 0xf6, 0x35, 0x6a, 0x89, 0xf4, 0x8a, 0xb1, 0x6c, 0x7a, 0x64, 0x3d,
0x4c, 0xc7, 0x94, 0x4c, 0x39, 0x0f, 0x23, 0xf1, 0xc1, 0xe2, 0x5a, 0xf4, 0x3a, 0x12, 0xc5, 0x0c,
0x65, 0x07, 0x11, 0x04, 0x4c, 0x86, 0x3b, 0xcd, 0xe5, 0x40, 0x33, 0xd3, 0xab, 0xe9, 0x59, 0x49,
0xd4, 0x39, 0x07, 0xe7, 0x10, 0xe4, 0x03, 0x12, 0x04, 0x08, 0x90, 0x20, 0xf0, 0x21, 0xc7, 0xe4,
0xac, 0x4f, 0xc8, 0x17, 0x04, 0x08, 0x92, 0x7b, 0x72, 0x49, 0xce, 0x41, 0x3f, 0x66, 0x76, 0x1e,
0xbd, 0xe4, 0x92, 0x8c, 0x22, 0xc1, 0xc8, 0x6d, 0xa6, 0xba, 0xba, 0xaa, 0xba, 0x9e, 0x5d, 0xd5,
0x70, 0xe6, 0xd1, 0x18, 0x87, 0x07, 0xd6, 0x80, 0x90, 0xd0, 0x59, 0x1b, 0x85, 0x24, 0x22, 0x08,
0xf9, 0xae, 0xf7, 0x78, 0x4c, 0xc5, 0xdf, 0x1a, 0x5f, 0x5f, 0x6e, 0x0e, 0x88, 0xef, 0x93, 0x40,
0xc0, 0x96, 0x9b, 0x69, 0x8c, 0xe5, 0xb6, 0x1b, 0x44, 0x38, 0x0c, 0x6c, 0x2f, 0x5e, 0xa5, 0x83,
0x7d, 0xec, 0xdb, 0xf2, 0x4f, 0x77, 0xec, 0xc8, 0x4e, 0xd3, 0x37, 0x7e, 0xaa, 0xc1, 0xd2, 0xce,
0x3e, 0x79, 0xb2, 0x41, 0x3c, 0x0f, 0x0f, 0x22, 0x97, 0x04, 0xd4, 0xc4, 0x8f, 0xc6, 0x98, 0x46,
0xe8, 0x3a, 0x54, 0x76, 0x6d, 0x8a, 0xbb, 0xda, 0x8a, 0x76, 0xb5, 0xb1, 0xfe, 0xfa, 0x5a, 0x46,
0x12, 0x29, 0xc2, 0x1d, 0x3a, 0xbc, 0x69, 0x53, 0x6c, 0x72, 0x4c, 0x84, 0xa0, 0xe2, 0xec, 0xf6,
0x7b, 0xdd, 0xd2, 0x8a, 0x76, 0xb5, 0x6c, 0xf2, 0x6f, 0xf4, 0x36, 0xb4, 0x06, 0x09, 0xed, 0x7e,
0x8f, 0x76, 0xcb, 0x2b, 0xe5, 0xab, 0x65, 0x33, 0x0b, 0x34, 0x7e, 0xaf, 0xc1, 0x6b, 0x05, 0x31,
0xe8, 0x88, 0x04, 0x14, 0xa3, 0x1b, 0x30, 0x4f, 0x23, 0x3b, 0x1a, 0x53, 0x29, 0xc9, 0x37, 0x94,
0x92, 0xec, 0x70, 0x14, 0x53, 0xa2, 0x16, 0xd9, 0x96, 0x14, 0x6c, 0xd1, 0xfb, 0x70, 0xce, 0x0d,
0xee, 0x60, 0x9f, 0x84, 0x07, 0xd6, 0x08, 0x87, 0x03, 0x1c, 0x44, 0xf6, 0x10, 0xc7, 0x32, 0x9e,
0x8d, 0xd7, 0xb6, 0x27, 0x4b, 0xc6, 0xef, 0x34, 0x58, 0x64, 0x92, 0x6e, 0xdb, 0x61, 0xe4, 0xbe,
0x00, 0x7d, 0x19, 0xd0, 0x4c, 0xcb, 0xd8, 0x2d, 0xf3, 0xb5, 0x0c, 0x8c, 0xe1, 0x8c, 0x62, 0xf6,
0xec, 0x6c, 0x15, 0x2e, 0x6e, 0x06, 0x66, 0xfc, 0x56, 0x1a, 0x36, 0x2d, 0xe7, 0x69, 0x14, 0x9a,
0xe7, 0x59, 0x2a, 0xf2, 0x3c, 0x89, 0x3a, 0x9f, 0x6b, 0xb0, 0x78, 0x9b, 0xd8, 0xce, 0xc4, 0xf0,
0xff, 0x7b, 0x75, 0x7e, 0x04, 0xf3, 0x22, 0x4a, 0xba, 0x15, 0xce, 0xeb, 0x52, 0x96, 0x97, 0x8c,
0xa0, 0x89, 0x84, 0x3b, 0x1c, 0x60, 0xca, 0x4d, 0xc6, 0xaf, 0x34, 0xe8, 0x9a, 0xd8, 0xc3, 0x36,
0xc5, 0x2f, 0xf3, 0x14, 0x4b, 0x30, 0x1f, 0x10, 0x07, 0xf7, 0x7b, 0xfc, 0x14, 0x65, 0x53, 0xfe,
0x19, 0x7f, 0x97, 0x1a, 0x7e, 0xc5, 0x1d, 0x36, 0x65, 0x85, 0xea, 0x49, 0xac, 0xf0, 0x7c, 0x62,
0x85, 0x57, 0xfd, 0xa4, 0x13, 0x4b, 0x55, 0x33, 0x96, 0xfa, 0x31, 0x9c, 0xdf, 0x08, 0xb1, 0x1d,
0xe1, 0x1f, 0xb2, 0x34, 0xbf, 0xb1, 0x6f, 0x07, 0x01, 0xf6, 0xe2, 0x23, 0xe4, 0x99, 0x6b, 0x0a,
0xe6, 0x5d, 0x58, 0x18, 0x85, 0xe4, 0xe9, 0x41, 0x22, 0x77, 0xfc, 0x6b, 0xfc, 0x46, 0x83, 0x65,
0x15, 0xed, 0xd3, 0x64, 0x84, 0x2b, 0xd0, 0x09, 0x85, 0x70, 0xd6, 0x40, 0xd0, 0xe3, 0x5c, 0xeb,
0x66, 0x5b, 0x82, 0x25, 0x17, 0x74, 0x09, 0xda, 0x21, 0xa6, 0x63, 0x6f, 0x82, 0x57, 0xe6, 0x78,
0x2d, 0x01, 0x95, 0x68, 0xc6, 0x57, 0x1a, 0x9c, 0xdf, 0xc4, 0x51, 0x62, 0x3d, 0xc6, 0x0e, 0xbf,
0xa2, 0xd9, 0xf5, 0xd7, 0x1a, 0x74, 0x72, 0x82, 0xa2, 0x15, 0x68, 0xa4, 0x70, 0xa4, 0x81, 0xd2,
0x20, 0xf4, 0x5d, 0xa8, 0x32, 0xdd, 0x61, 0x2e, 0x52, 0x7b, 0xdd, 0x58, 0x2b, 0x16, 0xf7, 0xb5,
0x2c, 0x55, 0x53, 0x6c, 0x40, 0xd7, 0xe0, 0xac, 0x22, 0xb3, 0x4a, 0xf1, 0x51, 0x31, 0xb1, 0x1a,
0x7f, 0xd0, 0x60, 0x59, 0xa5, 0xcc, 0xd3, 0x18, 0xfc, 0x3e, 0x2c, 0x25, 0xa7, 0xb1, 0x1c, 0x4c,
0x07, 0xa1, 0x3b, 0xe2, 0x61, 0xc6, 0x8b, 0x41, 0x63, 0xfd, 0xe2, 0xd1, 0xe7, 0xa1, 0xe6, 0x62,
0x42, 0xa2, 0x97, 0xa2, 0x60, 0xfc, 0x5c, 0x83, 0xc5, 0x4d, 0x1c, 0xed, 0xe0, 0xa1, 0x8f, 0x83,
0xa8, 0x1f, 0xec, 0x91, 0x93, 0x1b, 0xfe, 0x0d, 0x00, 0x2a, 0xe9, 0x24, 0x85, 0x2a, 0x05, 0x99,
0xc5, 0x09, 0x8c, 0x7f, 0x96, 0xa1, 0x91, 0x12, 0x06, 0xbd, 0x0e, 0xf5, 0x84, 0x82, 0x34, 0xed,
0x04, 0x50, 0xa0, 0x58, 0x52, 0xb8, 0x55, 0xce, 0x3d, 0xca, 0x45, 0xf7, 0x98, 0x92, 0xc1, 0xd1,
0x79, 0xa8, 0xf9, 0xd8, 0xb7, 0xa8, 0xfb, 0x0c, 0xcb, 0x8c, 0xb1, 0xe0, 0x63, 0x7f, 0xc7, 0x7d,
0x86, 0xd9, 0x52, 0x30, 0xf6, 0xad, 0x90, 0x3c, 0xa1, 0xdd, 0x79, 0xb1, 0x14, 0x8c, 0x7d, 0x93,
0x3c, 0xa1, 0xe8, 0x02, 0x80, 0x1b, 0x38, 0xf8, 0xa9, 0x15, 0xd8, 0x3e, 0xee, 0x2e, 0xf0, 0x88,
0xab, 0x73, 0xc8, 0x96, 0xed, 0x63, 0x96, 0x2b, 0xf8, 0x4f, 0xbf, 0xd7, 0xad, 0x89, 0x8d, 0xf2,
0x97, 0x1d, 0x55, 0xc6, 0x69, 0xbf, 0xd7, 0xad, 0x8b, 0x7d, 0x09, 0x00, 0x7d, 0x02, 0x2d, 0x79,
0x6e, 0x4b, 0xf8, 0x32, 0x70, 0x5f, 0x5e, 0x51, 0xd9, 0x5e, 0x2a, 0x50, 0x78, 0x72, 0x93, 0xa6,
0xfe, 0xd0, 0x65, 0x68, 0x0f, 0x88, 0x3f, 0xb2, 0xb9, 0x76, 0x6e, 0x85, 0xc4, 0xef, 0x36, 0xb8,
0x9d, 0x72, 0x50, 0x74, 0x1d, 0xce, 0x0e, 0x78, 0xde, 0x72, 0x6e, 0x1e, 0x6c, 0x24, 0x4b, 0xdd,
0xe6, 0x8a, 0x76, 0xb5, 0x66, 0xaa, 0x96, 0xd0, 0x77, 0xe2, 0x20, 0x6b, 0x71, 0xc1, 0xde, 0x52,
0x7b, 0x76, 0x5a, 0x32, 0x81, 0xcf, 0xaf, 0xc2, 0x79, 0x17, 0x3c, 0x4d, 0xb8, 0x7c, 0x0b, 0xaa,
0x6e, 0xb0, 0x47, 0xe2, 0xe8, 0x78, 0xf3, 0x10, 0x0d, 0x71, 0x66, 0x02, 0xdb, 0xf8, 0x53, 0x19,
0x96, 0x3e, 0x76, 0x1c, 0x55, 0x0d, 0x38, 0x7e, 0x28, 0x4c, 0x5c, 0xaa, 0x94, 0x71, 0xa9, 0x59,
0xf2, 0xe0, 0xbb, 0x70, 0x26, 0x97, 0xdf, 0xa5, 0x67, 0xd6, 0x4d, 0x3d, 0x9b, 0xe1, 0xfb, 0x3d,
0xf4, 0x0e, 0xe8, 0xd9, 0x1c, 0x2f, 0xab, 0x5b, 0xdd, 0xec, 0x64, 0xb2, 0x7c, 0xbf, 0x87, 0xbe,
0x0d, 0xaf, 0x0d, 0x3d, 0xb2, 0x6b, 0x7b, 0x16, 0xc5, 0xb6, 0x87, 0x1d, 0x6b, 0x12, 0x58, 0xf3,
0xdc, 0x07, 0x16, 0xc5, 0xf2, 0x0e, 0x5f, 0xdd, 0x49, 0x82, 0x6c, 0x93, 0x79, 0x1e, 0x7e, 0x68,
0x8d, 0x08, 0xe5, 0x11, 0xc3, 0x7d, 0xba, 0x91, 0xcf, 0xa2, 0x49, 0xff, 0x73, 0x87, 0x0e, 0xb7,
0x25, 0x26, 0xf3, 0x3d, 0xfc, 0x30, 0xfe, 0x43, 0x9f, 0xc3, 0x92, 0x52, 0x00, 0xda, 0xad, 0xcd,
0x66, 0xa9, 0x73, 0x0a, 0x01, 0xa9, 0xf1, 0x57, 0x0d, 0xce, 0x9b, 0xd8, 0x27, 0x8f, 0xf1, 0xd7,
0xd6, 0x76, 0xc6, 0xdf, 0x4a, 0xb0, 0xf4, 0x23, 0x3b, 0x1a, 0xec, 0xf7, 0x7c, 0x09, 0xa4, 0x2f,
0xe7, 0x80, 0xb9, 0x6c, 0x5a, 0x29, 0x66, 0xd3, 0x24, 0xfc, 0xaa, 0x2a, 0xa3, 0xb2, 0x46, 0x78,
0xed, 0x8b, 0xf8, 0xbc, 0x93, 0xf0, 0x4b, 0x5d, 0x43, 0xe7, 0x4f, 0x70, 0x0d, 0x45, 0x1b, 0xd0,
0xc2, 0x4f, 0x07, 0xde, 0xd8, 0xc1, 0x96, 0xe0, 0xbe, 0xc0, 0xb9, 0xbf, 0xa1, 0xe0, 0x9e, 0xf6,
0xa8, 0xa6, 0xdc, 0xd4, 0xe7, 0x29, 0xe0, 0xb9, 0x06, 0xe7, 0x85, 0x96, 0xb1, 0x17, 0xd9, 0x2f,
0x57, 0xd1, 0x89, 0x1a, 0x2b, 0xc7, 0x51, 0xa3, 0xf1, 0x65, 0x19, 0x3a, 0xf2, 0x80, 0xac, 0xf9,
0x98, 0xa1, 0x86, 0xe6, 0x2c, 0x5a, 0x2a, 0x5a, 0x74, 0x16, 0x71, 0xe3, 0x4b, 0x5f, 0x25, 0x75,
0xe9, 0xbb, 0x00, 0xb0, 0xe7, 0x8d, 0xe9, 0xbe, 0x15, 0xb9, 0x7e, 0x5c, 0x41, 0xeb, 0x1c, 0x72,
0xcf, 0xf5, 0x31, 0xfa, 0x18, 0x9a, 0xbb, 0x6e, 0xe0, 0x91, 0xa1, 0x35, 0xb2, 0xa3, 0x7d, 0xca,
0x93, 0x90, 0xda, 0x62, 0xb7, 0x5c, 0xec, 0x39, 0x37, 0x39, 0xae, 0xd9, 0x10, 0x7b, 0xb6, 0xd9,
0x16, 0xf4, 0x06, 0x34, 0x58, 0x19, 0x26, 0x7b, 0xa2, 0x12, 0x2f, 0x08, 0x16, 0xc1, 0xd8, 0xbf,
0xbb, 0xc7, 0x6b, 0xf1, 0xf7, 0xa1, 0xce, 0x8a, 0x02, 0xf5, 0xc8, 0x30, 0x4e, 0x32, 0x47, 0xd1,
0x9f, 0x6c, 0x40, 0x1f, 0x41, 0xdd, 0x61, 0x8e, 0xc0, 0x77, 0xd7, 0xa7, 0x9a, 0x81, 0x3b, 0xcb,
0x6d, 0x32, 0xe4, 0x66, 0x98, 0xec, 0x30, 0xfe, 0x5d, 0x82, 0xb3, 0xcc, 0x06, 0x71, 0xa2, 0x3a,
0xb9, 0x1f, 0x5d, 0x00, 0x70, 0x68, 0x64, 0x65, 0x7c, 0xa9, 0xee, 0xd0, 0x68, 0x4b, 0xb8, 0xd3,
0x87, 0xb1, 0xab, 0x94, 0xa7, 0x5f, 0x07, 0x73, 0x3e, 0x51, 0x8c, 0xba, 0x93, 0xb4, 0xe0, 0xe8,
0x07, 0xd0, 0xf6, 0x88, 0xed, 0x58, 0x03, 0x12, 0x38, 0xa2, 0x36, 0x54, 0x79, 0xf1, 0x7f, 0x5b,
0x25, 0xc2, 0xbd, 0xd0, 0x1d, 0x0e, 0x71, 0xb8, 0x11, 0xe3, 0x9a, 0x2d, 0x8f, 0x0f, 0x20, 0xe4,
0x2f, 0xba, 0x08, 0x2d, 0x4a, 0xc6, 0xe1, 0x00, 0xc7, 0x07, 0x15, 0x17, 0xab, 0xa6, 0x00, 0x6e,
0xa9, 0x43, 0x67, 0x41, 0x71, 0x87, 0xfc, 0x8b, 0x06, 0x4b, 0xb2, 0x25, 0x3d, 0xbd, 0xee, 0xa7,
0xc5, 0x70, 0xec, 0xf0, 0xe5, 0x43, 0xba, 0x9c, 0xca, 0x0c, 0x5d, 0x4e, 0x55, 0xd1, 0xa8, 0x66,
0x2f, 0xd2, 0xf3, 0xf9, 0x8b, 0xb4, 0x71, 0x0f, 0x5a, 0x49, 0x1d, 0xe0, 0x11, 0x7e, 0x11, 0x5a,
0x42, 0x2c, 0x8b, 0xa9, 0x14, 0x3b, 0x71, 0x97, 0x2a, 0x80, 0xb7, 0x39, 0x8c, 0x51, 0x4d, 0xea,
0x8c, 0xb8, 0x1c, 0xd5, 0xcd, 0x14, 0xc4, 0xf8, 0x63, 0x09, 0xf4, 0x74, 0x05, 0xe5, 0x94, 0x67,
0x69, 0x7f, 0xaf, 0x40, 0x47, 0x0e, 0x50, 0x93, 0x32, 0x26, 0x1b, 0xd2, 0x47, 0x69, 0x72, 0x3d,
0xf4, 0x01, 0x2c, 0x09, 0xc4, 0x42, 0xd9, 0x13, 0x8d, 0xe9, 0x39, 0xbe, 0x6a, 0xe6, 0xee, 0x2d,
0xd3, 0xaf, 0x0d, 0x95, 0x53, 0x5c, 0x1b, 0x8a, 0xd7, 0x9a, 0xea, 0xc9, 0xae, 0x35, 0xc6, 0x9f,
0xcb, 0xd0, 0x9e, 0x44, 0xc8, 0xcc, 0x5a, 0x9b, 0x65, 0xb0, 0xb7, 0x05, 0xfa, 0xa4, 0xf3, 0xe3,
0xb7, 0xe5, 0x43, 0x83, 0x3c, 0xdf, 0xf3, 0x75, 0x46, 0xb9, 0x56, 0xf9, 0x16, 0xb4, 0xa4, 0xce,
0xad, 0x74, 0x71, 0x79, 0x4b, 0x45, 0x2c, 0xe3, 0x61, 0x66, 0x33, 0x55, 0x6b, 0x28, 0xfa, 0x10,
0xea, 0x3c, 0xee, 0xa3, 0x83, 0x11, 0x96, 0x21, 0xff, 0xba, 0x8a, 0x06, 0xf3, 0xbc, 0x7b, 0x07,
0x23, 0x6c, 0xd6, 0x3c, 0xf9, 0x75, 0xda, 0x3a, 0x7f, 0x03, 0x16, 0x43, 0x11, 0xda, 0x8e, 0x95,
0x51, 0xdf, 0x02, 0x57, 0xdf, 0xb9, 0x78, 0x71, 0x3b, 0xad, 0xc6, 0x29, 0x5d, 0x7c, 0x6d, 0x6a,
0x17, 0xff, 0xcb, 0x12, 0x2c, 0x31, 0xd9, 0x6f, 0xda, 0x9e, 0x1d, 0x0c, 0xf0, 0xec, 0x0d, 0xe9,
0x7f, 0xa7, 0x98, 0x16, 0x32, 0x61, 0x45, 0x91, 0x09, 0xb3, 0x45, 0xa1, 0x9a, 0x2f, 0x0a, 0x6f,
0x42, 0x43, 0xd2, 0x70, 0x48, 0x80, 0xb9, 0xb2, 0x6b, 0x26, 0x08, 0x50, 0x8f, 0x04, 0xbc, 0x85,
0x65, 0xfb, 0xf9, 0xea, 0x02, 0x5f, 0x5d, 0x70, 0x68, 0xc4, 0x97, 0x2e, 0x00, 0x3c, 0xb6, 0x3d,
0xd7, 0xe1, 0x4e, 0xc2, 0xd5, 0x54, 0x33, 0xeb, 0x1c, 0xc2, 0x54, 0x60, 0xfc, 0x42, 0x83, 0xa5,
0x4f, 0xed, 0xc0, 0x21, 0x7b, 0x7b, 0xa7, 0xcf, 0xaf, 0x1b, 0x10, 0x37, 0xa8, 0xfd, 0xe3, 0x34,
0x6d, 0x99, 0x4d, 0xc6, 0x97, 0x25, 0x40, 0x29, 0x7b, 0x9d, 0x5c, 0x9a, 0x4b, 0xd0, 0xce, 0x68,
0x3e, 0x79, 0xbf, 0x48, 0xab, 0x9e, 0xb2, 0xba, 0xb7, 0x2b, 0x58, 0x59, 0x21, 0xb6, 0x29, 0x09,
0xb8, 0x19, 0x67, 0xae, 0x7b, 0xbb, 0xb1, 0x98, 0x6c, 0x2b, 0xb3, 0xd4, 0xc4, 0x90, 0xf1, 0xd8,
0x0b, 0x12, 0x4b, 0x52, 0xd6, 0x54, 0xe4, 0x3b, 0xb6, 0xb8, 0x6e, 0xe8, 0x34, 0xdb, 0xac, 0x51,
0xe3, 0x5f, 0x1a, 0x9c, 0x91, 0xbf, 0x2c, 0x7e, 0x87, 0x38, 0x2e, 0x10, 0x24, 0xf0, 0xdc, 0x20,
0xf1, 0x28, 0x99, 0x91, 0x04, 0x50, 0xba, 0xcc, 0xa7, 0xd0, 0x91, 0x48, 0x49, 0x86, 0x9d, 0xd1,
0x1a, 0x6d, 0xb1, 0x2f, 0xc9, 0xad, 0x97, 0xa0, 0x4d, 0xf6, 0xf6, 0xd2, 0xfc, 0x84, 0x9b, 0xb7,
0x24, 0x54, 0x32, 0xfc, 0x0c, 0xf4, 0x18, 0xed, 0xb8, 0x39, 0xbd, 0x23, 0x37, 0x26, 0x5d, 0xe0,
0xcf, 0x34, 0xe8, 0x66, 0x33, 0x7c, 0xea, 0xf8, 0xc7, 0x77, 0x84, 0xef, 0x65, 0x87, 0x08, 0x97,
0x0e, 0x91, 0x67, 0xc2, 0x47, 0xde, 0xaa, 0x56, 0x9f, 0x41, 0x3b, 0x9b, 0x8a, 0x51, 0x13, 0x6a,
0x5b, 0x24, 0xfa, 0xe4, 0xa9, 0x4b, 0x23, 0x7d, 0x0e, 0xb5, 0x01, 0xb6, 0x48, 0xb4, 0x1d, 0x62,
0x8a, 0x83, 0x48, 0xd7, 0x10, 0xc0, 0xfc, 0xdd, 0xa0, 0xe7, 0xd2, 0x87, 0x7a, 0x09, 0x9d, 0x95,
0x03, 0x4e, 0xdb, 0xeb, 0xcb, 0xbc, 0xa4, 0x97, 0xd9, 0xf6, 0xe4, 0xaf, 0x82, 0x74, 0x68, 0x26,
0x28, 0x9b, 0xdb, 0x9f, 0xeb, 0x55, 0x54, 0x87, 0xaa, 0xf8, 0x9c, 0x5f, 0xbd, 0x0b, 0x7a, 0xde,
0xe1, 0x50, 0x03, 0x16, 0xf6, 0x45, 0xbc, 0xea, 0x73, 0xa8, 0x03, 0x0d, 0x6f, 0x12, 0x2a, 0xba,
0xc6, 0x00, 0xc3, 0x70, 0x34, 0x90, 0x41, 0xa3, 0x97, 0x18, 0x37, 0x66, 0xb5, 0x1e, 0x79, 0x12,
0xe8, 0xe5, 0xd5, 0xcf, 0xa0, 0x99, 0x9e, 0xda, 0xa0, 0x1a, 0x54, 0xb6, 0x48, 0x80, 0xf5, 0x39,
0x46, 0x76, 0x33, 0x24, 0x4f, 0xdc, 0x60, 0x28, 0xce, 0x70, 0x2b, 0x24, 0xcf, 0x70, 0xa0, 0x97,
0xd8, 0x02, 0xf3, 0x4b, 0xb6, 0x50, 0x66, 0x0b, 0xc2, 0x49, 0xf5, 0xca, 0xea, 0xfb, 0x50, 0x8b,
0x4b, 0x02, 0x3a, 0x03, 0xad, 0xcc, 0xf3, 0x88, 0x3e, 0x87, 0x90, 0xb8, 0x4e, 0x4e, 0x92, 0xbf,
0xae, 0xad, 0xff, 0x03, 0x00, 0xc4, 0xad, 0x84, 0x90, 0xd0, 0x41, 0x23, 0x40, 0x9b, 0x38, 0xda,
0x20, 0xfe, 0x88, 0x04, 0xb1, 0x48, 0x14, 0x5d, 0x9f, 0x52, 0xb4, 0x8b, 0xa8, 0xf2, 0x94, 0xcb,
0x97, 0xa7, 0xec, 0xc8, 0xa1, 0x1b, 0x73, 0xc8, 0xe7, 0x1c, 0x59, 0xc7, 0x72, 0xcf, 0x1d, 0x3c,
0x8c, 0x67, 0xeb, 0x87, 0x70, 0xcc, 0xa1, 0xc6, 0x1c, 0x73, 0x15, 0x5b, 0xfe, 0xec, 0x44, 0xa1,
0x1b, 0x0c, 0xe3, 0x81, 0x97, 0x31, 0x87, 0x1e, 0xc1, 0xb9, 0x4d, 0xcc, 0xb9, 0xbb, 0x34, 0x72,
0x07, 0x34, 0x66, 0xb8, 0x3e, 0x9d, 0x61, 0x01, 0xf9, 0x98, 0x2c, 0x3d, 0xe8, 0xe4, 0xde, 0x80,
0xd1, 0xaa, 0xd2, 0xdf, 0x95, 0xef, 0xd5, 0xcb, 0xef, 0xce, 0x84, 0x9b, 0x70, 0x73, 0xa1, 0x9d,
0x7d, 0x1f, 0x45, 0xef, 0x4c, 0x23, 0x50, 0x78, 0x50, 0x5a, 0x5e, 0x9d, 0x05, 0x35, 0x61, 0x75,
0x1f, 0xda, 0xd9, 0x17, 0x38, 0x35, 0x2b, 0xe5, 0x2b, 0xdd, 0xf2, 0x61, 0xb3, 0x46, 0x63, 0x0e,
0xfd, 0x04, 0xce, 0x14, 0x9e, 0xbd, 0xd0, 0x37, 0x55, 0xe4, 0xa7, 0xbd, 0x8e, 0x1d, 0xc5, 0x41,
0x4a, 0x3f, 0xd1, 0xe2, 0x74, 0xe9, 0x0b, 0xef, 0x9f, 0xb3, 0x4b, 0x9f, 0x22, 0x7f, 0x98, 0xf4,
0xc7, 0xe6, 0x30, 0x06, 0x54, 0x7c, 0xf8, 0x42, 0xef, 0xa9, 0x58, 0x4c, 0x7d, 0x7c, 0x5b, 0x5e,
0x9b, 0x15, 0x3d, 0x31, 0xf9, 0x98, 0x47, 0x6b, 0xfe, 0x89, 0x48, 0xc9, 0x76, 0xea, 0x9b, 0x97,
0x9a, 0xed, 0xf4, 0x57, 0x1d, 0xe1, 0xd4, 0xd9, 0x11, 0xb6, 0xda, 0x56, 0xca, 0x97, 0x16, 0xb5,
0x53, 0xab, 0x27, 0xe2, 0xc6, 0x1c, 0xb2, 0x00, 0x36, 0x71, 0x74, 0x07, 0x47, 0xa1, 0x3b, 0xa0,
0xe8, 0xb2, 0x32, 0xc4, 0x27, 0x08, 0x31, 0x8f, 0x2b, 0x47, 0xe2, 0xc5, 0x0c, 0xd6, 0xbf, 0x02,
0xa8, 0x73, 0xed, 0xb2, 0x2a, 0xfd, 0xff, 0x84, 0xfb, 0x02, 0x12, 0xee, 0x03, 0xe8, 0xe4, 0x5e,
0x1a, 0xd4, 0x09, 0x57, 0xfd, 0x1c, 0x71, 0x54, 0xe4, 0xed, 0x02, 0x2a, 0x8e, 0xc3, 0xd5, 0x21,
0x30, 0x75, 0x6c, 0x7e, 0x14, 0x8f, 0x07, 0xd0, 0xc9, 0x8d, 0xa3, 0xd5, 0x27, 0x50, 0xcf, 0xac,
0x67, 0x38, 0x41, 0x71, 0x0c, 0xab, 0x3e, 0xc1, 0xd4, 0x71, 0xed, 0x51, 0x3c, 0xbe, 0x80, 0x66,
0x7a, 0x38, 0x87, 0xae, 0x4c, 0xcb, 0xad, 0xb9, 0x16, 0xe7, 0xe5, 0x67, 0xd6, 0x17, 0x5f, 0x79,
0x1e, 0x40, 0x27, 0x37, 0x3f, 0x53, 0x5b, 0x57, 0x3d, 0x64, 0x3b, 0x8a, 0xfa, 0xd7, 0x28, 0x57,
0xde, 0xfc, 0xe0, 0xfe, 0xfa, 0xd0, 0x8d, 0xf6, 0xc7, 0xbb, 0xec, 0x94, 0xd7, 0x04, 0xe6, 0x7b,
0x2e, 0x91, 0x5f, 0xd7, 0xe2, 0xa4, 0x71, 0x8d, 0x53, 0xba, 0xc6, 0xa5, 0x1d, 0xed, 0xee, 0xce,
0xf3, 0xdf, 0x1b, 0xff, 0x09, 0x00, 0x00, 0xff, 0xff, 0xa6, 0x2a, 0xbe, 0x5a, 0x78, 0x28, 0x00,
0x00,
}
// Reference imports to suppress errors if they are not otherwise used.

14
internal/querycoord/meta.go
View File

@ -64,6 +64,7 @@ type Meta interface {
setSegmentInfos(segmentInfos map[UniqueID]*querypb.SegmentInfo) error
showSegmentInfos(collectionID UniqueID, partitionIDs []UniqueID) []*querypb.SegmentInfo
getSegmentInfoByID(segmentID UniqueID) (*querypb.SegmentInfo, error)
getSegmentInfosByNode(nodeID int64) []*querypb.SegmentInfo
getPartitionStatesByID(collectionID UniqueID, partitionID UniqueID) (*querypb.PartitionStates, error)
@ -713,6 +714,19 @@ func (m *MetaReplica) getSegmentInfoByID(segmentID UniqueID) (*querypb.SegmentIn
return nil, errors.New("getSegmentInfoByID: can't find segmentID in segmentInfos")
}
func (m *MetaReplica) getSegmentInfosByNode(nodeID int64) []*querypb.SegmentInfo {
m.segmentMu.RLock()
defer m.segmentMu.RUnlock()
segmentInfos := make([]*querypb.SegmentInfo, 0)
for _, info := range m.segmentInfos {
if info.NodeID == nodeID {
segmentInfos = append(segmentInfos, proto.Clone(info).(*querypb.SegmentInfo))
}
}
return segmentInfos
}
func (m *MetaReplica) getCollectionInfoByID(collectionID UniqueID) (*querypb.CollectionInfo, error) {
m.collectionMu.RLock()

5
internal/querycoord/meta_test.go
View File

@ -240,6 +240,11 @@ func TestMetaFunc(t *testing.T) {
assert.Equal(t, defaultSegmentID, infos[0].SegmentID)
})
t.Run("Test GetSegmentInfoByNode", func(t *testing.T) {
infos := meta.getSegmentInfosByNode(nodeID)
assert.Equal(t, 1, len(infos))
})
t.Run("Test getQueryChannel", func(t *testing.T) {
info, err := meta.getQueryChannelInfoByID(defaultCollectionID)
assert.NotNil(t, info.QueryChannelID)

174
internal/querycoord/task.go
View File

@ -453,7 +453,7 @@ func (lct *loadCollectionTask) execute(ctx context.Context) error {
}
err = assignInternalTask(ctx, collectionID, lct, lct.meta, lct.cluster, loadSegmentReqs, watchDmChannelReqs, watchDeltaChannelReqs, false)
err = assignInternalTask(ctx, collectionID, lct, lct.meta, lct.cluster, loadSegmentReqs, watchDmChannelReqs, watchDeltaChannelReqs, false, nil)
if err != nil {
log.Warn("loadCollectionTask: assign child task failed", zap.Int64("collectionID", collectionID))
lct.setResultInfo(err)
@ -778,7 +778,7 @@ func (lpt *loadPartitionTask) execute(ctx context.Context) error {
}
}
err := assignInternalTask(ctx, collectionID, lpt, lpt.meta, lpt.cluster, loadSegmentReqs, watchDmReqs, watchDeltaReqs, false)
err := assignInternalTask(ctx, collectionID, lpt, lpt.meta, lpt.cluster, loadSegmentReqs, watchDmReqs, watchDeltaReqs, false, nil)
if err != nil {
log.Warn("loadPartitionTask: assign child task failed", zap.Int64("collectionID", collectionID), zap.Int64s("partitionIDs", partitionIDs))
lpt.setResultInfo(err)
@ -1627,7 +1627,7 @@ func (ht *handoffTask) execute(ctx context.Context) error {
ht.setResultInfo(err)
return err
}
err = assignInternalTask(ctx, collectionID, ht, ht.meta, ht.cluster, []*querypb.LoadSegmentsRequest{loadSegmentReq}, nil, watchDeltaChannelReqs, true)
err = assignInternalTask(ctx, collectionID, ht, ht.meta, ht.cluster, []*querypb.LoadSegmentsRequest{loadSegmentReq}, nil, watchDeltaChannelReqs, true, nil)
if err != nil {
log.Error("handoffTask: assign child task failed", zap.Any("segmentInfo", segmentInfo))
ht.setResultInfo(err)
@ -1699,6 +1699,7 @@ func (lbt *loadBalanceTask) timestamp() Timestamp {
func (lbt *loadBalanceTask) preExecute(context.Context) error {
lbt.setResultInfo(nil)
log.Debug("start do loadBalanceTask",
zap.Int32("trigger type", int32(lbt.triggerCondition)),
zap.Int64s("sourceNodeIDs", lbt.SourceNodeIDs),
zap.Any("balanceReason", lbt.BalanceReason),
zap.Int64("taskID", lbt.getTaskID()))
@ -1838,7 +1839,7 @@ func (lbt *loadBalanceTask) execute(ctx context.Context) error {
}
}
}
err = assignInternalTask(ctx, collectionID, lbt, lbt.meta, lbt.cluster, loadSegmentReqs, watchDmChannelReqs, watchDeltaChannelReqs, true)
err = assignInternalTask(ctx, collectionID, lbt, lbt.meta, lbt.cluster, loadSegmentReqs, watchDmChannelReqs, watchDeltaChannelReqs, true, lbt.SourceNodeIDs)
if err != nil {
log.Warn("loadBalanceTask: assign child task failed", zap.Int64("collectionID", collectionID), zap.Int64s("partitionIDs", partitionIDs))
lbt.setResultInfo(err)
@ -1849,12 +1850,154 @@ func (lbt *loadBalanceTask) execute(ctx context.Context) error {
}
}
//TODO::
//if lbt.triggerCondition == querypb.TriggerCondition_loadBalance {
// return nil
//}
//TODO:: use request.DstNodeIDs to balance
if lbt.triggerCondition == querypb.TriggerCondition_loadBalance {
if len(lbt.SourceNodeIDs) == 0 {
err := errors.New("loadBalanceTask: empty source Node list to balance")
log.Error(err.Error())
lbt.setResultInfo(err)
return err
}
balancedSegmentInfos := make(map[UniqueID]*querypb.SegmentInfo)
balancedSegmentIDs := make([]UniqueID, 0)
for _, nodeID := range lbt.SourceNodeIDs {
nodeExist := lbt.cluster.hasNode(nodeID)
if !nodeExist {
err := fmt.Errorf("loadBalanceTask: query node %d is not exist to balance", nodeID)
log.Error(err.Error())
lbt.setResultInfo(err)
return err
}
segmentInfos := lbt.meta.getSegmentInfosByNode(nodeID)
for _, info := range segmentInfos {
balancedSegmentInfos[info.SegmentID] = info
balancedSegmentIDs = append(balancedSegmentIDs, info.SegmentID)
}
}
// check balanced sealedSegmentIDs in request whether exist in query node
for _, segmentID := range lbt.SealedSegmentIDs {
if _, ok := balancedSegmentInfos[segmentID]; !ok {
err := fmt.Errorf("loadBalanceTask: unloaded segment %d", segmentID)
log.Warn(err.Error())
lbt.setResultInfo(err)
return err
}
}
if len(lbt.SealedSegmentIDs) != 0 {
balancedSegmentIDs = lbt.SealedSegmentIDs
}
col2PartitionIDs := make(map[UniqueID][]UniqueID)
par2Segments := make(map[UniqueID][]*querypb.SegmentInfo)
for _, segmentID := range balancedSegmentIDs {
info := balancedSegmentInfos[segmentID]
collectionID := info.CollectionID
partitionID := info.PartitionID
if _, ok := col2PartitionIDs[collectionID]; !ok {
col2PartitionIDs[collectionID] = make([]UniqueID, 0)
}
if _, ok := par2Segments[partitionID]; !ok {
col2PartitionIDs[collectionID] = append(col2PartitionIDs[collectionID], partitionID)
par2Segments[partitionID] = make([]*querypb.SegmentInfo, 0)
}
par2Segments[partitionID] = append(par2Segments[partitionID], info)
}
for collectionID, partitionIDs := range col2PartitionIDs {
segmentsToLoad := make([]UniqueID, 0)
loadSegmentReqs := make([]*querypb.LoadSegmentsRequest, 0)
watchDeltaChannelReqs := make([]*querypb.WatchDeltaChannelsRequest, 0)
collectionInfo, err := lbt.meta.getCollectionInfoByID(collectionID)
if err != nil {
log.Error("loadBalanceTask: can't find collectionID in meta", zap.Int64("collectionID", collectionID), zap.Error(err))
lbt.setResultInfo(err)
return err
}
for _, partitionID := range partitionIDs {
getRecoveryInfoRequest := &datapb.GetRecoveryInfoRequest{
Base: lbt.Base,
CollectionID: collectionID,
PartitionID: partitionID,
}
recoveryInfo, err := lbt.dataCoord.GetRecoveryInfo(ctx, getRecoveryInfoRequest)
if err != nil {
lbt.setResultInfo(err)
return err
}
segmentID2Binlog := make(map[UniqueID]*datapb.SegmentBinlogs)
for _, binlog := range recoveryInfo.Binlogs {
segmentID2Binlog[binlog.SegmentID] = binlog
}
for _, segmentInfo := range par2Segments[partitionID] {
segmentID := segmentInfo.SegmentID
if _, ok := segmentID2Binlog[segmentID]; !ok {
log.Warn("loadBalanceTask: can't find binlog of segment to balance, may be has been compacted", zap.Int64("segmentID", segmentID))
continue
}
segmentBingLog := segmentID2Binlog[segmentID]
segmentLoadInfo := &querypb.SegmentLoadInfo{
SegmentID: segmentID,
PartitionID: partitionID,
CollectionID: collectionID,
BinlogPaths: segmentBingLog.FieldBinlogs,
NumOfRows: segmentBingLog.NumOfRows,
Statslogs: segmentBingLog.Statslogs,
Deltalogs: segmentBingLog.Deltalogs,
}
msgBase := proto.Clone(lbt.Base).(*commonpb.MsgBase)
msgBase.MsgType = commonpb.MsgType_LoadSegments
loadSegmentReq := &querypb.LoadSegmentsRequest{
Base: msgBase,
Infos: []*querypb.SegmentLoadInfo{segmentLoadInfo},
Schema: collectionInfo.Schema,
LoadCondition: querypb.TriggerCondition_grpcRequest,
}
segmentsToLoad = append(segmentsToLoad, segmentID)
loadSegmentReqs = append(loadSegmentReqs, loadSegmentReq)
}
// init delta channels for sealed segments.
if len(loadSegmentReqs) != 0 && len(watchDeltaChannelReqs) != len(recoveryInfo.Channels) {
for _, info := range recoveryInfo.Channels {
deltaChannel, err := rootcoord.ConvertChannelName(info.ChannelName, Params.DmlChannelPrefix, Params.DeltaChannelPrefix)
if err != nil {
return err
}
deltaInfo := proto.Clone(info).(*datapb.VchannelInfo)
deltaInfo.ChannelName = deltaChannel
msgBase := proto.Clone(lbt.Base).(*commonpb.MsgBase)
msgBase.MsgType = commonpb.MsgType_WatchDeltaChannels
watchDeltaRequest := &querypb.WatchDeltaChannelsRequest{
Base: msgBase,
CollectionID: collectionID,
Infos: []*datapb.VchannelInfo{deltaInfo},
}
watchDeltaChannelReqs = append(watchDeltaChannelReqs, watchDeltaRequest)
}
}
}
// TODO:: assignInternalTask with multi collection
err = assignInternalTask(ctx, collectionID, lbt, lbt.meta, lbt.cluster, loadSegmentReqs, nil, watchDeltaChannelReqs, false, lbt.SourceNodeIDs)
if err != nil {
log.Warn("loadBalanceTask: assign child task failed", zap.Int64("collectionID", collectionID))
lbt.setResultInfo(err)
return err
}
}
log.Debug("loadBalanceTask: assign child task done", zap.Any("balance request", lbt.LoadBalanceRequest))
}
log.Debug("loadBalanceTask Execute done",
zap.Int32("trigger type", int32(lbt.triggerCondition)),
zap.Int64s("sourceNodeIDs", lbt.SourceNodeIDs),
zap.Any("balanceReason", lbt.BalanceReason),
zap.Int64("taskID", lbt.getTaskID()))
@ -1862,18 +2005,21 @@ func (lbt *loadBalanceTask) execute(ctx context.Context) error {
}
func (lbt *loadBalanceTask) postExecute(context.Context) error {
if lbt.result.ErrorCode == commonpb.ErrorCode_Success {
if lbt.result.ErrorCode != commonpb.ErrorCode_Success {
lbt.childTasks = []task{}
}
if lbt.triggerCondition == querypb.TriggerCondition_nodeDown {
for _, id := range lbt.SourceNodeIDs {
err := lbt.cluster.removeNodeInfo(id)
if err != nil {
//TODO:: clear node info after removeNodeInfo failed
log.Error("loadBalanceTask: occur error when removing node info from cluster", zap.Int64("nodeID", id))
}
}
} else {
lbt.childTasks = []task{}
}
log.Debug("loadBalanceTask postExecute done",
zap.Int32("trigger type", int32(lbt.triggerCondition)),
zap.Int64s("sourceNodeIDs", lbt.SourceNodeIDs),
zap.Any("balanceReason", lbt.BalanceReason),
zap.Int64("taskID", lbt.getTaskID()))
@ -2057,7 +2203,7 @@ func assignInternalTask(ctx context.Context,
loadSegmentRequests []*querypb.LoadSegmentsRequest,
watchDmChannelRequests []*querypb.WatchDmChannelsRequest,
watchDeltaChannelRequests []*querypb.WatchDeltaChannelsRequest,
wait bool) error {
wait bool, excludeNodeIDs []int64) error {
sp, _ := trace.StartSpanFromContext(ctx)
defer sp.Finish()
segmentsToLoad := make([]UniqueID, 0)
@ -2068,13 +2214,13 @@ func assignInternalTask(ctx context.Context,
for _, req := range watchDmChannelRequests {
channelsToWatch = append(channelsToWatch, req.Infos[0].ChannelName)
}
segment2Nodes, err := shuffleSegmentsToQueryNode(segmentsToLoad, cluster, wait, nil)
segment2Nodes, err := shuffleSegmentsToQueryNode(segmentsToLoad, cluster, wait, excludeNodeIDs)
if err != nil {
log.Error("assignInternalTask: segment to node failed", zap.Any("segments map", segment2Nodes), zap.Int64("collectionID", collectionID))
return err
}
log.Debug("assignInternalTask: segment to node", zap.Any("segments map", segment2Nodes), zap.Int64("collectionID", collectionID))
watchRequest2Nodes, err := shuffleChannelsToQueryNode(channelsToWatch, cluster, wait, nil)
watchRequest2Nodes, err := shuffleChannelsToQueryNode(channelsToWatch, cluster, wait, excludeNodeIDs)
if err != nil {
log.Error("assignInternalTask: watch request to node failed", zap.Any("request map", watchRequest2Nodes), zap.Int64("collectionID", collectionID))
return err

138
internal/querycoord/task_test.go
View File

@ -694,7 +694,7 @@ func Test_AssignInternalTask(t *testing.T) {
loadSegmentRequests = append(loadSegmentRequests, req)
}
err = assignInternalTask(queryCoord.loopCtx, defaultCollectionID, loadCollectionTask, queryCoord.meta, queryCoord.cluster, loadSegmentRequests, nil, nil, false)
err = assignInternalTask(queryCoord.loopCtx, defaultCollectionID, loadCollectionTask, queryCoord.meta, queryCoord.cluster, loadSegmentRequests, nil, nil, false, nil)
assert.Nil(t, err)
assert.NotEqual(t, 1, len(loadCollectionTask.getChildTask()))
@ -789,3 +789,139 @@ func Test_handoffSegmentFail(t *testing.T) {
err = removeAllSession()
assert.Nil(t, err)
}
func TestLoadBalanceSegmentsTask(t *testing.T) {
refreshParams()
ctx := context.Background()
queryCoord, err := startQueryCoord(ctx)
assert.Nil(t, err)
node1, err := startQueryNodeServer(ctx)
assert.Nil(t, err)
waitQueryNodeOnline(queryCoord.cluster, node1.queryNodeID)
t.Run("Test LoadCollection", func(t *testing.T) {
loadCollectionTask := genLoadCollectionTask(ctx, queryCoord)
err = queryCoord.scheduler.Enqueue(loadCollectionTask)
assert.Nil(t, err)
waitTaskFinalState(loadCollectionTask, taskExpired)
})
node2, err := startQueryNodeServer(ctx)
assert.Nil(t, err)
waitQueryNodeOnline(queryCoord.cluster, node2.queryNodeID)
t.Run("Test LoadBalanceBySegmentID", func(t *testing.T) {
baseTask := newBaseTask(ctx, querypb.TriggerCondition_loadBalance)
loadBalanceTask := &loadBalanceTask{
baseTask: baseTask,
LoadBalanceRequest: &querypb.LoadBalanceRequest{
Base: &commonpb.MsgBase{
MsgType: commonpb.MsgType_LoadBalanceSegments,
},
SourceNodeIDs: []int64{node1.queryNodeID},
SealedSegmentIDs: []UniqueID{defaultSegmentID},
},
rootCoord: queryCoord.rootCoordClient,
dataCoord: queryCoord.dataCoordClient,
cluster: queryCoord.cluster,
meta: queryCoord.meta,
}
err = queryCoord.scheduler.Enqueue(loadBalanceTask)
assert.Nil(t, err)
waitTaskFinalState(loadBalanceTask, taskExpired)
})
t.Run("Test LoadBalanceByNotExistSegmentID", func(t *testing.T) {
baseTask := newBaseTask(ctx, querypb.TriggerCondition_loadBalance)
loadBalanceTask := &loadBalanceTask{
baseTask: baseTask,
LoadBalanceRequest: &querypb.LoadBalanceRequest{
Base: &commonpb.MsgBase{
MsgType: commonpb.MsgType_LoadBalanceSegments,
},
SourceNodeIDs: []int64{node1.queryNodeID},
SealedSegmentIDs: []UniqueID{defaultSegmentID + 100},
},
rootCoord: queryCoord.rootCoordClient,
dataCoord: queryCoord.dataCoordClient,
cluster: queryCoord.cluster,
meta: queryCoord.meta,
}
err = queryCoord.scheduler.Enqueue(loadBalanceTask)
assert.Nil(t, err)
waitTaskFinalState(loadBalanceTask, taskFailed)
})
t.Run("Test LoadBalanceByNode", func(t *testing.T) {
baseTask := newBaseTask(ctx, querypb.TriggerCondition_loadBalance)
loadBalanceTask := &loadBalanceTask{
baseTask: baseTask,
LoadBalanceRequest: &querypb.LoadBalanceRequest{
Base: &commonpb.MsgBase{
MsgType: commonpb.MsgType_LoadBalanceSegments,
},
SourceNodeIDs: []int64{node1.queryNodeID},
},
rootCoord: queryCoord.rootCoordClient,
dataCoord: queryCoord.dataCoordClient,
cluster: queryCoord.cluster,
meta: queryCoord.meta,
}
err = queryCoord.scheduler.Enqueue(loadBalanceTask)
assert.Nil(t, err)
waitTaskFinalState(loadBalanceTask, taskExpired)
})
t.Run("Test LoadBalanceWithEmptySourceNode", func(t *testing.T) {
baseTask := newBaseTask(ctx, querypb.TriggerCondition_loadBalance)
loadBalanceTask := &loadBalanceTask{
baseTask: baseTask,
LoadBalanceRequest: &querypb.LoadBalanceRequest{
Base: &commonpb.MsgBase{
MsgType: commonpb.MsgType_LoadBalanceSegments,
},
},
rootCoord: queryCoord.rootCoordClient,
dataCoord: queryCoord.dataCoordClient,
cluster: queryCoord.cluster,
meta: queryCoord.meta,
}
err = queryCoord.scheduler.Enqueue(loadBalanceTask)
assert.Nil(t, err)
waitTaskFinalState(loadBalanceTask, taskFailed)
})
t.Run("Test LoadBalanceByNotExistNode", func(t *testing.T) {
baseTask := newBaseTask(ctx, querypb.TriggerCondition_loadBalance)
loadBalanceTask := &loadBalanceTask{
baseTask: baseTask,
LoadBalanceRequest: &querypb.LoadBalanceRequest{
Base: &commonpb.MsgBase{
MsgType: commonpb.MsgType_LoadBalanceSegments,
},
SourceNodeIDs: []int64{node1.queryNodeID + 100},
},
rootCoord: queryCoord.rootCoordClient,
dataCoord: queryCoord.dataCoordClient,
cluster: queryCoord.cluster,
meta: queryCoord.meta,
}
err = queryCoord.scheduler.Enqueue(loadBalanceTask)
assert.Nil(t, err)
waitTaskFinalState(loadBalanceTask, taskFailed)
})
t.Run("Test ReleaseCollection", func(t *testing.T) {
releaseCollectionTask := genReleaseCollectionTask(ctx, queryCoord)
err = queryCoord.scheduler.processTask(releaseCollectionTask)
assert.Nil(t, err)
})
node1.stop()
node2.stop()
queryCoord.Stop()
err = removeAllSession()
assert.Nil(t, err)
}

1
internal/querynode/impl.go
View File

@ -515,6 +515,7 @@ func (node *QueryNode) GetSegmentInfo(ctx context.Context, in *queryPb.GetSegmen
}
infos = append(infos, streamingSegmentInfos...)
log.Debug("GetSegmentInfo: get segment info from query node", zap.Int64("nodeID", node.session.ServerID), zap.Any("segment infos", infos))
return &queryPb.GetSegmentInfoResponse{
Status: &commonpb.Status{
ErrorCode: commonpb.ErrorCode_Success,