forked from gitea/gitea
46 lines
784 B
Go
46 lines
784 B
Go
|
package testfixtures
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
// MySQL is the MySQL helper for this package
|
||
|
|
type MySQL struct {
|
||
|
|
baseHelper
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*MySQL) paramType() int {
|
||
|
|
return paramTypeQuestion
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*MySQL) quoteKeyword(str string) string {
|
||
|
|
return fmt.Sprintf("`%s`", str)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*MySQL) databaseName(db *sql.DB) (dbName string) {
|
||
|
|
db.QueryRow("SELECT DATABASE()").Scan(&dbName)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *MySQL) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error {
|
||
|
|
// re-enable after load
|
||
|
|
defer db.Exec("SET FOREIGN_KEY_CHECKS = 1")
|
||
|
|
|
||
|
|
tx, err := db.Begin()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err = tx.Exec("SET FOREIGN_KEY_CHECKS = 0"); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if err = loadFn(tx); err != nil {
|
||
|
|
tx.Rollback()
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return tx.Commit()
|
||
|
|
}
|