parent
e5725dc249
commit
c70325c8c9
10
db.go
10
db.go
|
@ -58,10 +58,10 @@ func (t *LibSqlDB) Migrate() error {
|
||||||
// check if migration table exists
|
// check if migration table exists
|
||||||
var migrationsCheck string
|
var migrationsCheck string
|
||||||
//goland:noinspection SqlResolve
|
//goland:noinspection SqlResolve
|
||||||
err := t.db.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name='migrations'").Scan(&migrationsCheck)
|
err := t.DB.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name='migrations'").Scan(&migrationsCheck)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
_, err := t.db.Exec("CREATE TABLE migrations (name TEXT NOT NULL)")
|
_, err := t.DB.Exec("CREATE TABLE migrations (name TEXT NOT NULL)")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("error creating migrations table | %w", err)
|
return fmt.Errorf("error creating migrations table | %w", err)
|
||||||
}
|
}
|
||||||
|
@ -72,14 +72,14 @@ func (t *LibSqlDB) Migrate() error {
|
||||||
|
|
||||||
for _, migration := range migrations {
|
for _, migration := range migrations {
|
||||||
var migrationInHistory string
|
var migrationInHistory string
|
||||||
err = t.db.QueryRow("SELECT name FROM migrations WHERE name = ?", migration.name).Scan(&migrationInHistory)
|
err = t.DB.QueryRow("SELECT name FROM migrations WHERE name = ?", migration.name).Scan(&migrationInHistory)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
_, err := t.db.Exec(migration.query)
|
_, err := t.DB.Exec(migration.query)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("error running migration: %s | %w", migration.name, err)
|
return fmt.Errorf("error running migration: %s | %w", migration.name, err)
|
||||||
}
|
}
|
||||||
_, err = t.db.Exec("INSERT INTO migrations (name) VALUES (?)", migration.name)
|
_, err = t.DB.Exec("INSERT INTO migrations (name) VALUES (?)", migration.name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("error inserting migration: %s into migrations table | %w", migration.name, err)
|
return fmt.Errorf("error inserting migration: %s into migrations table | %w", migration.name, err)
|
||||||
}
|
}
|
||||||
|
|
|
@ -76,7 +76,7 @@ func NewLibSqlDB(primaryUrl string, migrationFiles embed.FS, opts ...Options) (*
|
||||||
return nil, fmt.Errorf("error setting up migrations | %w", err)
|
return nil, fmt.Errorf("error setting up migrations | %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
l.db = db
|
l.DB = db
|
||||||
l.connector = connector
|
l.connector = connector
|
||||||
l.dir = dir
|
l.dir = dir
|
||||||
|
|
||||||
|
|
|
@ -51,14 +51,14 @@ func NewLibSqlDB(primaryUrl string, migrationFiles embed.FS, opts ...Options) (*
|
||||||
}
|
}
|
||||||
|
|
||||||
return &LibSqlDB{
|
return &LibSqlDB{
|
||||||
db: db,
|
DB: db,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *LibSqlDB) Close() error {
|
func (t *LibSqlDB) Close() error {
|
||||||
var resultError *multierror.Error
|
var resultError *multierror.Error
|
||||||
|
|
||||||
if err := t.db.Close(); err != nil {
|
if err := t.DB.Close(); err != nil {
|
||||||
resultError = multierror.Append(resultError, fmt.Errorf("failed to close database: %w", err))
|
resultError = multierror.Append(resultError, fmt.Errorf("failed to close database: %w", err))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue