Feat: migration on startup

1 files changed, 59 insertions, 0 deletions

diff --git a/storage/migrate.go b/storage/migrate.go
new file mode 100644
index 0000000..d97b99d
--- /dev/null
+++ b/storage/migrate.go
@@ -0,0 +1,59 @@
+package storage
+
+import (
+	"embed"
+	"fmt"
+	"io/fs"
+	"strings"
+)
+
+//go:embed migrations/*
+var migrationsFS embed.FS
+
+func (p *ProviderSQL) Migrate() {
+	// Get the embedded filesystem
+	migrationsDir, err := fs.Sub(migrationsFS, "migrations")
+	if err != nil {
+		p.logger.Error("Failed to get embedded migrations directory;", "error", err)
+	}
+	// List all .up.sql files
+	files, err := migrationsFS.ReadDir("migrations")
+	if err != nil {
+		p.logger.Error("Failed to read migrations directory;", "error", err)
+	}
+	// Execute each .up.sql file
+	for _, file := range files {
+		if strings.HasSuffix(file.Name(), ".up.sql") {
+			err := p.executeMigration(migrationsDir, file.Name())
+			if err != nil {
+				p.logger.Error("Failed to execute migration %s: %v", file.Name(), err)
+			}
+		}
+	}
+	p.logger.Info("All migrations executed successfully!")
+}
+
+func (p *ProviderSQL) executeMigration(migrationsDir fs.FS, fileName string) error {
+	// Open the migration file
+	migrationFile, err := migrationsDir.Open(fileName)
+	if err != nil {
+		return fmt.Errorf("failed to open migration file %s: %w", fileName, err)
+	}
+	defer migrationFile.Close()
+	// Read the migration file content
+	migrationContent, err := fs.ReadFile(migrationsDir, fileName)
+	if err != nil {
+		return fmt.Errorf("failed to read migration file %s: %w", fileName, err)
+	}
+	// Execute the migration content
+	return p.executeSQL(migrationContent)
+}
+
+func (p *ProviderSQL) executeSQL(sqlContent []byte) error {
+	// Connect to the database (example using a simple connection)
+	_, err := p.db.Exec(string(sqlContent))
+	if err != nil {
+		return fmt.Errorf("failed to execute SQL: %w", err)
+	}
+	return nil
+}