serverrpc.NewServerRPC: set unaryLoger and streamLoger as grpc interceptors

Signed-off-by: Hui Zhu <[email protected]>

Author: teawater

serverrpc/server.go

@@ -3,10 +3,13 @@ package serverrpc
 import (
 	"net"
 
+	"github.com/golang/glog"
 	"github.com/hyperhq/hypercontainer-utils/hlog"
 	"github.com/hyperhq/hyperd/daemon"
 	"github.com/hyperhq/hyperd/types"
+	"golang.org/x/net/context"
 	"google.golang.org/grpc"
+	"time"
 )
 
 // ServerRPC is the main server for gRPC
@@ -15,10 +18,47 @@ type ServerRPC struct {
 	daemon *daemon.Daemon
 }
 
+type re interface {
+	String() string
+}
+
+func unaryLoger(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
+	reqMsg := req.(re).String()
+	glog.V(3).Infof("%s with request %s", info.FullMethod, reqMsg)
+
+	start := time.Now()
+	resp, err = handler(ctx, req)
+	elapsed := time.Now().Sub(start)
+
+	if err == nil {
+		glog.V(3).Infof("%s elapsed %s done %v with request %s", info.FullMethod, elapsed, resp.(re).String(), reqMsg)
+	} else {
+		glog.Errorf("%s elapsed %s failed %v with request %s", info.FullMethod, elapsed, err, reqMsg)
+	}
+
+	return resp, err
+}
+
+func streamLoger(srv interface{}, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
+	glog.V(3).Infof("%s with ServerStream %v", info.FullMethod, ss)
+
+	start := time.Now()
+	err := handler(srv, ss)
+	elapsed := time.Now().Sub(start)
+
+	if err == nil {
+		glog.V(3).Infof("%s elapsed %s done with ServerStream %v", info.FullMethod, elapsed, ss)
+	} else {
+		glog.Errorf("%s elapsed %s failed %v with ServerStream %v", info.FullMethod, elapsed, err, ss)
+	}
+
+	return err
+}
+
 // NewServerRPC creates a new ServerRPC
 func NewServerRPC(d *daemon.Daemon) *ServerRPC {
 	s := &ServerRPC{
-		server: grpc.NewServer(),
+		server: grpc.NewServer(grpc.UnaryInterceptor(unaryLoger), grpc.StreamInterceptor(streamLoger)),
 		daemon: d,
 	}
 	s.registerServer()