Home Explore Help
Register Sign In
oscarleiva
/
energylink
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: 02a89c752b
Branches Tags
energylink
/
pkg
/
services
/
sqlstore
/
migrations
/
datasource_mig.go

datasource_mig.go 4.5 KB
History Raw

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182 
  1. package migrations
    2. 

  2. 

  3. import . "github.com/grafana/grafana/pkg/services/sqlstore/migrator"
    4. 

  4. 

  5. func addDataSourceMigration(mg *Migrator) {
    6. 

  6. 	mg.AddMigration("create data_source table", new(AddTableMigration).
    7. 

  7. 		Name("data_source").WithColumns(
    8. 

  8. 		&Column{Name: "id", Type: DB_BigInt, IsPrimaryKey: true, IsAutoIncrement: true},
    9. 

  9. 		&Column{Name: "account_id", Type: DB_BigInt, Nullable: false},
    10. 

  10. 		&Column{Name: "version", Type: DB_Int, Nullable: false},
    11. 

  11. 		&Column{Name: "type", Type: DB_NVarchar, Length: 255, Nullable: false},
    12. 

  12. 		&Column{Name: "name", Type: DB_NVarchar, Length: 255, Nullable: false},
    13. 

  13. 		&Column{Name: "access", Type: DB_NVarchar, Length: 255, Nullable: false},
    14. 

  14. 		&Column{Name: "url", Type: DB_NVarchar, Length: 255, Nullable: false},
    15. 

  15. 		&Column{Name: "password", Type: DB_NVarchar, Length: 255, Nullable: true},
    16. 

  16. 		&Column{Name: "user", Type: DB_NVarchar, Length: 255, Nullable: true},
    17. 

  17. 		&Column{Name: "database", Type: DB_NVarchar, Length: 255, Nullable: true},
    18. 

  18. 		&Column{Name: "basic_auth", Type: DB_Bool, Nullable: false},
    19. 

  19. 		&Column{Name: "basic_auth_user", Type: DB_NVarchar, Length: 255, Nullable: true},
    20. 

  20. 		&Column{Name: "basic_auth_password", Type: DB_NVarchar, Length: 255, Nullable: true},
    21. 

  21. 		&Column{Name: "is_default", Type: DB_Bool, Nullable: false},
    22. 

  22. 		&Column{Name: "created", Type: DB_DateTime, Nullable: false},
    23. 

  23. 		&Column{Name: "updated", Type: DB_DateTime, Nullable: false},
    24. 

  24. 	))
    25. 

  25. 

  26. 	//-------  indexes ------------------
    27. 

  27. 	mg.AddMigration("add index data_source.account_id", new(AddIndexMigration).
    28. 

  28. 		Table("data_source").Columns("account_id"))
    29. 

  29. 

  30. 	mg.AddMigration("add unique index data_source.account_id_name", new(AddIndexMigration).
    31. 

  31. 		Table("data_source").Columns("account_id", "name").Unique())
    32. 

  32. 

  33. 	// ---------------------
    34. 

  34. 	// account -> org changes
    35. 

  35. 

  36. 	//-------  drop indexes ------------------
    37. 

  37. 	mg.AddMigration("drop index data_source.account_id", new(DropIndexMigration).
    38. 

  38. 		Table("data_source").Columns("account_id"))
    39. 

  39. 

  40. 	mg.AddMigration("drop unique index data_source.account_id_name", new(DropIndexMigration).
    41. 

  41. 		Table("data_source").Columns("account_id", "name").Unique())
    42. 

  42. 

  43. 	//------- rename table ------------------
    44. 

  44. 	mg.AddMigration("rename table data_source to data_source_old", new(RenameTableMigration).
    45. 

  45. 		Rename("data_source", "data_source_old"))
    46. 

  46. 

  47. 	//------- recreate table with new column names ------------------
    48. 

  48. 	mg.AddMigration("create data_source table v2", new(AddTableMigration).
    49. 

  49. 		Name("data_source").WithColumns(
    50. 

  50. 		&Column{Name: "id", Type: DB_BigInt, IsPrimaryKey: true, IsAutoIncrement: true},
    51. 

  51. 		&Column{Name: "org_id", Type: DB_BigInt, Nullable: false},
    52. 

  52. 		&Column{Name: "version", Type: DB_Int, Nullable: false},
    53. 

  53. 		&Column{Name: "type", Type: DB_NVarchar, Length: 255, Nullable: false},
    54. 

  54. 		&Column{Name: "name", Type: DB_NVarchar, Length: 255, Nullable: false},
    55. 

  55. 		&Column{Name: "access", Type: DB_NVarchar, Length: 255, Nullable: false},
    56. 

  56. 		&Column{Name: "url", Type: DB_NVarchar, Length: 255, Nullable: false},
    57. 

  57. 		&Column{Name: "password", Type: DB_NVarchar, Length: 255, Nullable: true},
    58. 

  58. 		&Column{Name: "user", Type: DB_NVarchar, Length: 255, Nullable: true},
    59. 

  59. 		&Column{Name: "database", Type: DB_NVarchar, Length: 255, Nullable: true},
    60. 

  60. 		&Column{Name: "basic_auth", Type: DB_Bool, Nullable: false},
    61. 

  61. 		&Column{Name: "basic_auth_user", Type: DB_NVarchar, Length: 255, Nullable: true},
    62. 

  62. 		&Column{Name: "basic_auth_password", Type: DB_NVarchar, Length: 255, Nullable: true},
    63. 

  63. 		&Column{Name: "is_default", Type: DB_Bool, Nullable: false},
    64. 

  64. 		&Column{Name: "json_data", Type: DB_Text, Nullable: true},
    65. 

  65. 		&Column{Name: "created", Type: DB_DateTime, Nullable: false},
    66. 

  66. 		&Column{Name: "updated", Type: DB_DateTime, Nullable: false},
    67. 

  67. 	))
    68. 

  68. 

  69. 	//------- data_source table indexes ------------------
    70. 

  70. 	mg.AddMigration("add index data_source.org_id", new(AddIndexMigration).
    71. 

  71. 		Table("data_source").Columns("org_id"))
    72. 

  72. 

  73. 	mg.AddMigration("add unique index data_source.org_id_name", new(AddIndexMigration).
    74. 

  74. 		Table("data_source").Columns("org_id", "name").Unique())
    75. 

  75. 

  76. 	//------- copy data from table -------------------
    77. 

  77. 	mg.AddMigration("copy data from data_source_old table", new(CopyTableDataMigration).
    78. 

  78. 		Source("data_source_old", "id, account_id, version, type, name, access, url, password, user, database, basic_auth, basic_auth_user, basic_auth_password, is_default, created, updated").
    79. 

  79. 		Target("data_source", "id, org_id, version, type, name, access, url, password, user, database, basic_auth, basic_auth_user, basic_auth_password, is_default, created, updated"))
    80. 

  80. 

  81. 	mg.AddMigration("Drop old table data_source_old", new(DropTableMigration).Table("data_source_old"))
    82. 

  82. }
    83.