feat(grpc): add client interceptor option

purposeinplay · Jan 30, 2023 · bbfb852 · bbfb852
1 parent a1e44dc
commit bbfb852
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/.golangci.yml b/.golangci.yml
@@ -102,7 +102,7 @@ linters-settings:
       - name: max-public-structs
         disabled: true
       - name: line-length-limit
-        arguments: [ 80 ]
+        arguments: [ 100 ]
       - name: argument-limit
         disabled: true
       - name: cyclomatic

diff --git a/grpc/grpcclient/grpc_client_conn.go b/grpc/grpcclient/grpc_client_conn.go
@@ -82,3 +82,13 @@ func WithContextDialer(
 		)
 	})
 }
+
+// WithClientUnaryInterceptor adds an interceptor for client calls.
+func WithClientUnaryInterceptor(interceptor grpc.UnaryClientInterceptor) OptionConn {
+	return newFuncConnOption(func(o *connOptions) {
+		o.dialOptions = append(
+			o.dialOptions,
+			grpc.WithUnaryInterceptor(interceptor),
+		)
+	})
+}