527 lines
12 KiB
Go
527 lines
12 KiB
Go
package postgres
|
|
|
|
import (
|
|
"database/sql"
|
|
)
|
|
|
|
var migrations = []struct {
|
|
name string
|
|
stmt string
|
|
}{
|
|
{
|
|
name: "create-table-users",
|
|
stmt: createTableUsers,
|
|
},
|
|
{
|
|
name: "create-table-repos",
|
|
stmt: createTableRepos,
|
|
},
|
|
{
|
|
name: "alter-table-repos-add-column-no-fork",
|
|
stmt: alterTableReposAddColumnNoFork,
|
|
},
|
|
{
|
|
name: "alter-table-repos-add-column-no-pulls",
|
|
stmt: alterTableReposAddColumnNoPulls,
|
|
},
|
|
{
|
|
name: "create-table-perms",
|
|
stmt: createTablePerms,
|
|
},
|
|
{
|
|
name: "create-index-perms-user",
|
|
stmt: createIndexPermsUser,
|
|
},
|
|
{
|
|
name: "create-index-perms-repo",
|
|
stmt: createIndexPermsRepo,
|
|
},
|
|
{
|
|
name: "create-table-builds",
|
|
stmt: createTableBuilds,
|
|
},
|
|
{
|
|
name: "create-index-builds-in-progress",
|
|
stmt: createIndexBuildsInProgress,
|
|
},
|
|
{
|
|
name: "create-index-builds-repo",
|
|
stmt: createIndexBuildsRepo,
|
|
},
|
|
{
|
|
name: "create-index-builds-author",
|
|
stmt: createIndexBuildsAuthor,
|
|
},
|
|
{
|
|
name: "create-index-builds-sender",
|
|
stmt: createIndexBuildsSender,
|
|
},
|
|
{
|
|
name: "create-index-builds-ref",
|
|
stmt: createIndexBuildsRef,
|
|
},
|
|
{
|
|
name: "create-table-stages",
|
|
stmt: createTableStages,
|
|
},
|
|
{
|
|
name: "create-index-stages-build",
|
|
stmt: createIndexStagesBuild,
|
|
},
|
|
{
|
|
name: "create-index-stages-status",
|
|
stmt: createIndexStagesStatus,
|
|
},
|
|
{
|
|
name: "create-table-steps",
|
|
stmt: createTableSteps,
|
|
},
|
|
{
|
|
name: "create-index-steps-stage",
|
|
stmt: createIndexStepsStage,
|
|
},
|
|
{
|
|
name: "create-table-logs",
|
|
stmt: createTableLogs,
|
|
},
|
|
{
|
|
name: "create-table-cron",
|
|
stmt: createTableCron,
|
|
},
|
|
{
|
|
name: "create-index-cron-repo",
|
|
stmt: createIndexCronRepo,
|
|
},
|
|
{
|
|
name: "create-index-cron-next",
|
|
stmt: createIndexCronNext,
|
|
},
|
|
{
|
|
name: "create-table-secrets",
|
|
stmt: createTableSecrets,
|
|
},
|
|
{
|
|
name: "create-index-secrets-repo",
|
|
stmt: createIndexSecretsRepo,
|
|
},
|
|
{
|
|
name: "create-index-secrets-repo-name",
|
|
stmt: createIndexSecretsRepoName,
|
|
},
|
|
{
|
|
name: "create-table-nodes",
|
|
stmt: createTableNodes,
|
|
},
|
|
}
|
|
|
|
// Migrate performs the database migration. If the migration fails
|
|
// and error is returned.
|
|
func Migrate(db *sql.DB) error {
|
|
if err := createTable(db); err != nil {
|
|
return err
|
|
}
|
|
completed, err := selectCompleted(db)
|
|
if err != nil && err != sql.ErrNoRows {
|
|
return err
|
|
}
|
|
for _, migration := range migrations {
|
|
if _, ok := completed[migration.name]; ok {
|
|
|
|
continue
|
|
}
|
|
|
|
if _, err := db.Exec(migration.stmt); err != nil {
|
|
return err
|
|
}
|
|
if err := insertMigration(db, migration.name); err != nil {
|
|
return err
|
|
}
|
|
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func createTable(db *sql.DB) error {
|
|
_, err := db.Exec(migrationTableCreate)
|
|
return err
|
|
}
|
|
|
|
func insertMigration(db *sql.DB, name string) error {
|
|
_, err := db.Exec(migrationInsert, name)
|
|
return err
|
|
}
|
|
|
|
func selectCompleted(db *sql.DB) (map[string]struct{}, error) {
|
|
migrations := map[string]struct{}{}
|
|
rows, err := db.Query(migrationSelect)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var name string
|
|
if err := rows.Scan(&name); err != nil {
|
|
return nil, err
|
|
}
|
|
migrations[name] = struct{}{}
|
|
}
|
|
return migrations, nil
|
|
}
|
|
|
|
//
|
|
// migration table ddl and sql
|
|
//
|
|
|
|
var migrationTableCreate = `
|
|
CREATE TABLE IF NOT EXISTS migrations (
|
|
name VARCHAR(255)
|
|
,UNIQUE(name)
|
|
)
|
|
`
|
|
|
|
var migrationInsert = `
|
|
INSERT INTO migrations (name) VALUES ($1)
|
|
`
|
|
|
|
var migrationSelect = `
|
|
SELECT name FROM migrations
|
|
`
|
|
|
|
//
|
|
// 001_create_table_user.sql
|
|
//
|
|
|
|
var createTableUsers = `
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
user_id SERIAL PRIMARY KEY
|
|
,user_login VARCHAR(250)
|
|
,user_email VARCHAR(500)
|
|
,user_admin BOOLEAN
|
|
,user_active BOOLEAN
|
|
,user_machine BOOLEAN
|
|
,user_avatar VARCHAR(2000)
|
|
,user_syncing BOOLEAN
|
|
,user_synced INTEGER
|
|
,user_created INTEGER
|
|
,user_updated INTEGER
|
|
,user_last_login INTEGER
|
|
,user_oauth_token VARCHAR(500)
|
|
,user_oauth_refresh VARCHAR(500)
|
|
,user_oauth_expiry INTEGER
|
|
,user_hash VARCHAR(500)
|
|
,UNIQUE(user_login)
|
|
,UNIQUE(user_hash)
|
|
);
|
|
`
|
|
|
|
//
|
|
// 002_create_table_repos.sql
|
|
//
|
|
|
|
var createTableRepos = `
|
|
CREATE TABLE IF NOT EXISTS repos (
|
|
repo_id SERIAL PRIMARY KEY
|
|
,repo_uid VARCHAR(250)
|
|
,repo_user_id INTEGER
|
|
,repo_namespace VARCHAR(250)
|
|
,repo_name VARCHAR(250)
|
|
,repo_slug VARCHAR(250)
|
|
,repo_scm VARCHAR(50)
|
|
,repo_clone_url VARCHAR(2000)
|
|
,repo_ssh_url VARCHAR(2000)
|
|
,repo_html_url VARCHAR(2000)
|
|
,repo_active BOOLEAN
|
|
,repo_private BOOLEAN
|
|
,repo_visibility VARCHAR(50)
|
|
,repo_branch VARCHAR(250)
|
|
,repo_counter INTEGER
|
|
,repo_config VARCHAR(500)
|
|
,repo_timeout INTEGER
|
|
,repo_trusted BOOLEAN
|
|
,repo_protected BOOLEAN
|
|
,repo_synced INTEGER
|
|
,repo_created INTEGER
|
|
,repo_updated INTEGER
|
|
,repo_version INTEGER
|
|
,repo_signer VARCHAR(50)
|
|
,repo_secret VARCHAR(50)
|
|
,UNIQUE(repo_slug)
|
|
,UNIQUE(repo_uid)
|
|
);
|
|
`
|
|
|
|
var alterTableReposAddColumnNoFork = `
|
|
ALTER TABLE repos ADD COLUMN repo_no_forks BOOLEAN NOT NULL DEFAULT false;
|
|
`
|
|
|
|
var alterTableReposAddColumnNoPulls = `
|
|
ALTER TABLE repos ADD COLUMN repo_no_pulls BOOLEAN NOT NULL DEFAULT false;
|
|
`
|
|
|
|
//
|
|
// 003_create_table_perms.sql
|
|
//
|
|
|
|
var createTablePerms = `
|
|
CREATE TABLE IF NOT EXISTS perms (
|
|
perm_user_id INTEGER
|
|
,perm_repo_uid VARCHAR(250)
|
|
,perm_read BOOLEAN
|
|
,perm_write BOOLEAN
|
|
,perm_admin BOOLEAN
|
|
,perm_synced INTEGER
|
|
,perm_created INTEGER
|
|
,perm_updated INTEGER
|
|
,PRIMARY KEY(perm_user_id, perm_repo_uid)
|
|
);
|
|
`
|
|
|
|
var createIndexPermsUser = `
|
|
CREATE INDEX IF NOT EXISTS ix_perms_user ON perms (perm_user_id);
|
|
`
|
|
|
|
var createIndexPermsRepo = `
|
|
CREATE INDEX IF NOT EXISTS ix_perms_repo ON perms (perm_repo_uid);
|
|
`
|
|
|
|
//
|
|
// 004_create_table_builds.sql
|
|
//
|
|
|
|
var createTableBuilds = `
|
|
CREATE TABLE IF NOT EXISTS builds (
|
|
build_id SERIAL PRIMARY KEY
|
|
,build_repo_id INTEGER
|
|
,build_config_id INTEGER
|
|
,build_trigger VARCHAR(250)
|
|
,build_number INTEGER
|
|
,build_parent INTEGER
|
|
,build_status VARCHAR(50)
|
|
,build_error VARCHAR(500)
|
|
,build_event VARCHAR(50)
|
|
,build_action VARCHAR(50)
|
|
,build_link VARCHAR(2000)
|
|
,build_timestamp INTEGER
|
|
,build_title VARCHAR(2000)
|
|
,build_message VARCHAR(2000)
|
|
,build_before VARCHAR(50)
|
|
,build_after VARCHAR(50)
|
|
,build_ref VARCHAR(500)
|
|
,build_source_repo VARCHAR(250)
|
|
,build_source VARCHAR(500)
|
|
,build_target VARCHAR(500)
|
|
,build_author VARCHAR(500)
|
|
,build_author_name VARCHAR(500)
|
|
,build_author_email VARCHAR(500)
|
|
,build_author_avatar VARCHAR(2000)
|
|
,build_sender VARCHAR(500)
|
|
,build_deploy VARCHAR(500)
|
|
,build_params VARCHAR(4000)
|
|
,build_started INTEGER
|
|
,build_finished INTEGER
|
|
,build_created INTEGER
|
|
,build_updated INTEGER
|
|
,build_version INTEGER
|
|
,UNIQUE(build_repo_id, build_number)
|
|
);
|
|
`
|
|
|
|
var createIndexBuildsInProgress = `
|
|
CREATE INDEX IF NOT EXISTS ix_build_in_progress ON builds (build_status)
|
|
WHERE build_status IN ('pending', 'running');
|
|
`
|
|
|
|
var createIndexBuildsRepo = `
|
|
CREATE INDEX IF NOT EXISTS ix_build_repo ON builds (build_repo_id);
|
|
`
|
|
|
|
var createIndexBuildsAuthor = `
|
|
CREATE INDEX IF NOT EXISTS ix_build_author ON builds (build_author);
|
|
`
|
|
|
|
var createIndexBuildsSender = `
|
|
CREATE INDEX IF NOT EXISTS ix_build_sender ON builds (build_sender);
|
|
`
|
|
|
|
var createIndexBuildsRef = `
|
|
CREATE INDEX IF NOT EXISTS ix_build_ref ON builds (build_repo_id, build_ref);
|
|
|
|
CREATE INDEX IF NOT EXISTS ix_build_incomplete ON builds (build_status)
|
|
WHERE build_status IN ('pending', 'running');
|
|
`
|
|
|
|
//
|
|
// 005_create_table_stages.sql
|
|
//
|
|
|
|
var createTableStages = `
|
|
CREATE TABLE IF NOT EXISTS stages (
|
|
stage_id SERIAL PRIMARY KEY
|
|
,stage_repo_id INTEGER
|
|
,stage_build_id INTEGER
|
|
,stage_number INTEGER
|
|
,stage_name VARCHAR(50)
|
|
,stage_kind VARCHAR(50)
|
|
,stage_type VARCHAR(50)
|
|
,stage_status VARCHAR(50)
|
|
,stage_error VARCHAR(500)
|
|
,stage_errignore BOOLEAN
|
|
,stage_exit_code INTEGER
|
|
,stage_limit INTEGER
|
|
,stage_os VARCHAR(50)
|
|
,stage_arch VARCHAR(50)
|
|
,stage_variant VARCHAR(10)
|
|
,stage_kernel VARCHAR(50)
|
|
,stage_machine VARCHAR(500)
|
|
,stage_started INTEGER
|
|
,stage_stopped INTEGER
|
|
,stage_created INTEGER
|
|
,stage_updated INTEGER
|
|
,stage_version INTEGER
|
|
,stage_on_success BOOLEAN
|
|
,stage_on_failure BOOLEAN
|
|
,stage_depends_on TEXT
|
|
,stage_labels TEXT
|
|
,UNIQUE(stage_build_id, stage_number)
|
|
);
|
|
`
|
|
|
|
var createIndexStagesBuild = `
|
|
CREATE INDEX IF NOT EXISTS ix_stages_build ON stages (stage_build_id);
|
|
`
|
|
|
|
var createIndexStagesStatus = `
|
|
CREATE INDEX IF NOT EXISTS ix_build_in_progress ON stages (stage_status)
|
|
WHERE stage_status IN ('pending', 'running');
|
|
`
|
|
|
|
//
|
|
// 006_create_table_steps.sql
|
|
//
|
|
|
|
var createTableSteps = `
|
|
CREATE TABLE IF NOT EXISTS steps (
|
|
step_id SERIAL PRIMARY KEY
|
|
,step_stage_id INTEGER
|
|
,step_number INTEGER
|
|
,step_name VARCHAR(100)
|
|
,step_status VARCHAR(50)
|
|
,step_error VARCHAR(500)
|
|
,step_errignore BOOLEAN
|
|
,step_exit_code INTEGER
|
|
,step_started INTEGER
|
|
,step_stopped INTEGER
|
|
,step_version INTEGER
|
|
,UNIQUE(step_stage_id, step_number)
|
|
);
|
|
`
|
|
|
|
var createIndexStepsStage = `
|
|
CREATE INDEX IF NOT EXISTS ix_steps_stage ON steps (step_stage_id);
|
|
`
|
|
|
|
//
|
|
// 007_create_table_logs.sql
|
|
//
|
|
|
|
var createTableLogs = `
|
|
CREATE TABLE IF NOT EXISTS logs (
|
|
log_id SERIAL PRIMARY KEY
|
|
,log_data BYTEA
|
|
);
|
|
`
|
|
|
|
//
|
|
// 008_create_table_cron.sql
|
|
//
|
|
|
|
var createTableCron = `
|
|
CREATE TABLE IF NOT EXISTS cron (
|
|
cron_id SERIAL PRIMARY KEY
|
|
,cron_repo_id INTEGER
|
|
,cron_name VARCHAR(50)
|
|
,cron_expr VARCHAR(50)
|
|
,cron_next INTEGER
|
|
,cron_prev INTEGER
|
|
,cron_event VARCHAR(50)
|
|
,cron_branch VARCHAR(250)
|
|
,cron_target VARCHAR(250)
|
|
,cron_disabled BOOLEAN
|
|
,cron_created INTEGER
|
|
,cron_updated INTEGER
|
|
,cron_version INTEGER
|
|
,UNIQUE(cron_repo_id, cron_name)
|
|
,FOREIGN KEY(cron_repo_id) REFERENCES repos(repo_id) ON DELETE CASCADE
|
|
);
|
|
`
|
|
|
|
var createIndexCronRepo = `
|
|
CREATE INDEX IF NOT EXISTS ix_cron_repo ON cron (cron_repo_id);
|
|
`
|
|
|
|
var createIndexCronNext = `
|
|
CREATE INDEX IF NOT EXISTS ix_cron_next ON cron (cron_next);
|
|
`
|
|
|
|
//
|
|
// 009_create_table_secrets.sql
|
|
//
|
|
|
|
var createTableSecrets = `
|
|
CREATE TABLE IF NOT EXISTS secrets (
|
|
secret_id SERIAL PRIMARY KEY
|
|
,secret_repo_id INTEGER
|
|
,secret_name VARCHAR(500)
|
|
,secret_data BYTEA
|
|
,secret_pull_request BOOLEAN
|
|
,secret_pull_request_push BOOLEAN
|
|
,UNIQUE(secret_repo_id, secret_name)
|
|
,FOREIGN KEY(secret_repo_id) REFERENCES repos(repo_id) ON DELETE CASCADE
|
|
);
|
|
`
|
|
|
|
var createIndexSecretsRepo = `
|
|
CREATE INDEX IF NOT EXISTS ix_secret_repo ON secrets (secret_repo_id);
|
|
`
|
|
|
|
var createIndexSecretsRepoName = `
|
|
CREATE INDEX IF NOT EXISTS ix_secret_repo_name ON secrets (secret_repo_id, secret_name);
|
|
`
|
|
|
|
//
|
|
// 010_create_table_nodes.sql
|
|
//
|
|
|
|
var createTableNodes = `
|
|
CREATE TABLE IF NOT EXISTS nodes (
|
|
node_id SERIAL PRIMARY KEY
|
|
,node_uid VARCHAR(500)
|
|
,node_provider VARCHAR(50)
|
|
,node_state VARCHAR(50)
|
|
,node_name VARCHAR(50)
|
|
,node_image VARCHAR(500)
|
|
,node_region VARCHAR(100)
|
|
,node_size VARCHAR(100)
|
|
,node_os VARCHAR(50)
|
|
,node_arch VARCHAR(50)
|
|
,node_kernel VARCHAR(50)
|
|
,node_variant VARCHAR(50)
|
|
,node_address VARCHAR(500)
|
|
,node_capacity INTEGER
|
|
,node_filter VARCHAR(2000)
|
|
,node_labels VARCHAR(2000)
|
|
,node_error VARCHAR(2000)
|
|
,node_ca_key BYTEA
|
|
,node_ca_cert BYTEA
|
|
,node_tls_key BYTEA
|
|
,node_tls_cert BYTEA
|
|
,node_tls_name VARCHAR(500)
|
|
,node_paused BOOLEAN
|
|
,node_protected BOOLEAN
|
|
,node_created INTEGER
|
|
,node_updated INTEGER
|
|
,node_pulled INTEGER
|
|
|
|
,UNIQUE(node_name)
|
|
);
|
|
`
|