OLD | NEW |
1 package db | 1 package db |
2 | 2 |
3 import ( | 3 import "skia.googlesource.com/buildbot.git/go/database" |
4 » "fmt" | |
5 » "strings" | |
6 | |
7 » "github.com/golang/glog" | |
8 » "skia.googlesource.com/buildbot.git/go/database" | |
9 » "skia.googlesource.com/buildbot.git/go/metadata" | |
10 ) | |
11 | 4 |
12 const ( | 5 const ( |
13 » // Key of the password for the readwrite user. | 6 » // Default database parameters. |
14 » METADATA_KEY = "readwrite" | 7 » PROD_DB_HOST = "173.194.104.24" |
15 | 8 » PROD_DB_PORT = 3306 |
16 » // Detfault database parameters. | 9 » PROD_DB_NAME = "skiacorrectness" |
17 » DEFAULT_DB_HOST = "173.194.104.24" | |
18 » DEFAULT_DB_PORT = "3306" | |
19 » DEFAULT_DB_NAME = "skiacorrectness" | |
20 | |
21 » // Template to generate the MySQL database connection string. | |
22 » // And 3306 is the default port for MySQL. | |
23 » DB_CONN_TMPL = "%s:%s@tcp(%s:%s)/%s?parseTime=true" | |
24 ) | 10 ) |
25 | 11 |
26 // MigrationSteps returns the migration (up and down) for the database. | 12 // MigrationSteps returns the migration (up and down) for the database. |
27 func MigrationSteps() []database.MigrationStep { | 13 func MigrationSteps() []database.MigrationStep { |
28 return migrationSteps | 14 return migrationSteps |
29 } | 15 } |
30 | 16 |
31 // GetConfig returns a DatabaseConfig instance for running in production if a | |
32 // metadata server is available. If 'local' is true it will always | |
33 // set the MySQL connection string to "" and thus use a local SQLite database | |
34 // when used with database.NewVersionedDB. | |
35 func GetConfig(mySQLConnStr string, sqlitePath string, local bool) *database.Dat
abaseConfig { | |
36 useMySQLConnStr := mySQLConnStr | |
37 | |
38 // We are in the production environment, so we look up the password. | |
39 if !local { | |
40 // First, get the password from the metadata server. | |
41 // See https://developers.google.com/compute/docs/metadata#custo
m. | |
42 password, err := metadata.Get(METADATA_KEY) | |
43 if err != nil { | |
44 glog.Fatalf("Failed to find metadata. Use 'local' flag w
hen running locally.") | |
45 } | |
46 useMySQLConnStr = fmt.Sprintf(mySQLConnStr, password) | |
47 } | |
48 | |
49 // If there is still a placeholder in the connection string, we | |
50 // set it to empty, so that the the local SQLite database kicks in. | |
51 if strings.Contains(useMySQLConnStr, "%s") { | |
52 useMySQLConnStr = "" | |
53 } | |
54 | |
55 return &database.DatabaseConfig{ | |
56 MySQLString: useMySQLConnStr, | |
57 SQLiteFilePath: sqlitePath, | |
58 MigrationSteps: migrationSteps, | |
59 } | |
60 } | |
61 | |
62 // GetConnectionString returns a MySQL connection string with the given | |
63 // parameters replace in the template. Only userName has to be provided. | |
64 // If host, port or dbName are empty the default (production) values will | |
65 // be used. | |
66 func GetConnectionString(userName, host, port, dbName string) string { | |
67 useHost, usePort, useDBName := host, port, dbName | |
68 if useHost == "" { | |
69 useHost = DEFAULT_DB_HOST | |
70 } | |
71 | |
72 if usePort == "" { | |
73 usePort = DEFAULT_DB_PORT | |
74 } | |
75 | |
76 if useDBName == "" { | |
77 useDBName = DEFAULT_DB_NAME | |
78 } | |
79 | |
80 return fmt.Sprintf(DB_CONN_TMPL, userName, "%s", useHost, usePort, useDB
Name) | |
81 } | |
82 | |
83 // migrationSteps define the steps it takes to migrate the db between versions. | 17 // migrationSteps define the steps it takes to migrate the db between versions. |
84 // Note: Only add to this list, once a step has landed in version control it | 18 // Note: Only add to this list, once a step has landed in version control it |
85 // must not be changed. | 19 // must not be changed. |
86 var migrationSteps = []database.MigrationStep{ | 20 var migrationSteps = []database.MigrationStep{ |
87 // version 1 | 21 // version 1 |
88 { | 22 { |
89 MySQLUp: []string{ | 23 MySQLUp: []string{ |
90 `CREATE TABLE expectations ( | 24 `CREATE TABLE expectations ( |
91 id INT NOT NULL AUTO_INCREMENT
PRIMARY KEY, | 25 id INT NOT NULL AUTO_INCREMENT
PRIMARY KEY, |
92 userid TEXT NOT NULL, | 26 userid TEXT NOT NULL, |
93 ts BIGINT NOT NULL, | 27 ts BIGINT NOT NULL, |
94 expectations MEDIUMTEXT NOT NULL | 28 expectations MEDIUMTEXT NOT NULL |
95 )`, | 29 )`, |
96 }, | 30 }, |
97 MySQLDown: []string{ | 31 MySQLDown: []string{ |
98 `DROP TABLE expectations`, | 32 `DROP TABLE expectations`, |
99 }, | 33 }, |
100 SQLiteUp: []string{ | |
101 `CREATE TABLE expectations ( | |
102 id INTEGER NOT NULL PRIMARY KEY A
UTOINCREMENT, | |
103 userid TEXT NOT NULL, | |
104 ts BIGINT NOT NULL, | |
105 expectations MEDIUXMTEXT NOT NULL | |
106 )`, | |
107 }, | |
108 SQLiteDown: []string{ | |
109 `DROP TABLE expectations`, | |
110 }, | |
111 }, | 34 }, |
112 | 35 |
113 // Use this is a template for more migration steps. | 36 // Use this is a template for more migration steps. |
114 // version x | 37 // version x |
115 // { | 38 // { |
116 // MySQLUp: , | 39 // MySQLUp: , |
117 // MySQLDown: , | 40 // MySQLDown: , |
118 // SQLiteUp: , | |
119 // SQLiteDown: , | |
120 // }, | 41 // }, |
121 } | 42 } |
OLD | NEW |