add context helpers

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

network/transport/context.go

@@ -0,0 +1,26 @@
+package transport
+
+import (
+	"context"
+)
+
+type transportKey struct{}
+
+func FromContext(ctx context.Context) (Transport, bool) {
+	c, ok := ctx.Value(transportKey{}).(Transport)
+	return c, ok
+}
+
+func NewContext(ctx context.Context, c Transport) context.Context {
+	return context.WithValue(ctx, transportKey{}, c)
+}
+
+// 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)
+	}
+}