client: add call option helper

Signed-off-by: Vasiliy Tolstov <[email protected]>
unistack-org · Jan 18, 2021 · 2b2bcf4 · 2b2bcf4
1 parent 77f517a
commit 2b2bcf4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/api/router/options.go b/api/router/options.go
@@ -5,13 +5,15 @@ import (
 
 	"github.com/unistack-org/micro/v3/api/resolver"
 	"github.com/unistack-org/micro/v3/api/resolver/vpath"
+	"github.com/unistack-org/micro/v3/logger"
 	"github.com/unistack-org/micro/v3/registry"
 )
 
 type Options struct {
 	Handler  string
 	Registry registry.Registry
 	Resolver resolver.Resolver
+	Logger   logger.Logger
 	Context  context.Context
 }
 

diff --git a/client/context.go b/client/context.go
@@ -32,3 +32,13 @@ func SetPublishOption(k, v interface{}) PublishOption {
 		o.Context = context.WithValue(o.Context, k, v)
 	}
 }
+
+// SetCallOption returns a function to setup a context with given value
+func SetCallOption(k, v interface{}) CallOption {
+	return func(o *CallOptions) {
+		if o.Context == nil {
+			o.Context = context.Background()
+		}
+		o.Context = context.WithValue(o.Context, k, v)
+	}
+}