Merge pull request #1384 from mattn/raise-go121-cleanup
Raise minimum Go version to 1.21
This commit is contained in:
2
go.mod
2
go.mod
@@ -1,6 +1,6 @@
|
|||||||
module github.com/mattn/go-sqlite3
|
module github.com/mattn/go-sqlite3
|
||||||
|
|
||||||
go 1.19
|
go 1.21
|
||||||
|
|
||||||
retract (
|
retract (
|
||||||
[v2.0.0+incompatible, v2.0.6+incompatible] // Accidental; no major changes or features.
|
[v2.0.0+incompatible, v2.0.6+incompatible] // Accidental; no major changes or features.
|
||||||
|
|||||||
@@ -1,120 +0,0 @@
|
|||||||
// Copyright (C) 2019 Yasuhiro Matsumoto <mattn.jp@gmail.com>.
|
|
||||||
//
|
|
||||||
// Use of this source code is governed by an MIT-style
|
|
||||||
// license that can be found in the LICENSE file.
|
|
||||||
|
|
||||||
//go:build go1.13 && cgo
|
|
||||||
// +build go1.13,cgo
|
|
||||||
|
|
||||||
package sqlite3
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"database/sql"
|
|
||||||
"database/sql/driver"
|
|
||||||
"errors"
|
|
||||||
"os"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestBeginTxCancel(t *testing.T) {
|
|
||||||
srcTempFilename := TempFilename(t)
|
|
||||||
defer os.Remove(srcTempFilename)
|
|
||||||
|
|
||||||
db, err := sql.Open("sqlite3", srcTempFilename)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
db.SetMaxOpenConns(10)
|
|
||||||
db.SetMaxIdleConns(5)
|
|
||||||
|
|
||||||
defer db.Close()
|
|
||||||
initDatabase(t, db, 100)
|
|
||||||
|
|
||||||
// create several go-routines to expose racy issue
|
|
||||||
for i := 0; i < 1000; i++ {
|
|
||||||
func() {
|
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
|
||||||
conn, err := db.Conn(ctx)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer func() {
|
|
||||||
if err := conn.Close(); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
err = conn.Raw(func(driverConn any) error {
|
|
||||||
d, ok := driverConn.(driver.ConnBeginTx)
|
|
||||||
if !ok {
|
|
||||||
t.Fatal("unexpected: wrong type")
|
|
||||||
}
|
|
||||||
// checks that conn.Raw can be used to get *SQLiteConn
|
|
||||||
if _, ok = driverConn.(*SQLiteConn); !ok {
|
|
||||||
t.Fatalf("conn.Raw() driverConn type=%T, expected *SQLiteConn", driverConn)
|
|
||||||
}
|
|
||||||
|
|
||||||
go cancel() // make it cancel concurrently with exec("BEGIN");
|
|
||||||
tx, err := d.BeginTx(ctx, driver.TxOptions{})
|
|
||||||
switch err {
|
|
||||||
case nil:
|
|
||||||
switch err := tx.Rollback(); err {
|
|
||||||
case nil, sql.ErrTxDone:
|
|
||||||
default:
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
case context.Canceled:
|
|
||||||
default:
|
|
||||||
// must not fail with "cannot start a transaction within a transaction"
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStmtReadonly(t *testing.T) {
|
|
||||||
db, err := sql.Open("sqlite3", ":memory:")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = db.Exec("CREATE TABLE t (count INT)")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
isRO := func(query string) bool {
|
|
||||||
c, err := db.Conn(context.Background())
|
|
||||||
if err != nil {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
var ro bool
|
|
||||||
c.Raw(func(dc any) error {
|
|
||||||
stmt, err := dc.(*SQLiteConn).Prepare(query)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if stmt == nil {
|
|
||||||
return errors.New("stmt is nil")
|
|
||||||
}
|
|
||||||
ro = stmt.(*SQLiteStmt).Readonly()
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
return ro // On errors ro will remain false.
|
|
||||||
}
|
|
||||||
|
|
||||||
if !isRO(`select * from t`) {
|
|
||||||
t.Error("select not seen as read-only")
|
|
||||||
}
|
|
||||||
if isRO(`insert into t values (1), (2)`) {
|
|
||||||
t.Error("insert seen as read-only")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -3,15 +3,14 @@
|
|||||||
// Use of this source code is governed by an MIT-style
|
// Use of this source code is governed by an MIT-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
//go:build cgo && go1.8
|
//go:build cgo
|
||||||
// +build cgo,go1.8
|
// +build cgo
|
||||||
|
|
||||||
package sqlite3
|
package sqlite3
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql/driver"
|
|
||||||
|
|
||||||
"context"
|
"context"
|
||||||
|
"database/sql/driver"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Ping implement Pinger.
|
// Ping implement Pinger.
|
||||||
@@ -3,16 +3,17 @@
|
|||||||
// Use of this source code is governed by an MIT-style
|
// Use of this source code is governed by an MIT-style
|
||||||
// license that can be found in the LICENSE file.
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
//go:build go1.8 && cgo
|
//go:build cgo
|
||||||
// +build go1.8,cgo
|
// +build cgo
|
||||||
|
|
||||||
package sqlite3
|
package sqlite3
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
|
"database/sql/driver"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
|
||||||
"math/rand"
|
"math/rand"
|
||||||
"os"
|
"os"
|
||||||
"sync"
|
"sync"
|
||||||
@@ -223,282 +224,125 @@ func TestQueryRowContextCancelParallel(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
defer db.Close()
|
||||||
|
initDatabase(t, db, 100)
|
||||||
|
|
||||||
|
const query = `SELECT key_id FROM test_table ORDER BY key2 ASC`
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
for i := 0; i < 100; i++ {
|
||||||
|
wg.Add(1)
|
||||||
|
go func() {
|
||||||
|
defer wg.Done()
|
||||||
|
var keyID string
|
||||||
|
for j := 0; j < 100; j++ {
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
row := db.QueryRowContext(ctx, query)
|
||||||
|
cancel()
|
||||||
|
_ = row.Scan(&keyID)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
wg.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBeginTxCancel(t *testing.T) {
|
||||||
|
srcTempFilename := TempFilename(t)
|
||||||
|
defer os.Remove(srcTempFilename)
|
||||||
|
|
||||||
|
db, err := sql.Open("sqlite3", srcTempFilename)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
db.SetMaxOpenConns(10)
|
db.SetMaxOpenConns(10)
|
||||||
db.SetMaxIdleConns(5)
|
db.SetMaxIdleConns(5)
|
||||||
|
|
||||||
defer db.Close()
|
defer db.Close()
|
||||||
initDatabase(t, db, 100)
|
initDatabase(t, db, 100)
|
||||||
|
|
||||||
const query = `SELECT key_id FROM test_table ORDER BY key2 ASC`
|
// create several go-routines to expose racy issue
|
||||||
wg := sync.WaitGroup{}
|
for i := 0; i < 1000; i++ {
|
||||||
defer wg.Wait()
|
func() {
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
testCtx, cancel := context.WithCancel(context.Background())
|
conn, err := db.Conn(ctx)
|
||||||
defer cancel()
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
for i := 0; i < 10; i++ {
|
}
|
||||||
wg.Add(1)
|
defer func() {
|
||||||
go func() {
|
if err := conn.Close(); err != nil {
|
||||||
defer wg.Done()
|
t.Error(err)
|
||||||
|
|
||||||
var keyID string
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-testCtx.Done():
|
|
||||||
return
|
|
||||||
default:
|
|
||||||
}
|
}
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
}()
|
||||||
row := db.QueryRowContext(ctx, query)
|
|
||||||
|
|
||||||
cancel()
|
err = conn.Raw(func(driverConn any) error {
|
||||||
_ = row.Scan(&keyID) // see TestQueryRowContextCancel
|
d, ok := driverConn.(driver.ConnBeginTx)
|
||||||
|
if !ok {
|
||||||
|
t.Fatal("unexpected: wrong type")
|
||||||
|
}
|
||||||
|
// checks that conn.Raw can be used to get *SQLiteConn
|
||||||
|
if _, ok = driverConn.(*SQLiteConn); !ok {
|
||||||
|
t.Fatalf("conn.Raw() driverConn type=%T, expected *SQLiteConn", driverConn)
|
||||||
|
}
|
||||||
|
|
||||||
|
go cancel() // make it cancel concurrently with exec("BEGIN");
|
||||||
|
tx, err := d.BeginTx(ctx, driver.TxOptions{})
|
||||||
|
switch err {
|
||||||
|
case nil:
|
||||||
|
switch err := tx.Rollback(); err {
|
||||||
|
case nil, sql.ErrTxDone:
|
||||||
|
default:
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
case context.Canceled:
|
||||||
|
default:
|
||||||
|
// must not fail with "cannot start a transaction within a transaction"
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
|
|
||||||
var keyID string
|
|
||||||
for i := 0; i < 10000; i++ {
|
|
||||||
// note that testCtx is not cancelled during query execution
|
|
||||||
row := db.QueryRowContext(testCtx, query)
|
|
||||||
|
|
||||||
if err := row.Scan(&keyID); err != nil {
|
|
||||||
t.Fatal(i, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestExecCancel(t *testing.T) {
|
func TestStmtReadonly(t *testing.T) {
|
||||||
db, err := sql.Open("sqlite3", ":memory:")
|
db, err := sql.Open("sqlite3", ":memory:")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer db.Close()
|
|
||||||
|
|
||||||
if _, err = db.Exec("create table foo (id integer primary key)"); err != nil {
|
_, err = db.Exec("CREATE TABLE t (count INT)")
|
||||||
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
for n := 0; n < 100; n++ {
|
isRO := func(query string) bool {
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
c, err := db.Conn(context.Background())
|
||||||
_, err = db.ExecContext(ctx, "insert into foo (id) values (?)", n)
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
return false
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func doTestOpenContext(t *testing.T, option string) (string, error) {
|
var ro bool
|
||||||
tempFilename := TempFilename(t)
|
c.Raw(func(dc any) error {
|
||||||
url := tempFilename + option
|
stmt, err := dc.(*SQLiteConn).Prepare(query)
|
||||||
|
if err != nil {
|
||||||
defer func() {
|
return err
|
||||||
err := os.Remove(tempFilename)
|
|
||||||
if err != nil {
|
|
||||||
t.Error("temp file remove error:", err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
db, err := sql.Open("sqlite3", url)
|
|
||||||
if err != nil {
|
|
||||||
return "Failed to open database:", err
|
|
||||||
}
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
err = db.Close()
|
|
||||||
if err != nil {
|
|
||||||
t.Error("db close error:", err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
err = db.PingContext(ctx)
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
return "ping error:", err
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
_, err = db.ExecContext(ctx, "drop table foo")
|
|
||||||
cancel()
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
_, err = db.ExecContext(ctx, "create table foo (id integer)")
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
return "Failed to create table:", err
|
|
||||||
}
|
|
||||||
|
|
||||||
if stat, err := os.Stat(tempFilename); err != nil || stat.IsDir() {
|
|
||||||
return "Failed to create ./foo.db", nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return "", nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestOpenContext(t *testing.T) {
|
|
||||||
cases := map[string]bool{
|
|
||||||
"": true,
|
|
||||||
"?_txlock=immediate": true,
|
|
||||||
"?_txlock=deferred": true,
|
|
||||||
"?_txlock=exclusive": true,
|
|
||||||
"?_txlock=bogus": false,
|
|
||||||
}
|
|
||||||
for option, expectedPass := range cases {
|
|
||||||
result, err := doTestOpenContext(t, option)
|
|
||||||
if result == "" {
|
|
||||||
if !expectedPass {
|
|
||||||
errmsg := fmt.Sprintf("_txlock error not caught at dbOpen with option: %s", option)
|
|
||||||
t.Fatal(errmsg)
|
|
||||||
}
|
}
|
||||||
} else if expectedPass {
|
if stmt == nil {
|
||||||
if err == nil {
|
return errors.New("stmt is nil")
|
||||||
t.Fatal(result)
|
|
||||||
} else {
|
|
||||||
t.Fatal(result, err)
|
|
||||||
}
|
}
|
||||||
}
|
ro = stmt.(*SQLiteStmt).Readonly()
|
||||||
}
|
return nil
|
||||||
}
|
})
|
||||||
|
return ro // On errors ro will remain false.
|
||||||
func TestFileCopyTruncate(t *testing.T) {
|
}
|
||||||
var err error
|
|
||||||
tempFilename := TempFilename(t)
|
if !isRO(`select * from t`) {
|
||||||
|
t.Error("select not seen as read-only")
|
||||||
defer func() {
|
}
|
||||||
err = os.Remove(tempFilename)
|
if isRO(`insert into t values (1), (2)`) {
|
||||||
if err != nil {
|
t.Error("insert seen as read-only")
|
||||||
t.Error("temp file remove error:", err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
db, err := sql.Open("sqlite3", tempFilename)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("open error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
err = db.Close()
|
|
||||||
if err != nil {
|
|
||||||
t.Error("db close error:", err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
err = db.PingContext(ctx)
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("ping error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
_, err = db.ExecContext(ctx, "drop table foo")
|
|
||||||
cancel()
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
_, err = db.ExecContext(ctx, "create table foo (id integer)")
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("create table error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// copy db to new file
|
|
||||||
var data []byte
|
|
||||||
data, err = ioutil.ReadFile(tempFilename)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("read file error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var f *os.File
|
|
||||||
f, err = os.Create(tempFilename + "-db-copy")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("create file error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
err = os.Remove(tempFilename + "-db-copy")
|
|
||||||
if err != nil {
|
|
||||||
t.Error("temp file moved remove error:", err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
_, err = f.Write(data)
|
|
||||||
if err != nil {
|
|
||||||
f.Close()
|
|
||||||
t.Fatal("write file error:", err)
|
|
||||||
}
|
|
||||||
err = f.Close()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("close file error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// truncate current db file
|
|
||||||
f, err = os.OpenFile(tempFilename, os.O_WRONLY|os.O_TRUNC, 0666)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("open file error:", err)
|
|
||||||
}
|
|
||||||
err = f.Close()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("close file error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// test db after file truncate
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
err = db.PingContext(ctx)
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("ping error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
_, err = db.ExecContext(ctx, "drop table foo")
|
|
||||||
cancel()
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("drop table no error")
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
_, err = db.ExecContext(ctx, "create table foo (id integer)")
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("create table error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = db.Close()
|
|
||||||
if err != nil {
|
|
||||||
t.Error("db close error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// test copied file
|
|
||||||
db, err = sql.Open("sqlite3", tempFilename+"-db-copy")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("open error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
err = db.Close()
|
|
||||||
if err != nil {
|
|
||||||
t.Error("db close error:", err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
err = db.PingContext(ctx)
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("ping error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
_, err = db.ExecContext(ctx, "drop table foo")
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("drop table error:", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel = context.WithTimeout(context.Background(), 55*time.Second)
|
|
||||||
_, err = db.ExecContext(ctx, "create table foo (id integer)")
|
|
||||||
cancel()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("create table error:", err)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Reference in New Issue
Block a user