/
admin.go
126 lines (105 loc) · 4.53 KB
/
admin.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// Copyright (c) 2017 Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
package mysql
import (
"fmt"
"time"
)
const (
readSchemaVersionQuery = `SELECT curr_version from schema_version where db_name=?`
writeSchemaVersionQuery = `REPLACE into schema_version(db_name, creation_time, curr_version, min_compatible_version) VALUES (?,?,?,?)`
writeSchemaUpdateHistoryQuery = `INSERT into schema_update_history(year, month, update_time, old_version, new_version, manifest_md5, description) VALUES(?,?,?,?,?,?,?)`
createSchemaVersionTableQuery = `CREATE TABLE schema_version(db_name VARCHAR(255) not null PRIMARY KEY, ` +
`creation_time DATETIME(6), ` +
`curr_version VARCHAR(64), ` +
`min_compatible_version VARCHAR(64));`
createSchemaUpdateHistoryTableQuery = `CREATE TABLE schema_update_history(` +
`year int not null, ` +
`month int not null, ` +
`update_time DATETIME(6) not null, ` +
`description VARCHAR(255), ` +
`manifest_md5 VARCHAR(64), ` +
`new_version VARCHAR(64), ` +
`old_version VARCHAR(64), ` +
`PRIMARY KEY (year, month, update_time));`
//NOTE we have to use %v because somehow mysql doesn't work with ? here
createDatabaseQuery = "CREATE database %v CHARACTER SET UTF8"
dropDatabaseQuery = "Drop database %v"
listTablesQuery = "SHOW TABLES FROM %v"
dropTableQuery = "DROP TABLE %v"
)
// CreateSchemaVersionTables sets up the schema version tables
func (mdb *db) CreateSchemaVersionTables() error {
if err := mdb.Exec(createSchemaVersionTableQuery); err != nil {
return err
}
return mdb.Exec(createSchemaUpdateHistoryTableQuery)
}
// ReadSchemaVersion returns the current schema version for the keyspace
func (mdb *db) ReadSchemaVersion(database string) (string, error) {
var version string
err := mdb.db.Get(&version, readSchemaVersionQuery, database)
return version, err
}
// UpdateSchemaVersion updates the schema version for the keyspace
func (mdb *db) UpdateSchemaVersion(database string, newVersion string, minCompatibleVersion string) error {
return mdb.Exec(writeSchemaVersionQuery, database, time.Now(), newVersion, minCompatibleVersion)
}
// WriteSchemaUpdateLog adds an entry to the schema update history table
func (mdb *db) WriteSchemaUpdateLog(oldVersion string, newVersion string, manifestMD5 string, desc string) error {
now := time.Now().UTC()
return mdb.Exec(writeSchemaUpdateHistoryQuery, now.Year(), int(now.Month()), now, oldVersion, newVersion, manifestMD5, desc)
}
// Exec executes a sql statement
func (mdb *db) Exec(stmt string, args ...interface{}) error {
_, err := mdb.db.Exec(stmt, args...)
return err
}
// ListTables returns a list of tables in this database
func (mdb *db) ListTables(database string) ([]string, error) {
var tables []string
err := mdb.db.Select(&tables, fmt.Sprintf(listTablesQuery, database))
return tables, err
}
// DropTable drops a given table from the database
func (mdb *db) DropTable(name string) error {
return mdb.Exec(fmt.Sprintf(dropTableQuery, name))
}
// DropAllTables drops all tables from this database
func (mdb *db) DropAllTables(database string) error {
tables, err := mdb.ListTables(database)
if err != nil {
return err
}
for _, tab := range tables {
if err := mdb.DropTable(tab); err != nil {
return err
}
}
return nil
}
// CreateDatabase creates a database if it doesn't exist
func (mdb *db) CreateDatabase(name string) error {
return mdb.Exec(fmt.Sprintf(createDatabaseQuery, name))
}
// DropDatabase drops a database
func (mdb *db) DropDatabase(name string) error {
return mdb.Exec(fmt.Sprintf(dropDatabaseQuery, name))
}