mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-23 23:03:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

Reuse existing database connections

Browse source
This commit is contained in:
Till Faelligen 2020-10-05 18:21:05 +02:00
parent eb25ba72ef
commit 12ce66dee2
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
internal/sqlutil/trace.go
View file

@ -96,10 +96,16 @@ func trackGoID(query string) {
logrus.Warnf("unsafe goid: SQL executed not on an ExclusiveWriter: %s", q) logrus.Warnf("unsafe goid: SQL executed not on an ExclusiveWriter: %s", q)
} }
var dbConns = make(map[string]*sql.DB)
// Open opens a database specified by its database driver name and a driver-specific data source name, // Open opens a database specified by its database driver name and a driver-specific data source name,
// usually consisting of at least a database name and connection information. Includes tracing driver // usually consisting of at least a database name and connection information. Includes tracing driver
// if DENDRITE_TRACE_SQL=1 // if DENDRITE_TRACE_SQL=1
func Open(dbProperties *config.DatabaseOptions) (*sql.DB, error) { func Open(dbProperties *config.DatabaseOptions) (*sql.DB, error) {
if conn, ok := dbConns[string(dbProperties.ConnectionString)]; ok {
logrus.Debug("Reusing existing database connection")
return conn, nil
}
var err error var err error
var driverName, dsn string var driverName, dsn string
switch { switch {
@ -134,6 +140,7 @@ func Open(dbProperties *config.DatabaseOptions) (*sql.DB, error) {
db.SetMaxIdleConns(dbProperties.MaxIdleConns()) db.SetMaxIdleConns(dbProperties.MaxIdleConns())
db.SetConnMaxLifetime(dbProperties.ConnMaxLifetime()) db.SetConnMaxLifetime(dbProperties.ConnMaxLifetime())
} }
dbConns[string(dbProperties.ConnectionString)] = db
return db, nil return db, nil
} }