123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222 |
- // Copyright (C) 2018 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.
- // +build sqlite_unlock_notify
- package sqlite3
- import (
- "database/sql"
- "fmt"
- "os"
- "sync"
- "testing"
- "time"
- )
- func TestUnlockNotify(t *testing.T) {
- tempFilename := TempFilename(t)
- defer os.Remove(tempFilename)
- dsn := fmt.Sprintf("file:%s?cache=shared&mode=rwc&_busy_timeout=%d", tempFilename, 500)
- db, err := sql.Open("sqlite3", dsn)
- if err != nil {
- t.Fatal("Failed to open database:", err)
- }
- defer db.Close()
- _, err = db.Exec("CREATE TABLE foo(id INTEGER, status INTEGER)")
- if err != nil {
- t.Fatal("Failed to create table:", err)
- }
- tx, err := db.Begin()
- if err != nil {
- t.Fatal("Failed to begin transaction:", err)
- }
- _, err = tx.Exec("INSERT INTO foo(id, status) VALUES(1, 100)")
- if err != nil {
- t.Fatal("Failed to insert null:", err)
- }
- _, err = tx.Exec("UPDATE foo SET status = 200 WHERE id = 1")
- if err != nil {
- t.Fatal("Failed to update table:", err)
- }
- wg := sync.WaitGroup{}
- wg.Add(1)
- timer := time.NewTimer(500 * time.Millisecond)
- go func() {
- <-timer.C
- err := tx.Commit()
- if err != nil {
- t.Fatal("Failed to commit transaction:", err)
- }
- wg.Done()
- }()
- rows, err := db.Query("SELECT count(*) from foo")
- if err != nil {
- t.Fatal("Unable to query foo table:", err)
- }
- if rows.Next() {
- var count int
- if err := rows.Scan(&count); err != nil {
- t.Fatal("Failed to Scan rows", err)
- }
- }
- if err := rows.Err(); err != nil {
- t.Fatal("Failed at the call to Next:", err)
- }
- wg.Wait()
- }
- func TestUnlockNotifyMany(t *testing.T) {
- tempFilename := TempFilename(t)
- defer os.Remove(tempFilename)
- dsn := fmt.Sprintf("file:%s?cache=shared&mode=rwc&_busy_timeout=%d", tempFilename, 500)
- db, err := sql.Open("sqlite3", dsn)
- if err != nil {
- t.Fatal("Failed to open database:", err)
- }
- defer db.Close()
- _, err = db.Exec("CREATE TABLE foo(id INTEGER, status INTEGER)")
- if err != nil {
- t.Fatal("Failed to create table:", err)
- }
- tx, err := db.Begin()
- if err != nil {
- t.Fatal("Failed to begin transaction:", err)
- }
- _, err = tx.Exec("INSERT INTO foo(id, status) VALUES(1, 100)")
- if err != nil {
- t.Fatal("Failed to insert null:", err)
- }
- _, err = tx.Exec("UPDATE foo SET status = 200 WHERE id = 1")
- if err != nil {
- t.Fatal("Failed to update table:", err)
- }
- wg := sync.WaitGroup{}
- wg.Add(1)
- timer := time.NewTimer(500 * time.Millisecond)
- go func() {
- <-timer.C
- err := tx.Commit()
- if err != nil {
- t.Fatal("Failed to commit transaction:", err)
- }
- wg.Done()
- }()
- const concurrentQueries = 1000
- wg.Add(concurrentQueries)
- for i := 0; i < concurrentQueries; i++ {
- go func() {
- rows, err := db.Query("SELECT count(*) from foo")
- if err != nil {
- t.Fatal("Unable to query foo table:", err)
- }
- if rows.Next() {
- var count int
- if err := rows.Scan(&count); err != nil {
- t.Fatal("Failed to Scan rows", err)
- }
- }
- if err := rows.Err(); err != nil {
- t.Fatal("Failed at the call to Next:", err)
- }
- wg.Done()
- }()
- }
- wg.Wait()
- }
- func TestUnlockNotifyDeadlock(t *testing.T) {
- tempFilename := TempFilename(t)
- defer os.Remove(tempFilename)
- dsn := fmt.Sprintf("file:%s?cache=shared&mode=rwc&_busy_timeout=%d", tempFilename, 500)
- db, err := sql.Open("sqlite3", dsn)
- if err != nil {
- t.Fatal("Failed to open database:", err)
- }
- defer db.Close()
- _, err = db.Exec("CREATE TABLE foo(id INTEGER, status INTEGER)")
- if err != nil {
- t.Fatal("Failed to create table:", err)
- }
- tx, err := db.Begin()
- if err != nil {
- t.Fatal("Failed to begin transaction:", err)
- }
- _, err = tx.Exec("INSERT INTO foo(id, status) VALUES(1, 100)")
- if err != nil {
- t.Fatal("Failed to insert null:", err)
- }
- _, err = tx.Exec("UPDATE foo SET status = 200 WHERE id = 1")
- if err != nil {
- t.Fatal("Failed to update table:", err)
- }
- wg := sync.WaitGroup{}
- wg.Add(1)
- timer := time.NewTimer(500 * time.Millisecond)
- go func() {
- <-timer.C
- err := tx.Commit()
- if err != nil {
- t.Fatal("Failed to commit transaction:", err)
- }
- wg.Done()
- }()
- wg.Add(1)
- go func() {
- tx2, err := db.Begin()
- if err != nil {
- t.Fatal("Failed to begin transaction:", err)
- }
- defer tx2.Rollback()
- _, err = tx2.Exec("DELETE FROM foo")
- if err != nil {
- t.Fatal("Failed to delete table:", err)
- }
- err = tx2.Commit()
- if err != nil {
- t.Fatal("Failed to commit transaction:", err)
- }
- wg.Done()
- }()
- rows, err := tx.Query("SELECT count(*) from foo")
- if err != nil {
- t.Fatal("Unable to query foo table:", err)
- }
- if rows.Next() {
- var count int
- if err := rows.Scan(&count); err != nil {
- t.Fatal("Failed to Scan rows", err)
- }
- }
- if err := rows.Err(); err != nil {
- t.Fatal("Failed at the call to Next:", err)
- }
- wg.Wait()
- }
|