session+firewall: pass context to GetSessionByID

This commit is contained in:
Elle Mouton 2025-02-25 13:31:10 +02:00
parent aa5674cabd
commit 9642ce1656
No known key found for this signature in database
GPG key ID: D7D916376026F177
8 changed files with 24 additions and 17 deletions

View file

@ -190,7 +190,7 @@ func (p *PrivacyMapper) checkAndReplaceIncomingRequest(ctx context.Context,
uri string, req proto.Message, sessionID session.ID) (proto.Message,
error) {
session, err := p.sessionDB.GetSessionByID(sessionID)
session, err := p.sessionDB.GetSessionByID(ctx, sessionID)
if err != nil {
return nil, err
}
@ -220,7 +220,7 @@ func (p *PrivacyMapper) checkAndReplaceIncomingRequest(ctx context.Context,
func (p *PrivacyMapper) replaceOutgoingResponse(ctx context.Context, uri string,
resp proto.Message, sessionID session.ID) (proto.Message, error) {
session, err := p.sessionDB.GetSessionByID(sessionID)
session, err := p.sessionDB.GetSessionByID(ctx, sessionID)
if err != nil {
return nil, err
}

View file

@ -386,7 +386,7 @@ func (r *RuleEnforcer) initRule(ctx context.Context, reqID uint64, name string,
return nil, err
}
session, err := r.sessionDB.GetSessionByID(sessionID)
session, err := r.sessionDB.GetSessionByID(ctx, sessionID)
if err != nil {
return nil, err
}