Selaa lähdekoodia

Simplify cmdline parsing

Lucas Stadler 9 vuotta sitten
vanhempi
commit
c011d755a2
1 muutettua tiedostoa jossa 7 lisäystä ja 10 poistoa
  1. 7 10
      go/linguaevalia/linguaevalia.go

+ 7 - 10
go/linguaevalia/linguaevalia.go

@ -239,21 +239,18 @@ help		- Display this help message.
239 239
		os.Args[0])
240 240
}
241 241
242
func parseCommand() (string, []string) {
243
	if len(os.Args) == 1 {
244
		return "server", []string{}
245
	} else {
246
		return os.Args[1], os.Args[2:]
247
	}
248
}
249
250 242
var language = flag.String("l", "", "The language to use for code passed via stdin.")
251 243
var host = flag.String("h", "localhost", "The host to listen on.")
252 244
var port = flag.Int("p", 8000, "The port to listen on.")
253 245
254 246
func main() {
255
	cmd, args := parseCommand()
256
	flag.CommandLine.Parse(args)
247
	var cmd string
248
	if len(os.Args) == 1 {
249
		cmd = "server"
250
	} else {
251
		cmd = os.Args[1]
252
	}
253
	flag.Parse()
257 254
258 255
	switch cmd {
259 256
	case "server":