forked from k1LoW/runn
/
db.go
53 lines (46 loc) · 983 Bytes
/
db.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package testutil
import (
"database/sql"
"fmt"
"os"
"strings"
"testing"
"github.com/xo/dburl"
"modernc.org/sqlite"
)
func SQLite(t *testing.T) (*sql.DB, string) {
t.Helper()
p, err := os.CreateTemp("", "tmp")
if err != nil {
t.Fatal(err)
}
t.Cleanup(func() {
_ = os.Remove(p.Name())
})
dsn := fmt.Sprintf("sqlite://%s", p.Name())
db, err := dburl.Open(normalizeDSN(dsn))
if err != nil {
t.Fatal(err)
}
return db, dsn
}
var dsnRep = strings.NewReplacer("sqlite://", "moderncsqlite://", "sqlite3://", "moderncsqlite://", "sq://", "moderncsqlite://")
func normalizeDSN(dsn string) string {
if !contains(sql.Drivers(), "sqlite3") { // sqlite3 => github.com/mattn/go-sqlite3
return dsnRep.Replace(dsn)
}
return dsn
}
func init() {
if !contains(sql.Drivers(), "moderncsqlite") {
sql.Register("moderncsqlite", &sqlite.Driver{})
}
}
func contains(s []string, e string) bool {
for _, v := range s {
if e == v {
return true
}
}
return false
}