Add client interceptors

ofpiyush · Sep 18, 2020 · 2688b1f · 2688b1f
1 parent 9a12f11
commit 2688b1f
Show file tree

Hide file tree

Showing 21 changed files with 1,463 additions and 316 deletions.
diff --git a/client_options.go b/client_options.go
@@ -22,8 +22,9 @@ type ClientOption func(*ClientOptions)
 
 // ClientOptions encapsulate the configurable parameters on a Twirp client.
 type ClientOptions struct {
-	Hooks      *ClientHooks
-	pathPrefix *string
+	Interceptors []Interceptor
+	Hooks        *ClientHooks
+	pathPrefix   *string
 }
 
 func (opts *ClientOptions) PathPrefix() string {
@@ -33,6 +34,13 @@ func (opts *ClientOptions) PathPrefix() string {
 	return *opts.pathPrefix
 }
 
+// WithClientInterceptors defines the interceptors for a Twirp client.
+func WithClientInterceptors(interceptors ...Interceptor) ClientOption {
+	return func(o *ClientOptions) {
+		o.Interceptors = append(o.Interceptors, interceptors...)
+	}
+}
+
 // WithClientHooks defines the hooks for a Twirp client.
 func WithClientHooks(hooks *ClientHooks) ClientOption {
 	return func(o *ClientOptions) {

diff --git a/clientcompat/internal/clientcompat/clientcompat.twirp.go b/clientcompat/internal/clientcompat/clientcompat.twirp.go