mirror of https://github.com/pressly/goose.git
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package dialectquery
|
|
|
|
import "fmt"
|
|
|
|
type Ydb struct{}
|
|
|
|
var _ Querier = (*Ydb)(nil)
|
|
|
|
func (c *Ydb) CreateTable(tableName string) string {
|
|
q := `CREATE TABLE %s (
|
|
version_id Uint64,
|
|
is_applied Bool,
|
|
tstamp Timestamp,
|
|
|
|
PRIMARY KEY(version_id)
|
|
)`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (c *Ydb) InsertVersion(tableName string) string {
|
|
q := `INSERT INTO %s (
|
|
version_id,
|
|
is_applied,
|
|
tstamp
|
|
) VALUES (
|
|
CAST($1 AS Uint64),
|
|
$2,
|
|
CurrentUtcTimestamp()
|
|
)`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (c *Ydb) DeleteVersion(tableName string) string {
|
|
q := `DELETE FROM %s WHERE version_id = $1`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (c *Ydb) 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 (c *Ydb) ListMigrations(tableName string) string {
|
|
q := `
|
|
SELECT version_id, is_applied, tstamp AS __discard_column_tstamp
|
|
FROM %s ORDER BY __discard_column_tstamp DESC`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|
|
|
|
func (c *Ydb) GetLatestVersion(tableName string) string {
|
|
q := `SELECT MAX(version_id) FROM %s`
|
|
return fmt.Sprintf(q, tableName)
|
|
}
|