diff --git a/Makefile b/Makefile index bf36806194d19050d4190fdffb82f9e26d581831..45c2564b2d784ba126ddbe31aabed116f16a64de 100644 --- a/Makefile +++ b/Makefile @@ -80,13 +80,14 @@ verifiers: getdeps cppcheck fmt static-check ruleguard master: build-cpp @echo "Building each component's binary to './bin'" - @echo "Building master ..." - @mkdir -p $(INSTALL_PATH) && go env -w CGO_ENABLED="0" && GO111MODULE=on $(GO) build -o $(INSTALL_PATH)/master $(PWD)/cmd/master/main.go 1>/dev/null + @echo "Building masterservice ..." + @mkdir -p $(INSTALL_PATH) && go env -w CGO_ENABLED="1" && GO111MODULE=on $(GO) build -o $(INSTALL_PATH)/masterservice $(PWD)/cmd/masterservice/main.go 1>/dev/null + # Builds various components locally. proxynode: build-cpp @echo "Building each component's binary to './bin'" - @echo "Building proxy ..." + @echo "Building proxy node ..." @mkdir -p $(INSTALL_PATH) && go env -w CGO_ENABLED="0" && GO111MODULE=on $(GO) build -o $(INSTALL_PATH)/proxynode $(PWD)/cmd/proxy/node/proxy_node.go 1>/dev/null # Builds various components locally. @@ -95,6 +96,7 @@ querynode: build-cpp @echo "Building query node ..." @mkdir -p $(INSTALL_PATH) && go env -w CGO_ENABLED="1" && GO111MODULE=on $(GO) build -o $(INSTALL_PATH)/querynode $(PWD)/cmd/querynode/query_node.go 1>/dev/null + # Builds various components locally. writenode: build-cpp @echo "Building each component's binary to './bin'" @@ -110,8 +112,13 @@ datanode: build-cpp # Builds various components locally. indexnode: build-cpp @echo "Building each component's binary to './bin'" - @echo "Building indexbuilder ..." - @mkdir -p $(INSTALL_PATH) && go env -w CGO_ENABLED="1" && GO111MODULE=on $(GO) build -o $(INSTALL_PATH)/indexbuilder $(PWD)/cmd/indexbuilder/indexbuilder.go 1>/dev/null + @echo "Building distributed indexnode ..." + @mkdir -p $(INSTALL_PATH) && go env -w CGO_ENABLED="1" && GO111MODULE=on $(GO) build -o $(INSTALL_PATH)/distributed/indexnode $(PWD)/cmd/distributed/indexnode/main.go 1>/dev/null + +indexservice: build-cpp + @echo "Building each component's binary to './bin'" + @echo "Building distributed indexservice ..." + @mkdir -p $(INSTALL_PATH) && go env -w CGO_ENABLED="1" && GO111MODULE=on $(GO) build -o $(INSTALL_PATH)/distributed/indexservice $(PWD)/cmd/distributed/indexservice/main.go 1>/dev/null # Builds various components locally. diff --git a/internal/distributed/indexnode/client/client.go b/internal/distributed/indexnode/client/client.go index 9e70820123a428d77f85e091726d20b5676ec98f..0a69c840b29e7a6da891db9d2d5b9981d192438a 100644 --- a/internal/distributed/indexnode/client/client.go +++ b/internal/distributed/indexnode/client/client.go @@ -2,11 +2,13 @@ package grpcindexnodeclient import ( "context" + "errors" "log" "time" "github.com/zilliztech/milvus-distributed/internal/proto/commonpb" "github.com/zilliztech/milvus-distributed/internal/proto/indexpb" + "github.com/zilliztech/milvus-distributed/internal/proto/internalpb2" "google.golang.org/grpc" ) @@ -15,6 +17,34 @@ type Client struct { nodeAddress string } +func (c Client) GetComponentStates() (*internalpb2.ComponentStates, error) { + return c.grpcClient.GetComponentStates(context.Background(), &commonpb.Empty{}) +} + +func (c Client) GetTimeTickChannel() (string, error) { + resp, err := c.grpcClient.GetTimeTickChannel(context.Background(), &commonpb.Empty{}) + + if err != nil { + return "", err + } + if resp.Status.ErrorCode != commonpb.ErrorCode_SUCCESS { + return "", errors.New(resp.Status.Reason) + } + return resp.Value, nil +} + +func (c Client) GetStatisticsChannel() (string, error) { + resp, err := c.grpcClient.GetStatisticsChannel(context.Background(), &commonpb.Empty{}) + + if err != nil { + return "", err + } + if resp.Status.ErrorCode != commonpb.ErrorCode_SUCCESS { + return "", errors.New(resp.Status.Reason) + } + return resp.Value, nil +} + func (c Client) Init() error { return nil } diff --git a/internal/distributed/indexnode/service.go b/internal/distributed/indexnode/service.go index 8336eb24495b66970ce9b929ad028a79b3e27390..a7d4bfacda8cf4c1cd60ae4986776cf940b24cb6 100644 --- a/internal/distributed/indexnode/service.go +++ b/internal/distributed/indexnode/service.go @@ -11,6 +11,8 @@ import ( "github.com/zilliztech/milvus-distributed/internal/indexnode" "github.com/zilliztech/milvus-distributed/internal/proto/commonpb" "github.com/zilliztech/milvus-distributed/internal/proto/indexpb" + "github.com/zilliztech/milvus-distributed/internal/proto/internalpb2" + "github.com/zilliztech/milvus-distributed/internal/proto/milvuspb" "github.com/zilliztech/milvus-distributed/internal/util/typeutil" "google.golang.org/grpc" ) @@ -25,6 +27,42 @@ type Server struct { loopWg sync.WaitGroup } +func (s *Server) GetComponentStates(ctx context.Context, empty *commonpb.Empty) (*internalpb2.ComponentStates, error) { + return s.node.GetComponentStates() +} + +func (s *Server) GetTimeTickChannel(ctx context.Context, empty *commonpb.Empty) (*milvuspb.StringResponse, error) { + ret, err := s.node.GetTimeTickChannel() + resp := &milvuspb.StringResponse{ + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_SUCCESS, + }, + } + if err != nil { + resp.Status.ErrorCode = commonpb.ErrorCode_UNEXPECTED_ERROR + resp.Status.Reason = err.Error() + } else { + resp.Value = ret + } + return resp, nil +} + +func (s *Server) GetStatisticsChannel(ctx context.Context, empty *commonpb.Empty) (*milvuspb.StringResponse, error) { + ret, err := s.node.GetStatisticsChannel() + resp := &milvuspb.StringResponse{ + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_SUCCESS, + }, + } + if err != nil { + resp.Status.ErrorCode = commonpb.ErrorCode_UNEXPECTED_ERROR + resp.Status.Reason = err.Error() + } else { + resp.Value = ret + } + return resp, nil +} + func (s *Server) registerNode() error { log.Printf("Registering node. IP = %s, Port = %d", indexnode.Params.NodeIP, indexnode.Params.NodePort) diff --git a/internal/distributed/indexservice/service.go b/internal/distributed/indexservice/service.go index 3d2a80ead3a9cf6c00b36fe7810eafe3bf8feca8..4a903a4a67c1d3eef86f5d96137b59a41891d530 100644 --- a/internal/distributed/indexservice/service.go +++ b/internal/distributed/indexservice/service.go @@ -7,10 +7,12 @@ import ( "strconv" "sync" + "github.com/zilliztech/milvus-distributed/internal/proto/internalpb2" + "github.com/zilliztech/milvus-distributed/internal/proto/milvuspb" + "github.com/zilliztech/milvus-distributed/internal/indexservice" "github.com/zilliztech/milvus-distributed/internal/proto/commonpb" "github.com/zilliztech/milvus-distributed/internal/proto/indexpb" - "github.com/zilliztech/milvus-distributed/internal/proto/internalpb2" "github.com/zilliztech/milvus-distributed/internal/util/typeutil" "google.golang.org/grpc" ) @@ -28,6 +30,43 @@ type Server struct { loopWg sync.WaitGroup } +func (s *Server) GetComponentStates(ctx context.Context, empty *commonpb.Empty) (*internalpb2.ComponentStates, error) { + return s.server.GetComponentStates() +} + +func (s *Server) GetTimeTickChannel(ctx context.Context, empty *commonpb.Empty) (*milvuspb.StringResponse, error) { + resp := &milvuspb.StringResponse{ + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_SUCCESS, + }, + } + channel, err := s.server.GetTimeTickChannel() + if err != nil { + resp.Status.ErrorCode = commonpb.ErrorCode_UNEXPECTED_ERROR + resp.Status.Reason = err.Error() + return resp, nil + } + + resp.Value = channel + return resp, nil +} + +func (s *Server) GetStatisticsChannel(ctx context.Context, empty *commonpb.Empty) (*milvuspb.StringResponse, error) { + resp := &milvuspb.StringResponse{ + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_SUCCESS, + }, + } + channel, err := s.server.GetStatisticsChannel() + if err != nil { + resp.Status.ErrorCode = commonpb.ErrorCode_UNEXPECTED_ERROR + resp.Status.Reason = err.Error() + return resp, nil + } + resp.Value = channel + return resp, nil +} + func Init() error { indexservice.Params.Init() return nil @@ -48,18 +87,6 @@ func (s *Server) Stop() error { return nil } -func (s *Server) GetComponentStates() (*internalpb2.ComponentStates, error) { - panic("implement me") -} - -func (s *Server) GetTimeTickChannel() (string, error) { - panic("implement me") -} - -func (s *Server) GetStatisticsChannel() (string, error) { - panic("implement me") -} - func (s *Server) RegisterNode(ctx context.Context, req *indexpb.RegisterNodeRequest) (*indexpb.RegisterNodeResponse, error) { return s.server.RegisterNode(req) diff --git a/internal/indexnode/indexnode.go b/internal/indexnode/indexnode.go index 0588f58a2fff9c7bf2a73f0d51f613da22bad403..9b7007e60c8c2ce080f0b27a16bc86c33e19b19a 100644 --- a/internal/indexnode/indexnode.go +++ b/internal/indexnode/indexnode.go @@ -11,6 +11,7 @@ import ( miniokv "github.com/zilliztech/milvus-distributed/internal/kv/minio" "github.com/zilliztech/milvus-distributed/internal/proto/commonpb" "github.com/zilliztech/milvus-distributed/internal/proto/indexpb" + "github.com/zilliztech/milvus-distributed/internal/proto/internalpb2" "github.com/zilliztech/milvus-distributed/internal/util/retry" "github.com/zilliztech/milvus-distributed/internal/util/typeutil" ) @@ -23,6 +24,8 @@ type UniqueID = typeutil.UniqueID type Timestamp = typeutil.Timestamp type IndexNode struct { + state internalpb2.StateCode + loopCtx context.Context loopCancel func() @@ -162,3 +165,29 @@ func (i *IndexNode) BuildIndex(request *indexpb.BuildIndexCmd) (*commonpb.Status } return ret, nil } + +func (i *IndexNode) GetComponentStates() (*internalpb2.ComponentStates, error) { + + stateInfo := &internalpb2.ComponentInfo{ + NodeID: Params.NodeID, + Role: "IndexNode", + StateCode: i.state, + } + + ret := &internalpb2.ComponentStates{ + State: stateInfo, + SubcomponentStates: nil, // todo add subcomponents states + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_SUCCESS, + }, + } + return ret, nil +} + +func (i *IndexNode) GetTimeTickChannel() (string, error) { + return "", nil +} + +func (i *IndexNode) GetStatisticsChannel() (string, error) { + return "", nil +} diff --git a/internal/indexservice/indexservice.go b/internal/indexservice/indexservice.go index a0e6dd614a05fd67db95e7cf62232d1c135d6fcc..4f8eeb8f98f1ee2f00dd09272b7bafac93870b8b 100644 --- a/internal/indexservice/indexservice.go +++ b/internal/indexservice/indexservice.go @@ -26,8 +26,11 @@ const ( type IndexService struct { nodeClients *PriorityQueue + nodeStates map[UniqueID]*internalpb2.ComponentStates + state internalpb2.StateCode + + ID UniqueID - //factory method loopCtx context.Context loopCancel func() loopWg sync.WaitGroup @@ -85,6 +88,11 @@ func CreateIndexService(ctx context.Context) (*IndexService, error) { return nil, err } + i.ID, err = i.idAllocator.AllocOne() + if err != nil { + return nil, err + } + connectMinIOFn := func() error { option := &miniokv.Option{ Address: Params.MinIOAddress, @@ -139,7 +147,21 @@ func (i *IndexService) Stop() error { } func (i *IndexService) GetComponentStates() (*internalpb2.ComponentStates, error) { - panic("implement me") + + stateInfo := &internalpb2.ComponentInfo{ + NodeID: i.ID, + Role: "IndexService", + StateCode: i.state, + } + + ret := &internalpb2.ComponentStates{ + State: stateInfo, + SubcomponentStates: nil, // todo add subcomponents states + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_SUCCESS, + }, + } + return ret, nil } func (i *IndexService) GetTimeTickChannel() (string, error) { @@ -147,7 +169,7 @@ func (i *IndexService) GetTimeTickChannel() (string, error) { } func (i *IndexService) GetStatisticsChannel() (string, error) { - panic("implement me") + return "", nil } func (i *IndexService) BuildIndex(req *indexpb.BuildIndexRequest) (*indexpb.BuildIndexResponse, error) { diff --git a/internal/proto/index_service.proto b/internal/proto/index_service.proto index 1012a04554db054e73127526401e19a49dd9fa09..73de1857bccb55d0e2c7618c27243c41927b7db8 100644 --- a/internal/proto/index_service.proto +++ b/internal/proto/index_service.proto @@ -6,6 +6,7 @@ option go_package = "github.com/zilliztech/milvus-distributed/internal/proto/ind import "common.proto"; import "internal.proto"; +import "milvus.proto"; message RegisterNodeRequest { @@ -93,6 +94,10 @@ service IndexService { rpc GetIndexStates(IndexStatesRequest) returns (IndexStatesResponse) {} rpc GetIndexFilePaths(IndexFilePathsRequest) returns (IndexFilePathsResponse){} rpc NotifyBuildIndex(BuildIndexNotification) returns (common.Status) {} + + rpc GetComponentStates(common.Empty) returns (internal.ComponentStates) {} + rpc GetTimeTickChannel(common.Empty) returns(milvus.StringResponse) {} + rpc GetStatisticsChannel(common.Empty) returns(milvus.StringResponse){} } @@ -106,4 +111,7 @@ service IndexNode { */ rpc BuildIndex(BuildIndexCmd) returns (common.Status){} + rpc GetComponentStates(common.Empty) returns (internal.ComponentStates) {} + rpc GetTimeTickChannel(common.Empty) returns(milvus.StringResponse) {} + rpc GetStatisticsChannel(common.Empty) returns(milvus.StringResponse){} } diff --git a/internal/proto/indexpb/index_service.pb.go b/internal/proto/indexpb/index_service.pb.go index 58d0078325a0942de526f1d00834b429d1c7e229..40318a475de261cb97a88f70b742aa9eb8661962 100644 --- a/internal/proto/indexpb/index_service.pb.go +++ b/internal/proto/indexpb/index_service.pb.go @@ -9,6 +9,7 @@ import ( proto "github.com/golang/protobuf/proto" commonpb "github.com/zilliztech/milvus-distributed/internal/proto/commonpb" internalpb2 "github.com/zilliztech/milvus-distributed/internal/proto/internalpb2" + milvuspb "github.com/zilliztech/milvus-distributed/internal/proto/milvuspb" grpc "google.golang.org/grpc" codes "google.golang.org/grpc/codes" status "google.golang.org/grpc/status" @@ -704,55 +705,60 @@ func init() { func init() { proto.RegisterFile("index_service.proto", fileDescriptor_a5d2036b4df73e0a) } var fileDescriptor_a5d2036b4df73e0a = []byte{ - // 757 bytes of a gzipped FileDescriptorProto - 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xc4, 0x56, 0x5f, 0x4b, 0x1b, 0x4b, - 0x14, 0x77, 0xef, 0xc6, 0x78, 0x73, 0x92, 0x1b, 0x74, 0xbc, 0x37, 0x84, 0xdc, 0x8a, 0xba, 0xa0, - 0xa6, 0x42, 0x13, 0x89, 0xd8, 0x3e, 0x16, 0x63, 0x68, 0x09, 0x45, 0x91, 0x29, 0xf4, 0xa1, 0xa5, - 0x84, 0x49, 0x76, 0xa2, 0x03, 0x9b, 0xdd, 0xb8, 0x33, 0x2b, 0xd5, 0x97, 0x52, 0xe8, 0x63, 0xa1, - 0x0f, 0xfd, 0x14, 0xfd, 0x0e, 0xfd, 0x30, 0xfd, 0x28, 0x65, 0x67, 0x66, 0x37, 0xbb, 0xba, 0x26, - 0x29, 0x0a, 0x7d, 0xcb, 0x99, 0xfd, 0x9d, 0x3f, 0xf3, 0xfb, 0x9d, 0x73, 0x26, 0xb0, 0xca, 0x5c, - 0x9b, 0x7e, 0xe8, 0x71, 0xea, 0x5f, 0xb2, 0x01, 0x6d, 0x8c, 0x7d, 0x4f, 0x78, 0x08, 0x8d, 0x98, - 0x73, 0x19, 0x70, 0x65, 0x35, 0x24, 0xa2, 0x56, 0x1a, 0x78, 0xa3, 0x91, 0xe7, 0xaa, 0xb3, 0x5a, - 0x99, 0xb9, 0x82, 0xfa, 0x2e, 0x71, 0x94, 0x6d, 0x7d, 0x84, 0x55, 0x4c, 0xcf, 0x18, 0x17, 0xd4, - 0x3f, 0xf1, 0x6c, 0x8a, 0xe9, 0x45, 0x40, 0xb9, 0x40, 0x7b, 0x90, 0xeb, 0x13, 0x4e, 0xab, 0xc6, - 0x86, 0x51, 0x2f, 0xb6, 0x1e, 0x35, 0x52, 0x71, 0x75, 0xc0, 0x63, 0x7e, 0xd6, 0x26, 0x9c, 0x62, - 0x89, 0x44, 0x4f, 0x61, 0x89, 0xd8, 0xb6, 0x4f, 0x39, 0xaf, 0xfe, 0x35, 0xc5, 0xe9, 0x50, 0x61, - 0x70, 0x04, 0xb6, 0xbe, 0x1a, 0xf0, 0x6f, 0xba, 0x02, 0x3e, 0xf6, 0x5c, 0x4e, 0xd1, 0x3e, 0xe4, - 0xb9, 0x20, 0x22, 0xe0, 0xba, 0x88, 0xff, 0x33, 0xe3, 0xbd, 0x96, 0x10, 0xac, 0xa1, 0xa8, 0x0d, - 0x45, 0xe6, 0x32, 0xd1, 0x1b, 0x13, 0x9f, 0x8c, 0xa2, 0x4a, 0x36, 0x1b, 0x37, 0x68, 0xd1, 0x0c, - 0x74, 0x5d, 0x26, 0x4e, 0x25, 0x10, 0x03, 0x8b, 0x7f, 0x5b, 0x7b, 0x80, 0xba, 0x21, 0x73, 0x61, - 0x68, 0xca, 0x23, 0x46, 0x6a, 0xf0, 0xb7, 0xe4, 0xb3, 0xdb, 0x09, 0x0b, 0x32, 0xeb, 0x26, 0x8e, - 0x6d, 0x4b, 0x40, 0x41, 0x7a, 0x74, 0xdd, 0xa1, 0x87, 0x0e, 0x60, 0x31, 0x2c, 0x46, 0x71, 0x57, - 0x6e, 0xad, 0x67, 0x96, 0x3d, 0x49, 0x80, 0x15, 0x1a, 0x55, 0x61, 0x49, 0xc7, 0x93, 0x55, 0x9b, - 0x38, 0x32, 0x51, 0x05, 0xf2, 0x98, 0x12, 0xee, 0xb9, 0x55, 0x73, 0xc3, 0xa8, 0x17, 0xb0, 0xb6, - 0xac, 0x4f, 0x06, 0xac, 0xa6, 0x0a, 0xbd, 0x0f, 0x71, 0x07, 0xca, 0x89, 0x86, 0x9c, 0x99, 0xf5, - 0x62, 0x6b, 0xad, 0x71, 0xbb, 0x95, 0x1a, 0xf1, 0x25, 0xb1, 0x06, 0x5b, 0x3f, 0x0c, 0x58, 0x69, - 0x07, 0xcc, 0xb1, 0xe5, 0xa7, 0x88, 0xab, 0x35, 0x00, 0x9b, 0x08, 0xd2, 0x1b, 0x13, 0x71, 0xae, - 0xd8, 0x2a, 0xe0, 0x42, 0x78, 0x72, 0x1a, 0x1e, 0x84, 0x22, 0x89, 0xab, 0x31, 0x9d, 0x88, 0x64, - 0xde, 0x16, 0x49, 0x57, 0xf9, 0x8a, 0x5e, 0xbd, 0x21, 0x4e, 0x40, 0x4f, 0x09, 0xf3, 0x31, 0x84, - 0x5e, 0x4a, 0x24, 0xd4, 0x81, 0x92, 0x1a, 0x00, 0x1d, 0xc4, 0x9c, 0x37, 0x48, 0x51, 0xba, 0x69, - 0xa9, 0x07, 0x80, 0x92, 0xd5, 0xdf, 0x87, 0xc0, 0x3b, 0xf5, 0xb3, 0xfa, 0xf0, 0xcf, 0x24, 0xc9, - 0xd1, 0xc8, 0x4e, 0x42, 0x8d, 0xb4, 0xd4, 0xcf, 0xc0, 0xf4, 0xe9, 0x85, 0x6e, 0xdb, 0xad, 0x2c, - 0x09, 0x6e, 0x91, 0x8d, 0x43, 0x0f, 0xeb, 0xbb, 0x01, 0x95, 0xc9, 0xa7, 0x13, 0x4f, 0xb0, 0x21, - 0x1b, 0x10, 0xc1, 0x3c, 0xf7, 0x81, 0x6f, 0x83, 0xea, 0xb0, 0xac, 0x88, 0x1f, 0x32, 0x87, 0x6a, - 0x85, 0x4d, 0xa9, 0x70, 0x59, 0x9e, 0xbf, 0x60, 0x0e, 0x55, 0x32, 0x57, 0x20, 0xef, 0x7a, 0x36, - 0xed, 0x76, 0xaa, 0x39, 0x19, 0x42, 0x5b, 0xd6, 0x3e, 0xfc, 0xd7, 0x4d, 0x21, 0xe7, 0x19, 0xb1, - 0x2f, 0x06, 0xac, 0xa4, 0xbc, 0xe4, 0xac, 0xfd, 0xa9, 0xbb, 0x59, 0xdf, 0x0c, 0xa8, 0xdc, 0xbc, - 0xc4, 0x7d, 0xba, 0xa7, 0x03, 0x90, 0xc8, 0xa9, 0x26, 0x62, 0xeb, 0xce, 0x11, 0x4c, 0x72, 0x80, - 0x0b, 0xc3, 0xb8, 0xaa, 0x9f, 0x86, 0x5e, 0x44, 0xc7, 0x54, 0x90, 0x87, 0x5f, 0x44, 0xeb, 0x50, - 0x1c, 0x12, 0xe6, 0xf4, 0xfc, 0xe4, 0x36, 0x82, 0xf0, 0x48, 0x6d, 0xa4, 0xa8, 0x7d, 0x73, 0xbf, - 0xdb, 0xbe, 0x99, 0xc4, 0x2f, 0x66, 0x11, 0xdf, 0xfa, 0x9c, 0x83, 0x92, 0xaa, 0x59, 0x3d, 0x7c, - 0x68, 0x00, 0xa5, 0xe4, 0xf3, 0x81, 0x76, 0xb2, 0xd2, 0x66, 0x3c, 0x71, 0xb5, 0xfa, 0x6c, 0xa0, - 0x52, 0xd4, 0x5a, 0x40, 0xef, 0x01, 0x26, 0x95, 0xa3, 0xf9, 0x6e, 0x56, 0xdb, 0x9e, 0x05, 0x8b, - 0xc3, 0x0f, 0xa0, 0xfc, 0x92, 0x8a, 0xc4, 0x2e, 0x47, 0xdb, 0x77, 0x6a, 0x9f, 0x7a, 0x95, 0x6a, - 0x3b, 0x33, 0x71, 0x71, 0x12, 0x07, 0x56, 0xa2, 0x24, 0x93, 0x19, 0x7d, 0x3c, 0xb3, 0xc7, 0xe2, - 0x54, 0xbb, 0xf3, 0x40, 0x13, 0x8c, 0x2d, 0xcb, 0x2d, 0x74, 0x95, 0xe0, 0x6d, 0x77, 0x3a, 0x21, - 0xc9, 0xad, 0x55, 0x9b, 0x36, 0x35, 0xd6, 0x42, 0xeb, 0x9d, 0x6e, 0x74, 0x29, 0xf9, 0x49, 0x4a, - 0x9d, 0xcd, 0xe9, 0x59, 0x8e, 0x46, 0xf6, 0x8c, 0xe0, 0xed, 0xc3, 0xb7, 0xcf, 0xcf, 0x98, 0x38, - 0x0f, 0xfa, 0xe1, 0x97, 0xe6, 0x35, 0x73, 0x1c, 0x76, 0x2d, 0xe8, 0xe0, 0xbc, 0xa9, 0xbc, 0x9e, - 0xd8, 0x8c, 0x0b, 0x9f, 0xf5, 0x03, 0x41, 0xed, 0x66, 0xf4, 0x67, 0xa2, 0x29, 0x43, 0x35, 0x65, - 0xb6, 0x71, 0xbf, 0x9f, 0x97, 0xe6, 0xfe, 0xaf, 0x00, 0x00, 0x00, 0xff, 0xff, 0x5c, 0xbe, 0xea, - 0x0e, 0xa6, 0x09, 0x00, 0x00, + // 842 bytes of a gzipped FileDescriptorProto + 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xec, 0x56, 0xd1, 0x6e, 0x1b, 0x45, + 0x14, 0xcd, 0x76, 0x53, 0x17, 0x5f, 0x9b, 0xa8, 0x99, 0x94, 0xc8, 0x5a, 0xa8, 0xda, 0x2e, 0x6a, + 0x6b, 0x2a, 0x61, 0x57, 0x8e, 0x0a, 0x8f, 0xa8, 0x49, 0x20, 0xb2, 0x50, 0xa3, 0x68, 0x5a, 0x81, + 0xa8, 0x84, 0xa2, 0xf1, 0xee, 0x75, 0x32, 0x62, 0x77, 0x76, 0xbb, 0x33, 0xae, 0x48, 0x5f, 0x10, + 0xef, 0x48, 0x3c, 0xc0, 0x4f, 0xf0, 0x0f, 0x7c, 0x0c, 0xef, 0xfc, 0x04, 0xda, 0x99, 0x59, 0x7b, + 0x37, 0xde, 0xd8, 0x8e, 0x12, 0x89, 0x97, 0xbe, 0xf9, 0xce, 0x9e, 0x7b, 0xee, 0x9d, 0x73, 0xe6, + 0xce, 0x18, 0xb6, 0xb8, 0x08, 0xf1, 0xe7, 0x63, 0x89, 0xd9, 0x5b, 0x1e, 0x60, 0x2f, 0xcd, 0x12, + 0x95, 0x10, 0x12, 0xf3, 0xe8, 0xed, 0x44, 0x9a, 0xa8, 0xa7, 0x11, 0x5e, 0x3b, 0x48, 0xe2, 0x38, + 0x11, 0x66, 0xcd, 0xdb, 0xe0, 0x42, 0x61, 0x26, 0x58, 0x64, 0xe3, 0x76, 0x39, 0xc3, 0xff, 0x05, + 0xb6, 0x28, 0x9e, 0x70, 0xa9, 0x30, 0x3b, 0x4c, 0x42, 0xa4, 0xf8, 0x66, 0x82, 0x52, 0x91, 0xa7, + 0xb0, 0x3e, 0x62, 0x12, 0x3b, 0xce, 0x7d, 0xa7, 0xdb, 0x1a, 0x7c, 0xd2, 0xab, 0x54, 0xb1, 0xf4, + 0x2f, 0xe4, 0xc9, 0x2e, 0x93, 0x48, 0x35, 0x92, 0x7c, 0x01, 0xb7, 0x58, 0x18, 0x66, 0x28, 0x65, + 0xe7, 0xc6, 0x82, 0xa4, 0xe7, 0x06, 0x43, 0x0b, 0xb0, 0xff, 0xbb, 0x03, 0x77, 0xaa, 0x1d, 0xc8, + 0x34, 0x11, 0x12, 0xc9, 0x0e, 0x34, 0xa4, 0x62, 0x6a, 0x22, 0x6d, 0x13, 0x1f, 0xd7, 0xf2, 0xbd, + 0xd4, 0x10, 0x6a, 0xa1, 0x64, 0x17, 0x5a, 0x5c, 0x70, 0x75, 0x9c, 0xb2, 0x8c, 0xc5, 0x45, 0x27, + 0x0f, 0x7a, 0xe7, 0x44, 0xb2, 0x7a, 0x0c, 0x05, 0x57, 0x47, 0x1a, 0x48, 0x81, 0x4f, 0x7f, 0xfb, + 0x4f, 0x81, 0x0c, 0x73, 0x1d, 0x73, 0x6a, 0x94, 0x85, 0x22, 0x1e, 0x7c, 0xa0, 0xd5, 0x1d, 0xee, + 0xe7, 0x0d, 0xb9, 0x5d, 0x97, 0x4e, 0x63, 0x5f, 0x41, 0x53, 0x67, 0x0c, 0xc5, 0x38, 0x21, 0xcf, + 0xe0, 0x66, 0xde, 0x8c, 0xd1, 0x6e, 0x63, 0x70, 0xaf, 0xb6, 0xed, 0x59, 0x01, 0x6a, 0xd0, 0xa4, + 0x03, 0xb7, 0x2c, 0x9f, 0xee, 0xda, 0xa5, 0x45, 0x48, 0xb6, 0xa1, 0x41, 0x91, 0xc9, 0x44, 0x74, + 0xdc, 0xfb, 0x4e, 0xb7, 0x49, 0x6d, 0xe4, 0xff, 0xea, 0xc0, 0x56, 0xa5, 0xd1, 0xab, 0x08, 0xf7, + 0xcc, 0x24, 0x61, 0xae, 0x99, 0xdb, 0x6d, 0x0d, 0xee, 0xf6, 0xe6, 0x0f, 0x56, 0x6f, 0xba, 0x49, + 0x6a, 0xc1, 0xfe, 0xdf, 0x0e, 0x6c, 0xee, 0x4e, 0x78, 0x14, 0xea, 0x4f, 0x85, 0x56, 0x77, 0x01, + 0x42, 0xa6, 0xd8, 0x71, 0xca, 0xd4, 0xa9, 0x51, 0xab, 0x49, 0x9b, 0xf9, 0xca, 0x51, 0xbe, 0x90, + 0x9b, 0xa4, 0xce, 0x52, 0x9c, 0x99, 0xe4, 0xce, 0x9b, 0x64, 0xbb, 0xfc, 0x16, 0xcf, 0xbe, 0x63, + 0xd1, 0x04, 0x8f, 0x18, 0xcf, 0x28, 0xe4, 0x59, 0xc6, 0x24, 0xb2, 0x0f, 0x6d, 0x33, 0x0e, 0x96, + 0xc4, 0x5d, 0x95, 0xa4, 0xa5, 0xd3, 0xac, 0xd5, 0x01, 0x90, 0x72, 0xf7, 0x57, 0x11, 0xf0, 0x42, + 0xff, 0xfc, 0x11, 0x7c, 0x38, 0x2b, 0xb2, 0x17, 0x87, 0x65, 0xa8, 0x53, 0xb5, 0xfa, 0x4b, 0x70, + 0x33, 0x7c, 0x63, 0x8f, 0xed, 0xc3, 0x3a, 0x0b, 0xe6, 0xc4, 0xa6, 0x79, 0x86, 0xff, 0x97, 0x03, + 0xdb, 0xb3, 0x4f, 0x87, 0x89, 0xe2, 0x63, 0x1e, 0x30, 0xc5, 0x13, 0x71, 0xcd, 0xbb, 0x21, 0x5d, + 0xb8, 0x6d, 0x84, 0x1f, 0xf3, 0x08, 0xad, 0xc3, 0xae, 0x76, 0x78, 0x43, 0xaf, 0x7f, 0xc3, 0x23, + 0x34, 0x36, 0x6f, 0x43, 0x43, 0x24, 0x21, 0x0e, 0xf7, 0x3b, 0xeb, 0x9a, 0xc2, 0x46, 0xfe, 0x0e, + 0x7c, 0x34, 0xac, 0x20, 0x57, 0x19, 0xb1, 0xdf, 0x1c, 0xd8, 0xac, 0x64, 0xe9, 0x59, 0xfb, 0xbf, + 0xf6, 0xe6, 0xff, 0xe1, 0xc0, 0xf6, 0xf9, 0x4d, 0x5c, 0xe5, 0xf4, 0xec, 0x03, 0x94, 0x6a, 0x9a, + 0x89, 0x78, 0x78, 0xe1, 0x08, 0x96, 0x35, 0xa0, 0xcd, 0xf1, 0xb4, 0xab, 0x7f, 0x1c, 0x7b, 0x11, + 0xbd, 0x40, 0xc5, 0xae, 0xff, 0x22, 0xba, 0x07, 0xad, 0x31, 0xe3, 0xd1, 0x71, 0x56, 0xbe, 0x8d, + 0x20, 0x5f, 0x32, 0x37, 0x52, 0x71, 0x7c, 0xd7, 0x2f, 0x7b, 0x7c, 0x6b, 0x85, 0xbf, 0x59, 0x27, + 0xfc, 0xe0, 0xcf, 0x06, 0xb4, 0x4d, 0xcf, 0xe6, 0x19, 0x24, 0x01, 0xb4, 0xcb, 0xcf, 0x07, 0x79, + 0x5c, 0x57, 0xb6, 0xe6, 0x89, 0xf3, 0xba, 0xcb, 0x81, 0xc6, 0x51, 0x7f, 0x8d, 0xfc, 0x08, 0x30, + 0xeb, 0x9c, 0xac, 0xb6, 0x33, 0xef, 0xd1, 0x32, 0xd8, 0x94, 0x3e, 0x80, 0x8d, 0x03, 0x54, 0xa5, + 0xbb, 0x9c, 0x3c, 0xba, 0xd0, 0xfb, 0xca, 0xab, 0xe4, 0x3d, 0x5e, 0x8a, 0x9b, 0x16, 0x89, 0x60, + 0xb3, 0x28, 0x32, 0x9b, 0xd1, 0xcf, 0x96, 0x9e, 0xb1, 0x69, 0xa9, 0x27, 0xab, 0x40, 0x4b, 0x8a, + 0xdd, 0xd6, 0xb7, 0xd0, 0x59, 0x49, 0xb7, 0x27, 0x8b, 0x05, 0x29, 0xdf, 0x5a, 0xde, 0xa2, 0xa9, + 0xf1, 0xd7, 0xc8, 0x6b, 0x20, 0x07, 0xa8, 0xf6, 0x92, 0x38, 0x4d, 0x04, 0x0a, 0x65, 0x55, 0xf3, + 0x6a, 0x93, 0xbe, 0x8e, 0x53, 0x75, 0x36, 0xef, 0x86, 0xfd, 0x13, 0x70, 0x8e, 0xc3, 0x5f, 0x23, + 0xdf, 0x6b, 0xee, 0x57, 0x3c, 0xc6, 0x57, 0x3c, 0xf8, 0x69, 0xef, 0x94, 0x09, 0x81, 0xd1, 0x42, + 0xee, 0x4f, 0xab, 0xdf, 0x6c, 0xf0, 0x52, 0x65, 0x5c, 0x9c, 0x94, 0x34, 0xf9, 0x01, 0xee, 0x1c, + 0xa0, 0xae, 0xc3, 0xa5, 0xe2, 0x81, 0xbc, 0x3e, 0xea, 0xc1, 0xbf, 0x37, 0xec, 0xe4, 0xeb, 0x19, + 0x38, 0xac, 0x1c, 0xd7, 0x07, 0x8b, 0x65, 0xdf, 0x8b, 0xc3, 0xf7, 0x6a, 0x5f, 0x8a, 0x7a, 0xf7, + 0xf9, 0xeb, 0xaf, 0x4e, 0xb8, 0x3a, 0x9d, 0x8c, 0xf2, 0xec, 0xfe, 0x3b, 0x1e, 0x45, 0xfc, 0x9d, + 0xc2, 0xe0, 0xb4, 0x6f, 0x12, 0x3e, 0x0f, 0xb9, 0x54, 0x19, 0x1f, 0x4d, 0x14, 0x86, 0xfd, 0x62, + 0xeb, 0x7d, 0x4d, 0xd9, 0xd7, 0xea, 0xa7, 0xa3, 0x51, 0x43, 0x87, 0x3b, 0xff, 0x05, 0x00, 0x00, + 0xff, 0xff, 0xde, 0x02, 0xad, 0xb0, 0xd5, 0x0b, 0x00, 0x00, } // Reference imports to suppress errors if they are not otherwise used. @@ -778,6 +784,9 @@ type IndexServiceClient interface { GetIndexStates(ctx context.Context, in *IndexStatesRequest, opts ...grpc.CallOption) (*IndexStatesResponse, error) GetIndexFilePaths(ctx context.Context, in *IndexFilePathsRequest, opts ...grpc.CallOption) (*IndexFilePathsResponse, error) NotifyBuildIndex(ctx context.Context, in *BuildIndexNotification, opts ...grpc.CallOption) (*commonpb.Status, error) + GetComponentStates(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*internalpb2.ComponentStates, error) + GetTimeTickChannel(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*milvuspb.StringResponse, error) + GetStatisticsChannel(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*milvuspb.StringResponse, error) } type indexServiceClient struct { @@ -833,6 +842,33 @@ func (c *indexServiceClient) NotifyBuildIndex(ctx context.Context, in *BuildInde return out, nil } +func (c *indexServiceClient) GetComponentStates(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*internalpb2.ComponentStates, error) { + out := new(internalpb2.ComponentStates) + err := c.cc.Invoke(ctx, "/milvus.proto.index.IndexService/GetComponentStates", in, out, opts...) + if err != nil { + return nil, err + } + return out, nil +} + +func (c *indexServiceClient) GetTimeTickChannel(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*milvuspb.StringResponse, error) { + out := new(milvuspb.StringResponse) + err := c.cc.Invoke(ctx, "/milvus.proto.index.IndexService/GetTimeTickChannel", in, out, opts...) + if err != nil { + return nil, err + } + return out, nil +} + +func (c *indexServiceClient) GetStatisticsChannel(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*milvuspb.StringResponse, error) { + out := new(milvuspb.StringResponse) + err := c.cc.Invoke(ctx, "/milvus.proto.index.IndexService/GetStatisticsChannel", in, out, opts...) + if err != nil { + return nil, err + } + return out, nil +} + // IndexServiceServer is the server API for IndexService service. type IndexServiceServer interface { //* @@ -846,6 +882,9 @@ type IndexServiceServer interface { GetIndexStates(context.Context, *IndexStatesRequest) (*IndexStatesResponse, error) GetIndexFilePaths(context.Context, *IndexFilePathsRequest) (*IndexFilePathsResponse, error) NotifyBuildIndex(context.Context, *BuildIndexNotification) (*commonpb.Status, error) + GetComponentStates(context.Context, *commonpb.Empty) (*internalpb2.ComponentStates, error) + GetTimeTickChannel(context.Context, *commonpb.Empty) (*milvuspb.StringResponse, error) + GetStatisticsChannel(context.Context, *commonpb.Empty) (*milvuspb.StringResponse, error) } // UnimplementedIndexServiceServer can be embedded to have forward compatible implementations. @@ -867,6 +906,15 @@ func (*UnimplementedIndexServiceServer) GetIndexFilePaths(ctx context.Context, r func (*UnimplementedIndexServiceServer) NotifyBuildIndex(ctx context.Context, req *BuildIndexNotification) (*commonpb.Status, error) { return nil, status.Errorf(codes.Unimplemented, "method NotifyBuildIndex not implemented") } +func (*UnimplementedIndexServiceServer) GetComponentStates(ctx context.Context, req *commonpb.Empty) (*internalpb2.ComponentStates, error) { + return nil, status.Errorf(codes.Unimplemented, "method GetComponentStates not implemented") +} +func (*UnimplementedIndexServiceServer) GetTimeTickChannel(ctx context.Context, req *commonpb.Empty) (*milvuspb.StringResponse, error) { + return nil, status.Errorf(codes.Unimplemented, "method GetTimeTickChannel not implemented") +} +func (*UnimplementedIndexServiceServer) GetStatisticsChannel(ctx context.Context, req *commonpb.Empty) (*milvuspb.StringResponse, error) { + return nil, status.Errorf(codes.Unimplemented, "method GetStatisticsChannel not implemented") +} func RegisterIndexServiceServer(s *grpc.Server, srv IndexServiceServer) { s.RegisterService(&_IndexService_serviceDesc, srv) @@ -962,6 +1010,60 @@ func _IndexService_NotifyBuildIndex_Handler(srv interface{}, ctx context.Context return interceptor(ctx, in, info, handler) } +func _IndexService_GetComponentStates_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(commonpb.Empty) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(IndexServiceServer).GetComponentStates(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: "/milvus.proto.index.IndexService/GetComponentStates", + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(IndexServiceServer).GetComponentStates(ctx, req.(*commonpb.Empty)) + } + return interceptor(ctx, in, info, handler) +} + +func _IndexService_GetTimeTickChannel_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(commonpb.Empty) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(IndexServiceServer).GetTimeTickChannel(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: "/milvus.proto.index.IndexService/GetTimeTickChannel", + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(IndexServiceServer).GetTimeTickChannel(ctx, req.(*commonpb.Empty)) + } + return interceptor(ctx, in, info, handler) +} + +func _IndexService_GetStatisticsChannel_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(commonpb.Empty) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(IndexServiceServer).GetStatisticsChannel(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: "/milvus.proto.index.IndexService/GetStatisticsChannel", + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(IndexServiceServer).GetStatisticsChannel(ctx, req.(*commonpb.Empty)) + } + return interceptor(ctx, in, info, handler) +} + var _IndexService_serviceDesc = grpc.ServiceDesc{ ServiceName: "milvus.proto.index.IndexService", HandlerType: (*IndexServiceServer)(nil), @@ -986,6 +1088,18 @@ var _IndexService_serviceDesc = grpc.ServiceDesc{ MethodName: "NotifyBuildIndex", Handler: _IndexService_NotifyBuildIndex_Handler, }, + { + MethodName: "GetComponentStates", + Handler: _IndexService_GetComponentStates_Handler, + }, + { + MethodName: "GetTimeTickChannel", + Handler: _IndexService_GetTimeTickChannel_Handler, + }, + { + MethodName: "GetStatisticsChannel", + Handler: _IndexService_GetStatisticsChannel_Handler, + }, }, Streams: []grpc.StreamDesc{}, Metadata: "index_service.proto", @@ -1002,6 +1116,9 @@ type IndexNodeClient interface { // // @return Status BuildIndex(ctx context.Context, in *BuildIndexCmd, opts ...grpc.CallOption) (*commonpb.Status, error) + GetComponentStates(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*internalpb2.ComponentStates, error) + GetTimeTickChannel(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*milvuspb.StringResponse, error) + GetStatisticsChannel(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*milvuspb.StringResponse, error) } type indexNodeClient struct { @@ -1021,6 +1138,33 @@ func (c *indexNodeClient) BuildIndex(ctx context.Context, in *BuildIndexCmd, opt return out, nil } +func (c *indexNodeClient) GetComponentStates(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*internalpb2.ComponentStates, error) { + out := new(internalpb2.ComponentStates) + err := c.cc.Invoke(ctx, "/milvus.proto.index.IndexNode/GetComponentStates", in, out, opts...) + if err != nil { + return nil, err + } + return out, nil +} + +func (c *indexNodeClient) GetTimeTickChannel(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*milvuspb.StringResponse, error) { + out := new(milvuspb.StringResponse) + err := c.cc.Invoke(ctx, "/milvus.proto.index.IndexNode/GetTimeTickChannel", in, out, opts...) + if err != nil { + return nil, err + } + return out, nil +} + +func (c *indexNodeClient) GetStatisticsChannel(ctx context.Context, in *commonpb.Empty, opts ...grpc.CallOption) (*milvuspb.StringResponse, error) { + out := new(milvuspb.StringResponse) + err := c.cc.Invoke(ctx, "/milvus.proto.index.IndexNode/GetStatisticsChannel", in, out, opts...) + if err != nil { + return nil, err + } + return out, nil +} + // IndexNodeServer is the server API for IndexNode service. type IndexNodeServer interface { //* @@ -1030,6 +1174,9 @@ type IndexNodeServer interface { // // @return Status BuildIndex(context.Context, *BuildIndexCmd) (*commonpb.Status, error) + GetComponentStates(context.Context, *commonpb.Empty) (*internalpb2.ComponentStates, error) + GetTimeTickChannel(context.Context, *commonpb.Empty) (*milvuspb.StringResponse, error) + GetStatisticsChannel(context.Context, *commonpb.Empty) (*milvuspb.StringResponse, error) } // UnimplementedIndexNodeServer can be embedded to have forward compatible implementations. @@ -1039,6 +1186,15 @@ type UnimplementedIndexNodeServer struct { func (*UnimplementedIndexNodeServer) BuildIndex(ctx context.Context, req *BuildIndexCmd) (*commonpb.Status, error) { return nil, status.Errorf(codes.Unimplemented, "method BuildIndex not implemented") } +func (*UnimplementedIndexNodeServer) GetComponentStates(ctx context.Context, req *commonpb.Empty) (*internalpb2.ComponentStates, error) { + return nil, status.Errorf(codes.Unimplemented, "method GetComponentStates not implemented") +} +func (*UnimplementedIndexNodeServer) GetTimeTickChannel(ctx context.Context, req *commonpb.Empty) (*milvuspb.StringResponse, error) { + return nil, status.Errorf(codes.Unimplemented, "method GetTimeTickChannel not implemented") +} +func (*UnimplementedIndexNodeServer) GetStatisticsChannel(ctx context.Context, req *commonpb.Empty) (*milvuspb.StringResponse, error) { + return nil, status.Errorf(codes.Unimplemented, "method GetStatisticsChannel not implemented") +} func RegisterIndexNodeServer(s *grpc.Server, srv IndexNodeServer) { s.RegisterService(&_IndexNode_serviceDesc, srv) @@ -1062,6 +1218,60 @@ func _IndexNode_BuildIndex_Handler(srv interface{}, ctx context.Context, dec fun return interceptor(ctx, in, info, handler) } +func _IndexNode_GetComponentStates_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(commonpb.Empty) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(IndexNodeServer).GetComponentStates(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: "/milvus.proto.index.IndexNode/GetComponentStates", + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(IndexNodeServer).GetComponentStates(ctx, req.(*commonpb.Empty)) + } + return interceptor(ctx, in, info, handler) +} + +func _IndexNode_GetTimeTickChannel_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(commonpb.Empty) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(IndexNodeServer).GetTimeTickChannel(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: "/milvus.proto.index.IndexNode/GetTimeTickChannel", + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(IndexNodeServer).GetTimeTickChannel(ctx, req.(*commonpb.Empty)) + } + return interceptor(ctx, in, info, handler) +} + +func _IndexNode_GetStatisticsChannel_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(commonpb.Empty) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(IndexNodeServer).GetStatisticsChannel(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: "/milvus.proto.index.IndexNode/GetStatisticsChannel", + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(IndexNodeServer).GetStatisticsChannel(ctx, req.(*commonpb.Empty)) + } + return interceptor(ctx, in, info, handler) +} + var _IndexNode_serviceDesc = grpc.ServiceDesc{ ServiceName: "milvus.proto.index.IndexNode", HandlerType: (*IndexNodeServer)(nil), @@ -1070,6 +1280,18 @@ var _IndexNode_serviceDesc = grpc.ServiceDesc{ MethodName: "BuildIndex", Handler: _IndexNode_BuildIndex_Handler, }, + { + MethodName: "GetComponentStates", + Handler: _IndexNode_GetComponentStates_Handler, + }, + { + MethodName: "GetTimeTickChannel", + Handler: _IndexNode_GetTimeTickChannel_Handler, + }, + { + MethodName: "GetStatisticsChannel", + Handler: _IndexNode_GetStatisticsChannel_Handler, + }, }, Streams: []grpc.StreamDesc{}, Metadata: "index_service.proto", diff --git a/internal/util/typeutil/interface.go b/internal/util/typeutil/interface.go index 3f8516258d4006055f8bdf1163f584b6f8afd200..6691ff33d5158883558511c6beeee346c84f9df2 100644 --- a/internal/util/typeutil/interface.go +++ b/internal/util/typeutil/interface.go @@ -21,6 +21,7 @@ type Component interface { type IndexNodeInterface interface { Service + Component BuildIndex(req *indexpb.BuildIndexCmd) (*commonpb.Status, error) }