diff --git a/cmd/datanode/main.go b/cmd/datanode/main.go index 732a8cbb1bd404422464e4568fe004cc49f03d77..f20802305df85a146bb7f2f3c7e6e202c7c3dc57 100644 --- a/cmd/datanode/main.go +++ b/cmd/datanode/main.go @@ -12,6 +12,7 @@ import ( dnc "github.com/zilliztech/milvus-distributed/internal/distributed/datanode" dsc "github.com/zilliztech/milvus-distributed/internal/distributed/dataservice" msc "github.com/zilliztech/milvus-distributed/internal/distributed/masterservice" + ms "github.com/zilliztech/milvus-distributed/internal/masterservice" "github.com/zilliztech/milvus-distributed/internal/proto/commonpb" "github.com/zilliztech/milvus-distributed/internal/proto/internalpb2" ) @@ -32,6 +33,7 @@ func main() { log.Println("Datanode is", dn.Params.NodeID) // --- Master Service Client --- + ms.Params.Init() log.Println("Master service address:", dn.Params.MasterAddress) masterClient, err := msc.NewGrpcClient(dn.Params.MasterAddress, 20*time.Second) if err != nil { diff --git a/internal/masterservice/master_service.go b/internal/masterservice/master_service.go index 11841cbb8bf457b16fbb7b722d688ceb43510b3c..c15d476cceab88221aea66c8667641c511e0a4d2 100644 --- a/internal/masterservice/master_service.go +++ b/internal/masterservice/master_service.go @@ -717,6 +717,18 @@ func (c *Core) GetComponentStates() (*internalpb2.ComponentStates, error) { StateCode: code, ExtraInfo: nil, }, + Status: &commonpb.Status{ + ErrorCode: commonpb.ErrorCode_SUCCESS, + Reason: "", + }, + SubcomponentStates: []*internalpb2.ComponentInfo{ + { + NodeID: int64(Params.NodeID), + Role: typeutil.MasterServiceRole, + StateCode: code, + ExtraInfo: nil, + }, + }, }, nil }