ct/go/ctfe_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 (
	"flag"

	"go.skia.org/infra/ct/go/db"
	"go.skia.org/infra/go/common"
	"go.skia.org/infra/go/sklog"
)

var (
	local          = flag.Bool("local", false, "Running locally if true. As opposed to in production.")
	promptPassword = flag.Bool("password", false, "Prompt for root password.")
	targetVersion  = flag.Int("target_version", -1, "Migration target version. Defaults to latest defined version.")
)

func main() {
	defer common.LogPanic()
	// Set up flags.
	dbConf := db.DBConfigFromFlags()

	// Global init to initialize cloud logging and parse arguments.
	common.InitWithMust("ctfe_migratedb", common.CloudLoggingOpt())

	if *promptPassword {
		if err := dbConf.PromptForPassword(); err != nil {
			sklog.Fatal(err)
		}
	}
	if !*local {
		if err := dbConf.GetPasswordFromMetadata(); err != nil {
			sklog.Fatal(err)
		}
	}
	vdb, err := dbConf.NewVersionedDB()
	if err != nil {
		sklog.Fatal(err)
	}

	if *targetVersion < 0 {
		// Get the current database version
		maxDBVersion := vdb.MaxDBVersion()
		sklog.Infof("Latest database version: %d", maxDBVersion)

		dbVersion, err := vdb.DBVersion()
		if err != nil {
			sklog.Fatalf("Unable to retrieve database version. Error: %s", err)
		}
		sklog.Infof("Current database version: %d", dbVersion)

		if dbVersion < maxDBVersion {
			sklog.Infof("Migrating to version: %d", maxDBVersion)
			err = vdb.Migrate(maxDBVersion)
			if err != nil {
				sklog.Fatalf("Unable to retrieve database version. Error: %s", err)
			}
		}
	} else {
		sklog.Infof("Migrating to version: %d", *targetVersion)
		err = vdb.Migrate(*targetVersion)
		if err != nil {
			sklog.Fatalf("Unable to retrieve database version. Error: %s", err)
		}
	}
	sklog.Infoln("Database migration finished.")
}