update examples

examples/bench/client/client.go

@@ -1,6 +1,8 @@
 package main
 
 import (
+	"crypto/rand"
+	"encoding/base64"
 	"log"
 	"net"
 	"runtime"
@@ -38,6 +40,8 @@ func main() {
 		eachClientCoroutineNum = 10
 	)
 
+	arpc.EnablePool(true)
+
 	clients := make([]*arpc.Client, clientNum)
 
 	for i := 0; i < clientNum; i++ {
@@ -55,14 +59,17 @@ func main() {
 		for j := 0; j < eachClientCoroutineNum; j++ {
 			go func() {
 				var err error
+				var data = make([]byte, 512)
 				for k := 0; true; k++ {
-					req := &HelloReq{Msg: "hello from client.Call"}
+					rand.Read(data)
+					req := &HelloReq{Msg: base64.RawStdEncoding.EncodeToString(data)}
 					rsp := &HelloRsp{}
 					err = client.Call(method, req, rsp, time.Second*5)
 					if err != nil {
 						log.Printf("Call failed: %v", err)
+					} else if rsp.Msg != req.Msg {
+						log.Fatal("Call failed: not equal")
 					} else {
-						//log.Printf("Call Response: \"%v\"", rsp.Msg)
 						atomic.AddUint64(&qpsSec, 1)
 					}
 				}

examples/bench/server/server.go

@@ -35,6 +35,8 @@ func main() {
 	}
 
 	svr := arpc.NewServer()
+	svr.Handler.EnablePool(true)
+	svr.Handler.SetAsyncResponse(true)
 	svr.Handler.Handle("Hello", OnHello)
 	svr.Serve(ln)
 }

examples/bench_nbio/client/client.go

@@ -1,6 +1,8 @@
 package main
 
 import (
+	"crypto/rand"
+	"encoding/base64"
 	"log"
 	"net"
 	"runtime"
@@ -38,6 +40,8 @@ func main() {
 		eachClientCoroutineNum = 10
 	)
 
+	arpc.EnablePool(true)
+
 	clients := make([]*arpc.Client, clientNum)
 
 	for i := 0; i < clientNum; i++ {
@@ -55,14 +59,17 @@ func main() {
 		for j := 0; j < eachClientCoroutineNum; j++ {
 			go func() {
 				var err error
+				var data = make([]byte, 512)
 				for k := 0; true; k++ {
-					req := &HelloReq{Msg: "hello from client.Call"}
+					rand.Read(data)
+					req := &HelloReq{Msg: base64.RawStdEncoding.EncodeToString(data)}
 					rsp := &HelloRsp{}
 					err = client.Call(method, req, rsp, time.Second*5)
 					if err != nil {
 						log.Printf("Call failed: %v", err)
+					} else if rsp.Msg != req.Msg {
+						log.Fatal("Call failed: not equal")
 					} else {
-						//log.Printf("Call Response: \"%v\"", rsp.Msg)
 						atomic.AddUint64(&qpsSec, 1)
 					}
 				}

examples/bench_nbio/server/server.go

@@ -70,7 +70,11 @@ func onData(c *nbio.Conn, data []byte) {
 func main() {
 	nlog.SetLogger(log.DefaultLogger)
 
+	arpc.BufferPool = mempool.DefaultMemPool
+
+	handler.EnablePool(true)
 	handler.SetAsyncWrite(false)
+	handler.SetAsyncResponse(true)
 
 	// register router
 	handler.Handle("Hello", func(ctx *arpc.Context) {
@@ -86,6 +90,9 @@ func main() {
 
 	g.OnOpen(onOpen)
 	g.OnData(onData)
+	g.OnWriteBufferRelease(func(c *nbio.Conn, b []byte) {
+		mempool.Free(b)
+	})
 
 	err := g.Start()
 	if err != nil {

examples/nbio/server/server.go

@@ -60,6 +60,8 @@ func onData(c *nbio.Conn, data []byte) {
 func main() {
 	nlog.SetLogger(log.DefaultLogger)
 
+	arpc.BufferPool = mempool.DefaultMemPool
+
 	handler.EnablePool(true)
 	handler.SetAsyncWrite(false)
 
@@ -83,6 +85,9 @@ func main() {
 
 	g.OnOpen(onOpen)
 	g.OnData(onData)
+	g.OnWriteBufferRelease(func(c *nbio.Conn, b []byte) {
+		mempool.Free(b)
+	})
 
 	err := g.Start()
 	if err != nil {