mirror of
https://github.com/J3vb/OwnCord.git
synced 2026-09-03 03:50:00 +03:00
jevb
ce4326766a
Fix 70+ errcheck violations by adding explicit error discards (`_ =`) for unchecked return values across test helpers and deferred Close() calls. Remove unused `senderID` field from broadcastMsg and unused `defaultCleanupMaxWindow` const. Apply De Morgan's law, remove empty branch, and simplify redundant type declaration per staticcheck.
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
// Package db provides database access for the OwnCord server.
|
|
// It uses modernc.org/sqlite — a pure-Go SQLite driver requiring no CGO.
|
|
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/owncord/server/migrations"
|
|
_ "modernc.org/sqlite" // register the sqlite3 driver
|
|
)
|
|
|
|
// DB wraps *sql.DB and exposes the subset of methods needed by the server.
|
|
type DB struct {
|
|
sqlDB *sql.DB
|
|
}
|
|
|
|
// Open opens (or creates) a SQLite database at path, enables WAL mode and
|
|
// foreign key enforcement, and returns a ready-to-use DB.
|
|
func Open(path string) (*DB, error) {
|
|
sqlDB, err := sql.Open("sqlite", path)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("opening sqlite db: %w", err)
|
|
}
|
|
|
|
// Verify the connection is actually usable.
|
|
if err := sqlDB.Ping(); err != nil {
|
|
_ = sqlDB.Close()
|
|
return nil, fmt.Errorf("pinging sqlite db: %w", err)
|
|
}
|
|
|
|
// In-memory databases are per-connection in SQLite; pin to one connection
|
|
// so all callers share the same in-memory state.
|
|
if path == ":memory:" {
|
|
sqlDB.SetMaxOpenConns(1)
|
|
}
|
|
|
|
// Enable WAL mode for better concurrent read performance.
|
|
if _, err := sqlDB.Exec("PRAGMA journal_mode=WAL;"); err != nil {
|
|
_ = sqlDB.Close()
|
|
return nil, fmt.Errorf("enabling WAL mode: %w", err)
|
|
}
|
|
|
|
// Enforce foreign key constraints.
|
|
if _, err := sqlDB.Exec("PRAGMA foreign_keys=ON;"); err != nil {
|
|
_ = sqlDB.Close()
|
|
return nil, fmt.Errorf("enabling foreign keys: %w", err)
|
|
}
|
|
|
|
return &DB{sqlDB: sqlDB}, nil
|
|
}
|
|
|
|
// Migrate runs all SQL migration files from the embedded migrations FS in
|
|
// lexicographic order, applying each file exactly once. It delegates to
|
|
// MigrateFS (defined in migrate.go) which maintains the schema_versions
|
|
// tracking table.
|
|
func Migrate(database *DB) error {
|
|
return MigrateFS(database, migrations.FS)
|
|
}
|
|
|
|
// Close releases the underlying database connection.
|
|
func (d *DB) Close() error {
|
|
return d.sqlDB.Close()
|
|
}
|
|
|
|
// QueryRow executes a query that returns at most one row.
|
|
func (d *DB) QueryRow(query string, args ...any) *sql.Row {
|
|
return d.sqlDB.QueryRow(query, args...)
|
|
}
|
|
|
|
// Exec executes a query that doesn't return rows.
|
|
func (d *DB) Exec(query string, args ...any) (sql.Result, error) {
|
|
return d.sqlDB.Exec(query, args...)
|
|
}
|
|
|
|
// Query executes a query that returns multiple rows.
|
|
func (d *DB) Query(query string, args ...any) (*sql.Rows, error) {
|
|
return d.sqlDB.Query(query, args...)
|
|
}
|
|
|
|
// Begin starts a database transaction.
|
|
func (d *DB) Begin() (*sql.Tx, error) {
|
|
return d.sqlDB.Begin()
|
|
}
|
|
|
|
// SQLDb returns the underlying *sql.DB for cases requiring direct access.
|
|
func (d *DB) SQLDb() *sql.DB {
|
|
return d.sqlDB
|
|
}
|