migrations

zmigrations/zmigration.go

+package zmigrations
+
+import (
+	"embed"
+	"fmt"
+	"git.barsukov.pro/barsukov/zgo/zdb"
+	"io/fs"
+	"log"
+)
+
+func Migrations(db *zdb.Pool, files embed.FS, loc string, reCreateSchema ...string) error {
+	var err error
+	var f []fs.DirEntry
+
+	if len(reCreateSchema) > 0 && len(reCreateSchema[0]) > 0 {
+		_ = db.WExec(fmt.Sprintf("DROP SCHEMA IF EXISTS %s CASCADE;", reCreateSchema[0]))
+		_ = db.WExec(fmt.Sprintf("CREATE SCHEMA IF NOT EXISTS %s;", reCreateSchema[0]))
+	}
+
+	if err = db.WExec("CREATE TABLE IF NOT EXISTS z_migrations(name text PRIMARY KEY, created_at timestamptz DEFAULT NOW());"); err != nil {
+		return err
+	}
+
+	if f, err = files.ReadDir(loc); err != nil {
+		return err
+	}
+
+	for _, m := range f {
+
+		var s string
+
+		if err = db.Get(&s, "SELECT name FROM z_migrations WHERE name = $1", m.Name()); err != nil {
+			if zdb.IsEmpty(err) {
+				var b []byte
+
+				if b, err = files.ReadFile(loc + m.Name()); err != nil {
+					return err
+				}
+
+				if err = db.WExec(string(b)); err != nil {
+					return fmt.Errorf("MIGRATION: %s - %v", m.Name(), err)
+				}
+
+				if err = db.WExec("INSERT INTO z_migrations(name) VALUES ($1)", m.Name()); err != nil {
+					return err
+				}
+
+				log.Printf("MIGRATION: %s", m.Name())
+			} else {
+				return err
+			}
+		}
+	}
+
+	return nil
+}