tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename lru store

Browse Source
This commit is contained in:
Jinzhu 2025-04-25 11:33:15 +08:00
parent 14dc8ed9e0
commit 2ec55b55c4
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
internal/stmt_store/stmt_store.go
View File

@ -37,10 +37,6 @@ type Store interface {
Delete(key string) Delete(key string)
} }
type LRUStore struct {
lru *lru.LRU[string, *Stmt]
}
const ( const (
defaultMaxSize = (1 << 63) - 1 defaultMaxSize = (1 << 63) - 1
defaultTTL = time.Hour * 24 defaultTTL = time.Hour * 24
@ -60,14 +56,18 @@ func New(size int, ttl time.Duration) Store {
go v.Close() go v.Close()
} }
} }
return &LRUStore{lru: lru.NewLRU[string, *Stmt](size, onEvicted, ttl)} return &lruStore{lru: lru.NewLRU[string, *Stmt](size, onEvicted, ttl)}
} }
func (s *LRUStore) Keys() []string { type lruStore struct {
lru *lru.LRU[string, *Stmt]
}
func (s *lruStore) Keys() []string {
return s.lru.Keys() return s.lru.Keys()
} }
func (s *LRUStore) Get(key string) (*Stmt, bool) { func (s *lruStore) Get(key string) (*Stmt, bool) {
stmt, ok := s.lru.Get(key) stmt, ok := s.lru.Get(key)
if ok && stmt != nil { if ok && stmt != nil {
<-stmt.prepared <-stmt.prepared
@ -75,11 +75,11 @@ func (s *LRUStore) Get(key string) (*Stmt, bool) {
return stmt, ok return stmt, ok
} }
func (s *LRUStore) Set(key string, value *Stmt) { func (s *lruStore) Set(key string, value *Stmt) {
s.lru.Add(key, value) s.lru.Add(key, value)
} }
func (s *LRUStore) Delete(key string) { func (s *lruStore) Delete(key string) {
s.lru.Remove(key) s.lru.Remove(key)
} }
@ -87,7 +87,7 @@ type ConnPool interface {
PrepareContext(ctx context.Context, query string) (*sql.Stmt, error) PrepareContext(ctx context.Context, query string) (*sql.Stmt, error)
} }
func (s *LRUStore) New(ctx context.Context, key string, isTransaction bool, conn ConnPool, locker sync.Locker) (_ *Stmt, err error) { func (s *lruStore) New(ctx context.Context, key string, isTransaction bool, conn ConnPool, locker sync.Locker) (_ *Stmt, err error) {
cacheStmt := &Stmt{ cacheStmt := &Stmt{
Transaction: isTransaction, Transaction: isTransaction,
prepared: make(chan struct{}), prepared: make(chan struct{}),