cloudfoundry-incubator/stratos

package datastore

import (
    "database/sql"
    "strings"

    "bitbucket.org/liamstask/goose/lib/goose"
)

func init() {
    RegisterMigration(20170818120003, "InitialSchema", func(txn *sql.Tx, conf *goose.DBConf) error {
        binaryDataType := "BYTEA"
        if strings.Contains(conf.Driver.Name, "mysql") {
            binaryDataType = "BLOB"
        }

        createTokens := "CREATE TABLE IF NOT EXISTS tokens ("
        createTokens += "user_guid     VARCHAR(36) NOT NULL, "
        createTokens += "cnsi_guid     VARCHAR(36), "
        createTokens += "token_type    VARCHAR(4)  NOT NULL, "
        createTokens += "auth_token    " + binaryDataType + "       NOT NULL, "
        createTokens += "refresh_token " + binaryDataType + "       NOT NULL, "
        createTokens += "token_expiry  BIGINT      NOT NULL, "
        createTokens += "last_updated  TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP)"

        if strings.Contains(conf.Driver.Name, "postgres") {
            createTokens += " WITH (OIDS=FALSE);"
        } else {
            createTokens += ";"
        }

        _, err := txn.Exec(createTokens)
        if err != nil {
            return err
        }

        createCnsisTable := "CREATE TABLE IF NOT EXISTS cnsis ("
        createCnsisTable += "guid    VARCHAR(36)   NOT NULL UNIQUE,"
        createCnsisTable += "name                      VARCHAR(255)  NOT NULL,"
        createCnsisTable += "cnsi_type                 VARCHAR(3)    NOT NULL,"
        createCnsisTable += "api_endpoint              VARCHAR(255)  NOT NULL,"
        createCnsisTable += "auth_endpoint             VARCHAR(255)  NOT NULL,"
        createCnsisTable += "token_endpoint            VARCHAR(255)  NOT NULL,"
        createCnsisTable += "doppler_logging_endpoint  VARCHAR(255)  NOT NULL,"
        createCnsisTable += "skip_ssl_validation       BOOLEAN       NOT NULL DEFAULT FALSE,"
        createCnsisTable += "last_updated              TIMESTAMP     NOT NULL DEFAULT CURRENT_TIMESTAMP,"
        createCnsisTable += "PRIMARY KEY (guid) );"

        _, err = txn.Exec(createCnsisTable)
        if err != nil {
            return err
        }

        createIndex := "CREATE INDEX tokens_user_guid ON tokens (user_guid);"
        _, err = txn.Exec(createIndex)
        if err != nil {
            return err
        }
        createIndex = "CREATE INDEX tokens_cnsi_guid ON tokens (cnsi_guid);"
        _, err = txn.Exec(createIndex)
        if err != nil {
            return err
        }
        createIndex = "CREATE INDEX tokens_token_type ON tokens (token_type);"
        _, err = txn.Exec(createIndex)
        if err != nil {
            return err
        }
        createIndex = "CREATE INDEX cnsis_name ON cnsis (name);"
        _, err = txn.Exec(createIndex)
        if err != nil {
            return err
        }
        createIndex = "CREATE INDEX cnsis_cnsi_type ON cnsis (cnsi_type);"
        _, err = txn.Exec(createIndex)
        if err != nil {
            return err
        }

        return nil
    })
}