adding trace level logging to RPC layer

pagecr · Sep 19, 2013 · 514a209 · 514a209
1 parent ffbc03b
commit 514a209
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/rpc/bsonrpc/client.go b/rpc/bsonrpc/client.go
@@ -28,6 +28,9 @@ func NewClientCodec(conn io.ReadWriteCloser) (codec rpc.ClientCodec) {
 }
 
 func (cc *ccodec) WriteRequest(req *rpc.Request, v interface{}) (err error) {
+	log.Println(log.TRACE, "RPC Server Entered: WriteRequest")
+	defer log.Println(log.TRACE, "RPC Server Leaving: WriteRequest")
+
 	err = cc.enc.Encode(req)
 	if err != nil {
 		log.Println(log.ERROR, "RPC Client Error enconding request rpc request: ", err)
@@ -44,6 +47,9 @@ func (cc *ccodec) WriteRequest(req *rpc.Request, v interface{}) (err error) {
 }
 
 func (cc *ccodec) ReadResponseHeader(res *rpc.Response) (err error) {
+	log.Println(log.TRACE, "RPC Server Entered: ReadResponseHeader")
+	defer log.Println(log.TRACE, "RPC Server Leaving: ReadResponseHeader")
+
 	err = cc.dec.Decode(res)
 
 	if err != nil {
@@ -53,6 +59,9 @@ func (cc *ccodec) ReadResponseHeader(res *rpc.Response) (err error) {
 }
 
 func (cc *ccodec) ReadResponseBody(v interface{}) (err error) {
+	log.Println(log.TRACE, "RPC Server Entered: ReadResponseBody")
+	defer log.Println(log.TRACE, "RPC Server Leaving: ReadResponseBody")
+
 	if v == nil {
 		err = errors.New("Response object cannot be nil")
 		if err != nil {
@@ -70,6 +79,9 @@ func (cc *ccodec) ReadResponseBody(v interface{}) (err error) {
 }
 
 func (cc *ccodec) Close() (err error) {
+	log.Println(log.TRACE, "RPC Server Entered: Close")
+	defer log.Println(log.TRACE, "RPC Server Leaving: Close")
+
 	err = cc.conn.Close()
 
 	if err != nil {

diff --git a/rpc/bsonrpc/server.go b/rpc/bsonrpc/server.go
@@ -27,6 +27,9 @@ func NewServerCodec(conn io.ReadWriteCloser) (codec rpc.ServerCodec) {
 }
 
 func (sc *scodec) ReadRequestHeader(rq *rpc.Request) (err error) {
+	log.Println(log.TRACE, "RPC Server Entered: ReadRequestHeader")
+	defer log.Println(log.TRACE, "RPC Server Leaving: ReadRequestHeader")
+
 	err = sc.dec.Decode(rq)
 	if err != nil && err != io.EOF {
 		log.Println(log.ERROR, "RPC Server Error decoding request header: ", err)
@@ -35,6 +38,9 @@ func (sc *scodec) ReadRequestHeader(rq *rpc.Request) (err error) {
 }
 
 func (sc *scodec) ReadRequestBody(v interface{}) (err error) {
+	log.Println(log.TRACE, "RPC Server Entered: ReadRequestBody")
+	defer log.Println(log.TRACE, "RPC Server Leaving: ReadRequestBody")
+
 	err = sc.dec.Decode(v)
 	if err != nil {
 		log.Println(log.ERROR, "RPC Server Error decoding request body: ", err)
@@ -43,6 +49,9 @@ func (sc *scodec) ReadRequestBody(v interface{}) (err error) {
 }
 
 func (sc *scodec) WriteResponse(rs *rpc.Response, v interface{}) (err error) {
+	log.Println(log.TRACE, "RPC Server Entered: WriteResponse")
+	defer log.Println(log.TRACE, "RPC Server Leaving: WriteResponse")
+
 	err = sc.enc.Encode(rs)
 	if err != nil {
 		log.Println(log.ERROR, "RPC Server Error encoding rpc response: ", err)
@@ -57,6 +66,9 @@ func (sc *scodec) WriteResponse(rs *rpc.Response, v interface{}) (err error) {
 }
 
 func (sc *scodec) Close() (err error) {
+	log.Println(log.TRACE, "RPC Server Entered: Close")
+	defer log.Println(log.TRACE, "RPC Server Leaving: Close")
+
 	err = sc.conn.Close()
 	if err != nil {
 		log.Println(log.ERROR, "RPC Server Error closing connection: ", err)