mirror of
https://github.com/pressly/goose.git
synced 2025-09-04 19:37:19 +00:00
71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package dialects
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/pressly/goose/v3/database/dialect"
|
|
)
|
|
|
|
// NewPostgres returns a new [dialect.Querier] for PostgreSQL dialect.
|
|
func NewPostgres() dialect.QuerierExtender {
|
|
return &postgres{}
|
|
}
|
|
|
|
type postgres struct{}
|
|
|
|
var _ dialect.QuerierExtender = (*postgres)(nil)
|
|
|
|
func (p *postgres) CreateTable(tableName string) string {
|
|
q := `CREATE TABLE %s (
|
|
id integer PRIMARY KEY GENERATED BY DEFAULT AS IDENTITY,
|
|
version_id bigint NOT NULL,
|
|
is_applied boolean NOT NULL,
|
|
tstamp timestamp NOT NULL DEFAULT now()
|
|
)`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (p *postgres) InsertVersion(tableName string) string {
|
|
q := `INSERT INTO %s (version_id, is_applied) VALUES ($1, $2)`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (p *postgres) DeleteVersion(tableName string) string {
|
|
q := `DELETE FROM %s WHERE version_id=$1`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (p *postgres) GetMigrationByVersion(tableName string) string {
|
|
q := `SELECT tstamp, is_applied FROM %s WHERE version_id=$1 ORDER BY tstamp DESC LIMIT 1`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (p *postgres) ListMigrations(tableName string) string {
|
|
q := `SELECT version_id, is_applied from %s ORDER BY id DESC`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (p *postgres) GetLatestVersion(tableName string) string {
|
|
q := `SELECT max(version_id) FROM %s`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (p *postgres) TableExists(tableName string) string {
|
|
schemaName, tableName := parseTableIdentifier(tableName)
|
|
if schemaName != "" {
|
|
q := `SELECT EXISTS ( SELECT 1 FROM pg_tables WHERE schemaname = '%s' AND tablename = '%s' )`
|
|
return fmt.Sprintf(q, schemaName, tableName)
|
|
}
|
|
q := `SELECT EXISTS ( SELECT 1 FROM pg_tables WHERE (current_schema() IS NULL OR schemaname = current_schema()) AND tablename = '%s' )`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func parseTableIdentifier(name string) (schema, table string) {
|
|
schema, table, found := strings.Cut(name, ".")
|
|
if !found {
|
|
return "", name
|
|
}
|
|
return schema, table
|
|
}
|