mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-23 23:03:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

Take first argument if flags package doesn't find any args

Browse source
This commit is contained in:
Neil Alexander 2020-10-20 14:58:58 +01:00
parent 705339d098
commit 26c237a36c
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 15 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
cmd/dendrite-polylith-multi/main.go
View file

@ -29,7 +29,13 @@ type entrypoint func(base *setup.BaseDendrite, cfg *config.Dendrite)
func main() {
cfg := setup.ParseFlags(true)
component := flag.Arg(0)
component := ""
if flag.NFlag() > 0 {
component = flag.Arg(0) // ./dendrite-polylith-multi --config=... clientapi
} else if len(os.Args) > 1 {
component = os.Args[1] // ./dendrite-polylith-multi clientapi
}
components := map[string]entrypoint{
"appservice": personalities.Appservice,
@ -47,13 +53,19 @@ func main() {
start, ok := components[component]
if !ok {
if component == "" {
logrus.Errorf("No component specified")
logrus.Info("The first argument on the command line must be the name of the component to run")
} else {
logrus.Errorf("Unknown component %q specified", component)
}
var list []string
for c := range components {
list = append(list, c)
}
logrus.Errorf("Unknown component %q specified", component)
logrus.Infof("Valid components: %s", strings.Join(list, ", "))
os.Exit(1)
}