GOMODULES/pgx
1
0
Fork 0
mirror of https://github.com/jackc/pgx.git synced 2025-05-31 11:42:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

have lru.Get() always check if context is already expired

Browse Source
This commit is contained in:
Georges Varouchas 2021-11-08 21:00:24 +01:00 committed by Jack Christensen
parent 141f132ae7
commit cd7dcd5802
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
stmtcache/lru.go
View File

@ -53,6 +53,14 @@ func (c *LRU) Get(ctx context.Context, sql string) (*pgconn.StatementDescription
} }
} }
if ctx != context.Background() {
select {
case <-ctx.Done():
return nil, ctx.Err()
default:
}
}
if el, ok := c.m[sql]; ok { if el, ok := c.m[sql]; ok {
c.l.MoveToFront(el) c.l.MoveToFront(el)
return el.Value.(*pgconn.StatementDescription), nil return el.Value.(*pgconn.StatementDescription), nil