24 lines
492 B
Go
24 lines
492 B
Go
package with
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"strings"
|
|
"time"
|
|
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
func Sqlite(ctx context.Context, conn string, foo func(db *sql.DB) error) error {
|
|
conn = strings.TrimPrefix(conn, "sqlite://")
|
|
return _sql(ctx, "sqlite", conn, func(db *sql.DB) error {
|
|
if err := Timeout(ctx, 2*time.Second, func(ctx context.Context) error {
|
|
_, err := db.ExecContext(ctx, `PRAGMA journal_mode=WAL`)
|
|
return err
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
return foo(db)
|
|
})
|
|
}
|