Merge pull request #55 from philippseith/feature/Add_HubContext.Context

new: Hub.Context / HubContext.Context
philippseith · Aug 30, 2021 · 87ae82d · 87ae82d
2 parents 652a4bd + 18b1170
commit 87ae82d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/hub.go b/hub.go
@@ -1,6 +1,9 @@
 package signalr
 
-import "sync"
+import (
+	"context"
+	"sync"
+)
 
 // HubInterface is a hubs interface
 type HubInterface interface {
@@ -42,6 +45,11 @@ func (h *Hub) ConnectionID() string {
 	return h.context.ConnectionID()
 }
 
+// Context is the context.Context of the current connection
+func (h *Hub) Context() context.Context {
+	return h.context.Context()
+}
+
 // Abort aborts the current connection
 func (h *Hub) Abort() {
 	h.context.Abort()

diff --git a/hubcontext.go b/hubcontext.go
@@ -17,6 +17,7 @@ type HubContext interface {
 	Groups() GroupManager
 	Items() *sync.Map
 	ConnectionID() string
+	Context() context.Context
 	Abort()
 	Logger() (info StructuredLogger, dbg StructuredLogger)
 }
@@ -46,6 +47,10 @@ func (c *connectionHubContext) ConnectionID() string {
 	return c.connection.ConnectionID()
 }
 
+func (c *connectionHubContext) Context() context.Context {
+	return c.connection.Context()
+}
+
 func (c *connectionHubContext) Abort() {
 	c.abort()
 }