mirror of
https://github.com/yanislav-igonin/micrach
synced 2026-07-27 05:14:17 +03:00
db8a46cdf1
* chore: add agents md * docs: expand repository and setup guides * docs: plan project modernization * docs: add Wave 1 implementation plan * test: characterize legacy behavior * build: update Go and Fiber v2 * fix(gateway): preserve duplicate auth behavior * refactor(db): migrate to pgx v5 * refactor: replace deprecated I/O APIs * build: add local PostgreSQL compose * build: add multi-stage image * ci: replace legacy deployment workflow * fix(deps): update vulnerable Go modules * docs: document modern development workflow
104 lines
2.0 KiB
Go
104 lines
2.0 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
Config "micrach/config"
|
|
Files "micrach/files"
|
|
|
|
"github.com/jackc/pgx/v5/pgconn"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
var Pool *pgxpool.Pool
|
|
|
|
type MigrationsMap map[int]string
|
|
type Migration struct {
|
|
ID int
|
|
Name string
|
|
}
|
|
|
|
func Init(ctx context.Context) {
|
|
var err error
|
|
Pool, err = pgxpool.New(ctx, Config.Db.Url)
|
|
if err != nil {
|
|
log.Println("database - offline")
|
|
log.Panicln(err)
|
|
}
|
|
|
|
log.Println("database - online")
|
|
}
|
|
|
|
func Migrate(ctx context.Context) {
|
|
dbMigrations := getDbMigrations(ctx)
|
|
sqlMigrations := Files.GetFullFilePathsInFolder("migrations")
|
|
for _, m := range sqlMigrations {
|
|
filename := filepath.Base(m)
|
|
splitted := strings.Split(filename, "-")
|
|
id, err := strconv.Atoi(splitted[0])
|
|
if err != nil {
|
|
log.Panicln(err)
|
|
}
|
|
// Get name without extension
|
|
name := strings.Split(splitted[1], ".")[0]
|
|
|
|
_, isMigrationInDb := dbMigrations[id]
|
|
if !isMigrationInDb {
|
|
sql := Files.ReadFileText(m)
|
|
runMigration(ctx, id, name, sql)
|
|
log.Println("migration - " + name + " - online")
|
|
}
|
|
}
|
|
|
|
log.Println("migrations - online")
|
|
}
|
|
|
|
func runMigration(ctx context.Context, mid int, mname, msql string) {
|
|
_, err := Pool.Exec(ctx, msql)
|
|
if err != nil {
|
|
log.Panicln(err)
|
|
}
|
|
|
|
_, err = Pool.Exec(ctx, `INSERT INTO migrations (id, name) VALUES ($1, $2)`, mid, mname)
|
|
if err != nil {
|
|
log.Panicln(err)
|
|
}
|
|
}
|
|
|
|
func getDbMigrations(ctx context.Context) MigrationsMap {
|
|
rows, err := Pool.Query(ctx, `SELECT id, name FROM migrations`)
|
|
if err != nil {
|
|
if isUndefinedTable(err) {
|
|
return make(MigrationsMap)
|
|
}
|
|
log.Panicln(err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
migrationsMap := make(MigrationsMap)
|
|
for rows.Next() {
|
|
var m Migration
|
|
err = rows.Scan(&m.ID, &m.Name)
|
|
if err != nil {
|
|
log.Panicln(err)
|
|
}
|
|
|
|
migrationsMap[m.ID] = m.Name
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
log.Panicln(err)
|
|
}
|
|
|
|
return migrationsMap
|
|
}
|
|
|
|
func isUndefinedTable(err error) bool {
|
|
var pgErr *pgconn.PgError
|
|
return errors.As(err, &pgErr) && pgErr.Code == "42P01"
|
|
}
|