enhance: remove the rpc level of coordinator

Signed-off-by: chyezh <[email protected]>
milvus-io · Nov 20, 2024 · 7cb3836 · 7cb3836
1 parent 7144219
commit 7cb3836
Show file tree

Hide file tree

Showing 15 changed files with 2,383 additions and 46 deletions.
diff --git a/cmd/milvus/util.go b/cmd/milvus/util.go
@@ -20,6 +20,7 @@ import (
 	"go.uber.org/zap"
 
 	"github.com/milvus-io/milvus/cmd/roles"
+	"github.com/milvus-io/milvus/internal/coordinator/coordclient"
 	"github.com/milvus-io/milvus/internal/util/sessionutil"
 	"github.com/milvus-io/milvus/internal/util/streamingutil"
 	"github.com/milvus-io/milvus/pkg/log"
@@ -179,7 +180,12 @@ func GetMilvusRoles(args []string, flags *flag.FlagSet) *roles.MilvusRoles {
 		fmt.Fprintf(os.Stderr, "Unknown server type = %s\n%s", serverType, getHelp())
 		os.Exit(-1)
 	}
-
+	coordclient.EnableLocalClientRole(&coordclient.LocalClientRoleConfig{
+		ServerType:       serverType,
+		EnableQueryCoord: role.EnableQueryCoord,
+		EnableDataCoord:  role.EnableDataCoord,
+		EnableRootCoord:  role.EnableRootCoord,
+	})
 	return role
 }