func main()

in benchmark/benchserver/main.go [41:78]

33 lines of code
8 McCabe index (conditional complexity)


func main() {
	flag.Parse()

	var adHosts []string
	if len(*advertiseHosts) > 0 {
		adHosts = strings.Split(*advertiseHosts, ",")
	}

	server := benchmark.NewServer(
		benchmark.WithServiceName(*serviceName),
		benchmark.WithAdvertiseHosts(adHosts),
	)

	fmt.Println(server.HostPort())

	rdr := bufio.NewReader(os.Stdin)
	for {
		line, err := rdr.ReadString('\n')
		if err != nil {
			if err == io.EOF {
				return
			}
			log.Fatalf("stdin read failed: %v", err)
		}

		line = strings.TrimSuffix(line, "\n")
		switch line {
		case "count-raw":
			fmt.Println(server.RawCalls())
		case "count-thrift":
			fmt.Println(server.ThriftCalls())
		case "quit":
			return
		default:
			log.Fatalf("unrecognized command: %v", line)
		}
	}
}