mirror of
https://github.com/therootcompany/golib.git
synced 2026-04-24 12:48:00 +00:00
feat(litemigrate): add SQLite backend for sqlmigrate
This commit is contained in:
parent
9614dea7df
commit
9fb5b4eda6
5
database/sqlmigrate/litemigrate/go.mod
Normal file
5
database/sqlmigrate/litemigrate/go.mod
Normal file
@ -0,0 +1,5 @@
|
||||
module github.com/therootcompany/golib/database/sqlmigrate/litemigrate
|
||||
|
||||
go 1.26.1
|
||||
|
||||
require github.com/therootcompany/golib/database/sqlmigrate v1.0.1
|
||||
87
database/sqlmigrate/litemigrate/litemigrate.go
Normal file
87
database/sqlmigrate/litemigrate/litemigrate.go
Normal file
@ -0,0 +1,87 @@
|
||||
// Package litemigrate implements sqlmigrate.Migrator for SQLite
|
||||
// using database/sql. The caller imports the driver:
|
||||
//
|
||||
// import _ "modernc.org/sqlite"
|
||||
//
|
||||
// db, err := sql.Open("sqlite", "app.db?_pragma=foreign_keys(1)")
|
||||
//
|
||||
// SQLite disables foreign key enforcement by default. The _pragma DSN
|
||||
// parameter enables it on every connection the pool opens.
|
||||
package litemigrate
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/therootcompany/golib/database/sqlmigrate"
|
||||
)
|
||||
|
||||
// Migrator implements sqlmigrate.Migrator using a *sql.DB with SQLite.
|
||||
type Migrator struct {
|
||||
DB *sql.DB
|
||||
}
|
||||
|
||||
// New creates a Migrator from the given database handle.
|
||||
func New(db *sql.DB) *Migrator {
|
||||
return &Migrator{DB: db}
|
||||
}
|
||||
|
||||
var _ sqlmigrate.Migrator = (*Migrator)(nil)
|
||||
|
||||
// ExecUp runs the up migration SQL inside a transaction.
|
||||
func (m *Migrator) ExecUp(ctx context.Context, mig sqlmigrate.Migration) error {
|
||||
return m.execInTx(ctx, mig.Up)
|
||||
}
|
||||
|
||||
// ExecDown runs the down migration SQL inside a transaction.
|
||||
func (m *Migrator) ExecDown(ctx context.Context, mig sqlmigrate.Migration) error {
|
||||
return m.execInTx(ctx, mig.Down)
|
||||
}
|
||||
|
||||
func (m *Migrator) execInTx(ctx context.Context, sqlStr string) error {
|
||||
tx, err := m.DB.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("%w: begin: %w", sqlmigrate.ErrExecFailed, err)
|
||||
}
|
||||
defer func() { _ = tx.Rollback() }()
|
||||
|
||||
if _, err := tx.ExecContext(ctx, sqlStr); err != nil {
|
||||
return fmt.Errorf("%w: exec: %w", sqlmigrate.ErrExecFailed, err)
|
||||
}
|
||||
|
||||
if err := tx.Commit(); err != nil {
|
||||
return fmt.Errorf("%w: commit: %w", sqlmigrate.ErrExecFailed, err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Applied returns all applied migrations from the _migrations table.
|
||||
// Returns an empty slice if the table does not exist.
|
||||
func (m *Migrator) Applied(ctx context.Context) ([]sqlmigrate.AppliedMigration, error) {
|
||||
rows, err := m.DB.QueryContext(ctx, "SELECT id, name FROM _migrations ORDER BY name")
|
||||
if err != nil {
|
||||
// SQLite reports "no such table: _migrations" — stable across versions
|
||||
if strings.Contains(err.Error(), "no such table") {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, fmt.Errorf("%w: %w", sqlmigrate.ErrQueryApplied, err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var applied []sqlmigrate.AppliedMigration
|
||||
for rows.Next() {
|
||||
var a sqlmigrate.AppliedMigration
|
||||
if err := rows.Scan(&a.ID, &a.Name); err != nil {
|
||||
return nil, fmt.Errorf("%w: scanning row: %w", sqlmigrate.ErrQueryApplied, err)
|
||||
}
|
||||
applied = append(applied, a)
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, fmt.Errorf("%w: reading rows: %w", sqlmigrate.ErrQueryApplied, err)
|
||||
}
|
||||
|
||||
return applied, nil
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user