75
storage/postgres/postgres.go
Normal file
75
storage/postgres/postgres.go
Normal file
@@ -0,0 +1,75 @@
|
||||
// go:build postgres
|
||||
|
||||
package postgres
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"embed"
|
||||
|
||||
"github.com/golang-migrate/migrate/v4"
|
||||
"github.com/golang-migrate/migrate/v4/database/postgres"
|
||||
"github.com/golang-migrate/migrate/v4/source/iofs"
|
||||
)
|
||||
|
||||
type PostgresStorage struct {
|
||||
db *sql.DB
|
||||
fs embed.FS
|
||||
}
|
||||
|
||||
func NewStoragePostgres(fs embed.FS) func(*sql.DB) (interface{}, error) {
|
||||
return func(db *sql.DB) (interface{}, error) {
|
||||
return &PostgresStorage{db: db, fs: fs}, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (s *PostgresStorage) MigrateUp() error {
|
||||
driver, err := postgres.WithInstance(s.db, &postgres.Config{
|
||||
MigrationsTable: postgres.DefaultMigrationsTable,
|
||||
DatabaseName: "template",
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
source, err := iofs.New(s.fs, "migrations/postgres")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// TODO: pass own logger
|
||||
m, err := migrate.NewWithInstance("fs", source, "template", driver)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err = m.Up(); err != nil && err != migrate.ErrNoChange {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *PostgresStorage) MigrateDown() error {
|
||||
driver, err := postgres.WithInstance(s.db, &postgres.Config{
|
||||
MigrationsTable: postgres.DefaultMigrationsTable,
|
||||
DatabaseName: "template",
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
source, err := iofs.New(s.fs, "migrations/postgres")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// TODO: pass own logger
|
||||
m, err := migrate.NewWithInstance("fs", source, "template", driver)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err = m.Down(); err != nil && err != migrate.ErrNoChange {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
5
storage/postgres/query.go
Normal file
5
storage/postgres/query.go
Normal file
@@ -0,0 +1,5 @@
|
||||
package postgres
|
||||
|
||||
const (
|
||||
// TODO: fill queries
|
||||
)
|
Reference in New Issue
Block a user