diff --git a/pkg/database/migrate/201402211147_github_enterprise_support.go b/pkg/database/migrate/201402211147_github_enterprise_support.go index 273786772..dcdd5e8f1 100644 --- a/pkg/database/migrate/201402211147_github_enterprise_support.go +++ b/pkg/database/migrate/201402211147_github_enterprise_support.go @@ -14,6 +14,9 @@ func (r *Rev3) Up(op Operation) error { return err } _, err = op.AddColumn("settings", "github_apiurl VARCHAR(255)") + + op.Exec("update settings set github_domain=?", "github.com") + op.Exec("update settings set github_apiurl=?", "https://api.github.com") return err } diff --git a/pkg/database/migrate/all.go b/pkg/database/migrate/all.go index 3933a93ca..ec5facdca 100644 --- a/pkg/database/migrate/all.go +++ b/pkg/database/migrate/all.go @@ -4,6 +4,7 @@ func (m *Migration) All() *Migration { // List all migrations here m.Add(RenamePrivelegedToPrivileged) + m.Add(GitHubEnterpriseSupport) // m.Add(...) // ... diff --git a/pkg/database/migrate/migrate.go b/pkg/database/migrate/migrate.go index f4a286d7b..63cef3914 100644 --- a/pkg/database/migrate/migrate.go +++ b/pkg/database/migrate/migrate.go @@ -64,6 +64,12 @@ type Operation interface { DropColumns(tableName string, columnsToDrop []string) (sql.Result, error) RenameColumns(tableName string, columnChanges map[string]string) (sql.Result, error) + + Exec(query string, args ...interface{}) (sql.Result, error) + + Query(query string, args ...interface{}) (*sql.Rows, error) + + QueryRow(query string, args ...interface{}) *sql.Row } type Revision interface { diff --git a/pkg/database/migrate/sqlite.go b/pkg/database/migrate/sqlite.go index 512a1e2af..140988a47 100644 --- a/pkg/database/migrate/sqlite.go +++ b/pkg/database/migrate/sqlite.go @@ -15,6 +15,18 @@ func SQLite(tx *sql.Tx) Operation { return &SQLiteDriver{Tx: tx} } +func (s *SQLiteDriver) Exec(query string, args ...interface{}) (sql.Result, error) { + return s.Tx.Exec(query, args) +} + +func (s *SQLiteDriver) Query(query string, args ...interface{}) (*sql.Rows, error) { + return s.Tx.Query(query, args) +} + +func (s *SQLiteDriver) QueryRow(query string, args ...interface{}) *sql.Row { + return s.Tx.QueryRow(query, args) +} + func (s *SQLiteDriver) CreateTable(tableName string, args []string) (sql.Result, error) { return s.Tx.Exec(fmt.Sprintf("CREATE TABLE %s (%s);", tableName, strings.Join(args, ", "))) }