x/dal
1
1
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

[wip] help gc (hopefully) after closing connections

Browse source 
Signed-off-by: Anton Nesterov <anton@demiurg.io>
This commit is contained in:
Anton Nesterov 2024-08-12 15:53:12 +02:00
parent b1f9cead34
commit 0648807cb2
No known key found for this signature in database
GPG key ID: 59121E8AE2851FB5
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
pkg/adapter/DBAdapter.go
View file

@ -31,9 +31,11 @@ func (a *DBAdapter) Open(url string) (*sql.DB, error) {
if a.MaxAttempts == 0 { if a.MaxAttempts == 0 {
a.MaxAttempts = 6 a.MaxAttempts = 6
} }
if a.ConnectionLiveTime == 0 { if a.ConnectionLiveTime == 0 {
a.ConnectionLiveTime = 120 a.ConnectionLiveTime = 120
} }
if a.dbs == nil { if a.dbs == nil {
a.dbs = &DBMap{ a.dbs = &DBMap{
Connections: make(map[string]*sql.DB), Connections: make(map[string]*sql.DB),
@ -65,8 +67,9 @@ func (a *DBAdapter) Open(url string) (*sql.DB, error) {
return a.Open(url) return a.Open(url)
} }
} }
a.dbs.ConnectionAttempts[url] = 0
return a.dbs.Connections[url], nil attempts[url] = 0
return connections[url], nil
} }
func (a *DBAdapter) GetDB(url string) *sql.DB { func (a *DBAdapter) GetDB(url string) *sql.DB {
@ -77,8 +80,11 @@ func (a *DBAdapter) GetDB(url string) *sql.DB {
} }
func (a *DBAdapter) Close() { func (a *DBAdapter) Close() {
for _, db := range a.dbs.Connections { for url, db := range a.dbs.Connections {
db.Close() db.Close()
delete(a.dbs.Connections, url)
delete(a.dbs.ConnectionAttempts, url)
delete(a.dbs.ConnectionTime, url)
} }
a.dbs = nil a.dbs = nil
} }