mirror of
https://github.com/steveiliop56/tinyauth.git
synced 2026-05-11 06:48:11 +00:00
Scott McKendry
36bfcd45c1
removes the sqlite dependency for tests, also brings back the option for users to run zero persistence instances of tinyauth. adds new mapErr fn for sqlc wrapper gen to prevent sql errors from leaking out of the store implementation.
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/tinyauthapp/tinyauth/internal/assets"
|
|
"github.com/tinyauthapp/tinyauth/internal/repository"
|
|
"github.com/tinyauthapp/tinyauth/internal/repository/memory"
|
|
"github.com/tinyauthapp/tinyauth/internal/repository/sqlite"
|
|
|
|
"github.com/golang-migrate/migrate/v4"
|
|
"github.com/golang-migrate/migrate/v4/database/sqlite3"
|
|
"github.com/golang-migrate/migrate/v4/source/iofs"
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
func (app *BootstrapApp) SetupStore() (repository.Store, error) {
|
|
switch app.config.Database.Driver {
|
|
case "memory":
|
|
return memory.New(), nil
|
|
case "sqlite", "":
|
|
return app.setupSQLite(app.config.Database.Path)
|
|
default:
|
|
return nil, fmt.Errorf("unknown database driver %q: valid values are sqlite, memory", app.config.Database.Driver)
|
|
}
|
|
}
|
|
|
|
func (app *BootstrapApp) setupSQLite(databasePath string) (repository.Store, error) {
|
|
dir := filepath.Dir(databasePath)
|
|
|
|
if err := os.MkdirAll(dir, 0750); err != nil {
|
|
return nil, fmt.Errorf("failed to create database directory %s: %w", dir, err)
|
|
}
|
|
|
|
db, err := sql.Open("sqlite", databasePath)
|
|
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
|
|
// Close the database if there is an error during migration
|
|
defer func() {
|
|
if err != nil {
|
|
db.Close()
|
|
}
|
|
}()
|
|
|
|
// Limit to 1 connection to sequence writes, this may need to be revisited in the future
|
|
// if the sqlite connection starts being a bottleneck
|
|
db.SetMaxOpenConns(1)
|
|
|
|
migrations, err := iofs.New(assets.Migrations, "migrations/sqlite")
|
|
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create migrations: %w", err)
|
|
}
|
|
|
|
target, err := sqlite3.WithInstance(db, &sqlite3.Config{})
|
|
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create sqlite3 instance: %w", err)
|
|
}
|
|
|
|
migrator, err := migrate.NewWithInstance("iofs", migrations, "sqlite3", target)
|
|
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create migrator: %w", err)
|
|
}
|
|
|
|
if err := migrator.Up(); err != nil && err != migrate.ErrNoChange {
|
|
return nil, fmt.Errorf("failed to migrate database: %w", err)
|
|
}
|
|
|
|
app.db = db
|
|
|
|
return sqlite.NewStore(sqlite.New(db)), nil
|
|
}
|