feat: add migration

internal/persistence/connection.go

@@ -0,0 +1,22 @@
+package persistence
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/jackc/pgx/v5"
+)
+
+func NewConnection() (*pgx.Conn, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
+	defer cancel()
+
+	conn, err := pgx.Connect(ctx, os.Getenv("ORBIS_POSTGRES_DB"))
+	if err != nil {
+		return nil, fmt.Errorf("failed to connect to orbis postgres database: %w", err)
+	}
+
+	return conn, nil
+}

internal/persistence/migrations.go

@@ -0,0 +1,59 @@
+package persistence
+
+import (
+	"database/sql"
+	"embed"
+	"errors"
+	"fmt"
+	"os"
+
+	"github.com/golang-migrate/migrate/v4"
+	migratepgx "github.com/golang-migrate/migrate/v4/database/pgx/v5"
+	"github.com/golang-migrate/migrate/v4/source"
+	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/golang-migrate/migrate/v4/source/iofs"
+)
+
+const migrationSource = "migrations"
+
+//go:embed migrations/*.sql
+var migrations embed.FS
+
+func Migrate() error {
+	db, err := sql.Open("pgx", os.Getenv("ORBIS_POSTGRES_DB"))
+	if err != nil {
+		return fmt.Errorf("failed to establish connection to database: %w", err)
+	}
+
+	driver, err := migratepgx.WithInstance(db, &migratepgx.Config{})
+	if err != nil {
+		return fmt.Errorf("failed to install postgres driver for migrations: %w", err)
+	}
+
+	migrationSource, err := NewEmbedDriver(migrationSource, migrations)
+	if err != nil {
+		return fmt.Errorf("failed to setup embedded driver for migrations: %w", err)
+	}
+
+	migration, err := migrate.NewWithInstance("iofs", migrationSource, "postgres", driver)
+	if err != nil {
+		return err
+	}
+
+	if err := migration.Up(); err != nil {
+		if !errors.Is(err, migrate.ErrNoChange) {
+			return fmt.Errorf("failed to migrate database: %w", err)
+		}
+	}
+
+	return nil
+}
+
+func NewEmbedDriver(path string, files embed.FS) (source.Driver, error) {
+	driver, err := iofs.New(files, migrationSource)
+	if err != nil {
+		return nil, err
+	}
+
+	return driver, nil
+}

internal/persistence/migrations/001_create_initial.up.sql

@@ -0,0 +1,3 @@
+CREATE TABLE test_table (
+    id UUID primary key not null
+);