49 lines
766 B
Go
49 lines
766 B
Go
package database
|
|
|
|
import (
|
|
_ "embed"
|
|
"log"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/jmoiron/sqlx"
|
|
_ "github.com/lib/pq"
|
|
"github.com/spf13/viper"
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
var (
|
|
//go:embed init/users.sql
|
|
userTable string
|
|
//go:embed init/otp.sql
|
|
otpTable string
|
|
)
|
|
|
|
func InitDB() (*sqlx.DB, error) {
|
|
driver := viper.GetString("database.driver")
|
|
dsn := viper.GetString("database.dsn")
|
|
|
|
db, err := sqlx.Open(driver, dsn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := db.Ping(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
log.Println("Connected to database!")
|
|
return db, nil
|
|
}
|
|
|
|
func MigrateDB(db *sqlx.DB) error {
|
|
_, err := db.Exec(userTable)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = db.Exec(otpTable)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|