Overhaul database package

golden/go/correctness_migratedb/main.go

 package main
 
 // Executes database migrations to the latest target version. In production this
 // requires the root password for MySQL. The user will be prompted for that so
 // it is not entered via the command line.
 
 import (
-        "bufio"
         "flag"
-        "fmt"
-        "os"
-        "strings"
 
         "github.com/golang/glog"
         "skia.googlesource.com/buildbot.git/go/common"
         "skia.googlesource.com/buildbot.git/go/database"
         "skia.googlesource.com/buildbot.git/golden/go/db"
 )
 
+var (
+        local = flag.Bool("local", false, "Running locally if true. As opposed to in production.")
+)
+
 func main() {
-        defaultConnStr := db.GetConnectionString("root", "", "", "")
-
-        // flags
-        dbConnString := flag.String("db_conn_string", defaultConnStr, "\n\tDatabase string to open connect to the MySQL database. "+
-                "\n\tNeeds to follow the format of the golang-mysql driver (https://github.com/go-sql-driver/mysql."+
-                "\n\tIf the string contains %s the user will be prompted to enter a password which will then be used for subtitution.")
+        // Set up flags.
+        database.SetupFlags(db.PROD_DB_HOST, db.PROD_DB_PORT, database.USER_ROOT, db.PROD_DB_NAME)
 
         // Global init to initialize glog and parse arguments.
         common.Init()
 
-        var connectionStr = *dbConnString
-
-        // if it contains formatting information read the password from stdin.
-        if strings.Contains(connectionStr, "%s") {
-                glog.Infof("Using connection string: %s", connectionStr)
-                reader := bufio.NewReader(os.Stdin)
-                fmt.Print("Enter password for MySQL: ")
-                password, err := reader.ReadString('\n')
-                if err != nil {
-                        glog.Fatalf("Unable to read password. Error: %s", err)
-                }
-                connectionStr = fmt.Sprintf(connectionStr, strings.TrimRight(password, "\n"))
-        }
-
-        conf := &database.DatabaseConfig{
-                MySQLString:    connectionStr,
-                MigrationSteps: db.MigrationSteps(),
+        conf, err := database.ConfigFromFlagsAndMetadata(*local, db.MigrationSteps())
+        if err != nil {
+                glog.Fatal(err)
         }
         vdb := database.NewVersionedDB(conf)
 
         // Get the current database version
         maxDBVersion := vdb.MaxDBVersion()
         glog.Infof("Latest database version: %d", maxDBVersion)
 
         dbVersion, err := vdb.DBVersion()
         if err != nil {
                 glog.Fatalf("Unable to retrieve database version. Error: %s", err)
         }
         glog.Infof("Current database version: %d", dbVersion)
 
         if dbVersion < maxDBVersion {
                 glog.Infof("Migrating to version: %d", maxDBVersion)
                 err = vdb.Migrate(maxDBVersion)
                 if err != nil {
                         glog.Fatalf("Unable to retrieve database version. Error: %s", err)
                 }
         }
 
         glog.Infoln("Database migration finished.")
 }