From 4783c6d9a33b40b140e0f93ae6a4dfd28a593df7 Mon Sep 17 00:00:00 2001 From: Vasiliy Tolstov Date: Tue, 19 Jan 2021 00:45:55 +0300 Subject: [PATCH] client: add option helper Signed-off-by: Vasiliy Tolstov --- client/context.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/client/context.go b/client/context.go index 01b03a32..f4da4464 100644 --- a/client/context.go +++ b/client/context.go @@ -42,3 +42,13 @@ func SetCallOption(k, v interface{}) CallOption { o.Context = context.WithValue(o.Context, k, v) } } + +// SetOption returns a function to setup a context with given value +func SetOption(k, v interface{}) Option { + return func(o *Options) { + if o.Context == nil { + o.Context = context.Background() + } + o.Context = context.WithValue(o.Context, k, v) + } +}