35 行
564 B
Go
35 行
564 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"strings"
|
|
|
|
"code.widuu.com/widuu/micro_service/demo1/cmd"
|
|
)
|
|
|
|
var (
|
|
port = flag.String("port", ":1234", "run port")
|
|
runMode = flag.String("mode", "", "run mode")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
if !strings.Contains(*port, ":") {
|
|
*port = ":" + *port
|
|
}
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
switch *runMode {
|
|
case "client":
|
|
cmd.RunClient(*port)
|
|
return
|
|
case "server":
|
|
cmd.RunServer(*port, ctx)
|
|
return
|
|
default:
|
|
go cmd.RunServer(*port, ctx)
|
|
cmd.RunClient(*port)
|
|
}
|
|
}
|