52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package sqlite
|
||
|
||
import (
|
||
"database/sql"
|
||
"fmt"
|
||
"messenger/internal/pkg/logger"
|
||
"time"
|
||
|
||
_ "modernc.org/sqlite"
|
||
)
|
||
|
||
type DB struct {
|
||
*sql.DB
|
||
}
|
||
|
||
func NewDB(dbPath string) (*DB, error) {
|
||
// Подключение к SQLite с оптимизациями
|
||
db, err := sql.Open("sqlite", fmt.Sprintf("%s?_journal=WAL&_foreign_keys=on&_busy_timeout=5000", dbPath))
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to open database: %w", err)
|
||
}
|
||
|
||
// Настройка пула соединений
|
||
db.SetMaxOpenConns(25)
|
||
db.SetMaxIdleConns(10)
|
||
db.SetConnMaxLifetime(5 * time.Minute)
|
||
|
||
// Проверка соединения
|
||
if err := db.Ping(); err != nil {
|
||
return nil, fmt.Errorf("failed to ping database: %w", err)
|
||
}
|
||
|
||
// Автоматически выполняем миграции
|
||
if err := RunMigrations(db, "./migrations"); err != nil {
|
||
logger.Error("Failed to run migrations", "error", err)
|
||
// Не возвращаем ошибку, продолжаем работу
|
||
}
|
||
|
||
logger.Info("SQLite database connected", "path", dbPath)
|
||
|
||
return &DB{DB: db}, nil
|
||
}
|
||
|
||
func (db *DB) Close() error {
|
||
logger.Info("Closing SQLite database connection")
|
||
return db.DB.Close()
|
||
}
|
||
|
||
// BeginTx начинает транзакцию
|
||
func (db *DB) BeginTx() (*sql.Tx, error) {
|
||
return db.DB.Begin()
|
||
} |