gRPC logrus interceptor return log warn if error

dynastymasra · Feb 14, 2020 · f4c7ee4 · f4c7ee4
1 parent a4583b6
commit f4c7ee4
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/grpc/interceptor.go b/grpc/interceptor.go
@@ -88,6 +88,16 @@ func LogrusUnaryInterceptor(logger *logrus.Entry) grpc.UnaryServerInterceptor {
 		responseTime := time.Now()
 		deltaTime := responseTime.Sub(startTime)
 
+		if err != nil {
+			log.WithFields(logrus.Fields{
+				"response_time": responseTime.Format(time.RFC3339),
+				"delta_time":    deltaTime,
+				"error":         err,
+			}).Warnln("gRPC request")
+
+			return resp, err
+		}
+
 		log.WithFields(logrus.Fields{
 			"response_time": responseTime.Format(time.RFC3339),
 			"delta_time":    deltaTime,
@@ -116,6 +126,16 @@ func LogrusStreamInterceptor(logger *logrus.Entry) grpc.StreamServerInterceptor
 		responseTime := time.Now()
 		deltaTime := responseTime.Sub(startTime)
 
+		if err != nil {
+			log.WithFields(logrus.Fields{
+				"response_time": responseTime.Format(time.RFC3339),
+				"delta_time":    deltaTime,
+				"error":         err,
+			}).Warnln("gRPC request")
+
+			return err
+		}
+
 		log.WithFields(logrus.Fields{
 			"response_time": responseTime.Format(time.RFC3339),
 			"delta_time":    deltaTime,