chore(bench): update

Change-Id: I39f3d7468fb121ed894787df0fc87604e0e82784

Author: andeya

examples/bench/client.go

@@ -16,10 +16,13 @@ import (
 
 //go:generate go build $GOFILE
 
-var concurrency = flag.Int("c", 1, "concurrency")
-var total = flag.Int("n", 1, "total requests for all clients")
-var host = flag.String("s", "127.0.0.1:8972", "server ip and port")
-var debugAddr = flag.String("d", "127.0.0.1:9982", "server ip and port")
+var (
+	concurrency = flag.Int("c", 1, "concurrency")
+	total       = flag.Int("n", 1, "total requests for all clients")
+	host        = flag.String("s", "127.0.0.1:8972", "server ip and port")
+	debugAddr   = flag.String("d", "127.0.0.1:9982", "server ip and port")
+	network     = flag.String("network", "tcp", "network")
+)
 
 func main() {
 	flag.Parse()
@@ -43,6 +46,7 @@ func main() {
 
 	serviceMethod := "Hello.Say"
 	client := erpc.NewPeer(erpc.PeerConfig{
+		Network:          *network,
 		DefaultBodyCodec: "protobuf",
 	})
 

examples/bench/mclient.go

@@ -16,9 +16,12 @@ import (
 
 //go:generate go build $GOFILE
 
-var concurrency = flag.Int("c", 1, "concurrency")
-var total = flag.Int("n", 1, "total requests for all clients")
-var host = flag.String("s", "127.0.0.1:8972", "server ip and port")
+var (
+	concurrency = flag.Int("c", 1, "concurrency")
+	total       = flag.Int("n", 1, "total requests for all clients")
+	host        = flag.String("s", "127.0.0.1:8972", "server ip and port")
+	network     = flag.String("network", "tcp", "network")
+)
 
 func main() {
 	flag.Parse()
@@ -38,6 +41,7 @@ func main() {
 
 	serviceMethod := "Hello.Say"
 	client := erpc.NewPeer(erpc.PeerConfig{
+		Network:          *network,
 		DefaultBodyCodec: "protobuf",
 	})
 

examples/bench/server.go

@@ -14,27 +14,11 @@ import (
 
 //go:generate go build $GOFILE
 
-type Hello struct {
-	erpc.CallCtx
-}
-
-func (t *Hello) Say(args *msg.BenchmarkMessage) (*msg.BenchmarkMessage, *erpc.Status) {
-	s := "OK"
-	var i int32 = 100
-	args.Field1 = s
-	args.Field2 = i
-	if *delay > 0 {
-		time.Sleep(*delay)
-	} else {
-		runtime.Gosched()
-	}
-	return args, nil
-}
-
 var (
 	port      = flag.Int64("p", 8972, "listened port")
 	delay     = flag.Duration("delay", 0, "delay to mock business processing")
 	debugAddr = flag.String("d", "127.0.0.1:9981", "server ip and port")
+	network   = flag.String("network", "tcp", "network")
 )
 
 func main() {
@@ -49,9 +33,27 @@ func main() {
 
 	erpc.SetServiceMethodMapper(erpc.RPCServiceMethodMapper)
 	server := erpc.NewPeer(erpc.PeerConfig{
+		Network:          *network,
 		DefaultBodyCodec: "protobuf",
 		ListenPort:       uint16(*port),
 	})
 	server.RouteCall(new(Hello))
 	server.ListenAndServe()
 }
+
+type Hello struct {
+	erpc.CallCtx
+}
+
+func (t *Hello) Say(args *msg.BenchmarkMessage) (*msg.BenchmarkMessage, *erpc.Status) {
+	s := "OK"
+	var i int32 = 100
+	args.Field1 = s
+	args.Field2 = i
+	if *delay > 0 {
+		time.Sleep(*delay)
+	} else {
+		runtime.Gosched()
+	}
+	return args, nil
+}