File size: 835 Bytes
9853396 | 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 | package sqlite
import (
"database/sql"
"database/sql/driver"
"fmt"
"modernc.org/sqlite"
)
// https://github.com/ent/ent/discussions/1667
type sqliteDriver struct {
*sqlite.Driver
}
func (d sqliteDriver) Open(name string) (driver.Conn, error) {
conn, err := d.Driver.Open(name)
if err != nil {
return conn, err
}
//nolint:forcetypeassert
c := conn.(interface {
Exec(stmt string, args []driver.Value) (driver.Result, error)
})
if _, err := c.Exec("PRAGMA foreign_keys = on;", nil); err != nil {
if err := conn.Close(); err != nil {
return nil, fmt.Errorf("failed to close connection: %w", err)
}
return nil, fmt.Errorf("failed to enable foreign keys: %w", err)
}
return conn, nil
}
func init() {
sql.Register("sqlite3", sqliteDriver{Driver: &sqlite.Driver{}})
}
|