diff --git a/internal/proto/query_coord.proto b/internal/proto/query_coord.proto index decddc1fdc..324d1f5260 100644 --- a/internal/proto/query_coord.proto +++ b/internal/proto/query_coord.proto @@ -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 ----------------- diff --git a/internal/proto/querypb/query_coord.pb.go b/internal/proto/querypb/query_coord.pb.go index 73106fd21f..bf7fca0d62 100644 --- a/internal/proto/querypb/query_coord.pb.go +++ b/internal/proto/querypb/query_coord.pb.go @@ -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. diff --git a/internal/querycoord/meta.go b/internal/querycoord/meta.go index 7539ec2cd8..dca2a827e8 100644 --- a/internal/querycoord/meta.go +++ b/internal/querycoord/meta.go @@ -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() diff --git a/internal/querycoord/meta_test.go b/internal/querycoord/meta_test.go index b260db4deb..85ad7cba2a 100644 --- a/internal/querycoord/meta_test.go +++ b/internal/querycoord/meta_test.go @@ -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) diff --git a/internal/querycoord/task.go b/internal/querycoord/task.go index 01354b497a..d88b1e3fc2 100644 --- a/internal/querycoord/task.go +++ b/internal/querycoord/task.go @@ -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 diff --git a/internal/querycoord/task_test.go b/internal/querycoord/task_test.go index 4cdfd0c8a0..0c90ad42ec 100644 --- a/internal/querycoord/task_test.go +++ b/internal/querycoord/task_test.go @@ -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) +} diff --git a/internal/querynode/impl.go b/internal/querynode/impl.go index 071a513118..ce4ddb4c75 100644 --- a/internal/querynode/impl.go +++ b/internal/querynode/impl.go @@ -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,