mirror of https://github.com/milvus-io/milvus.git
25 lines
533 B
Go
25 lines
533 B
Go
package command
|
|
|
|
import (
|
|
"flag"
|
|
|
|
"github.com/milvus-io/milvus/cmd/tools/migration/console"
|
|
)
|
|
|
|
type commandParser struct {
|
|
configYaml string
|
|
}
|
|
|
|
func (c *commandParser) formatYaml(args []string, flags *flag.FlagSet) {
|
|
flags.StringVar(&c.configYaml, "config", "", "set config yaml")
|
|
}
|
|
|
|
func (c *commandParser) parse(args []string, flags *flag.FlagSet) {
|
|
console.AbnormalExitIf(flags.Parse(args[1:]), false)
|
|
}
|
|
|
|
func (c *commandParser) format(args []string, flags *flag.FlagSet) {
|
|
c.formatYaml(args, flags)
|
|
c.parse(args, flags)
|
|
}
|