mirror of https://github.com/pressly/goose.git
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
)
|
|
|
|
// SqlDialect abstracts the details of specific SQL dialects
|
|
// for goose's few SQL specific statements
|
|
type SqlDialect interface {
|
|
createVersionTableSql() string // sql string to create the goose_db_version table
|
|
insertVersionSql() string // sql string to insert the initial version table row
|
|
dbVersionQuery(db *sql.DB) (*sql.Rows, error)
|
|
}
|
|
|
|
// drivers that we don't know about can ask for a dialect by name
|
|
func DialectByName(d string) SqlDialect {
|
|
switch d {
|
|
case "postgres":
|
|
return &PostgresDialect{}
|
|
case "mysql":
|
|
return &MySqlDialect{}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
////////////////////////////
|
|
// Postgres
|
|
////////////////////////////
|
|
|
|
type PostgresDialect struct{}
|
|
|
|
func (pg *PostgresDialect) createVersionTableSql() string {
|
|
return `CREATE TABLE goose_db_version (
|
|
id serial NOT NULL,
|
|
version_id bigint NOT NULL,
|
|
is_applied boolean NOT NULL,
|
|
tstamp timestamp NULL default now(),
|
|
PRIMARY KEY(id)
|
|
);`
|
|
}
|
|
|
|
func (pg *PostgresDialect) insertVersionSql() string {
|
|
return "INSERT INTO goose_db_version (version_id, is_applied) VALUES (0, true);"
|
|
}
|
|
|
|
func (pg *PostgresDialect) dbVersionQuery(db *sql.DB) (*sql.Rows, error) {
|
|
rows, err := db.Query("SELECT version_id, is_applied from goose_db_version ORDER BY id DESC")
|
|
|
|
// XXX: check for postgres specific error indicating the table doesn't exist.
|
|
// for now, assume any error is because the table doesn't exist,
|
|
// in which case we'll try to create it.
|
|
if err != nil {
|
|
return nil, ErrTableDoesNotExist
|
|
}
|
|
|
|
return rows, err
|
|
}
|
|
|
|
////////////////////////////
|
|
// MySQL
|
|
////////////////////////////
|
|
|
|
type MySqlDialect struct{}
|
|
|
|
func (m *MySqlDialect) createVersionTableSql() string {
|
|
return `CREATE TABLE goose_db_version (
|
|
id serial NOT NULL,
|
|
version_id bigint NOT NULL,
|
|
is_applied boolean NOT NULL,
|
|
tstamp timestamp NULL default now(),
|
|
PRIMARY KEY(id)
|
|
);`
|
|
}
|
|
|
|
func (m *MySqlDialect) insertVersionSql() string {
|
|
return "INSERT INTO goose_db_version (version_id, is_applied) VALUES (0, true);"
|
|
}
|
|
|
|
func (m *MySqlDialect) dbVersionQuery(db *sql.DB) (*sql.Rows, error) {
|
|
rows, err := db.Query("SELECT version_id, is_applied from goose_db_version ORDER BY id DESC")
|
|
|
|
// XXX: check for mysql specific error indicating the table doesn't exist.
|
|
// for now, assume any error is because the table doesn't exist,
|
|
// in which case we'll try to create it.
|
|
if err != nil {
|
|
return nil, ErrTableDoesNotExist
|
|
}
|
|
|
|
return rows, err
|
|
}
|