diff options
author | Martin Tournoij <martin@arp242.net> | 2020-12-28 07:52:08 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-12-28 08:52:08 +0900 |
commit | 3cbdae750e52afa881060732446298f98131e834 (patch) | |
tree | c181c1cdeb093cfe1b4d74c8ece55a5fdcd303ee /sqlite3_go113_test.go | |
parent | Update amalgamation code (#896) (diff) | |
download | golite-3cbdae750e52afa881060732446298f98131e834.tar.gz golite-3cbdae750e52afa881060732446298f98131e834.tar.xz |
Export sqlite3_stmt_readonly() via SQLiteStmt.Readonly() (#895)
This can be used like in the test; I wrote a little wrapper around
sql.DB which uses this, and allows concurrent reads but just one single
write. This is perhaps a better generic "table locked"-solution than
setting the connections to 1 and/or cache=shared (although even better
would be to design your app in such a way that this doesn't happpen in
the first place, but even then a little seat belt isn't a bad thing).
The parsing adds about 0.1ms to 0.2ms of overhead in the wrapper, which
isn't too bad (and it caches the results, so only needs to do this
once).
At any rate, I can't really access functions from sqlite3-binding.c from
my application, so expose it via SQLiteStmt.
Diffstat (limited to 'sqlite3_go113_test.go')
-rw-r--r-- | sqlite3_go113_test.go | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/sqlite3_go113_test.go b/sqlite3_go113_test.go index 6f74e6b..a010cb7 100644 --- a/sqlite3_go113_test.go +++ b/sqlite3_go113_test.go @@ -11,6 +11,7 @@ import ( "context" "database/sql" "database/sql/driver" + "errors" "os" "testing" ) @@ -76,3 +77,43 @@ func TestBeginTxCancel(t *testing.T) { }() } } + +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 interface{}) 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") + } +} |