Mihai Doarna
100 lines
4.2 KiB
Go
100 lines
4.2 KiB
Go
package migrations
|
|
|
|
import . "github.com/grafana/grafana/pkg/services/sqlstore/migrator"
|
|
|
|
func addTeamMigrations(mg *Migrator) {
|
|
teamV1 := Table{
|
|
Name: "team",
|
|
Columns: []*Column{
|
|
{Name: "id", Type: DB_BigInt, IsPrimaryKey: true, IsAutoIncrement: true},
|
|
{Name: "name", Type: DB_NVarchar, Length: 190, Nullable: false},
|
|
{Name: "org_id", Type: DB_BigInt},
|
|
{Name: "created", Type: DB_DateTime, Nullable: false},
|
|
{Name: "updated", Type: DB_DateTime, Nullable: false},
|
|
},
|
|
Indices: []*Index{
|
|
{Cols: []string{"org_id"}},
|
|
{Cols: []string{"org_id", "name"}, Type: UniqueIndex},
|
|
},
|
|
}
|
|
|
|
mg.AddMigration("create team table", NewAddTableMigration(teamV1))
|
|
|
|
//------- indexes ------------------
|
|
mg.AddMigration("add index team.org_id", NewAddIndexMigration(teamV1, teamV1.Indices[0]))
|
|
mg.AddMigration("add unique index team_org_id_name", NewAddIndexMigration(teamV1, teamV1.Indices[1]))
|
|
|
|
mg.AddMigration("Add column uid in team", NewAddColumnMigration(teamV1, &Column{
|
|
Name: "uid", Type: DB_NVarchar, Length: 40, Nullable: true,
|
|
}))
|
|
|
|
mg.AddMigration("Update uid column values in team", NewRawSQLMigration("").
|
|
SQLite("UPDATE team SET uid=printf('t%09d',id) WHERE uid IS NULL;").
|
|
Postgres("UPDATE team SET uid='t' || lpad('' || id::text,9,'0') WHERE uid IS NULL;").
|
|
Mysql("UPDATE team SET uid=concat('t',lpad(id,9,'0')) WHERE uid IS NULL;"))
|
|
|
|
mg.AddMigration("Add unique index team_org_id_uid", NewAddIndexMigration(teamV1, &Index{
|
|
Cols: []string{"org_id", "uid"}, Type: UniqueIndex,
|
|
}))
|
|
|
|
mg.AddMigration("Add column external_uid in team", NewAddColumnMigration(teamV1, &Column{
|
|
Name: "external_uid", Type: DB_NVarchar, Length: 256, Nullable: true,
|
|
}))
|
|
|
|
mg.AddMigration("Add column is_provisioned in team", NewAddColumnMigration(teamV1, &Column{
|
|
Name: "is_provisioned", Type: DB_Bool, Nullable: true,
|
|
}))
|
|
|
|
teamMemberV1 := Table{
|
|
Name: "team_member",
|
|
Columns: []*Column{
|
|
{Name: "id", Type: DB_BigInt, IsPrimaryKey: true, IsAutoIncrement: true},
|
|
{Name: "org_id", Type: DB_BigInt},
|
|
{Name: "team_id", Type: DB_BigInt},
|
|
{Name: "user_id", Type: DB_BigInt},
|
|
{Name: "created", Type: DB_DateTime, Nullable: false},
|
|
{Name: "updated", Type: DB_DateTime, Nullable: false},
|
|
},
|
|
Indices: []*Index{
|
|
{Cols: []string{"org_id"}},
|
|
{Cols: []string{"org_id", "team_id", "user_id"}, Type: UniqueIndex},
|
|
{Cols: []string{"team_id"}},
|
|
{Cols: []string{"user_id", "org_id"}},
|
|
},
|
|
}
|
|
|
|
mg.AddMigration("create team member table", NewAddTableMigration(teamMemberV1))
|
|
|
|
//------- indexes ------------------
|
|
mg.AddMigration("add index team_member.org_id", NewAddIndexMigration(teamMemberV1, teamMemberV1.Indices[0]))
|
|
mg.AddMigration("add unique index team_member_org_id_team_id_user_id", NewAddIndexMigration(teamMemberV1, teamMemberV1.Indices[1]))
|
|
mg.AddMigration("add index team_member.team_id", NewAddIndexMigration(teamMemberV1, teamMemberV1.Indices[2]))
|
|
|
|
// add column email
|
|
mg.AddMigration("Add column email to team table", NewAddColumnMigration(teamV1, &Column{
|
|
Name: "email", Type: DB_NVarchar, Nullable: true, Length: 190,
|
|
}))
|
|
|
|
mg.AddMigration("Add column external to team_member table", NewAddColumnMigration(teamMemberV1, &Column{
|
|
Name: "external", Type: DB_Bool, Nullable: true,
|
|
}))
|
|
|
|
mg.AddMigration("Add column permission to team_member table", NewAddColumnMigration(teamMemberV1, &Column{
|
|
Name: "permission", Type: DB_SmallInt, Nullable: true,
|
|
}))
|
|
mg.AddMigration("add unique index team_member_user_id_org_id", NewAddIndexMigration(teamMemberV1, teamMemberV1.Indices[3]))
|
|
|
|
mg.AddMigration("Add column uid in team_member", NewAddColumnMigration(teamMemberV1, &Column{
|
|
Name: "uid", Type: DB_NVarchar, Length: 40, Nullable: true,
|
|
}))
|
|
|
|
mg.AddMigration("Update uid column values in team_member", NewRawSQLMigration("").
|
|
SQLite("UPDATE team_member SET uid=printf('tm%09d',id) WHERE uid IS NULL OR uid = '';").
|
|
Postgres("UPDATE team_member SET uid='tm' || lpad('' || id::text,9,'0') WHERE uid IS NULL OR uid = '';").
|
|
Mysql("UPDATE team_member SET uid=concat('tm',lpad(id,9,'0')) WHERE uid IS NULL OR uid = '';"))
|
|
|
|
mg.AddMigration("Add unique index team_member_uid", NewAddIndexMigration(teamMemberV1, &Index{
|
|
Cols: []string{"uid"}, Type: UniqueIndex,
|
|
}))
|
|
}
|