diff --git a/pkg/database/client.go b/pkg/database/client.go new file mode 100644 index 0000000..89ff2ac --- /dev/null +++ b/pkg/database/client.go @@ -0,0 +1,68 @@ +package database + +import ( + "context" + "embed" + "fmt" + "git.anthrove.art/Anthrove/otter-space-sdk/v2/pkg/models" + migrate "github.com/rubenv/sql-migrate" + log "github.com/sirupsen/logrus" + "gorm.io/driver/postgres" + "gorm.io/gorm" +) + +//go:embed migrations/*.sql +var embedMigrations embed.FS +var client *gorm.DB + +func Connect(_ context.Context, config models.DatabaseConfig) error { + var localSSL string + + if config.SSL { + localSSL = "require" + } else { + localSSL = "disable" + } + + dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=%s", config.Endpoint, config.Username, config.Password, config.Database, config.Port, localSSL, config.Timezone) + sqlDB, err := gorm.Open(postgres.Open(dsn), &gorm.Config{}) + + if err != nil { + return err + } + + client = sqlDB + err = migrateDatabase(sqlDB, config) + if err != nil { + return err + } + + return nil +} + +func migrateDatabase(dbPool *gorm.DB, config models.DatabaseConfig) error { + dialect := "postgres" + migrations := &migrate.EmbedFileSystemMigrationSource{FileSystem: embedMigrations, Root: "migrations"} + + db, err := dbPool.DB() + if err != nil { + return fmt.Errorf("postgres migration: %v", err) + } + + n, err := migrate.Exec(db, dialect, migrations, migrate.Up) + if err != nil { + return fmt.Errorf("postgres migration: %v", err) + } + + if config.Debug { + if n != 0 { + log.Infof("postgres migration: applied %d migrations!", n) + + } else { + log.Info("postgres migration: nothing to migrate") + + } + } + + return nil +}