Mix start with more role (#19251)

Signed-off-by: xige-16 <xi.ge@zilliz.com>

Signed-off-by: xige-16 <xi.ge@zilliz.com>
pull/19252/head
xige-16 2022-09-16 20:38:49 +08:00 committed by GitHub
parent d8b11121a8
commit 0ba47bd0d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 2 deletions

View File

@ -24,8 +24,15 @@ const (
type run struct {
serverType string
// flags
svrAlias string
enableRootCoord, enableQueryCoord, enableIndexCoord, enableDataCoord bool
svrAlias string
enableRootCoord bool
enableQueryCoord bool
enableIndexCoord bool
enableDataCoord bool
enableQueryNode bool
enableDataNode bool
enableIndexNode bool
enableProxy bool
}
func (c *run) getHelp() string {
@ -79,6 +86,10 @@ func (c *run) execute(args []string, flags *flag.FlagSet) {
role.EnableQueryCoord = c.enableQueryCoord
role.EnableDataCoord = c.enableDataCoord
role.EnableIndexCoord = c.enableIndexCoord
role.EnableQueryNode = c.enableQueryNode
role.EnableDataNode = c.enableDataNode
role.EnableIndexNode = c.enableIndexNode
role.EnableProxy = c.enableProxy
default:
fmt.Fprintf(os.Stderr, "Unknown server type = %s\n%s", c.serverType, c.getHelp())
os.Exit(-1)
@ -119,6 +130,11 @@ func (c *run) formatFlags(args []string, flags *flag.FlagSet) {
flags.BoolVar(&c.enableIndexCoord, typeutil.IndexCoordRole, false, "enable index coordinator")
flags.BoolVar(&c.enableDataCoord, typeutil.DataCoordRole, false, "enable data coordinator")
flags.BoolVar(&c.enableQueryNode, typeutil.QueryNodeRole, false, "enable query node")
flags.BoolVar(&c.enableDataNode, typeutil.DataNodeRole, false, "enable data node")
flags.BoolVar(&c.enableIndexNode, typeutil.IndexNodeRole, false, "enable index node")
flags.BoolVar(&c.enableProxy, typeutil.ProxyRole, false, "enable proxy node")
initMaxprocs(c.serverType, flags)
if err := flags.Parse(args[3:]); err != nil {
os.Exit(-1)