322394fe63
* feat: use go1.16 embedding for resources We want to embed everything that can be easily embedded. We should, at a minimum, replace the downloading of resources and bindata. Ref: https://github.com/ooni/probe/issues/1367. * fix: get rid of bindata and use go embed instead * fix: start unbreaking some automatic tests * fix: fetch resources as part of the mobile build * fix: convert more stuff to go1.16 I still expect many breakages, but we'll fix them. * fix: make the windows CI green * fix: get resources before running QA * fix: go1.16 uses modules by default * hopefully fix all other outstanding issues * fix(QA/telegram.py): add another DC IP address * Apply suggestions from code review
72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
"embed"
|
|
"io/ioutil"
|
|
|
|
"github.com/apex/log"
|
|
migrate "github.com/rubenv/sql-migrate"
|
|
"upper.io/db.v3/lib/sqlbuilder"
|
|
"upper.io/db.v3/sqlite"
|
|
)
|
|
|
|
//go:embed migrations/*.sql
|
|
var efs embed.FS
|
|
|
|
func readAsset(path string) ([]byte, error) {
|
|
filep, err := efs.Open(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return ioutil.ReadAll(filep)
|
|
}
|
|
|
|
func readAssetDir(path string) ([]string, error) {
|
|
var out []string
|
|
lst, err := efs.ReadDir(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, e := range lst {
|
|
out = append(out, e.Name())
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
// RunMigrations runs the database migrations
|
|
func RunMigrations(db *sql.DB) error {
|
|
log.Debugf("running migrations")
|
|
migrations := &migrate.AssetMigrationSource{
|
|
Asset: readAsset,
|
|
AssetDir: readAssetDir,
|
|
Dir: "migrations",
|
|
}
|
|
n, err := migrate.Exec(db, "sqlite3", migrations, migrate.Up)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Debugf("performed %d migrations", n)
|
|
return nil
|
|
}
|
|
|
|
// Connect to the database
|
|
func Connect(path string) (db sqlbuilder.Database, err error) {
|
|
settings := sqlite.ConnectionURL{
|
|
Database: path,
|
|
Options: map[string]string{"_foreign_keys": "1"},
|
|
}
|
|
sess, err := sqlite.Open(settings)
|
|
if err != nil {
|
|
log.WithError(err).Error("failed to open the DB")
|
|
return nil, err
|
|
}
|
|
|
|
err = RunMigrations(sess.Driver().(*sql.DB))
|
|
if err != nil {
|
|
log.WithError(err).Error("failed to run DB migration")
|
|
return nil, err
|
|
}
|
|
return sess, err
|
|
}
|