package datastore import ( "database/sql" "os" "github.com/drone/drone/shared/envconfig" "github.com/drone/drone/store" "github.com/drone/drone/store/migration" _ "github.com/go-sql-driver/mysql" _ "github.com/lib/pq" _ "github.com/mattn/go-sqlite3" "github.com/rubenv/sql-migrate" "github.com/russross/meddler" log "github.com/Sirupsen/logrus" ) // Load opens a new database connection with the specified driver // and connection string specified in the environment variables. func Load(env envconfig.Env) store.Store { var ( driver = env.String("DATABASE_DRIVER", "sqlite3") config = env.String("DATABASE_CONFIG", "drone.sqlite") ) log.Infof("using database driver %s", driver) log.Infof("using database config %s", config) return New(driver, config) } func New(driver, config string) store.Store { conn := Open(driver, config) return store.New( driver, &nodestore{conn}, &userstore{conn}, &repostore{conn}, &keystore{conn}, &buildstore{conn}, &jobstore{conn}, &logstore{conn}, ) } // Open opens a new database connection with the specified // driver and connection string and returns a store. func Open(driver, config string) *sql.DB { db, err := sql.Open(driver, config) if err != nil { log.Errorln(err) log.Fatalln("database connection failed") } setupMeddler(driver) if err := setupDatabase(driver, db); err != nil { log.Errorln(err) log.Fatalln("migration failed") } return db } // OpenTest opens a new database connection for testing purposes. // The database driver and connection string are provided by // environment variables, with fallback to in-memory sqlite. func openTest() *sql.DB { var ( driver = "sqlite3" config = ":memory:" ) if os.Getenv("DATABASE_DRIVER") != "" { driver = os.Getenv("DATABASE_DRIVER") config = os.Getenv("DATABASE_CONFIG") } return Open(driver, config) } // helper function to setup the databsae by performing // automated database migration steps. func setupDatabase(driver string, db *sql.DB) error { var migrations = &migrate.AssetMigrationSource{ Asset: migration.Asset, AssetDir: migration.AssetDir, Dir: driver, } _, err := migrate.Exec(db, driver, migrations, migrate.Up) return err } // helper function to setup the meddler default driver // based on the selected driver name. func setupMeddler(driver string) { switch driver { case "sqlite3": meddler.Default = meddler.SQLite case "mysql": meddler.Default = meddler.MySQL case "postgres": meddler.Default = meddler.PostgreSQL } }