mirror of
https://github.com/jackc/pgx.git
synced 2025-05-31 11:42:24 +00:00
832b4f9771
When a conn is going to execute a query, the first thing it does is to deallocate any invalidated prepared statements from the statement cache. However, the statements were removed from the cache regardless of whether the deallocation succeeded. This would cause subsequent calls of the same SQL to fail with "prepared statement already exists" error. This problem is easy to trigger by running a query with a context that is already canceled. This commit changes the deallocate invalidated cached statements logic so that the statements are only removed from the cache if the deallocation was successful on the server. https://github.com/jackc/pgx/issues/1847
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package stmtcache
|
|
|
|
import (
|
|
"math"
|
|
|
|
"github.com/jackc/pgx/v5/pgconn"
|
|
)
|
|
|
|
// UnlimitedCache implements Cache with no capacity limit.
|
|
type UnlimitedCache struct {
|
|
m map[string]*pgconn.StatementDescription
|
|
invalidStmts []*pgconn.StatementDescription
|
|
}
|
|
|
|
// NewUnlimitedCache creates a new UnlimitedCache.
|
|
func NewUnlimitedCache() *UnlimitedCache {
|
|
return &UnlimitedCache{
|
|
m: make(map[string]*pgconn.StatementDescription),
|
|
}
|
|
}
|
|
|
|
// Get returns the statement description for sql. Returns nil if not found.
|
|
func (c *UnlimitedCache) Get(sql string) *pgconn.StatementDescription {
|
|
return c.m[sql]
|
|
}
|
|
|
|
// Put stores sd in the cache. Put panics if sd.SQL is "". Put does nothing if sd.SQL already exists in the cache.
|
|
func (c *UnlimitedCache) Put(sd *pgconn.StatementDescription) {
|
|
if sd.SQL == "" {
|
|
panic("cannot store statement description with empty SQL")
|
|
}
|
|
|
|
if _, present := c.m[sd.SQL]; present {
|
|
return
|
|
}
|
|
|
|
c.m[sd.SQL] = sd
|
|
}
|
|
|
|
// Invalidate invalidates statement description identified by sql. Does nothing if not found.
|
|
func (c *UnlimitedCache) Invalidate(sql string) {
|
|
if sd, ok := c.m[sql]; ok {
|
|
delete(c.m, sql)
|
|
c.invalidStmts = append(c.invalidStmts, sd)
|
|
}
|
|
}
|
|
|
|
// InvalidateAll invalidates all statement descriptions.
|
|
func (c *UnlimitedCache) InvalidateAll() {
|
|
for _, sd := range c.m {
|
|
c.invalidStmts = append(c.invalidStmts, sd)
|
|
}
|
|
|
|
c.m = make(map[string]*pgconn.StatementDescription)
|
|
}
|
|
|
|
// GetInvalidated returns a slice of all statement descriptions invalidated since the last call to RemoveInvalidated.
|
|
func (c *UnlimitedCache) GetInvalidated() []*pgconn.StatementDescription {
|
|
return c.invalidStmts
|
|
}
|
|
|
|
// RemoveInvalidated removes all invalidated statement descriptions. No other calls to Cache must be made between a
|
|
// call to GetInvalidated and RemoveInvalidated or RemoveInvalidated may remove statement descriptions that were
|
|
// never seen by the call to GetInvalidated.
|
|
func (c *UnlimitedCache) RemoveInvalidated() {
|
|
c.invalidStmts = nil
|
|
}
|
|
|
|
// Len returns the number of cached prepared statement descriptions.
|
|
func (c *UnlimitedCache) Len() int {
|
|
return len(c.m)
|
|
}
|
|
|
|
// Cap returns the maximum number of cached prepared statement descriptions.
|
|
func (c *UnlimitedCache) Cap() int {
|
|
return math.MaxInt
|
|
}
|