package main

import (
	"database/sql"

	_ "github.com/go-sql-driver/mysql"
	_ "github.com/lib/pq"
	log "github.com/sirupsen/logrus"
)

// OpenDB opens db.
func OpenDB(dsn string, driver string) (*sql.DB, error) {
	db, err := sql.Open(driver, dsn)
	if err != nil {
		return nil, err
	}
	return db, nil
}

// MustExec must execute sql or fatal.
func MustExec(db *sql.DB, query string, args ...interface{}) {
	tx, err := db.Begin()
	if err != nil {
		log.Fatalf("Start transaction error: %v", err)
	}
	_, err = tx.Exec(query, args...)
	if err != nil {
		log.Fatalf("Execute failed SQL: %s, error: %v", query, err)
	}
	err = tx.Commit()
	if err != nil {
		log.Fatalf("Commit error: %v", err)
	}
}