unistack-org/micro-broker-nats
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 8 Wiki Activity

provide setServerSubscriberOption to brokers

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2019-12-03 02:23:42 +03:00
parent e144e143de
commit 2d6163afd0
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
context.go
View File

@ -4,6 +4,7 @@ import (
"context" "context"
"github.com/micro/go-micro/broker" "github.com/micro/go-micro/broker"
"github.com/micro/go-micro/server"
) )
// setSubscribeOption returns a function to setup a context with given value // setSubscribeOption returns a function to setup a context with given value
@ -26,6 +27,16 @@ func setBrokerOption(k, v interface{}) broker.Option {
} }
} }
// setBrokerOption returns a function to setup a context with given value
func setServerSubscriberOption(k, v interface{}) server.SubscriberOption {
return func(o *server.SubscriberOptions) {
if o.Context == nil {
o.Context = context.Background()
}
o.Context = context.WithValue(o.Context, k, v)
}
}
// setPublishOption returns a function to setup a context with given value // setPublishOption returns a function to setup a context with given value
func setPublishOption(k, v interface{}) broker.PublishOption { func setPublishOption(k, v interface{}) broker.PublishOption {
return func(o *broker.PublishOptions) { return func(o *broker.PublishOptions) {