forked from Shiloh/githaven
39 lines
743 B
Go
39 lines
743 B
Go
|
package testfixtures
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
paramTypeDollar = iota + 1
|
||
|
|
paramTypeQuestion
|
||
|
|
paramTypeColon
|
||
|
|
)
|
||
|
|
|
||
|
|
type loadFunction func(tx *sql.Tx) error
|
||
|
|
|
||
|
|
// Helper is the generic interface for the database helper
|
||
|
|
type Helper interface {
|
||
|
|
init(*sql.DB) error
|
||
|
|
disableReferentialIntegrity(*sql.DB, loadFunction) error
|
||
|
|
paramType() int
|
||
|
|
databaseName(*sql.DB) string
|
||
|
|
quoteKeyword(string) string
|
||
|
|
whileInsertOnTable(*sql.Tx, string, func() error) error
|
||
|
|
}
|
||
|
|
|
||
|
|
type baseHelper struct{}
|
||
|
|
|
||
|
|
func (*baseHelper) init(_ *sql.DB) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*baseHelper) quoteKeyword(str string) string {
|
||
|
|
return fmt.Sprintf(`"%s"`, str)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*baseHelper) whileInsertOnTable(_ *sql.Tx, _ string, fn func() error) error {
|
||
|
|
return fn()
|
||
|
|
}
|