Refactor context handling in HandleCount and HandleReq functions
- Updated context creation in HandleCount and HandleReq to use context.Background() instead of the connection context, isolating timeouts to prevent affecting long-lived websocket connections. - Improved comments for clarity on the purpose of the context changes. - bump version to v0.17.17
This commit is contained in:
@@ -25,7 +25,7 @@ func (l *Listener) HandleCount(msg []byte) (err error) {
|
|||||||
if _, err = env.Unmarshal(msg); chk.E(err) {
|
if _, err = env.Unmarshal(msg); chk.E(err) {
|
||||||
return normalize.Error.Errorf(err.Error())
|
return normalize.Error.Errorf(err.Error())
|
||||||
}
|
}
|
||||||
log.D.C(func() string { return fmt.Sprintf("COUNT sub=%s filters=%d", env.Subscription, len(env.Filters)) })
|
log.D.C(func() string { return fmt.Sprintf("COUNT sub=%s filters=%d", env.Subscription, len(env.Filters)) })
|
||||||
|
|
||||||
// If ACL is active, send a challenge (same as REQ path)
|
// If ACL is active, send a challenge (same as REQ path)
|
||||||
if acl.Registry.Active.Load() != "none" {
|
if acl.Registry.Active.Load() != "none" {
|
||||||
@@ -43,14 +43,15 @@ func (l *Listener) HandleCount(msg []byte) (err error) {
|
|||||||
// allowed to read
|
// allowed to read
|
||||||
}
|
}
|
||||||
|
|
||||||
// Use a bounded context for counting
|
// Use a bounded context for counting, isolated from the connection context
|
||||||
ctx, cancel := context.WithTimeout(l.ctx, 30*time.Second)
|
// to prevent count timeouts from affecting the long-lived websocket connection
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
// Aggregate count across all provided filters
|
// Aggregate count across all provided filters
|
||||||
var total int
|
var total int
|
||||||
var approx bool // database returns false per implementation
|
var approx bool // database returns false per implementation
|
||||||
for _, f := range env.Filters {
|
for _, f := range env.Filters {
|
||||||
if f == nil {
|
if f == nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -86,9 +86,10 @@ func (l *Listener) HandleReq(msg []byte) (err error) {
|
|||||||
// user has read access or better, continue
|
// user has read access or better, continue
|
||||||
}
|
}
|
||||||
var events event.S
|
var events event.S
|
||||||
// Create a single context for all filter queries, tied to the connection context, to prevent leaks and support timely cancellation
|
// Create a single context for all filter queries, isolated from the connection context
|
||||||
|
// to prevent query timeouts from affecting the long-lived websocket connection
|
||||||
queryCtx, queryCancel := context.WithTimeout(
|
queryCtx, queryCancel := context.WithTimeout(
|
||||||
l.ctx, 30*time.Second,
|
context.Background(), 30*time.Second,
|
||||||
)
|
)
|
||||||
defer queryCancel()
|
defer queryCancel()
|
||||||
|
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
v0.17.16
|
v0.17.17
|
||||||
Reference in New Issue
Block a user