changes

mysql.go

@@ -0,0 +1,65 @@
+package goblocks
+
+import (
+	"database/sql"
+	"log/slog"
+	"sync"
+	"time"
+
+	_ "github.com/go-sql-driver/mysql"
+)
+
+const maxOpenDbConn = 10
+const maxIdleDbConn = 5
+const maxDbLifetime = time.Minute * 5
+
+var (
+	mysqlDBs   = make(map[string]*sql.DB)
+	mysqlMutex sync.RWMutex
+)
+
+func (a *App) NewMySQL(name string) (*sql.DB, error) {
+	mysqlMutex.Lock()
+	defer mysqlMutex.Unlock()
+
+	if db, exists := mysqlDBs[name]; exists {
+		return db, nil
+	}
+
+	d, err := sql.Open("mysql", a.Datasource(name))
+	if err != nil {
+		slog.Error("error connecting to database", "error", err, "name", name)
+		return nil, err
+	}
+
+	d.SetMaxOpenConns(maxOpenDbConn)
+	d.SetMaxIdleConns(maxIdleDbConn)
+	d.SetConnMaxLifetime(maxDbLifetime)
+
+	if err := d.Ping(); err != nil {
+		slog.Error("error pinging database", "error", err, "name", name)
+		return nil, err
+	}
+
+	mysqlDBs[name] = d
+	slog.Info("connected to database", "name", name)
+	return d, nil
+}
+
+func (a *App) GetMySQL(name string) (*sql.DB, bool) {
+	mysqlMutex.RLock()
+	defer mysqlMutex.RUnlock()
+	db, exists := mysqlDBs[name]
+	return db, exists
+}
+
+func (a *App) CloseMySQLDBs() {
+	mysqlMutex.Lock()
+	defer mysqlMutex.Unlock()
+
+	for name, db := range mysqlDBs {
+		db.Close()
+		delete(mysqlDBs, name)
+		slog.Info("closed database connection", "name", name)
+	}
+}