diff --git a/broker/memory/memory.go b/broker/memory/memory.go index 1b301baf..9fa1551c 100644 --- a/broker/memory/memory.go +++ b/broker/memory/memory.go @@ -373,6 +373,10 @@ func (m *memoryEvent) SetError(err error) { m.err = err } +func (m *memoryEvent) Context() context.Context { + return m.opts.Context +} + func (m *memorySubscriber) Options() broker.SubscribeOptions { return m.opts } diff --git a/network/tunnel/broker/broker.go b/network/tunnel/broker/broker.go index ea14c1f9..3066bd44 100644 --- a/network/tunnel/broker/broker.go +++ b/network/tunnel/broker/broker.go @@ -305,6 +305,10 @@ func (t *tunEvent) SetError(err error) { t.err = err } +func (t *tunEvent) Context() context.Context { + return context.TODO() +} + // NewBroker returns new tunnel broker func NewBroker(opts ...broker.Option) (broker.Broker, error) { options := broker.NewOptions(opts...)