2019-11-18 12:03:31 +03:00
|
|
|
// Copyright (C) 2019 Yasuhiro Matsumoto <mattn.jp@gmail.com>.
|
2018-05-29 13:20:11 +03:00
|
|
|
// Copyright (C) 2018 G.J.R. Timmer <gjr.timmer@gmail.com>.
|
2014-08-18 11:56:31 +04:00
|
|
|
//
|
|
|
|
// Use of this source code is governed by an MIT-style
|
|
|
|
// license that can be found in the LICENSE file.
|
2014-08-18 12:00:59 +04:00
|
|
|
|
2022-01-09 16:38:53 +03:00
|
|
|
//go:build cgo
|
2018-05-27 01:05:45 +03:00
|
|
|
// +build cgo
|
|
|
|
|
2013-08-13 17:10:30 +04:00
|
|
|
package sqlite3
|
2011-11-11 16:36:22 +04:00
|
|
|
|
|
|
|
/*
|
2014-12-19 01:16:42 +03:00
|
|
|
#cgo CFLAGS: -std=gnu99
|
2018-05-27 01:05:45 +03:00
|
|
|
#cgo CFLAGS: -DSQLITE_ENABLE_RTREE
|
|
|
|
#cgo CFLAGS: -DSQLITE_THREADSAFE=1
|
|
|
|
#cgo CFLAGS: -DHAVE_USLEEP=1
|
|
|
|
#cgo CFLAGS: -DSQLITE_ENABLE_FTS3
|
|
|
|
#cgo CFLAGS: -DSQLITE_ENABLE_FTS3_PARENTHESIS
|
2016-09-05 21:58:10 +03:00
|
|
|
#cgo CFLAGS: -DSQLITE_TRACE_SIZE_LIMIT=15
|
2018-05-24 00:10:05 +03:00
|
|
|
#cgo CFLAGS: -DSQLITE_OMIT_DEPRECATED
|
2018-05-24 00:05:26 +03:00
|
|
|
#cgo CFLAGS: -DSQLITE_DEFAULT_WAL_SYNCHRONOUS=1
|
2018-05-23 23:43:42 +03:00
|
|
|
#cgo CFLAGS: -DSQLITE_ENABLE_UPDATE_DELETE_LIMIT
|
2016-10-28 17:22:18 +03:00
|
|
|
#cgo CFLAGS: -Wno-deprecated-declarations
|
2021-10-25 18:02:17 +03:00
|
|
|
#cgo openbsd CFLAGS: -I/usr/local/include
|
|
|
|
#cgo openbsd LDFLAGS: -L/usr/local/lib
|
2016-04-02 13:32:47 +03:00
|
|
|
#ifndef USE_LIBSQLITE3
|
2021-10-25 18:23:19 +03:00
|
|
|
#include "sqlite3-binding.h"
|
2016-04-02 13:32:47 +03:00
|
|
|
#else
|
|
|
|
#include <sqlite3.h>
|
|
|
|
#endif
|
2011-11-11 16:36:22 +04:00
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
|
2013-10-02 07:50:50 +04:00
|
|
|
#ifdef __CYGWIN__
|
|
|
|
# include <errno.h>
|
|
|
|
#endif
|
|
|
|
|
2013-04-06 18:06:48 +04:00
|
|
|
#ifndef SQLITE_OPEN_READWRITE
|
|
|
|
# define SQLITE_OPEN_READWRITE 0
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef SQLITE_OPEN_FULLMUTEX
|
|
|
|
# define SQLITE_OPEN_FULLMUTEX 0
|
|
|
|
#endif
|
|
|
|
|
2016-04-22 11:20:04 +03:00
|
|
|
#ifndef SQLITE_DETERMINISTIC
|
|
|
|
# define SQLITE_DETERMINISTIC 0
|
|
|
|
#endif
|
|
|
|
|
2023-12-08 18:22:05 +03:00
|
|
|
#if defined(HAVE_PREAD64) && defined(HAVE_PWRITE64)
|
|
|
|
# undef USE_PREAD
|
|
|
|
# undef USE_PWRITE
|
|
|
|
# define USE_PREAD64 1
|
|
|
|
# define USE_PWRITE64 1
|
|
|
|
#elif defined(HAVE_PREAD) && defined(HAVE_PWRITE)
|
|
|
|
# undef USE_PREAD
|
|
|
|
# undef USE_PWRITE
|
|
|
|
# define USE_PREAD64 1
|
|
|
|
# define USE_PWRITE64 1
|
|
|
|
#endif
|
|
|
|
|
2013-02-03 18:25:30 +04:00
|
|
|
static int
|
|
|
|
_sqlite3_open_v2(const char *filename, sqlite3 **ppDb, int flags, const char *zVfs) {
|
|
|
|
#ifdef SQLITE_OPEN_URI
|
|
|
|
return sqlite3_open_v2(filename, ppDb, flags | SQLITE_OPEN_URI, zVfs);
|
|
|
|
#else
|
|
|
|
return sqlite3_open_v2(filename, ppDb, flags, zVfs);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2011-11-11 16:36:22 +04:00
|
|
|
static int
|
|
|
|
_sqlite3_bind_text(sqlite3_stmt *stmt, int n, char *p, int np) {
|
|
|
|
return sqlite3_bind_text(stmt, n, p, np, SQLITE_TRANSIENT);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
|
|
_sqlite3_bind_blob(sqlite3_stmt *stmt, int n, void *p, int np) {
|
|
|
|
return sqlite3_bind_blob(stmt, n, p, np, SQLITE_TRANSIENT);
|
|
|
|
}
|
2012-12-07 07:58:08 +04:00
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdint.h>
|
|
|
|
|
2015-03-24 00:17:00 +03:00
|
|
|
static int
|
2015-08-07 06:13:52 +03:00
|
|
|
_sqlite3_exec(sqlite3* db, const char* pcmd, long long* rowid, long long* changes)
|
2015-03-24 00:17:00 +03:00
|
|
|
{
|
|
|
|
int rv = sqlite3_exec(db, pcmd, 0, 0, 0);
|
2015-08-07 06:13:52 +03:00
|
|
|
*rowid = (long long) sqlite3_last_insert_rowid(db);
|
|
|
|
*changes = (long long) sqlite3_changes(db);
|
2015-03-24 00:17:00 +03:00
|
|
|
return rv;
|
2012-12-07 07:58:08 +04:00
|
|
|
}
|
|
|
|
|
2018-09-30 05:06:56 +03:00
|
|
|
#ifdef SQLITE_ENABLE_UNLOCK_NOTIFY
|
2018-10-20 18:02:18 +03:00
|
|
|
extern int _sqlite3_step_blocking(sqlite3_stmt *stmt);
|
|
|
|
extern int _sqlite3_step_row_blocking(sqlite3_stmt* stmt, long long* rowid, long long* changes);
|
|
|
|
extern int _sqlite3_prepare_v2_blocking(sqlite3 *db, const char *zSql, int nBytes, sqlite3_stmt **ppStmt, const char **pzTail);
|
2018-09-30 05:06:56 +03:00
|
|
|
|
|
|
|
static int
|
2018-10-20 18:02:18 +03:00
|
|
|
_sqlite3_step_internal(sqlite3_stmt *stmt)
|
2018-09-30 05:06:56 +03:00
|
|
|
{
|
2018-10-20 18:02:18 +03:00
|
|
|
return _sqlite3_step_blocking(stmt);
|
2018-09-30 05:06:56 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2018-10-20 18:02:18 +03:00
|
|
|
_sqlite3_step_row_internal(sqlite3_stmt* stmt, long long* rowid, long long* changes)
|
2018-09-30 05:06:56 +03:00
|
|
|
{
|
2018-10-20 18:02:18 +03:00
|
|
|
return _sqlite3_step_row_blocking(stmt, rowid, changes);
|
2018-09-30 05:06:56 +03:00
|
|
|
}
|
|
|
|
|
2015-03-24 00:17:00 +03:00
|
|
|
static int
|
2018-10-20 18:02:18 +03:00
|
|
|
_sqlite3_prepare_v2_internal(sqlite3 *db, const char *zSql, int nBytes, sqlite3_stmt **ppStmt, const char **pzTail)
|
2018-09-30 05:06:56 +03:00
|
|
|
{
|
2018-10-20 18:02:18 +03:00
|
|
|
return _sqlite3_prepare_v2_blocking(db, zSql, nBytes, ppStmt, pzTail);
|
2018-09-30 05:06:56 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
#else
|
|
|
|
static int
|
2018-10-20 18:02:18 +03:00
|
|
|
_sqlite3_step_internal(sqlite3_stmt *stmt)
|
2018-09-30 05:06:56 +03:00
|
|
|
{
|
|
|
|
return sqlite3_step(stmt);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2018-10-20 18:02:18 +03:00
|
|
|
_sqlite3_step_row_internal(sqlite3_stmt* stmt, long long* rowid, long long* changes)
|
2015-03-24 00:17:00 +03:00
|
|
|
{
|
|
|
|
int rv = sqlite3_step(stmt);
|
|
|
|
sqlite3* db = sqlite3_db_handle(stmt);
|
2015-08-07 06:13:52 +03:00
|
|
|
*rowid = (long long) sqlite3_last_insert_rowid(db);
|
|
|
|
*changes = (long long) sqlite3_changes(db);
|
2015-03-24 00:17:00 +03:00
|
|
|
return rv;
|
2012-12-07 07:58:08 +04:00
|
|
|
}
|
|
|
|
|
2018-09-30 05:06:56 +03:00
|
|
|
static int
|
2018-10-20 18:02:18 +03:00
|
|
|
_sqlite3_prepare_v2_internal(sqlite3 *db, const char *zSql, int nBytes, sqlite3_stmt **ppStmt, const char **pzTail)
|
2018-09-30 05:06:56 +03:00
|
|
|
{
|
|
|
|
return sqlite3_prepare_v2(db, zSql, nBytes, ppStmt, pzTail);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2015-08-21 09:08:48 +03:00
|
|
|
void _sqlite3_result_text(sqlite3_context* ctx, const char* s) {
|
|
|
|
sqlite3_result_text(ctx, s, -1, &free);
|
|
|
|
}
|
|
|
|
|
|
|
|
void _sqlite3_result_blob(sqlite3_context* ctx, const void* b, int l) {
|
|
|
|
sqlite3_result_blob(ctx, b, l, SQLITE_TRANSIENT);
|
|
|
|
}
|
|
|
|
|
2015-12-30 17:29:15 +03:00
|
|
|
|
|
|
|
int _sqlite3_create_function(
|
|
|
|
sqlite3 *db,
|
|
|
|
const char *zFunctionName,
|
|
|
|
int nArg,
|
|
|
|
int eTextRep,
|
|
|
|
uintptr_t pApp,
|
|
|
|
void (*xFunc)(sqlite3_context*,int,sqlite3_value**),
|
|
|
|
void (*xStep)(sqlite3_context*,int,sqlite3_value**),
|
|
|
|
void (*xFinal)(sqlite3_context*)
|
|
|
|
) {
|
|
|
|
return sqlite3_create_function(db, zFunctionName, nArg, eTextRep, (void*) pApp, xFunc, xStep, xFinal);
|
|
|
|
}
|
|
|
|
|
2015-08-21 09:08:48 +03:00
|
|
|
void callbackTrampoline(sqlite3_context*, int, sqlite3_value**);
|
2017-11-05 05:18:06 +03:00
|
|
|
void stepTrampoline(sqlite3_context*, int, sqlite3_value**);
|
|
|
|
void doneTrampoline(sqlite3_context*);
|
2017-06-09 05:14:07 +03:00
|
|
|
|
|
|
|
int compareTrampoline(void*, int, char*, int, char*);
|
2017-07-03 21:51:48 +03:00
|
|
|
int commitHookTrampoline(void*);
|
|
|
|
void rollbackHookTrampoline(void*);
|
|
|
|
void updateHookTrampoline(void*, int, char*, char*, sqlite3_int64);
|
2017-11-05 02:47:52 +03:00
|
|
|
|
2018-08-25 21:25:16 +03:00
|
|
|
int authorizerTrampoline(void*, int, char*, char*, char*, char*);
|
|
|
|
|
2017-11-05 02:47:52 +03:00
|
|
|
#ifdef SQLITE_LIMIT_WORKER_THREADS
|
|
|
|
# define _SQLITE_HAS_LIMIT
|
|
|
|
# define SQLITE_LIMIT_LENGTH 0
|
|
|
|
# define SQLITE_LIMIT_SQL_LENGTH 1
|
|
|
|
# define SQLITE_LIMIT_COLUMN 2
|
|
|
|
# define SQLITE_LIMIT_EXPR_DEPTH 3
|
|
|
|
# define SQLITE_LIMIT_COMPOUND_SELECT 4
|
|
|
|
# define SQLITE_LIMIT_VDBE_OP 5
|
|
|
|
# define SQLITE_LIMIT_FUNCTION_ARG 6
|
|
|
|
# define SQLITE_LIMIT_ATTACHED 7
|
|
|
|
# define SQLITE_LIMIT_LIKE_PATTERN_LENGTH 8
|
|
|
|
# define SQLITE_LIMIT_VARIABLE_NUMBER 9
|
|
|
|
# define SQLITE_LIMIT_TRIGGER_DEPTH 10
|
|
|
|
# define SQLITE_LIMIT_WORKER_THREADS 11
|
2017-11-05 14:01:16 +03:00
|
|
|
# else
|
|
|
|
# define SQLITE_LIMIT_WORKER_THREADS 11
|
2017-11-05 02:47:52 +03:00
|
|
|
#endif
|
|
|
|
|
|
|
|
static int _sqlite3_limit(sqlite3* db, int limitId, int newLimit) {
|
|
|
|
#ifndef _SQLITE_HAS_LIMIT
|
|
|
|
return -1;
|
|
|
|
#else
|
|
|
|
return sqlite3_limit(db, limitId, newLimit);
|
|
|
|
#endif
|
|
|
|
}
|
2019-12-17 09:58:28 +03:00
|
|
|
|
|
|
|
#if SQLITE_VERSION_NUMBER < 3012000
|
|
|
|
static int sqlite3_system_errno(sqlite3 *db) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
2011-11-11 16:36:22 +04:00
|
|
|
*/
|
|
|
|
import "C"
|
|
|
|
import (
|
2017-09-06 05:54:16 +03:00
|
|
|
"context"
|
2012-01-20 20:44:24 +04:00
|
|
|
"database/sql"
|
|
|
|
"database/sql/driver"
|
2012-03-01 20:52:03 +04:00
|
|
|
"errors"
|
2014-11-30 01:08:02 +03:00
|
|
|
"fmt"
|
2012-03-01 20:52:03 +04:00
|
|
|
"io"
|
2015-03-04 16:49:17 +03:00
|
|
|
"net/url"
|
2015-08-21 09:08:48 +03:00
|
|
|
"reflect"
|
2014-11-14 11:13:35 +03:00
|
|
|
"runtime"
|
2015-01-02 09:31:46 +03:00
|
|
|
"strconv"
|
2012-04-07 08:17:54 +04:00
|
|
|
"strings"
|
2017-06-17 22:22:09 +03:00
|
|
|
"sync"
|
2019-12-17 09:58:28 +03:00
|
|
|
"syscall"
|
2012-04-07 08:17:54 +04:00
|
|
|
"time"
|
2011-11-11 16:36:22 +04:00
|
|
|
"unsafe"
|
|
|
|
)
|
|
|
|
|
2016-11-04 18:40:06 +03:00
|
|
|
// SQLiteTimestampFormats is timestamp formats understood by both this module
|
|
|
|
// and SQLite. The first format in the slice will be used when saving time
|
|
|
|
// values into the database. When parsing a string from a timestamp or datetime
|
|
|
|
// column, the formats are tried in order.
|
2012-12-30 04:36:29 +04:00
|
|
|
var SQLiteTimestampFormats = []string{
|
2015-10-10 08:59:25 +03:00
|
|
|
// By default, store timestamps with whatever timezone they come with.
|
|
|
|
// When parsed, they will be returned with the same timezone.
|
|
|
|
"2006-01-02 15:04:05.999999999-07:00",
|
|
|
|
"2006-01-02T15:04:05.999999999-07:00",
|
2012-12-30 04:36:29 +04:00
|
|
|
"2006-01-02 15:04:05.999999999",
|
2012-12-30 04:51:15 +04:00
|
|
|
"2006-01-02T15:04:05.999999999",
|
2012-12-30 04:36:29 +04:00
|
|
|
"2006-01-02 15:04:05",
|
2012-12-30 04:51:15 +04:00
|
|
|
"2006-01-02T15:04:05",
|
|
|
|
"2006-01-02 15:04",
|
|
|
|
"2006-01-02T15:04",
|
2012-12-30 04:36:29 +04:00
|
|
|
"2006-01-02",
|
|
|
|
}
|
2012-04-07 08:17:54 +04:00
|
|
|
|
2018-04-17 12:13:35 +03:00
|
|
|
const (
|
|
|
|
columnDate string = "date"
|
|
|
|
columnDatetime string = "datetime"
|
|
|
|
columnTimestamp string = "timestamp"
|
|
|
|
)
|
|
|
|
|
2022-01-09 16:38:53 +03:00
|
|
|
// This variable can be replaced with -ldflags like below:
|
|
|
|
// go build -ldflags="-X 'github.com/mattn/go-sqlite3.driverName=my-sqlite3'"
|
|
|
|
var driverName = "sqlite3"
|
|
|
|
|
2011-11-11 16:36:22 +04:00
|
|
|
func init() {
|
2022-01-09 16:38:53 +03:00
|
|
|
if driverName != "" {
|
|
|
|
sql.Register(driverName, &SQLiteDriver{})
|
|
|
|
}
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2016-02-23 09:20:57 +03:00
|
|
|
// Version returns SQLite library version information.
|
2016-11-04 18:40:06 +03:00
|
|
|
func Version() (libVersion string, libVersionNumber int, sourceID string) {
|
2014-10-13 06:05:49 +04:00
|
|
|
libVersion = C.GoString(C.sqlite3_libversion())
|
|
|
|
libVersionNumber = int(C.sqlite3_libversion_number())
|
2016-11-04 18:40:06 +03:00
|
|
|
sourceID = C.GoString(C.sqlite3_sourceid())
|
|
|
|
return libVersion, libVersionNumber, sourceID
|
2014-10-13 06:05:49 +04:00
|
|
|
}
|
|
|
|
|
2017-07-03 21:51:48 +03:00
|
|
|
const (
|
2018-08-30 11:25:56 +03:00
|
|
|
// used by authorizer and pre_update_hook
|
2017-07-03 21:51:48 +03:00
|
|
|
SQLITE_DELETE = C.SQLITE_DELETE
|
|
|
|
SQLITE_INSERT = C.SQLITE_INSERT
|
|
|
|
SQLITE_UPDATE = C.SQLITE_UPDATE
|
2018-08-30 11:25:56 +03:00
|
|
|
|
|
|
|
// used by authorzier - as return value
|
|
|
|
SQLITE_OK = C.SQLITE_OK
|
|
|
|
SQLITE_IGNORE = C.SQLITE_IGNORE
|
|
|
|
SQLITE_DENY = C.SQLITE_DENY
|
|
|
|
|
|
|
|
// different actions query tries to do - passed as argument to authorizer
|
2018-08-25 21:25:16 +03:00
|
|
|
SQLITE_CREATE_INDEX = C.SQLITE_CREATE_INDEX
|
|
|
|
SQLITE_CREATE_TABLE = C.SQLITE_CREATE_TABLE
|
|
|
|
SQLITE_CREATE_TEMP_INDEX = C.SQLITE_CREATE_TEMP_INDEX
|
|
|
|
SQLITE_CREATE_TEMP_TABLE = C.SQLITE_CREATE_TEMP_TABLE
|
|
|
|
SQLITE_CREATE_TEMP_TRIGGER = C.SQLITE_CREATE_TEMP_TRIGGER
|
|
|
|
SQLITE_CREATE_TEMP_VIEW = C.SQLITE_CREATE_TEMP_VIEW
|
|
|
|
SQLITE_CREATE_TRIGGER = C.SQLITE_CREATE_TRIGGER
|
|
|
|
SQLITE_CREATE_VIEW = C.SQLITE_CREATE_VIEW
|
2018-08-30 11:25:56 +03:00
|
|
|
SQLITE_CREATE_VTABLE = C.SQLITE_CREATE_VTABLE
|
2018-08-25 21:25:16 +03:00
|
|
|
SQLITE_DROP_INDEX = C.SQLITE_DROP_INDEX
|
|
|
|
SQLITE_DROP_TABLE = C.SQLITE_DROP_TABLE
|
|
|
|
SQLITE_DROP_TEMP_INDEX = C.SQLITE_DROP_TEMP_INDEX
|
|
|
|
SQLITE_DROP_TEMP_TABLE = C.SQLITE_DROP_TEMP_TABLE
|
|
|
|
SQLITE_DROP_TEMP_TRIGGER = C.SQLITE_DROP_TEMP_TRIGGER
|
|
|
|
SQLITE_DROP_TEMP_VIEW = C.SQLITE_DROP_TEMP_VIEW
|
|
|
|
SQLITE_DROP_TRIGGER = C.SQLITE_DROP_TRIGGER
|
|
|
|
SQLITE_DROP_VIEW = C.SQLITE_DROP_VIEW
|
2018-08-30 11:25:56 +03:00
|
|
|
SQLITE_DROP_VTABLE = C.SQLITE_DROP_VTABLE
|
2018-08-25 21:25:16 +03:00
|
|
|
SQLITE_PRAGMA = C.SQLITE_PRAGMA
|
|
|
|
SQLITE_READ = C.SQLITE_READ
|
|
|
|
SQLITE_SELECT = C.SQLITE_SELECT
|
|
|
|
SQLITE_TRANSACTION = C.SQLITE_TRANSACTION
|
|
|
|
SQLITE_ATTACH = C.SQLITE_ATTACH
|
|
|
|
SQLITE_DETACH = C.SQLITE_DETACH
|
|
|
|
SQLITE_ALTER_TABLE = C.SQLITE_ALTER_TABLE
|
|
|
|
SQLITE_REINDEX = C.SQLITE_REINDEX
|
|
|
|
SQLITE_ANALYZE = C.SQLITE_ANALYZE
|
|
|
|
SQLITE_FUNCTION = C.SQLITE_FUNCTION
|
|
|
|
SQLITE_SAVEPOINT = C.SQLITE_SAVEPOINT
|
|
|
|
SQLITE_COPY = C.SQLITE_COPY
|
2018-09-11 04:29:35 +03:00
|
|
|
/*SQLITE_RECURSIVE = C.SQLITE_RECURSIVE*/
|
2017-07-03 21:51:48 +03:00
|
|
|
)
|
|
|
|
|
2022-01-28 18:37:05 +03:00
|
|
|
// Standard File Control Opcodes
|
|
|
|
// See: https://www.sqlite.org/c3ref/c_fcntl_begin_atomic_write.html
|
|
|
|
const (
|
|
|
|
SQLITE_FCNTL_LOCKSTATE = int(1)
|
|
|
|
SQLITE_FCNTL_GET_LOCKPROXYFILE = int(2)
|
|
|
|
SQLITE_FCNTL_SET_LOCKPROXYFILE = int(3)
|
|
|
|
SQLITE_FCNTL_LAST_ERRNO = int(4)
|
|
|
|
SQLITE_FCNTL_SIZE_HINT = int(5)
|
|
|
|
SQLITE_FCNTL_CHUNK_SIZE = int(6)
|
|
|
|
SQLITE_FCNTL_FILE_POINTER = int(7)
|
|
|
|
SQLITE_FCNTL_SYNC_OMITTED = int(8)
|
|
|
|
SQLITE_FCNTL_WIN32_AV_RETRY = int(9)
|
|
|
|
SQLITE_FCNTL_PERSIST_WAL = int(10)
|
|
|
|
SQLITE_FCNTL_OVERWRITE = int(11)
|
|
|
|
SQLITE_FCNTL_VFSNAME = int(12)
|
|
|
|
SQLITE_FCNTL_POWERSAFE_OVERWRITE = int(13)
|
|
|
|
SQLITE_FCNTL_PRAGMA = int(14)
|
|
|
|
SQLITE_FCNTL_BUSYHANDLER = int(15)
|
|
|
|
SQLITE_FCNTL_TEMPFILENAME = int(16)
|
|
|
|
SQLITE_FCNTL_MMAP_SIZE = int(18)
|
|
|
|
SQLITE_FCNTL_TRACE = int(19)
|
|
|
|
SQLITE_FCNTL_HAS_MOVED = int(20)
|
|
|
|
SQLITE_FCNTL_SYNC = int(21)
|
|
|
|
SQLITE_FCNTL_COMMIT_PHASETWO = int(22)
|
|
|
|
SQLITE_FCNTL_WIN32_SET_HANDLE = int(23)
|
|
|
|
SQLITE_FCNTL_WAL_BLOCK = int(24)
|
|
|
|
SQLITE_FCNTL_ZIPVFS = int(25)
|
|
|
|
SQLITE_FCNTL_RBU = int(26)
|
|
|
|
SQLITE_FCNTL_VFS_POINTER = int(27)
|
|
|
|
SQLITE_FCNTL_JOURNAL_POINTER = int(28)
|
|
|
|
SQLITE_FCNTL_WIN32_GET_HANDLE = int(29)
|
|
|
|
SQLITE_FCNTL_PDB = int(30)
|
|
|
|
SQLITE_FCNTL_BEGIN_ATOMIC_WRITE = int(31)
|
|
|
|
SQLITE_FCNTL_COMMIT_ATOMIC_WRITE = int(32)
|
|
|
|
SQLITE_FCNTL_ROLLBACK_ATOMIC_WRITE = int(33)
|
|
|
|
SQLITE_FCNTL_LOCK_TIMEOUT = int(34)
|
|
|
|
SQLITE_FCNTL_DATA_VERSION = int(35)
|
|
|
|
SQLITE_FCNTL_SIZE_LIMIT = int(36)
|
|
|
|
SQLITE_FCNTL_CKPT_DONE = int(37)
|
|
|
|
SQLITE_FCNTL_RESERVE_BYTES = int(38)
|
|
|
|
SQLITE_FCNTL_CKPT_START = int(39)
|
|
|
|
SQLITE_FCNTL_EXTERNAL_READER = int(40)
|
|
|
|
SQLITE_FCNTL_CKSM_FILE = int(41)
|
|
|
|
)
|
|
|
|
|
2018-08-22 03:43:37 +03:00
|
|
|
// SQLiteDriver implements driver.Driver.
|
2011-11-11 16:36:22 +04:00
|
|
|
type SQLiteDriver struct {
|
2013-08-25 07:36:35 +04:00
|
|
|
Extensions []string
|
|
|
|
ConnectHook func(*SQLiteConn) error
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2018-08-22 03:43:37 +03:00
|
|
|
// SQLiteConn implements driver.Conn.
|
2011-11-11 16:36:22 +04:00
|
|
|
type SQLiteConn struct {
|
2017-08-01 18:06:18 +03:00
|
|
|
mu sync.Mutex
|
2015-08-22 06:31:41 +03:00
|
|
|
db *C.sqlite3
|
|
|
|
loc *time.Location
|
|
|
|
txlock string
|
|
|
|
funcs []*functionInfo
|
|
|
|
aggregators []*aggInfo
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2018-08-22 03:43:37 +03:00
|
|
|
// SQLiteTx implements driver.Tx.
|
2011-11-11 16:36:22 +04:00
|
|
|
type SQLiteTx struct {
|
|
|
|
c *SQLiteConn
|
|
|
|
}
|
|
|
|
|
2018-08-22 03:43:37 +03:00
|
|
|
// SQLiteStmt implements driver.Stmt.
|
2013-01-31 11:48:30 +04:00
|
|
|
type SQLiteStmt struct {
|
2017-08-30 07:29:47 +03:00
|
|
|
mu sync.Mutex
|
2013-01-31 11:48:30 +04:00
|
|
|
c *SQLiteConn
|
|
|
|
s *C.sqlite3_stmt
|
|
|
|
t string
|
|
|
|
closed bool
|
2014-07-15 20:11:07 +04:00
|
|
|
cls bool
|
2013-01-31 11:48:30 +04:00
|
|
|
}
|
|
|
|
|
2018-08-22 03:43:37 +03:00
|
|
|
// SQLiteResult implements sql.Result.
|
2013-01-31 11:48:30 +04:00
|
|
|
type SQLiteResult struct {
|
2013-05-11 16:45:48 +04:00
|
|
|
id int64
|
2013-05-11 16:43:31 +04:00
|
|
|
changes int64
|
2013-01-31 11:48:30 +04:00
|
|
|
}
|
|
|
|
|
2018-08-22 03:43:37 +03:00
|
|
|
// SQLiteRows implements driver.Rows.
|
2013-01-31 11:48:30 +04:00
|
|
|
type SQLiteRows struct {
|
|
|
|
s *SQLiteStmt
|
|
|
|
nc int
|
|
|
|
cols []string
|
|
|
|
decltype []string
|
2014-07-15 20:11:07 +04:00
|
|
|
cls bool
|
2017-08-01 19:43:14 +03:00
|
|
|
closed bool
|
2019-09-06 01:26:33 +03:00
|
|
|
ctx context.Context // no better alternative to pass context into Next() method
|
2013-01-31 11:48:30 +04:00
|
|
|
}
|
|
|
|
|
2015-08-21 09:08:48 +03:00
|
|
|
type functionInfo struct {
|
2015-08-22 02:34:55 +03:00
|
|
|
f reflect.Value
|
|
|
|
argConverters []callbackArgConverter
|
|
|
|
variadicConverter callbackArgConverter
|
|
|
|
retConverter callbackRetConverter
|
2015-08-21 09:08:48 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
func (fi *functionInfo) Call(ctx *C.sqlite3_context, argv []*C.sqlite3_value) {
|
2015-08-22 06:31:41 +03:00
|
|
|
args, err := callbackConvertArgs(argv, fi.argConverters, fi.variadicConverter)
|
|
|
|
if err != nil {
|
|
|
|
callbackError(ctx, err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
ret := fi.f.Call(args)
|
2015-08-22 02:34:55 +03:00
|
|
|
|
2015-08-22 06:31:41 +03:00
|
|
|
if len(ret) == 2 && ret[1].Interface() != nil {
|
|
|
|
callbackError(ctx, ret[1].Interface().(error))
|
|
|
|
return
|
2015-08-22 02:34:55 +03:00
|
|
|
}
|
|
|
|
|
2015-08-22 06:31:41 +03:00
|
|
|
err = fi.retConverter(ctx, ret[0])
|
|
|
|
if err != nil {
|
|
|
|
callbackError(ctx, err)
|
|
|
|
return
|
2015-08-21 09:08:48 +03:00
|
|
|
}
|
2015-08-22 06:31:41 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
type aggInfo struct {
|
|
|
|
constructor reflect.Value
|
|
|
|
|
|
|
|
// Active aggregator objects for aggregations in flight. The
|
|
|
|
// aggregators are indexed by a counter stored in the aggregation
|
|
|
|
// user data space provided by sqlite.
|
|
|
|
active map[int64]reflect.Value
|
|
|
|
next int64
|
|
|
|
|
|
|
|
stepArgConverters []callbackArgConverter
|
|
|
|
stepVariadicConverter callbackArgConverter
|
|
|
|
|
|
|
|
doneRetConverter callbackRetConverter
|
|
|
|
}
|
2015-08-21 09:08:48 +03:00
|
|
|
|
2015-08-22 06:31:41 +03:00
|
|
|
func (ai *aggInfo) agg(ctx *C.sqlite3_context) (int64, reflect.Value, error) {
|
|
|
|
aggIdx := (*int64)(C.sqlite3_aggregate_context(ctx, C.int(8)))
|
|
|
|
if *aggIdx == 0 {
|
|
|
|
*aggIdx = ai.next
|
|
|
|
ret := ai.constructor.Call(nil)
|
|
|
|
if len(ret) == 2 && ret[1].Interface() != nil {
|
|
|
|
return 0, reflect.Value{}, ret[1].Interface().(error)
|
|
|
|
}
|
|
|
|
if ret[0].IsNil() {
|
|
|
|
return 0, reflect.Value{}, errors.New("aggregator constructor returned nil state")
|
2015-08-22 02:34:55 +03:00
|
|
|
}
|
2015-08-22 06:31:41 +03:00
|
|
|
ai.next++
|
|
|
|
ai.active[*aggIdx] = ret[0]
|
2015-08-22 02:34:55 +03:00
|
|
|
}
|
2015-08-22 06:31:41 +03:00
|
|
|
return *aggIdx, ai.active[*aggIdx], nil
|
|
|
|
}
|
2015-08-22 02:34:55 +03:00
|
|
|
|
2015-08-22 06:31:41 +03:00
|
|
|
func (ai *aggInfo) Step(ctx *C.sqlite3_context, argv []*C.sqlite3_value) {
|
|
|
|
_, agg, err := ai.agg(ctx)
|
|
|
|
if err != nil {
|
|
|
|
callbackError(ctx, err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
args, err := callbackConvertArgs(argv, ai.stepArgConverters, ai.stepVariadicConverter)
|
|
|
|
if err != nil {
|
|
|
|
callbackError(ctx, err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
ret := agg.MethodByName("Step").Call(args)
|
|
|
|
if len(ret) == 1 && ret[0].Interface() != nil {
|
|
|
|
callbackError(ctx, ret[0].Interface().(error))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
2015-08-21 09:08:48 +03:00
|
|
|
|
2015-08-22 06:31:41 +03:00
|
|
|
func (ai *aggInfo) Done(ctx *C.sqlite3_context) {
|
|
|
|
idx, agg, err := ai.agg(ctx)
|
|
|
|
if err != nil {
|
|
|
|
callbackError(ctx, err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
defer func() { delete(ai.active, idx) }()
|
|
|
|
|
|
|
|
ret := agg.MethodByName("Done").Call(nil)
|
2015-08-21 09:08:48 +03:00
|
|
|
if len(ret) == 2 && ret[1].Interface() != nil {
|
2015-08-22 06:31:41 +03:00
|
|
|
callbackError(ctx, ret[1].Interface().(error))
|
2015-08-21 09:08:48 +03:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2015-08-22 06:31:41 +03:00
|
|
|
err = ai.doneRetConverter(ctx, ret[0])
|
2015-08-21 23:38:22 +03:00
|
|
|
if err != nil {
|
2015-08-22 06:31:41 +03:00
|
|
|
callbackError(ctx, err)
|
2015-08-21 09:08:48 +03:00
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-01-31 11:48:30 +04:00
|
|
|
// Commit transaction.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (tx *SQLiteTx) Commit() error {
|
2016-11-08 06:19:13 +03:00
|
|
|
_, err := tx.c.exec(context.Background(), "COMMIT", nil)
|
2022-09-02 05:45:11 +03:00
|
|
|
if err != nil {
|
|
|
|
// sqlite3 may leave the transaction open in this scenario.
|
2016-04-18 14:49:17 +03:00
|
|
|
// However, database/sql considers the transaction complete once we
|
|
|
|
// return from Commit() - we must clean up to honour its semantics.
|
2022-09-02 05:45:11 +03:00
|
|
|
// We don't know if the ROLLBACK is strictly necessary, but according
|
|
|
|
// to sqlite's docs, there is no harm in calling ROLLBACK unnecessarily.
|
2016-11-08 06:19:13 +03:00
|
|
|
tx.c.exec(context.Background(), "ROLLBACK", nil)
|
2016-04-18 14:49:17 +03:00
|
|
|
}
|
2014-06-25 06:41:58 +04:00
|
|
|
return err
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2013-01-31 11:48:30 +04:00
|
|
|
// Rollback transaction.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (tx *SQLiteTx) Rollback() error {
|
2016-11-08 06:19:13 +03:00
|
|
|
_, err := tx.c.exec(context.Background(), "ROLLBACK", nil)
|
2014-06-25 06:41:58 +04:00
|
|
|
return err
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2017-06-09 05:14:07 +03:00
|
|
|
// RegisterCollation makes a Go function available as a collation.
|
|
|
|
//
|
|
|
|
// cmp receives two UTF-8 strings, a and b. The result should be 0 if
|
|
|
|
// a==b, -1 if a < b, and +1 if a > b.
|
|
|
|
//
|
|
|
|
// cmp must always return the same result given the same
|
|
|
|
// inputs. Additionally, it must have the following properties for all
|
|
|
|
// strings A, B and C: if A==B then B==A; if A==B and B==C then A==C;
|
|
|
|
// if A<B then B>A; if A<B and B<C then A<C.
|
|
|
|
//
|
|
|
|
// If cmp does not obey these constraints, sqlite3's behavior is
|
|
|
|
// undefined when the collation is used.
|
|
|
|
func (c *SQLiteConn) RegisterCollation(name string, cmp func(string, string) int) error {
|
|
|
|
handle := newHandle(c, cmp)
|
|
|
|
cname := C.CString(name)
|
|
|
|
defer C.free(unsafe.Pointer(cname))
|
2020-08-26 03:36:43 +03:00
|
|
|
rv := C.sqlite3_create_collation(c.db, cname, C.SQLITE_UTF8, handle, (*[0]byte)(unsafe.Pointer(C.compareTrampoline)))
|
2017-06-09 05:14:07 +03:00
|
|
|
if rv != C.SQLITE_OK {
|
|
|
|
return c.lastError()
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-07-03 21:51:48 +03:00
|
|
|
// RegisterCommitHook sets the commit hook for a connection.
|
|
|
|
//
|
|
|
|
// If the callback returns non-zero the transaction will become a rollback.
|
|
|
|
//
|
|
|
|
// If there is an existing commit hook for this connection, it will be
|
|
|
|
// removed. If callback is nil the existing hook (if any) will be removed
|
|
|
|
// without creating a new one.
|
|
|
|
func (c *SQLiteConn) RegisterCommitHook(callback func() int) {
|
|
|
|
if callback == nil {
|
|
|
|
C.sqlite3_commit_hook(c.db, nil, nil)
|
|
|
|
} else {
|
2020-08-26 03:36:43 +03:00
|
|
|
C.sqlite3_commit_hook(c.db, (*[0]byte)(C.commitHookTrampoline), newHandle(c, callback))
|
2017-07-03 21:51:48 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// RegisterRollbackHook sets the rollback hook for a connection.
|
|
|
|
//
|
|
|
|
// If there is an existing rollback hook for this connection, it will be
|
|
|
|
// removed. If callback is nil the existing hook (if any) will be removed
|
|
|
|
// without creating a new one.
|
|
|
|
func (c *SQLiteConn) RegisterRollbackHook(callback func()) {
|
|
|
|
if callback == nil {
|
|
|
|
C.sqlite3_rollback_hook(c.db, nil, nil)
|
|
|
|
} else {
|
2020-08-26 03:36:43 +03:00
|
|
|
C.sqlite3_rollback_hook(c.db, (*[0]byte)(C.rollbackHookTrampoline), newHandle(c, callback))
|
2017-07-03 21:51:48 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// RegisterUpdateHook sets the update hook for a connection.
|
|
|
|
//
|
|
|
|
// The parameters to the callback are the operation (one of the constants
|
|
|
|
// SQLITE_INSERT, SQLITE_DELETE, or SQLITE_UPDATE), the database name, the
|
|
|
|
// table name, and the rowid.
|
|
|
|
//
|
|
|
|
// If there is an existing update hook for this connection, it will be
|
|
|
|
// removed. If callback is nil the existing hook (if any) will be removed
|
|
|
|
// without creating a new one.
|
|
|
|
func (c *SQLiteConn) RegisterUpdateHook(callback func(int, string, string, int64)) {
|
|
|
|
if callback == nil {
|
|
|
|
C.sqlite3_update_hook(c.db, nil, nil)
|
|
|
|
} else {
|
2020-08-26 03:36:43 +03:00
|
|
|
C.sqlite3_update_hook(c.db, (*[0]byte)(C.updateHookTrampoline), newHandle(c, callback))
|
2017-07-03 21:51:48 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-25 21:25:16 +03:00
|
|
|
// RegisterAuthorizer sets the authorizer for connection.
|
|
|
|
//
|
|
|
|
// The parameters to the callback are the operation (one of the constants
|
|
|
|
// SQLITE_INSERT, SQLITE_DELETE, or SQLITE_UPDATE), and 1 to 3 arguments,
|
|
|
|
// depending on operation. More details see:
|
|
|
|
// https://www.sqlite.org/c3ref/c_alter_table.html
|
|
|
|
func (c *SQLiteConn) RegisterAuthorizer(callback func(int, string, string, string) int) {
|
|
|
|
if callback == nil {
|
|
|
|
C.sqlite3_set_authorizer(c.db, nil, nil)
|
|
|
|
} else {
|
2020-08-26 03:36:43 +03:00
|
|
|
C.sqlite3_set_authorizer(c.db, (*[0]byte)(C.authorizerTrampoline), newHandle(c, callback))
|
2018-08-25 21:25:16 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-08-21 09:08:48 +03:00
|
|
|
// RegisterFunc makes a Go function available as a SQLite function.
|
|
|
|
//
|
2015-08-22 03:12:18 +03:00
|
|
|
// The Go function can have arguments of the following types: any
|
2024-01-25 16:36:00 +03:00
|
|
|
// numeric type except complex, bool, []byte, string and any.
|
|
|
|
// any arguments are given the direct translation of the SQLite data type:
|
|
|
|
// int64 for INTEGER, float64 for FLOAT, []byte for BLOB, string for TEXT.
|
2015-08-22 03:12:18 +03:00
|
|
|
//
|
|
|
|
// The function can additionally be variadic, as long as the type of
|
|
|
|
// the variadic argument is one of the above.
|
2015-08-21 09:08:48 +03:00
|
|
|
//
|
|
|
|
// If pure is true. SQLite will assume that the function's return
|
|
|
|
// value depends only on its inputs, and make more aggressive
|
|
|
|
// optimizations in its queries.
|
2015-08-22 06:31:41 +03:00
|
|
|
//
|
|
|
|
// See _example/go_custom_funcs for a detailed example.
|
2024-01-25 16:36:00 +03:00
|
|
|
func (c *SQLiteConn) RegisterFunc(name string, impl any, pure bool) error {
|
2015-08-21 09:08:48 +03:00
|
|
|
var fi functionInfo
|
|
|
|
fi.f = reflect.ValueOf(impl)
|
|
|
|
t := fi.f.Type()
|
|
|
|
if t.Kind() != reflect.Func {
|
|
|
|
return errors.New("Non-function passed to RegisterFunc")
|
|
|
|
}
|
|
|
|
if t.NumOut() != 1 && t.NumOut() != 2 {
|
|
|
|
return errors.New("SQLite functions must return 1 or 2 values")
|
|
|
|
}
|
|
|
|
if t.NumOut() == 2 && !t.Out(1).Implements(reflect.TypeOf((*error)(nil)).Elem()) {
|
|
|
|
return errors.New("Second return value of SQLite function must be error")
|
|
|
|
}
|
|
|
|
|
2015-08-22 02:34:55 +03:00
|
|
|
numArgs := t.NumIn()
|
|
|
|
if t.IsVariadic() {
|
|
|
|
numArgs--
|
|
|
|
}
|
|
|
|
|
|
|
|
for i := 0; i < numArgs; i++ {
|
2015-08-21 23:38:22 +03:00
|
|
|
conv, err := callbackArg(t.In(i))
|
|
|
|
if err != nil {
|
|
|
|
return err
|
2015-08-21 09:08:48 +03:00
|
|
|
}
|
|
|
|
fi.argConverters = append(fi.argConverters, conv)
|
|
|
|
}
|
|
|
|
|
2015-08-22 02:34:55 +03:00
|
|
|
if t.IsVariadic() {
|
|
|
|
conv, err := callbackArg(t.In(numArgs).Elem())
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
fi.variadicConverter = conv
|
|
|
|
// Pass -1 to sqlite so that it allows any number of
|
|
|
|
// arguments. The call helper verifies that the minimum number
|
|
|
|
// of arguments is present for variadic functions.
|
|
|
|
numArgs = -1
|
|
|
|
}
|
|
|
|
|
2015-08-21 23:38:22 +03:00
|
|
|
conv, err := callbackRet(t.Out(0))
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
fi.retConverter = conv
|
|
|
|
|
2015-08-21 09:08:48 +03:00
|
|
|
// fi must outlast the database connection, or we'll have dangling pointers.
|
|
|
|
c.funcs = append(c.funcs, &fi)
|
|
|
|
|
|
|
|
cname := C.CString(name)
|
|
|
|
defer C.free(unsafe.Pointer(cname))
|
|
|
|
opts := C.SQLITE_UTF8
|
|
|
|
if pure {
|
|
|
|
opts |= C.SQLITE_DETERMINISTIC
|
|
|
|
}
|
2017-03-04 18:45:41 +03:00
|
|
|
rv := sqlite3CreateFunction(c.db, cname, C.int(numArgs), C.int(opts), newHandle(c, &fi), C.callbackTrampoline, nil, nil)
|
2015-08-22 06:31:41 +03:00
|
|
|
if rv != C.SQLITE_OK {
|
|
|
|
return c.lastError()
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2020-08-26 03:36:43 +03:00
|
|
|
func sqlite3CreateFunction(db *C.sqlite3, zFunctionName *C.char, nArg C.int, eTextRep C.int, pApp unsafe.Pointer, xFunc unsafe.Pointer, xStep unsafe.Pointer, xFinal unsafe.Pointer) C.int {
|
|
|
|
return C._sqlite3_create_function(db, zFunctionName, nArg, eTextRep, C.uintptr_t(uintptr(pApp)), (*[0]byte)(xFunc), (*[0]byte)(xStep), (*[0]byte)(xFinal))
|
2016-11-08 06:19:13 +03:00
|
|
|
}
|
|
|
|
|
2017-11-05 05:18:06 +03:00
|
|
|
// RegisterAggregator makes a Go type available as a SQLite aggregation function.
|
|
|
|
//
|
|
|
|
// Because aggregation is incremental, it's implemented in Go with a
|
|
|
|
// type that has 2 methods: func Step(values) accumulates one row of
|
|
|
|
// data into the accumulator, and func Done() ret finalizes and
|
|
|
|
// returns the aggregate value. "values" and "ret" may be any type
|
|
|
|
// supported by RegisterFunc.
|
|
|
|
//
|
|
|
|
// RegisterAggregator takes as implementation a constructor function
|
|
|
|
// that constructs an instance of the aggregator type each time an
|
|
|
|
// aggregation begins. The constructor must return a pointer to a
|
|
|
|
// type, or an interface that implements Step() and Done().
|
|
|
|
//
|
|
|
|
// The constructor function and the Step/Done methods may optionally
|
|
|
|
// return an error in addition to their other return values.
|
|
|
|
//
|
|
|
|
// See _example/go_custom_funcs for a detailed example.
|
2024-01-25 16:36:00 +03:00
|
|
|
func (c *SQLiteConn) RegisterAggregator(name string, impl any, pure bool) error {
|
2017-11-05 05:18:06 +03:00
|
|
|
var ai aggInfo
|
|
|
|
ai.constructor = reflect.ValueOf(impl)
|
|
|
|
t := ai.constructor.Type()
|
|
|
|
if t.Kind() != reflect.Func {
|
|
|
|
return errors.New("non-function passed to RegisterAggregator")
|
|
|
|
}
|
|
|
|
if t.NumOut() != 1 && t.NumOut() != 2 {
|
|
|
|
return errors.New("SQLite aggregator constructors must return 1 or 2 values")
|
|
|
|
}
|
|
|
|
if t.NumOut() == 2 && !t.Out(1).Implements(reflect.TypeOf((*error)(nil)).Elem()) {
|
|
|
|
return errors.New("Second return value of SQLite function must be error")
|
|
|
|
}
|
|
|
|
if t.NumIn() != 0 {
|
|
|
|
return errors.New("SQLite aggregator constructors must not have arguments")
|
|
|
|
}
|
|
|
|
|
|
|
|
agg := t.Out(0)
|
|
|
|
switch agg.Kind() {
|
|
|
|
case reflect.Ptr, reflect.Interface:
|
|
|
|
default:
|
|
|
|
return errors.New("SQlite aggregator constructor must return a pointer object")
|
|
|
|
}
|
|
|
|
stepFn, found := agg.MethodByName("Step")
|
|
|
|
if !found {
|
|
|
|
return errors.New("SQlite aggregator doesn't have a Step() function")
|
|
|
|
}
|
|
|
|
step := stepFn.Type
|
|
|
|
if step.NumOut() != 0 && step.NumOut() != 1 {
|
|
|
|
return errors.New("SQlite aggregator Step() function must return 0 or 1 values")
|
|
|
|
}
|
|
|
|
if step.NumOut() == 1 && !step.Out(0).Implements(reflect.TypeOf((*error)(nil)).Elem()) {
|
|
|
|
return errors.New("type of SQlite aggregator Step() return value must be error")
|
|
|
|
}
|
|
|
|
|
|
|
|
stepNArgs := step.NumIn()
|
|
|
|
start := 0
|
|
|
|
if agg.Kind() == reflect.Ptr {
|
|
|
|
// Skip over the method receiver
|
|
|
|
stepNArgs--
|
|
|
|
start++
|
|
|
|
}
|
|
|
|
if step.IsVariadic() {
|
|
|
|
stepNArgs--
|
|
|
|
}
|
|
|
|
for i := start; i < start+stepNArgs; i++ {
|
|
|
|
conv, err := callbackArg(step.In(i))
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
ai.stepArgConverters = append(ai.stepArgConverters, conv)
|
|
|
|
}
|
|
|
|
if step.IsVariadic() {
|
2019-05-10 17:23:32 +03:00
|
|
|
conv, err := callbackArg(step.In(start + stepNArgs).Elem())
|
2017-11-05 05:18:06 +03:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
ai.stepVariadicConverter = conv
|
|
|
|
// Pass -1 to sqlite so that it allows any number of
|
|
|
|
// arguments. The call helper verifies that the minimum number
|
|
|
|
// of arguments is present for variadic functions.
|
|
|
|
stepNArgs = -1
|
|
|
|
}
|
|
|
|
|
|
|
|
doneFn, found := agg.MethodByName("Done")
|
|
|
|
if !found {
|
|
|
|
return errors.New("SQlite aggregator doesn't have a Done() function")
|
|
|
|
}
|
|
|
|
done := doneFn.Type
|
|
|
|
doneNArgs := done.NumIn()
|
|
|
|
if agg.Kind() == reflect.Ptr {
|
|
|
|
// Skip over the method receiver
|
|
|
|
doneNArgs--
|
|
|
|
}
|
|
|
|
if doneNArgs != 0 {
|
|
|
|
return errors.New("SQlite aggregator Done() function must have no arguments")
|
|
|
|
}
|
|
|
|
if done.NumOut() != 1 && done.NumOut() != 2 {
|
|
|
|
return errors.New("SQLite aggregator Done() function must return 1 or 2 values")
|
|
|
|
}
|
|
|
|
if done.NumOut() == 2 && !done.Out(1).Implements(reflect.TypeOf((*error)(nil)).Elem()) {
|
|
|
|
return errors.New("second return value of SQLite aggregator Done() function must be error")
|
|
|
|
}
|
|
|
|
|
|
|
|
conv, err := callbackRet(done.Out(0))
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
ai.doneRetConverter = conv
|
|
|
|
ai.active = make(map[int64]reflect.Value)
|
|
|
|
ai.next = 1
|
|
|
|
|
|
|
|
// ai must outlast the database connection, or we'll have dangling pointers.
|
|
|
|
c.aggregators = append(c.aggregators, &ai)
|
|
|
|
|
|
|
|
cname := C.CString(name)
|
|
|
|
defer C.free(unsafe.Pointer(cname))
|
|
|
|
opts := C.SQLITE_UTF8
|
|
|
|
if pure {
|
|
|
|
opts |= C.SQLITE_DETERMINISTIC
|
|
|
|
}
|
|
|
|
rv := sqlite3CreateFunction(c.db, cname, C.int(stepNArgs), C.int(opts), newHandle(c, &ai), nil, C.stepTrampoline, C.doneTrampoline)
|
|
|
|
if rv != C.SQLITE_OK {
|
|
|
|
return c.lastError()
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2013-09-09 05:44:44 +04:00
|
|
|
// AutoCommit return which currently auto commit or not.
|
2013-08-23 09:11:15 +04:00
|
|
|
func (c *SQLiteConn) AutoCommit() bool {
|
2018-08-30 15:44:58 +03:00
|
|
|
c.mu.Lock()
|
|
|
|
defer c.mu.Unlock()
|
2013-08-23 09:26:33 +04:00
|
|
|
return int(C.sqlite3_get_autocommit(c.db)) != 0
|
2013-08-23 09:11:15 +04:00
|
|
|
}
|
|
|
|
|
2017-03-21 03:14:48 +03:00
|
|
|
func (c *SQLiteConn) lastError() error {
|
2017-04-01 19:12:21 +03:00
|
|
|
return lastError(c.db)
|
|
|
|
}
|
|
|
|
|
2019-12-17 09:58:28 +03:00
|
|
|
// Note: may be called with db == nil
|
2017-04-01 19:12:21 +03:00
|
|
|
func lastError(db *C.sqlite3) error {
|
2019-12-17 09:58:28 +03:00
|
|
|
rv := C.sqlite3_errcode(db) // returns SQLITE_NOMEM if db == nil
|
2017-03-20 17:23:24 +03:00
|
|
|
if rv == C.SQLITE_OK {
|
|
|
|
return nil
|
|
|
|
}
|
2019-12-17 09:58:28 +03:00
|
|
|
extrv := C.sqlite3_extended_errcode(db) // returns SQLITE_NOMEM if db == nil
|
|
|
|
errStr := C.GoString(C.sqlite3_errmsg(db)) // returns "out of memory" if db == nil
|
|
|
|
|
|
|
|
// https://www.sqlite.org/c3ref/system_errno.html
|
|
|
|
// sqlite3_system_errno is only meaningful if the error code was SQLITE_CANTOPEN,
|
|
|
|
// or it was SQLITE_IOERR and the extended code was not SQLITE_IOERR_NOMEM
|
|
|
|
var systemErrno syscall.Errno
|
|
|
|
if rv == C.SQLITE_CANTOPEN || (rv == C.SQLITE_IOERR && extrv != C.SQLITE_IOERR_NOMEM) {
|
|
|
|
systemErrno = syscall.Errno(C.sqlite3_system_errno(db))
|
|
|
|
}
|
|
|
|
|
2017-03-21 03:14:48 +03:00
|
|
|
return Error{
|
2017-03-20 17:23:24 +03:00
|
|
|
Code: ErrNo(rv),
|
2019-12-17 09:58:28 +03:00
|
|
|
ExtendedCode: ErrNoExtended(extrv),
|
|
|
|
SystemErrno: systemErrno,
|
|
|
|
err: errStr,
|
2013-11-19 13:13:19 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-11-04 18:40:06 +03:00
|
|
|
// Exec implements Execer.
|
2014-06-25 06:41:58 +04:00
|
|
|
func (c *SQLiteConn) Exec(query string, args []driver.Value) (driver.Result, error) {
|
2023-02-12 01:14:42 +03:00
|
|
|
list := make([]driver.NamedValue, len(args))
|
2016-11-04 08:24:22 +03:00
|
|
|
for i, v := range args {
|
2023-02-12 01:14:42 +03:00
|
|
|
list[i] = driver.NamedValue{
|
2016-11-04 08:24:22 +03:00
|
|
|
Ordinal: i + 1,
|
|
|
|
Value: v,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return c.exec(context.Background(), query, list)
|
|
|
|
}
|
|
|
|
|
2023-02-12 01:14:42 +03:00
|
|
|
func (c *SQLiteConn) exec(ctx context.Context, query string, args []driver.NamedValue) (driver.Result, error) {
|
2016-11-04 09:00:29 +03:00
|
|
|
start := 0
|
2014-06-25 22:54:09 +04:00
|
|
|
for {
|
2016-11-08 19:13:34 +03:00
|
|
|
s, err := c.prepare(ctx, query)
|
2014-06-25 22:54:09 +04:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
var res driver.Result
|
|
|
|
if s.(*SQLiteStmt).s != nil {
|
2023-02-12 01:14:42 +03:00
|
|
|
stmtArgs := make([]driver.NamedValue, 0, len(args))
|
2014-06-25 22:54:09 +04:00
|
|
|
na := s.NumInput()
|
2020-08-26 03:36:43 +03:00
|
|
|
if len(args)-start < na {
|
2017-01-07 16:22:02 +03:00
|
|
|
s.Close()
|
2017-03-05 16:16:51 +03:00
|
|
|
return nil, fmt.Errorf("not enough args to execute query: want %d got %d", na, len(args))
|
2014-08-18 13:23:58 +04:00
|
|
|
}
|
2020-05-14 17:28:04 +03:00
|
|
|
// consume the number of arguments used in the current
|
|
|
|
// statement and append all named arguments not
|
|
|
|
// contained therein
|
|
|
|
stmtArgs = append(stmtArgs, args[start:start+na]...)
|
|
|
|
for i := range args {
|
|
|
|
if (i < start || i >= na) && args[i].Name != "" {
|
|
|
|
stmtArgs = append(stmtArgs, args[i])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for i := range stmtArgs {
|
|
|
|
stmtArgs[i].Ordinal = i + 1
|
2016-11-04 09:00:29 +03:00
|
|
|
}
|
2020-05-14 17:28:04 +03:00
|
|
|
res, err = s.(*SQLiteStmt).exec(ctx, stmtArgs)
|
2014-06-25 22:54:09 +04:00
|
|
|
if err != nil && err != driver.ErrSkip {
|
|
|
|
s.Close()
|
|
|
|
return nil, err
|
|
|
|
}
|
2016-11-04 09:00:29 +03:00
|
|
|
start += na
|
2014-06-25 22:54:09 +04:00
|
|
|
}
|
|
|
|
tail := s.(*SQLiteStmt).t
|
2014-07-15 20:11:07 +04:00
|
|
|
s.Close()
|
2014-06-25 22:54:09 +04:00
|
|
|
if tail == "" {
|
2021-10-19 12:18:21 +03:00
|
|
|
if res == nil {
|
|
|
|
// https://github.com/mattn/go-sqlite3/issues/963
|
|
|
|
res = &SQLiteResult{0, 0}
|
|
|
|
}
|
2014-06-25 22:54:09 +04:00
|
|
|
return res, nil
|
|
|
|
}
|
|
|
|
query = tail
|
|
|
|
}
|
2014-06-25 06:41:58 +04:00
|
|
|
}
|
|
|
|
|
2016-11-04 18:40:06 +03:00
|
|
|
// Query implements Queryer.
|
2014-06-25 06:41:58 +04:00
|
|
|
func (c *SQLiteConn) Query(query string, args []driver.Value) (driver.Rows, error) {
|
2023-02-12 01:14:42 +03:00
|
|
|
list := make([]driver.NamedValue, len(args))
|
2016-11-04 08:24:22 +03:00
|
|
|
for i, v := range args {
|
2023-02-12 01:14:42 +03:00
|
|
|
list[i] = driver.NamedValue{
|
2016-11-04 08:24:22 +03:00
|
|
|
Ordinal: i + 1,
|
|
|
|
Value: v,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return c.query(context.Background(), query, list)
|
|
|
|
}
|
|
|
|
|
2023-02-12 01:14:42 +03:00
|
|
|
func (c *SQLiteConn) query(ctx context.Context, query string, args []driver.NamedValue) (driver.Rows, error) {
|
2016-11-04 09:00:29 +03:00
|
|
|
start := 0
|
2014-06-25 22:54:09 +04:00
|
|
|
for {
|
2023-02-12 01:14:42 +03:00
|
|
|
stmtArgs := make([]driver.NamedValue, 0, len(args))
|
2016-11-08 19:13:34 +03:00
|
|
|
s, err := c.prepare(ctx, query)
|
2014-06-25 22:54:09 +04:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2014-07-15 20:11:07 +04:00
|
|
|
s.(*SQLiteStmt).cls = true
|
2014-06-25 22:54:09 +04:00
|
|
|
na := s.NumInput()
|
2020-08-26 03:36:43 +03:00
|
|
|
if len(args)-start < na {
|
|
|
|
return nil, fmt.Errorf("not enough args to execute query: want %d got %d", na, len(args)-start)
|
2020-05-14 17:28:04 +03:00
|
|
|
}
|
|
|
|
// consume the number of arguments used in the current
|
|
|
|
// statement and append all named arguments not contained
|
|
|
|
// therein
|
|
|
|
stmtArgs = append(stmtArgs, args[start:start+na]...)
|
|
|
|
for i := range args {
|
|
|
|
if (i < start || i >= na) && args[i].Name != "" {
|
|
|
|
stmtArgs = append(stmtArgs, args[i])
|
|
|
|
}
|
2014-11-30 01:08:02 +03:00
|
|
|
}
|
2020-05-14 17:28:04 +03:00
|
|
|
for i := range stmtArgs {
|
|
|
|
stmtArgs[i].Ordinal = i + 1
|
2016-11-04 09:00:29 +03:00
|
|
|
}
|
2020-05-14 17:28:04 +03:00
|
|
|
rows, err := s.(*SQLiteStmt).query(ctx, stmtArgs)
|
2014-06-25 22:54:09 +04:00
|
|
|
if err != nil && err != driver.ErrSkip {
|
|
|
|
s.Close()
|
2016-11-08 07:22:46 +03:00
|
|
|
return rows, err
|
2014-06-25 22:54:09 +04:00
|
|
|
}
|
2016-11-04 09:00:29 +03:00
|
|
|
start += na
|
2014-06-25 22:54:09 +04:00
|
|
|
tail := s.(*SQLiteStmt).t
|
|
|
|
if tail == "" {
|
|
|
|
return rows, nil
|
|
|
|
}
|
2014-11-13 20:21:49 +03:00
|
|
|
rows.Close()
|
2014-06-25 22:54:09 +04:00
|
|
|
s.Close()
|
|
|
|
query = tail
|
|
|
|
}
|
2014-06-25 06:41:58 +04:00
|
|
|
}
|
|
|
|
|
2013-01-31 11:48:30 +04:00
|
|
|
// Begin transaction.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (c *SQLiteConn) Begin() (driver.Tx, error) {
|
2016-11-04 09:15:16 +03:00
|
|
|
return c.begin(context.Background())
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *SQLiteConn) begin(ctx context.Context) (driver.Tx, error) {
|
2016-11-08 06:19:13 +03:00
|
|
|
if _, err := c.exec(ctx, c.txlock, nil); err != nil {
|
2011-11-11 16:36:22 +04:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return &SQLiteTx{c}, nil
|
|
|
|
}
|
|
|
|
|
2013-01-31 11:48:30 +04:00
|
|
|
// Open database and return a new connection.
|
2018-05-29 12:33:43 +03:00
|
|
|
//
|
|
|
|
// A pragma can take either zero or one argument.
|
|
|
|
// The argument is may be either in parentheses or it may be separated from
|
|
|
|
// the pragma name by an equal sign. The two syntaxes yield identical results.
|
|
|
|
// In many pragmas, the argument is a boolean. The boolean can be one of:
|
2024-01-25 16:36:27 +03:00
|
|
|
//
|
|
|
|
// 1 yes true on
|
|
|
|
// 0 no false off
|
2018-05-29 12:33:43 +03:00
|
|
|
//
|
2016-02-23 09:20:57 +03:00
|
|
|
// You can specify a DSN string using a URI as the filename.
|
2018-05-29 14:02:28 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// test.db
|
|
|
|
// file:test.db?cache=shared&mode=memory
|
|
|
|
// :memory:
|
|
|
|
// file::memory:
|
2018-05-29 14:02:28 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// mode
|
|
|
|
// Access mode of the database.
|
|
|
|
// https://www.sqlite.org/c3ref/open.html
|
|
|
|
// Values:
|
|
|
|
// - ro
|
|
|
|
// - rw
|
|
|
|
// - rwc
|
|
|
|
// - memory
|
2018-05-29 14:02:28 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// cache
|
|
|
|
// SQLite Shared-Cache Mode
|
|
|
|
// https://www.sqlite.org/sharedcache.html
|
|
|
|
// Values:
|
|
|
|
// - shared
|
|
|
|
// - private
|
2018-05-29 14:23:39 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// immutable=Boolean
|
|
|
|
// The immutable parameter is a boolean query parameter that indicates
|
|
|
|
// that the database file is stored on read-only media. When immutable is set,
|
|
|
|
// SQLite assumes that the database file cannot be changed,
|
|
|
|
// even by a process with higher privilege,
|
|
|
|
// and so the database is opened read-only and all locking and change detection is disabled.
|
|
|
|
// Caution: Setting the immutable property on a database file that
|
|
|
|
// does in fact change can result in incorrect query results and/or SQLITE_CORRUPT errors.
|
2018-05-29 12:33:43 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// go-sqlite3 adds the following query parameters to those used by SQLite:
|
2018-05-29 12:33:43 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _loc=XXX
|
|
|
|
// Specify location of time format. It's possible to specify "auto".
|
2018-05-29 12:33:43 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _mutex=XXX
|
|
|
|
// Specify mutex mode. XXX can be "no", "full".
|
2018-05-29 15:01:33 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _txlock=XXX
|
|
|
|
// Specify locking behavior for transactions. XXX can be "immediate",
|
|
|
|
// "deferred", "exclusive".
|
2018-05-29 12:33:43 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _auto_vacuum=X | _vacuum=X
|
|
|
|
// 0 | none - Auto Vacuum disabled
|
|
|
|
// 1 | full - Auto Vacuum FULL
|
|
|
|
// 2 | incremental - Auto Vacuum Incremental
|
2018-05-29 12:46:53 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _busy_timeout=XXX"| _timeout=XXX
|
|
|
|
// Specify value for sqlite3_busy_timeout.
|
2018-05-29 13:09:56 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _case_sensitive_like=Boolean | _cslike=Boolean
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_case_sensitive_like
|
|
|
|
// Default or disabled the LIKE operation is case-insensitive.
|
|
|
|
// When enabling this options behaviour of LIKE will become case-sensitive.
|
2018-05-29 12:33:43 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _defer_foreign_keys=Boolean | _defer_fk=Boolean
|
|
|
|
// Defer Foreign Keys until outermost transaction is committed.
|
2018-05-29 13:19:46 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _foreign_keys=Boolean | _fk=Boolean
|
|
|
|
// Enable or disable enforcement of foreign keys.
|
2018-05-29 14:13:38 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _ignore_check_constraints=Boolean
|
|
|
|
// This pragma enables or disables the enforcement of CHECK constraints.
|
|
|
|
// The default setting is off, meaning that CHECK constraints are enforced by default.
|
2018-05-29 14:19:40 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _journal_mode=MODE | _journal=MODE
|
|
|
|
// Set journal mode for the databases associated with the current connection.
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_journal_mode
|
2018-05-29 14:29:06 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _locking_mode=X | _locking=X
|
|
|
|
// Sets the database connection locking-mode.
|
|
|
|
// The locking-mode is either NORMAL or EXCLUSIVE.
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_locking_mode
|
2018-05-29 12:33:43 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _query_only=Boolean
|
|
|
|
// The query_only pragma prevents all changes to database files when enabled.
|
2018-05-29 14:41:52 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _recursive_triggers=Boolean | _rt=Boolean
|
|
|
|
// Enable or disable recursive triggers.
|
2018-05-29 14:55:31 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _secure_delete=Boolean|FAST
|
|
|
|
// When secure_delete is on, SQLite overwrites deleted content with zeros.
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_secure_delete
|
2018-05-29 15:01:33 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _synchronous=X | _sync=X
|
|
|
|
// Change the setting of the "synchronous" flag.
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_synchronous
|
2018-05-29 15:01:33 +03:00
|
|
|
//
|
2024-01-25 16:36:27 +03:00
|
|
|
// _writable_schema=Boolean
|
|
|
|
// When this pragma is on, the SQLITE_MASTER tables in which database
|
|
|
|
// can be changed using ordinary UPDATE, INSERT, and DELETE statements.
|
|
|
|
// Warning: misuse of this pragma can easily result in a corrupt database file.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (d *SQLiteDriver) Open(dsn string) (driver.Conn, error) {
|
|
|
|
if C.sqlite3_threadsafe() == 0 {
|
|
|
|
return nil, errors.New("sqlite library was not compiled for thread-safe operation")
|
|
|
|
}
|
|
|
|
|
2018-05-29 12:53:36 +03:00
|
|
|
var pkey string
|
|
|
|
|
2018-05-29 12:33:43 +03:00
|
|
|
// Options
|
2015-03-04 16:49:17 +03:00
|
|
|
var loc *time.Location
|
2018-05-30 23:36:49 +03:00
|
|
|
authCreate := false
|
|
|
|
authUser := ""
|
|
|
|
authPass := ""
|
2018-06-05 14:45:32 +03:00
|
|
|
authCrypt := ""
|
|
|
|
authSalt := ""
|
2018-12-08 16:01:50 +03:00
|
|
|
mutex := C.int(C.SQLITE_OPEN_FULLMUTEX)
|
2015-04-10 19:32:18 +03:00
|
|
|
txlock := "BEGIN"
|
2018-05-29 12:33:43 +03:00
|
|
|
|
|
|
|
// PRAGMA's
|
|
|
|
autoVacuum := -1
|
2016-11-04 18:40:06 +03:00
|
|
|
busyTimeout := 5000
|
2018-05-29 12:46:53 +03:00
|
|
|
caseSensitiveLike := -1
|
2018-05-29 13:09:56 +03:00
|
|
|
deferForeignKeys := -1
|
2018-05-29 14:19:40 +03:00
|
|
|
foreignKeys := -1
|
2018-05-29 13:19:46 +03:00
|
|
|
ignoreCheckConstraints := -1
|
2019-09-24 00:50:49 +03:00
|
|
|
var journalMode string
|
2018-05-29 14:19:40 +03:00
|
|
|
lockingMode := "NORMAL"
|
2018-05-29 14:29:06 +03:00
|
|
|
queryOnly := -1
|
2017-07-09 17:32:14 +03:00
|
|
|
recursiveTriggers := -1
|
2018-05-29 14:41:52 +03:00
|
|
|
secureDelete := "DEFAULT"
|
2018-05-29 14:55:31 +03:00
|
|
|
synchronousMode := "NORMAL"
|
2018-05-29 15:01:33 +03:00
|
|
|
writableSchema := -1
|
2020-11-16 19:54:21 +03:00
|
|
|
vfsName := ""
|
2020-12-26 17:05:20 +03:00
|
|
|
var cacheSize *int64
|
2018-05-29 12:33:43 +03:00
|
|
|
|
2015-03-05 05:05:58 +03:00
|
|
|
pos := strings.IndexRune(dsn, '?')
|
|
|
|
if pos >= 1 {
|
|
|
|
params, err := url.ParseQuery(dsn[pos+1:])
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-30 23:36:49 +03:00
|
|
|
// Authentication
|
|
|
|
if _, ok := params["_auth"]; ok {
|
|
|
|
authCreate = true
|
|
|
|
}
|
|
|
|
if val := params.Get("_auth_user"); val != "" {
|
|
|
|
authUser = val
|
|
|
|
}
|
|
|
|
if val := params.Get("_auth_pass"); val != "" {
|
|
|
|
authPass = val
|
|
|
|
}
|
2018-06-05 14:45:32 +03:00
|
|
|
if val := params.Get("_auth_crypt"); val != "" {
|
|
|
|
authCrypt = val
|
|
|
|
}
|
|
|
|
if val := params.Get("_auth_salt"); val != "" {
|
|
|
|
authSalt = val
|
|
|
|
}
|
2018-05-30 23:36:49 +03:00
|
|
|
|
2015-03-21 21:02:03 +03:00
|
|
|
// _loc
|
|
|
|
if val := params.Get("_loc"); val != "" {
|
2018-05-29 14:06:07 +03:00
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "auto":
|
2015-03-05 05:05:58 +03:00
|
|
|
loc = time.Local
|
2018-05-29 14:06:07 +03:00
|
|
|
default:
|
2015-03-05 05:05:58 +03:00
|
|
|
loc, err = time.LoadLocation(val)
|
|
|
|
if err != nil {
|
2015-03-21 21:02:03 +03:00
|
|
|
return nil, fmt.Errorf("Invalid _loc: %v: %v", val, err)
|
2015-03-04 16:49:17 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-03-05 05:05:58 +03:00
|
|
|
|
2018-05-29 12:33:43 +03:00
|
|
|
// _mutex
|
|
|
|
if val := params.Get("_mutex"); val != "" {
|
2018-05-29 14:06:07 +03:00
|
|
|
switch strings.ToLower(val) {
|
2018-05-29 12:33:43 +03:00
|
|
|
case "no":
|
2018-12-08 16:01:50 +03:00
|
|
|
mutex = C.SQLITE_OPEN_NOMUTEX
|
2018-05-29 12:33:43 +03:00
|
|
|
case "full":
|
2018-12-08 16:01:50 +03:00
|
|
|
mutex = C.SQLITE_OPEN_FULLMUTEX
|
2018-05-29 12:33:43 +03:00
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("Invalid _mutex: %v", val)
|
2015-03-21 21:02:03 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-10 19:32:18 +03:00
|
|
|
// _txlock
|
|
|
|
if val := params.Get("_txlock"); val != "" {
|
2018-05-29 14:06:07 +03:00
|
|
|
switch strings.ToLower(val) {
|
2015-04-10 19:32:18 +03:00
|
|
|
case "immediate":
|
|
|
|
txlock = "BEGIN IMMEDIATE"
|
|
|
|
case "exclusive":
|
|
|
|
txlock = "BEGIN EXCLUSIVE"
|
|
|
|
case "deferred":
|
|
|
|
txlock = "BEGIN"
|
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("Invalid _txlock: %v", val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 12:46:53 +03:00
|
|
|
// Auto Vacuum (_vacuum)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_auto_vacuum
|
|
|
|
//
|
2018-05-29 15:11:49 +03:00
|
|
|
pkey = "" // Reset pkey
|
|
|
|
if _, ok := params["_auto_vacuum"]; ok {
|
|
|
|
pkey = "_auto_vacuum"
|
|
|
|
}
|
|
|
|
if _, ok := params["_vacuum"]; ok {
|
|
|
|
pkey = "_vacuum"
|
|
|
|
}
|
|
|
|
if val := params.Get(pkey); val != "" {
|
2018-05-29 12:33:43 +03:00
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "none":
|
|
|
|
autoVacuum = 0
|
|
|
|
case "1", "full":
|
|
|
|
autoVacuum = 1
|
|
|
|
case "2", "incremental":
|
|
|
|
autoVacuum = 2
|
|
|
|
default:
|
2018-05-29 15:11:49 +03:00
|
|
|
return nil, fmt.Errorf("Invalid _auto_vacuum: %v, expecting value of '0 NONE 1 FULL 2 INCREMENTAL'", val)
|
2018-05-29 12:33:43 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 12:46:53 +03:00
|
|
|
// Busy Timeout (_busy_timeout)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_busy_timeout
|
|
|
|
//
|
2018-05-29 13:01:16 +03:00
|
|
|
pkey = "" // Reset pkey
|
2018-05-29 12:53:36 +03:00
|
|
|
if _, ok := params["_busy_timeout"]; ok {
|
|
|
|
pkey = "_busy_timeout"
|
|
|
|
}
|
|
|
|
if _, ok := params["_timeout"]; ok {
|
|
|
|
pkey = "_timeout"
|
|
|
|
}
|
|
|
|
if val := params.Get(pkey); val != "" {
|
2018-05-29 12:33:43 +03:00
|
|
|
iv, err := strconv.ParseInt(val, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("Invalid _busy_timeout: %v: %v", val, err)
|
|
|
|
}
|
|
|
|
busyTimeout = int(iv)
|
|
|
|
}
|
|
|
|
|
2018-05-29 12:46:53 +03:00
|
|
|
// Case Sensitive Like (_cslike)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_case_sensitive_like
|
|
|
|
//
|
2018-05-29 15:11:49 +03:00
|
|
|
pkey = "" // Reset pkey
|
|
|
|
if _, ok := params["_case_sensitive_like"]; ok {
|
|
|
|
pkey = "_case_sensitive_like"
|
|
|
|
}
|
|
|
|
if _, ok := params["_cslike"]; ok {
|
|
|
|
pkey = "_cslike"
|
|
|
|
}
|
|
|
|
if val := params.Get(pkey); val != "" {
|
2018-05-29 12:46:53 +03:00
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "no", "false", "off":
|
|
|
|
caseSensitiveLike = 0
|
|
|
|
case "1", "yes", "true", "on":
|
|
|
|
caseSensitiveLike = 1
|
|
|
|
default:
|
2018-05-29 15:11:49 +03:00
|
|
|
return nil, fmt.Errorf("Invalid _case_sensitive_like: %v, expecting boolean value of '0 1 false true no yes off on'", val)
|
2018-05-29 12:46:53 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 13:19:57 +03:00
|
|
|
// Defer Foreign Keys (_defer_foreign_keys | _defer_fk)
|
2018-05-29 13:09:56 +03:00
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_defer_foreign_keys
|
|
|
|
//
|
|
|
|
pkey = "" // Reset pkey
|
|
|
|
if _, ok := params["_defer_foreign_keys"]; ok {
|
|
|
|
pkey = "_defer_foreign_keys"
|
|
|
|
}
|
|
|
|
if _, ok := params["_defer_fk"]; ok {
|
|
|
|
pkey = "_defer_fk"
|
|
|
|
}
|
|
|
|
if val := params.Get(pkey); val != "" {
|
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "no", "false", "off":
|
|
|
|
deferForeignKeys = 0
|
|
|
|
case "1", "yes", "true", "on":
|
|
|
|
deferForeignKeys = 1
|
|
|
|
default:
|
2018-05-29 13:19:57 +03:00
|
|
|
return nil, fmt.Errorf("Invalid _defer_foreign_keys: %v, expecting boolean value of '0 1 false true no yes off on'", val)
|
2018-05-29 13:09:56 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 13:19:57 +03:00
|
|
|
// Foreign Keys (_foreign_keys | _fk)
|
2018-05-29 12:46:53 +03:00
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_foreign_keys
|
|
|
|
//
|
2018-05-29 13:01:16 +03:00
|
|
|
pkey = "" // Reset pkey
|
2018-05-29 12:58:29 +03:00
|
|
|
if _, ok := params["_foreign_keys"]; ok {
|
|
|
|
pkey = "_foreign_keys"
|
|
|
|
}
|
|
|
|
if _, ok := params["_fk"]; ok {
|
|
|
|
pkey = "_fk"
|
|
|
|
}
|
|
|
|
if val := params.Get(pkey); val != "" {
|
2018-05-29 12:33:43 +03:00
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "no", "false", "off":
|
2017-04-01 19:12:21 +03:00
|
|
|
foreignKeys = 0
|
2018-05-29 12:33:43 +03:00
|
|
|
case "1", "yes", "true", "on":
|
|
|
|
foreignKeys = 1
|
2017-04-01 19:12:21 +03:00
|
|
|
default:
|
2018-05-29 12:46:53 +03:00
|
|
|
return nil, fmt.Errorf("Invalid _foreign_keys: %v, expecting boolean value of '0 1 false true no yes off on'", val)
|
2017-04-01 19:12:21 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 13:19:46 +03:00
|
|
|
// Ignore CHECK Constrains (_ignore_check_constraints)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_ignore_check_constraints
|
|
|
|
//
|
|
|
|
if val := params.Get("_ignore_check_constraints"); val != "" {
|
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "no", "false", "off":
|
|
|
|
ignoreCheckConstraints = 0
|
|
|
|
case "1", "yes", "true", "on":
|
|
|
|
ignoreCheckConstraints = 1
|
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("Invalid _ignore_check_constraints: %v, expecting boolean value of '0 1 false true no yes off on'", val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 15:11:49 +03:00
|
|
|
// Journal Mode (_journal_mode | _journal)
|
2018-05-29 14:13:38 +03:00
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_journal_mode
|
|
|
|
//
|
2018-05-29 15:11:49 +03:00
|
|
|
pkey = "" // Reset pkey
|
|
|
|
if _, ok := params["_journal_mode"]; ok {
|
|
|
|
pkey = "_journal_mode"
|
|
|
|
}
|
|
|
|
if _, ok := params["_journal"]; ok {
|
|
|
|
pkey = "_journal"
|
|
|
|
}
|
|
|
|
if val := params.Get(pkey); val != "" {
|
2018-05-29 14:13:38 +03:00
|
|
|
switch strings.ToUpper(val) {
|
2018-05-29 14:55:31 +03:00
|
|
|
case "DELETE", "TRUNCATE", "PERSIST", "MEMORY", "OFF":
|
|
|
|
journalMode = strings.ToUpper(val)
|
|
|
|
case "WAL":
|
2018-05-29 14:13:38 +03:00
|
|
|
journalMode = strings.ToUpper(val)
|
2018-05-29 14:55:31 +03:00
|
|
|
|
|
|
|
// For WAL Mode set Synchronous Mode to 'NORMAL'
|
|
|
|
// See https://www.sqlite.org/pragma.html#pragma_synchronous
|
|
|
|
synchronousMode = "NORMAL"
|
2018-05-29 14:13:38 +03:00
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("Invalid _journal: %v, expecting value of 'DELETE TRUNCATE PERSIST MEMORY WAL OFF'", val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 14:19:40 +03:00
|
|
|
// Locking Mode (_locking)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_locking_mode
|
|
|
|
//
|
2018-05-29 15:11:49 +03:00
|
|
|
pkey = "" // Reset pkey
|
|
|
|
if _, ok := params["_locking_mode"]; ok {
|
|
|
|
pkey = "_locking_mode"
|
|
|
|
}
|
|
|
|
if _, ok := params["_locking"]; ok {
|
|
|
|
pkey = "_locking"
|
|
|
|
}
|
2019-09-24 00:57:24 +03:00
|
|
|
if val := params.Get(pkey); val != "" {
|
2018-05-29 14:19:40 +03:00
|
|
|
switch strings.ToUpper(val) {
|
|
|
|
case "NORMAL", "EXCLUSIVE":
|
|
|
|
lockingMode = strings.ToUpper(val)
|
|
|
|
default:
|
2018-05-29 15:11:49 +03:00
|
|
|
return nil, fmt.Errorf("Invalid _locking_mode: %v, expecting value of 'NORMAL EXCLUSIVE", val)
|
2018-05-29 14:19:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 14:29:06 +03:00
|
|
|
// Query Only (_query_only)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_query_only
|
|
|
|
//
|
|
|
|
if val := params.Get("_query_only"); val != "" {
|
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "no", "false", "off":
|
|
|
|
queryOnly = 0
|
|
|
|
case "1", "yes", "true", "on":
|
|
|
|
queryOnly = 1
|
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("Invalid _query_only: %v, expecting boolean value of '0 1 false true no yes off on'", val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 12:46:53 +03:00
|
|
|
// Recursive Triggers (_recursive_triggers)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_recursive_triggers
|
|
|
|
//
|
2018-05-29 13:03:18 +03:00
|
|
|
pkey = "" // Reset pkey
|
|
|
|
if _, ok := params["_recursive_triggers"]; ok {
|
|
|
|
pkey = "_recursive_triggers"
|
|
|
|
}
|
|
|
|
if _, ok := params["_rt"]; ok {
|
|
|
|
pkey = "_rt"
|
|
|
|
}
|
|
|
|
if val := params.Get(pkey); val != "" {
|
2018-05-29 12:33:43 +03:00
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "no", "false", "off":
|
2017-07-09 17:32:14 +03:00
|
|
|
recursiveTriggers = 0
|
2018-05-29 12:33:43 +03:00
|
|
|
case "1", "yes", "true", "on":
|
|
|
|
recursiveTriggers = 1
|
2017-07-09 17:32:14 +03:00
|
|
|
default:
|
2018-05-29 12:46:53 +03:00
|
|
|
return nil, fmt.Errorf("Invalid _recursive_triggers: %v, expecting boolean value of '0 1 false true no yes off on'", val)
|
2017-07-09 17:32:14 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 14:41:52 +03:00
|
|
|
// Secure Delete (_secure_delete)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_secure_delete
|
|
|
|
//
|
|
|
|
if val := params.Get("_secure_delete"); val != "" {
|
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "no", "false", "off":
|
|
|
|
secureDelete = "OFF"
|
|
|
|
case "1", "yes", "true", "on":
|
|
|
|
secureDelete = "ON"
|
|
|
|
case "fast":
|
|
|
|
secureDelete = "FAST"
|
|
|
|
default:
|
2018-05-29 15:57:40 +03:00
|
|
|
return nil, fmt.Errorf("Invalid _secure_delete: %v, expecting boolean value of '0 1 false true no yes off on fast'", val)
|
2018-05-29 14:41:52 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 14:55:31 +03:00
|
|
|
// Synchronous Mode (_synchronous | _sync)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_synchronous
|
|
|
|
//
|
|
|
|
pkey = "" // Reset pkey
|
|
|
|
if _, ok := params["_synchronous"]; ok {
|
|
|
|
pkey = "_synchronous"
|
|
|
|
}
|
|
|
|
if _, ok := params["_sync"]; ok {
|
|
|
|
pkey = "_sync"
|
|
|
|
}
|
|
|
|
if val := params.Get(pkey); val != "" {
|
|
|
|
switch strings.ToUpper(val) {
|
|
|
|
case "0", "OFF", "1", "NORMAL", "2", "FULL", "3", "EXTRA":
|
|
|
|
synchronousMode = strings.ToUpper(val)
|
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("Invalid _synchronous: %v, expecting value of '0 OFF 1 NORMAL 2 FULL 3 EXTRA'", val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 15:01:33 +03:00
|
|
|
// Writable Schema (_writeable_schema)
|
|
|
|
//
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_writeable_schema
|
|
|
|
//
|
|
|
|
if val := params.Get("_writable_schema"); val != "" {
|
|
|
|
switch strings.ToLower(val) {
|
|
|
|
case "0", "no", "false", "off":
|
|
|
|
writableSchema = 0
|
|
|
|
case "1", "yes", "true", "on":
|
|
|
|
writableSchema = 1
|
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("Invalid _writable_schema: %v, expecting boolean value of '0 1 false true no yes off on'", val)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-26 17:05:20 +03:00
|
|
|
// Cache size (_cache_size)
|
|
|
|
//
|
|
|
|
// https://sqlite.org/pragma.html#pragma_cache_size
|
|
|
|
//
|
|
|
|
if val := params.Get("_cache_size"); val != "" {
|
|
|
|
iv, err := strconv.ParseInt(val, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("Invalid _cache_size: %v: %v", val, err)
|
|
|
|
}
|
|
|
|
cacheSize = &iv
|
|
|
|
}
|
|
|
|
|
2020-11-16 19:54:21 +03:00
|
|
|
if val := params.Get("vfs"); val != "" {
|
|
|
|
vfsName = val
|
|
|
|
}
|
|
|
|
|
2015-03-05 20:23:57 +03:00
|
|
|
if !strings.HasPrefix(dsn, "file:") {
|
2015-03-05 20:00:09 +03:00
|
|
|
dsn = dsn[:pos]
|
|
|
|
}
|
2015-03-04 16:49:17 +03:00
|
|
|
}
|
|
|
|
|
2011-11-11 16:36:22 +04:00
|
|
|
var db *C.sqlite3
|
|
|
|
name := C.CString(dsn)
|
|
|
|
defer C.free(unsafe.Pointer(name))
|
2020-11-16 19:54:21 +03:00
|
|
|
var vfs *C.char
|
|
|
|
if vfsName != "" {
|
|
|
|
vfs = C.CString(vfsName)
|
|
|
|
defer C.free(unsafe.Pointer(vfs))
|
|
|
|
}
|
2018-12-08 16:01:50 +03:00
|
|
|
rv := C._sqlite3_open_v2(name, &db,
|
|
|
|
mutex|C.SQLITE_OPEN_READWRITE|C.SQLITE_OPEN_CREATE,
|
2020-11-16 19:54:21 +03:00
|
|
|
vfs)
|
2011-11-11 16:36:22 +04:00
|
|
|
if rv != 0 {
|
2019-12-17 09:58:28 +03:00
|
|
|
// Save off the error _before_ closing the database.
|
|
|
|
// This is safe even if db is nil.
|
|
|
|
err := lastError(db)
|
2018-12-07 07:13:49 +03:00
|
|
|
if db != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
}
|
2019-12-17 09:58:28 +03:00
|
|
|
return nil, err
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
if db == nil {
|
|
|
|
return nil, errors.New("sqlite succeeded without returning a database")
|
|
|
|
}
|
2012-03-12 09:20:55 +04:00
|
|
|
|
2017-04-01 19:12:21 +03:00
|
|
|
exec := func(s string) error {
|
|
|
|
cs := C.CString(s)
|
|
|
|
rv := C.sqlite3_exec(db, cs, nil, nil, nil)
|
|
|
|
C.free(unsafe.Pointer(cs))
|
|
|
|
if rv != C.SQLITE_OK {
|
|
|
|
return lastError(db)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
2018-05-29 12:33:43 +03:00
|
|
|
|
2021-10-25 18:08:40 +03:00
|
|
|
// Busy timeout
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA busy_timeout = %d;", busyTimeout)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-30 23:36:49 +03:00
|
|
|
// USER AUTHENTICATION
|
|
|
|
//
|
|
|
|
// User Authentication is always performed even when
|
|
|
|
// sqlite_userauth is not compiled in, because without user authentication
|
|
|
|
// the authentication is a no-op.
|
|
|
|
//
|
|
|
|
// Workflow
|
|
|
|
// - Authenticate
|
|
|
|
// ON::SUCCESS => Continue
|
|
|
|
// ON::SQLITE_AUTH => Return error and exit Open(...)
|
|
|
|
//
|
|
|
|
// - Activate User Authentication
|
|
|
|
// Check if the user wants to activate User Authentication.
|
|
|
|
// If so then first create a temporary AuthConn to the database
|
2018-11-21 13:14:40 +03:00
|
|
|
// This is possible because we are already successfully authenticated.
|
2018-05-30 23:36:49 +03:00
|
|
|
//
|
|
|
|
// - Check if `sqlite_user`` table exists
|
|
|
|
// YES => Add the provided user from DSN as Admin User and
|
|
|
|
// activate user authentication.
|
|
|
|
// NO => Continue
|
|
|
|
//
|
|
|
|
|
|
|
|
// Create connection to SQLite
|
|
|
|
conn := &SQLiteConn{db: db, loc: loc, txlock: txlock}
|
|
|
|
|
2018-11-21 13:14:40 +03:00
|
|
|
// Password Cipher has to be registered before authentication
|
2018-06-05 14:45:32 +03:00
|
|
|
if len(authCrypt) > 0 {
|
|
|
|
switch strings.ToUpper(authCrypt) {
|
|
|
|
case "SHA1":
|
|
|
|
if err := conn.RegisterFunc("sqlite_crypt", CryptEncoderSHA1, true); err != nil {
|
|
|
|
return nil, fmt.Errorf("CryptEncoderSHA1: %s", err)
|
|
|
|
}
|
|
|
|
case "SSHA1":
|
|
|
|
if len(authSalt) == 0 {
|
|
|
|
return nil, fmt.Errorf("_auth_crypt=ssha1, requires _auth_salt")
|
|
|
|
}
|
|
|
|
if err := conn.RegisterFunc("sqlite_crypt", CryptEncoderSSHA1(authSalt), true); err != nil {
|
|
|
|
return nil, fmt.Errorf("CryptEncoderSSHA1: %s", err)
|
|
|
|
}
|
|
|
|
case "SHA256":
|
|
|
|
if err := conn.RegisterFunc("sqlite_crypt", CryptEncoderSHA256, true); err != nil {
|
|
|
|
return nil, fmt.Errorf("CryptEncoderSHA256: %s", err)
|
|
|
|
}
|
|
|
|
case "SSHA256":
|
|
|
|
if len(authSalt) == 0 {
|
|
|
|
return nil, fmt.Errorf("_auth_crypt=ssha256, requires _auth_salt")
|
|
|
|
}
|
|
|
|
if err := conn.RegisterFunc("sqlite_crypt", CryptEncoderSSHA256(authSalt), true); err != nil {
|
|
|
|
return nil, fmt.Errorf("CryptEncoderSSHA256: %s", err)
|
|
|
|
}
|
|
|
|
case "SHA384":
|
|
|
|
if err := conn.RegisterFunc("sqlite_crypt", CryptEncoderSHA384, true); err != nil {
|
|
|
|
return nil, fmt.Errorf("CryptEncoderSHA384: %s", err)
|
|
|
|
}
|
|
|
|
case "SSHA384":
|
|
|
|
if len(authSalt) == 0 {
|
|
|
|
return nil, fmt.Errorf("_auth_crypt=ssha384, requires _auth_salt")
|
|
|
|
}
|
|
|
|
if err := conn.RegisterFunc("sqlite_crypt", CryptEncoderSSHA384(authSalt), true); err != nil {
|
|
|
|
return nil, fmt.Errorf("CryptEncoderSSHA384: %s", err)
|
|
|
|
}
|
|
|
|
case "SHA512":
|
|
|
|
if err := conn.RegisterFunc("sqlite_crypt", CryptEncoderSHA512, true); err != nil {
|
|
|
|
return nil, fmt.Errorf("CryptEncoderSHA512: %s", err)
|
|
|
|
}
|
|
|
|
case "SSHA512":
|
|
|
|
if len(authSalt) == 0 {
|
|
|
|
return nil, fmt.Errorf("_auth_crypt=ssha512, requires _auth_salt")
|
|
|
|
}
|
|
|
|
if err := conn.RegisterFunc("sqlite_crypt", CryptEncoderSSHA512(authSalt), true); err != nil {
|
|
|
|
return nil, fmt.Errorf("CryptEncoderSSHA512: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-30 23:36:49 +03:00
|
|
|
// Preform Authentication
|
|
|
|
if err := conn.Authenticate(authUser, authPass); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-06-01 12:28:04 +03:00
|
|
|
// Register: authenticate
|
2018-05-30 23:36:49 +03:00
|
|
|
// Authenticate will perform an authentication of the provided username
|
|
|
|
// and password against the database.
|
|
|
|
//
|
|
|
|
// If a database contains the SQLITE_USER table, then the
|
|
|
|
// call to Authenticate must be invoked with an
|
|
|
|
// appropriate username and password prior to enable read and write
|
|
|
|
//access to the database.
|
|
|
|
//
|
|
|
|
// Return SQLITE_OK on success or SQLITE_ERROR if the username/password
|
|
|
|
// combination is incorrect or unknown.
|
|
|
|
//
|
|
|
|
// If the SQLITE_USER table is not present in the database file, then
|
|
|
|
// this interface is a harmless no-op returnning SQLITE_OK.
|
2018-06-01 18:59:42 +03:00
|
|
|
if err := conn.RegisterFunc("authenticate", conn.authenticate, true); err != nil {
|
2018-05-30 23:36:49 +03:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
//
|
2018-06-01 12:28:04 +03:00
|
|
|
// Register: auth_user_add
|
|
|
|
// auth_user_add can be used (by an admin user only)
|
2018-05-30 23:36:49 +03:00
|
|
|
// to create a new user. When called on a no-authentication-required
|
|
|
|
// database, this routine converts the database into an authentication-
|
|
|
|
// required database, automatically makes the added user an
|
|
|
|
// administrator, and logs in the current connection as that user.
|
|
|
|
// The AuthUserAdd only works for the "main" database, not
|
|
|
|
// for any ATTACH-ed databases. Any call to AuthUserAdd by a
|
|
|
|
// non-admin user results in an error.
|
2018-06-01 18:59:42 +03:00
|
|
|
if err := conn.RegisterFunc("auth_user_add", conn.authUserAdd, true); err != nil {
|
2018-05-30 23:36:49 +03:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
//
|
2018-06-01 12:28:04 +03:00
|
|
|
// Register: auth_user_change
|
|
|
|
// auth_user_change can be used to change a users
|
2018-05-30 23:36:49 +03:00
|
|
|
// login credentials or admin privilege. Any user can change their own
|
|
|
|
// login credentials. Only an admin user can change another users login
|
|
|
|
// credentials or admin privilege setting. No user may change their own
|
|
|
|
// admin privilege setting.
|
2018-06-01 18:59:42 +03:00
|
|
|
if err := conn.RegisterFunc("auth_user_change", conn.authUserChange, true); err != nil {
|
2018-05-30 23:36:49 +03:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
//
|
2018-06-01 12:28:04 +03:00
|
|
|
// Register: auth_user_delete
|
|
|
|
// auth_user_delete can be used (by an admin user only)
|
2018-05-30 23:36:49 +03:00
|
|
|
// to delete a user. The currently logged-in user cannot be deleted,
|
|
|
|
// which guarantees that there is always an admin user and hence that
|
|
|
|
// the database cannot be converted into a no-authentication-required
|
|
|
|
// database.
|
2018-06-01 18:59:42 +03:00
|
|
|
if err := conn.RegisterFunc("auth_user_delete", conn.authUserDelete, true); err != nil {
|
2018-06-01 12:28:04 +03:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register: auth_enabled
|
|
|
|
// auth_enabled can be used to check if user authentication is enabled
|
2018-06-01 18:59:42 +03:00
|
|
|
if err := conn.RegisterFunc("auth_enabled", conn.authEnabled, true); err != nil {
|
2018-05-30 23:36:49 +03:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-29 12:33:43 +03:00
|
|
|
// Auto Vacuum
|
2018-05-30 23:36:49 +03:00
|
|
|
// Moved auto_vacuum command, the user preference for auto_vacuum needs to be implemented directly after
|
|
|
|
// the authentication and before the sqlite_user table gets created if the user
|
|
|
|
// decides to activate User Authentication because
|
|
|
|
// auto_vacuum needs to be set before any tables are created
|
|
|
|
// and activating user authentication creates the internal table `sqlite_user`.
|
2018-05-29 12:33:43 +03:00
|
|
|
if autoVacuum > -1 {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA auto_vacuum = %d;", autoVacuum)); err != nil {
|
2017-04-01 19:12:21 +03:00
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
2018-05-29 12:33:43 +03:00
|
|
|
|
2018-05-30 23:36:49 +03:00
|
|
|
// Check if user wants to activate User Authentication
|
|
|
|
if authCreate {
|
|
|
|
// Before going any further, we need to check that the user
|
|
|
|
// has provided an username and password within the DSN.
|
|
|
|
// We are not allowed to continue.
|
2019-08-22 13:40:55 +03:00
|
|
|
if len(authUser) == 0 {
|
2018-05-30 23:36:49 +03:00
|
|
|
return nil, fmt.Errorf("Missing '_auth_user' while user authentication was requested with '_auth'")
|
|
|
|
}
|
2019-08-22 13:40:55 +03:00
|
|
|
if len(authPass) == 0 {
|
2018-05-30 23:36:49 +03:00
|
|
|
return nil, fmt.Errorf("Missing '_auth_pass' while user authentication was requested with '_auth'")
|
|
|
|
}
|
|
|
|
|
2018-05-31 15:55:22 +03:00
|
|
|
// Check if User Authentication is Enabled
|
2018-06-01 12:28:04 +03:00
|
|
|
authExists := conn.AuthEnabled()
|
2018-05-30 23:36:49 +03:00
|
|
|
if !authExists {
|
|
|
|
if err := conn.AuthUserAdd(authUser, authPass, true); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 13:10:05 +03:00
|
|
|
// Case Sensitive LIKE
|
2018-05-29 12:46:53 +03:00
|
|
|
if caseSensitiveLike > -1 {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA case_sensitive_like = %d;", caseSensitiveLike)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 13:09:56 +03:00
|
|
|
// Defer Foreign Keys
|
|
|
|
if deferForeignKeys > -1 {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA defer_foreign_keys = %d;", deferForeignKeys)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 12:33:43 +03:00
|
|
|
// Forgein Keys
|
|
|
|
if foreignKeys > -1 {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA foreign_keys = %d;", foreignKeys)); err != nil {
|
2017-07-09 17:32:14 +03:00
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
2018-05-29 12:33:43 +03:00
|
|
|
}
|
|
|
|
|
2018-05-29 13:19:46 +03:00
|
|
|
// Ignore CHECK Constraints
|
|
|
|
if ignoreCheckConstraints > -1 {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA ignore_check_constraints = %d;", ignoreCheckConstraints)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 14:13:38 +03:00
|
|
|
// Journal Mode
|
2019-09-24 00:50:49 +03:00
|
|
|
if journalMode != "" {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA journal_mode = %s;", journalMode)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
2018-05-29 14:13:38 +03:00
|
|
|
}
|
|
|
|
|
2018-05-29 14:19:40 +03:00
|
|
|
// Locking Mode
|
|
|
|
// Because the default is NORMAL and this is not changed in this package
|
|
|
|
// by using the compile time SQLITE_DEFAULT_LOCKING_MODE this PRAGMA can always be executed
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA locking_mode = %s;", lockingMode)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-29 14:29:06 +03:00
|
|
|
// Query Only
|
2018-05-29 15:57:25 +03:00
|
|
|
if queryOnly > -1 {
|
2018-05-29 14:29:06 +03:00
|
|
|
if err := exec(fmt.Sprintf("PRAGMA query_only = %d;", queryOnly)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 12:33:43 +03:00
|
|
|
// Recursive Triggers
|
|
|
|
if recursiveTriggers > -1 {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA recursive_triggers = %d;", recursiveTriggers)); err != nil {
|
2017-07-09 17:32:14 +03:00
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
2017-04-01 19:12:21 +03:00
|
|
|
|
2018-05-29 14:41:52 +03:00
|
|
|
// Secure Delete
|
|
|
|
//
|
|
|
|
// Because this package can set the compile time flag SQLITE_SECURE_DELETE with a build tag
|
|
|
|
// the default value for secureDelete var is 'DEFAULT' this way
|
|
|
|
// you can compile with secure_delete 'ON' and disable it for a specific database connection.
|
|
|
|
if secureDelete != "DEFAULT" {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA secure_delete = %s;", secureDelete)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-29 14:55:31 +03:00
|
|
|
// Synchronous Mode
|
|
|
|
//
|
|
|
|
// Because default is NORMAL this statement is always executed
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA synchronous = %s;", synchronousMode)); err != nil {
|
2021-02-15 16:57:26 +03:00
|
|
|
conn.Close()
|
2018-05-29 14:55:31 +03:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-29 15:01:33 +03:00
|
|
|
// Writable Schema
|
|
|
|
if writableSchema > -1 {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA writable_schema = %d;", writableSchema)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
2020-12-26 17:05:20 +03:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Cache Size
|
|
|
|
if cacheSize != nil {
|
|
|
|
if err := exec(fmt.Sprintf("PRAGMA cache_size = %d;", *cacheSize)); err != nil {
|
|
|
|
C.sqlite3_close_v2(db)
|
2018-05-29 15:01:33 +03:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-25 07:36:35 +04:00
|
|
|
if len(d.Extensions) > 0 {
|
2015-09-04 21:16:27 +03:00
|
|
|
if err := conn.loadExtensions(d.Extensions); err != nil {
|
2017-03-24 02:48:29 +03:00
|
|
|
conn.Close()
|
2015-09-04 21:16:27 +03:00
|
|
|
return nil, err
|
2013-08-25 07:36:35 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-23 08:58:54 +04:00
|
|
|
if d.ConnectHook != nil {
|
2013-08-25 07:04:51 +04:00
|
|
|
if err := d.ConnectHook(conn); err != nil {
|
2017-03-24 02:48:29 +03:00
|
|
|
conn.Close()
|
2013-08-25 07:04:51 +04:00
|
|
|
return nil, err
|
|
|
|
}
|
2013-08-23 08:58:54 +04:00
|
|
|
}
|
2014-11-14 11:13:35 +03:00
|
|
|
runtime.SetFinalizer(conn, (*SQLiteConn).Close)
|
2013-08-23 08:58:54 +04:00
|
|
|
return conn, nil
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2013-01-31 11:48:30 +04:00
|
|
|
// Close the connection.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (c *SQLiteConn) Close() error {
|
2013-10-24 17:21:37 +04:00
|
|
|
rv := C.sqlite3_close_v2(c.db)
|
2011-11-11 16:36:22 +04:00
|
|
|
if rv != C.SQLITE_OK {
|
2013-11-19 13:13:19 +04:00
|
|
|
return c.lastError()
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
2016-11-10 18:27:20 +03:00
|
|
|
deleteHandles(c)
|
2017-08-01 18:06:18 +03:00
|
|
|
c.mu.Lock()
|
2011-11-11 16:36:22 +04:00
|
|
|
c.db = nil
|
2017-08-01 18:06:18 +03:00
|
|
|
c.mu.Unlock()
|
2014-11-14 11:13:35 +03:00
|
|
|
runtime.SetFinalizer(c, nil)
|
2011-11-11 16:36:22 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-06-17 22:22:09 +03:00
|
|
|
func (c *SQLiteConn) dbConnOpen() bool {
|
|
|
|
if c == nil {
|
|
|
|
return false
|
|
|
|
}
|
2017-08-01 18:06:18 +03:00
|
|
|
c.mu.Lock()
|
|
|
|
defer c.mu.Unlock()
|
2017-06-17 22:22:09 +03:00
|
|
|
return c.db != nil
|
|
|
|
}
|
|
|
|
|
2016-02-23 09:20:57 +03:00
|
|
|
// Prepare the query string. Return a new statement.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (c *SQLiteConn) Prepare(query string) (driver.Stmt, error) {
|
2016-11-04 09:11:24 +03:00
|
|
|
return c.prepare(context.Background(), query)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *SQLiteConn) prepare(ctx context.Context, query string) (driver.Stmt, error) {
|
2011-11-11 16:36:22 +04:00
|
|
|
pquery := C.CString(query)
|
|
|
|
defer C.free(unsafe.Pointer(pquery))
|
|
|
|
var s *C.sqlite3_stmt
|
2013-09-09 05:44:44 +04:00
|
|
|
var tail *C.char
|
2018-11-02 07:03:02 +03:00
|
|
|
rv := C._sqlite3_prepare_v2_internal(c.db, pquery, C.int(-1), &s, &tail)
|
2011-11-11 16:36:22 +04:00
|
|
|
if rv != C.SQLITE_OK {
|
2013-11-19 13:13:19 +04:00
|
|
|
return nil, c.lastError()
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
var t string
|
2015-03-24 00:18:23 +03:00
|
|
|
if tail != nil && *tail != '\000' {
|
2013-09-09 05:44:44 +04:00
|
|
|
t = strings.TrimSpace(C.GoString(tail))
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
2016-11-04 09:00:29 +03:00
|
|
|
ss := &SQLiteStmt{c: c, s: s, t: t}
|
2014-11-14 11:13:35 +03:00
|
|
|
runtime.SetFinalizer(ss, (*SQLiteStmt).Close)
|
|
|
|
return ss, nil
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2017-07-17 15:29:07 +03:00
|
|
|
// Run-Time Limit Categories.
|
|
|
|
// See: http://www.sqlite.org/c3ref/c_limit_attached.html
|
|
|
|
const (
|
|
|
|
SQLITE_LIMIT_LENGTH = C.SQLITE_LIMIT_LENGTH
|
|
|
|
SQLITE_LIMIT_SQL_LENGTH = C.SQLITE_LIMIT_SQL_LENGTH
|
|
|
|
SQLITE_LIMIT_COLUMN = C.SQLITE_LIMIT_COLUMN
|
|
|
|
SQLITE_LIMIT_EXPR_DEPTH = C.SQLITE_LIMIT_EXPR_DEPTH
|
|
|
|
SQLITE_LIMIT_COMPOUND_SELECT = C.SQLITE_LIMIT_COMPOUND_SELECT
|
|
|
|
SQLITE_LIMIT_VDBE_OP = C.SQLITE_LIMIT_VDBE_OP
|
|
|
|
SQLITE_LIMIT_FUNCTION_ARG = C.SQLITE_LIMIT_FUNCTION_ARG
|
|
|
|
SQLITE_LIMIT_ATTACHED = C.SQLITE_LIMIT_ATTACHED
|
|
|
|
SQLITE_LIMIT_LIKE_PATTERN_LENGTH = C.SQLITE_LIMIT_LIKE_PATTERN_LENGTH
|
|
|
|
SQLITE_LIMIT_VARIABLE_NUMBER = C.SQLITE_LIMIT_VARIABLE_NUMBER
|
|
|
|
SQLITE_LIMIT_TRIGGER_DEPTH = C.SQLITE_LIMIT_TRIGGER_DEPTH
|
|
|
|
SQLITE_LIMIT_WORKER_THREADS = C.SQLITE_LIMIT_WORKER_THREADS
|
|
|
|
)
|
|
|
|
|
2018-05-12 03:48:25 +03:00
|
|
|
// GetFilename returns the absolute path to the file containing
|
|
|
|
// the requested schema. When passed an empty string, it will
|
|
|
|
// instead use the database's default schema: "main".
|
|
|
|
// See: sqlite3_db_filename, https://www.sqlite.org/c3ref/db_filename.html
|
|
|
|
func (c *SQLiteConn) GetFilename(schemaName string) string {
|
|
|
|
if schemaName == "" {
|
|
|
|
schemaName = "main"
|
|
|
|
}
|
|
|
|
return C.GoString(C.sqlite3_db_filename(c.db, C.CString(schemaName)))
|
|
|
|
}
|
|
|
|
|
2017-07-17 15:29:07 +03:00
|
|
|
// GetLimit returns the current value of a run-time limit.
|
|
|
|
// See: sqlite3_limit, http://www.sqlite.org/c3ref/limit.html
|
|
|
|
func (c *SQLiteConn) GetLimit(id int) int {
|
2018-11-02 07:03:02 +03:00
|
|
|
return int(C._sqlite3_limit(c.db, C.int(id), C.int(-1)))
|
2017-07-17 15:29:07 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// SetLimit changes the value of a run-time limits.
|
|
|
|
// Then this method returns the prior value of the limit.
|
|
|
|
// See: sqlite3_limit, http://www.sqlite.org/c3ref/limit.html
|
|
|
|
func (c *SQLiteConn) SetLimit(id int, newVal int) int {
|
2017-11-05 02:47:52 +03:00
|
|
|
return int(C._sqlite3_limit(c.db, C.int(id), C.int(newVal)))
|
2017-07-17 15:29:07 +03:00
|
|
|
}
|
|
|
|
|
2022-01-28 18:37:05 +03:00
|
|
|
// SetFileControlInt invokes the xFileControl method on a given database. The
|
|
|
|
// dbName is the name of the database. It will default to "main" if left blank.
|
|
|
|
// The op is one of the opcodes prefixed by "SQLITE_FCNTL_". The arg argument
|
|
|
|
// and return code are both opcode-specific. Please see the SQLite documentation.
|
|
|
|
//
|
|
|
|
// This method is not thread-safe as the returned error code can be changed by
|
|
|
|
// another call if invoked concurrently.
|
|
|
|
//
|
|
|
|
// See: sqlite3_file_control, https://www.sqlite.org/c3ref/file_control.html
|
|
|
|
func (c *SQLiteConn) SetFileControlInt(dbName string, op int, arg int) error {
|
|
|
|
if dbName == "" {
|
|
|
|
dbName = "main"
|
|
|
|
}
|
|
|
|
|
|
|
|
cDBName := C.CString(dbName)
|
|
|
|
defer C.free(unsafe.Pointer(cDBName))
|
|
|
|
|
|
|
|
cArg := C.int(arg)
|
|
|
|
rv := C.sqlite3_file_control(c.db, cDBName, C.int(op), unsafe.Pointer(&cArg))
|
|
|
|
if rv != C.SQLITE_OK {
|
|
|
|
return c.lastError()
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2013-01-31 11:48:30 +04:00
|
|
|
// Close the statement.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (s *SQLiteStmt) Close() error {
|
2017-08-30 07:29:47 +03:00
|
|
|
s.mu.Lock()
|
|
|
|
defer s.mu.Unlock()
|
2012-02-20 11:14:49 +04:00
|
|
|
if s.closed {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
s.closed = true
|
2017-06-17 22:22:09 +03:00
|
|
|
if !s.c.dbConnOpen() {
|
2013-02-13 05:32:40 +04:00
|
|
|
return errors.New("sqlite statement with already closed database connection")
|
|
|
|
}
|
2011-11-11 16:38:53 +04:00
|
|
|
rv := C.sqlite3_finalize(s.s)
|
2017-08-28 12:58:02 +03:00
|
|
|
s.s = nil
|
2011-11-11 16:36:22 +04:00
|
|
|
if rv != C.SQLITE_OK {
|
2013-11-19 13:13:19 +04:00
|
|
|
return s.c.lastError()
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
2014-11-14 11:13:35 +03:00
|
|
|
runtime.SetFinalizer(s, nil)
|
2011-11-11 16:36:22 +04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2016-11-04 18:40:06 +03:00
|
|
|
// NumInput return a number of parameters.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (s *SQLiteStmt) NumInput() int {
|
2016-11-04 09:00:29 +03:00
|
|
|
return int(C.sqlite3_bind_parameter_count(s.s))
|
2015-03-21 20:08:47 +03:00
|
|
|
}
|
|
|
|
|
2017-06-21 03:36:44 +03:00
|
|
|
var placeHolder = []byte{0}
|
2017-06-14 16:22:40 +03:00
|
|
|
|
2023-02-12 01:14:42 +03:00
|
|
|
func (s *SQLiteStmt) bind(args []driver.NamedValue) error {
|
2011-11-11 16:36:22 +04:00
|
|
|
rv := C.sqlite3_reset(s.s)
|
2011-11-11 16:38:53 +04:00
|
|
|
if rv != C.SQLITE_ROW && rv != C.SQLITE_OK && rv != C.SQLITE_DONE {
|
2013-11-19 13:13:19 +04:00
|
|
|
return s.c.lastError()
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2020-05-14 17:28:04 +03:00
|
|
|
bindIndices := make([][3]int, len(args))
|
|
|
|
prefixes := []string{":", "@", "$"}
|
2016-11-04 09:00:29 +03:00
|
|
|
for i, v := range args {
|
2020-05-14 17:28:04 +03:00
|
|
|
bindIndices[i][0] = args[i].Ordinal
|
2016-11-04 09:00:29 +03:00
|
|
|
if v.Name != "" {
|
2020-05-14 17:28:04 +03:00
|
|
|
for j := range prefixes {
|
|
|
|
cname := C.CString(prefixes[j] + v.Name)
|
|
|
|
bindIndices[i][j] = int(C.sqlite3_bind_parameter_index(s.s, cname))
|
|
|
|
C.free(unsafe.Pointer(cname))
|
2011-12-03 02:32:38 +04:00
|
|
|
}
|
2020-05-14 17:28:04 +03:00
|
|
|
args[i].Ordinal = bindIndices[i][0]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for i, arg := range args {
|
|
|
|
for j := range bindIndices[i] {
|
|
|
|
if bindIndices[i][j] == 0 {
|
|
|
|
continue
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
2020-05-14 17:28:04 +03:00
|
|
|
n := C.int(bindIndices[i][j])
|
|
|
|
switch v := arg.Value.(type) {
|
|
|
|
case nil:
|
2018-05-31 03:39:01 +03:00
|
|
|
rv = C.sqlite3_bind_null(s.s, n)
|
2020-05-14 17:28:04 +03:00
|
|
|
case string:
|
|
|
|
if len(v) == 0 {
|
|
|
|
rv = C._sqlite3_bind_text(s.s, n, (*C.char)(unsafe.Pointer(&placeHolder[0])), C.int(0))
|
|
|
|
} else {
|
|
|
|
b := []byte(v)
|
|
|
|
rv = C._sqlite3_bind_text(s.s, n, (*C.char)(unsafe.Pointer(&b[0])), C.int(len(b)))
|
|
|
|
}
|
|
|
|
case int64:
|
|
|
|
rv = C.sqlite3_bind_int64(s.s, n, C.sqlite3_int64(v))
|
|
|
|
case bool:
|
|
|
|
if v {
|
|
|
|
rv = C.sqlite3_bind_int(s.s, n, 1)
|
|
|
|
} else {
|
|
|
|
rv = C.sqlite3_bind_int(s.s, n, 0)
|
2018-05-31 03:39:01 +03:00
|
|
|
}
|
2020-05-14 17:28:04 +03:00
|
|
|
case float64:
|
|
|
|
rv = C.sqlite3_bind_double(s.s, n, C.double(v))
|
|
|
|
case []byte:
|
|
|
|
if v == nil {
|
|
|
|
rv = C.sqlite3_bind_null(s.s, n)
|
|
|
|
} else {
|
|
|
|
ln := len(v)
|
|
|
|
if ln == 0 {
|
|
|
|
v = placeHolder
|
|
|
|
}
|
|
|
|
rv = C._sqlite3_bind_blob(s.s, n, unsafe.Pointer(&v[0]), C.int(ln))
|
|
|
|
}
|
|
|
|
case time.Time:
|
|
|
|
b := []byte(v.Format(SQLiteTimestampFormats[0]))
|
|
|
|
rv = C._sqlite3_bind_text(s.s, n, (*C.char)(unsafe.Pointer(&b[0])), C.int(len(b)))
|
|
|
|
}
|
|
|
|
if rv != C.SQLITE_OK {
|
|
|
|
return s.c.lastError()
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-02-18 04:06:30 +04:00
|
|
|
// Query the statement with arguments. Return records.
|
2012-02-20 11:14:49 +04:00
|
|
|
func (s *SQLiteStmt) Query(args []driver.Value) (driver.Rows, error) {
|
2023-02-12 01:14:42 +03:00
|
|
|
list := make([]driver.NamedValue, len(args))
|
2016-11-04 08:24:22 +03:00
|
|
|
for i, v := range args {
|
2023-02-12 01:14:42 +03:00
|
|
|
list[i] = driver.NamedValue{
|
2016-11-04 08:24:22 +03:00
|
|
|
Ordinal: i + 1,
|
|
|
|
Value: v,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return s.query(context.Background(), list)
|
|
|
|
}
|
|
|
|
|
2023-02-12 01:14:42 +03:00
|
|
|
func (s *SQLiteStmt) query(ctx context.Context, args []driver.NamedValue) (driver.Rows, error) {
|
2011-11-11 16:36:22 +04:00
|
|
|
if err := s.bind(args); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2016-11-06 14:43:53 +03:00
|
|
|
|
|
|
|
rows := &SQLiteRows{
|
|
|
|
s: s,
|
|
|
|
nc: int(C.sqlite3_column_count(s.s)),
|
|
|
|
cols: nil,
|
|
|
|
decltype: nil,
|
|
|
|
cls: s.cls,
|
2017-08-01 19:43:14 +03:00
|
|
|
closed: false,
|
2019-09-06 01:26:33 +03:00
|
|
|
ctx: ctx,
|
Don't spawn interrupt goroutine if we know that context cannot be canceled
For a Go-only project the following code pattern
go func() {
select {
case <-ctx.Done():
// call some cancel
case <-done:
// work finished ok
}
}()
// do some work
close(done)
works good and fast - without high scheduling overhead because scheduler
usually puts spawned goroutine into run queue on the same OS thread and so
after done is closed control is passed to spawned goroutine without OS context
switch.
However in the presence of Cgo calls in "do some work" the situation can
become different - Cgo calls are treated by go runtime similarly to
system calls with the effect that goroutines spawned on original OS
thread tend to be migrated by scheduler to be executed on another OS
thread.
This in turn can bring high overhead for communicating on "done", which
ultimately can result in full context switch: if the spawned goroutine
had chance to run, already checked done and ctx to be not ready, and went
into sleep via wait on futex - showing as something like below in strace for
one read query (note futex calls):
27867 00:38:39.782146 stat(".../neo.sqlite-journal", 0x7f83809c4a20) = -1 ENOENT (No such file or directory)
27867 00:38:39.782165 pread64(3, "\0\0\0\33\0\0\10\235\0\0\10]\0\0\0\27", 16, 24) = 16
27871 00:38:39.782179 <... pselect6 resumed> ) = 0 (Timeout)
27868 00:38:39.782187 <... pselect6 resumed> ) = 0 (Timeout)
27871 00:38:39.782193 futex(0xc4200f8538, FUTEX_WAIT, 0, NULL <unfinished ...>
27868 00:38:39.782199 futex(0xc420013138, FUTEX_WAIT, 0, NULL <unfinished ...>
27867 00:38:39.782205 stat(".../neo.sqlite-wal", 0x7f83809c4a20) = -1 ENOENT (No such file or directory)
27867 00:38:39.782224 fstat(3, {st_mode=S_IFREG|0644, st_size=9031680, ...}) = 0
27867 00:38:39.782247 futex(0xc420013138, FUTEX_WAKE, 1 <unfinished ...>
27868 00:38:39.782259 <... futex resumed> ) = 0
27867 00:38:39.782265 <... futex resumed> ) = 1
27868 00:38:39.782270 pselect6(0, NULL, NULL, NULL, {tv_sec=0, tv_nsec=3000}, NULL <unfinished ...>
27867 00:38:39.782279 fcntl(3, F_SETLK, {l_type=F_UNLCK, l_whence=SEEK_SET, l_start=0, l_len=0}) = 0
27867 00:38:39.782315 fcntl(3, F_SETLK, {l_type=F_RDLCK, l_whence=SEEK_SET, l_start=1073741824, l_len=1}) = 0
27868 00:38:39.782336 <... pselect6 resumed> ) = 0 (Timeout)
27867 00:38:39.782342 fcntl(3, F_SETLK, {l_type=F_RDLCK, l_whence=SEEK_SET, l_start=1073741826, l_len=510} <unfinished ...>
27868 00:38:39.782348 futex(0xc4200f8538, FUTEX_WAKE, 1 <unfinished ...>
27867 00:38:39.782355 <... fcntl resumed> ) = 0
27871 00:38:39.782360 <... futex resumed> ) = 0
27868 00:38:39.782367 <... futex resumed> ) = 1
27871 00:38:39.782372 futex(0xc4200f8138, FUTEX_WAKE, 1 <unfinished ...>
27868 00:38:39.782377 pselect6(0, NULL, NULL, NULL, {tv_sec=0, tv_nsec=3000}, NULL <unfinished ...>
27871 00:38:39.782384 <... futex resumed> ) = 1
27870 00:38:39.782389 <... futex resumed> ) = 0
27867 00:38:39.782394 fcntl(3, F_SETLK, {l_type=F_UNLCK, l_whence=SEEK_SET, l_start=1073741824, l_len=1} <unfinished ...>
27870 00:38:39.782400 pselect6(0, NULL, NULL, NULL, {tv_sec=0, tv_nsec=3000}, NULL <unfinished ...>
27867 00:38:39.782408 <... fcntl resumed> ) = 0
Below link shows that go scheduler itself might be significantly improved for
cases when there are several Cgo calls made for a request in a server:
https://github.com/golang/go/issues/21827#issuecomment-329092317
in particular CGo-4 case should be closely related to this sqlite3 go package,
because for one query many CGo calls are made to SQLite.
However until there are proper scheduler fixes, let's make what could
be made to improve time to do queries:
If we know that the context under which a query is executed will never
be canceled - we know we can safely skip spawning the interrupt
goroutine and this was avoid ping-pong on done in between different OS
threads.
This brings the following speedup on my notebook with go1.10:
name old req/s new req/s delta
Exec 254k ± 1% 379k ± 1% +48.89% (p=0.000 n=10+10)
Query 90.6k ± 2% 96.4k ± 1% +6.37% (p=0.000 n=10+10)
Params 81.5k ± 1% 87.0k ± 1% +6.83% (p=0.000 n=10+10)
Stmt 122k ± 2% 129k ± 1% +6.07% (p=0.000 n=10+9)
Rows 2.98k ± 1% 3.06k ± 1% +2.77% (p=0.000 n=9+10)
StmtRows 3.10k ± 1% 3.13k ± 1% +1.12% (p=0.000 n=9+10)
name old time/op new time/op delta
CustomFunctions-4 10.6µs ± 1% 10.1µs ± 1% -5.01% (p=0.000 n=10+10)
2018-02-16 19:35:14 +03:00
|
|
|
}
|
2016-11-06 14:43:53 +03:00
|
|
|
|
|
|
|
return rows, nil
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2019-08-19 16:53:09 +03:00
|
|
|
// LastInsertId return last inserted ID.
|
2011-11-14 17:10:13 +04:00
|
|
|
func (r *SQLiteResult) LastInsertId() (int64, error) {
|
2013-05-11 16:43:31 +04:00
|
|
|
return r.id, nil
|
2011-11-14 17:10:13 +04:00
|
|
|
}
|
|
|
|
|
2016-11-04 18:40:06 +03:00
|
|
|
// RowsAffected return how many rows affected.
|
2011-11-14 17:10:13 +04:00
|
|
|
func (r *SQLiteResult) RowsAffected() (int64, error) {
|
2013-05-11 16:43:31 +04:00
|
|
|
return r.changes, nil
|
2011-11-14 17:10:13 +04:00
|
|
|
}
|
|
|
|
|
2016-11-04 18:40:06 +03:00
|
|
|
// Exec execute the statement with arguments. Return result object.
|
2012-02-20 11:14:49 +04:00
|
|
|
func (s *SQLiteStmt) Exec(args []driver.Value) (driver.Result, error) {
|
2023-02-12 01:14:42 +03:00
|
|
|
list := make([]driver.NamedValue, len(args))
|
2016-11-04 08:24:22 +03:00
|
|
|
for i, v := range args {
|
2023-02-12 01:14:42 +03:00
|
|
|
list[i] = driver.NamedValue{
|
2016-11-04 08:24:22 +03:00
|
|
|
Ordinal: i + 1,
|
|
|
|
Value: v,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return s.exec(context.Background(), list)
|
|
|
|
}
|
|
|
|
|
2020-08-28 18:43:21 +03:00
|
|
|
func isInterruptErr(err error) bool {
|
|
|
|
sqliteErr, ok := err.(Error)
|
|
|
|
if ok {
|
|
|
|
return sqliteErr.Code == ErrInterrupt
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2019-09-06 01:26:33 +03:00
|
|
|
// exec executes a query that doesn't return rows. Attempts to honor context timeout.
|
2023-02-12 01:14:42 +03:00
|
|
|
func (s *SQLiteStmt) exec(ctx context.Context, args []driver.NamedValue) (driver.Result, error) {
|
2019-09-06 01:26:33 +03:00
|
|
|
if ctx.Done() == nil {
|
|
|
|
return s.execSync(args)
|
|
|
|
}
|
|
|
|
|
|
|
|
type result struct {
|
|
|
|
r driver.Result
|
|
|
|
err error
|
|
|
|
}
|
|
|
|
resultCh := make(chan result)
|
|
|
|
go func() {
|
|
|
|
r, err := s.execSync(args)
|
|
|
|
resultCh <- result{r, err}
|
|
|
|
}()
|
2020-08-28 18:43:21 +03:00
|
|
|
var rv result
|
2019-09-06 01:26:33 +03:00
|
|
|
select {
|
2020-08-28 18:43:21 +03:00
|
|
|
case rv = <-resultCh:
|
2019-09-06 01:26:33 +03:00
|
|
|
case <-ctx.Done():
|
|
|
|
select {
|
2020-08-28 18:43:21 +03:00
|
|
|
case rv = <-resultCh: // no need to interrupt, operation completed in db
|
2019-09-06 01:26:33 +03:00
|
|
|
default:
|
|
|
|
// this is still racy and can be no-op if executed between sqlite3_* calls in execSync.
|
|
|
|
C.sqlite3_interrupt(s.c.db)
|
2020-08-28 18:43:21 +03:00
|
|
|
rv = <-resultCh // wait for goroutine completed
|
|
|
|
if isInterruptErr(rv.err) {
|
|
|
|
return nil, ctx.Err()
|
|
|
|
}
|
2019-09-06 01:26:33 +03:00
|
|
|
}
|
|
|
|
}
|
2020-08-28 18:43:21 +03:00
|
|
|
return rv.r, rv.err
|
2019-09-06 01:26:33 +03:00
|
|
|
}
|
|
|
|
|
2023-02-12 01:14:42 +03:00
|
|
|
func (s *SQLiteStmt) execSync(args []driver.NamedValue) (driver.Result, error) {
|
2011-11-11 16:36:22 +04:00
|
|
|
if err := s.bind(args); err != nil {
|
2014-11-16 17:51:46 +03:00
|
|
|
C.sqlite3_reset(s.s)
|
2015-03-19 07:29:43 +03:00
|
|
|
C.sqlite3_clear_bindings(s.s)
|
2011-11-11 16:36:22 +04:00
|
|
|
return nil, err
|
|
|
|
}
|
2016-11-06 14:46:27 +03:00
|
|
|
|
2015-08-07 06:13:52 +03:00
|
|
|
var rowid, changes C.longlong
|
2018-10-20 18:02:18 +03:00
|
|
|
rv := C._sqlite3_step_row_internal(s.s, &rowid, &changes)
|
2011-11-11 16:38:53 +04:00
|
|
|
if rv != C.SQLITE_ROW && rv != C.SQLITE_OK && rv != C.SQLITE_DONE {
|
2015-01-26 11:40:18 +03:00
|
|
|
err := s.c.lastError()
|
2014-11-16 17:51:46 +03:00
|
|
|
C.sqlite3_reset(s.s)
|
2015-03-19 07:29:43 +03:00
|
|
|
C.sqlite3_clear_bindings(s.s)
|
2015-01-26 11:40:18 +03:00
|
|
|
return nil, err
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
2016-11-06 14:46:27 +03:00
|
|
|
|
2016-11-08 06:19:13 +03:00
|
|
|
return &SQLiteResult{id: int64(rowid), changes: int64(changes)}, nil
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2020-12-28 02:52:08 +03:00
|
|
|
// Readonly reports if this statement is considered readonly by SQLite.
|
|
|
|
//
|
|
|
|
// See: https://sqlite.org/c3ref/stmt_readonly.html
|
|
|
|
func (s *SQLiteStmt) Readonly() bool {
|
|
|
|
return C.sqlite3_stmt_readonly(s.s) == 1
|
|
|
|
}
|
|
|
|
|
2013-01-31 11:48:30 +04:00
|
|
|
// Close the rows.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (rc *SQLiteRows) Close() error {
|
2017-08-30 07:29:47 +03:00
|
|
|
rc.s.mu.Lock()
|
2017-08-01 19:43:14 +03:00
|
|
|
if rc.s.closed || rc.closed {
|
2017-08-30 07:29:47 +03:00
|
|
|
rc.s.mu.Unlock()
|
2013-09-09 07:28:34 +04:00
|
|
|
return nil
|
|
|
|
}
|
2017-08-01 19:43:14 +03:00
|
|
|
rc.closed = true
|
2014-07-15 20:11:07 +04:00
|
|
|
if rc.cls {
|
2017-08-30 07:29:47 +03:00
|
|
|
rc.s.mu.Unlock()
|
2014-07-15 20:11:07 +04:00
|
|
|
return rc.s.Close()
|
|
|
|
}
|
2012-03-12 09:20:55 +04:00
|
|
|
rv := C.sqlite3_reset(rc.s.s)
|
|
|
|
if rv != C.SQLITE_OK {
|
2017-08-30 07:29:47 +03:00
|
|
|
rc.s.mu.Unlock()
|
2013-11-19 13:13:19 +04:00
|
|
|
return rc.s.c.lastError()
|
2012-03-12 09:20:55 +04:00
|
|
|
}
|
2017-08-30 07:29:47 +03:00
|
|
|
rc.s.mu.Unlock()
|
2012-03-12 09:20:55 +04:00
|
|
|
return nil
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
|
2016-11-04 18:40:06 +03:00
|
|
|
// Columns return column names.
|
2011-11-11 16:36:22 +04:00
|
|
|
func (rc *SQLiteRows) Columns() []string {
|
2017-08-30 07:29:47 +03:00
|
|
|
rc.s.mu.Lock()
|
|
|
|
defer rc.s.mu.Unlock()
|
|
|
|
if rc.s.s != nil && rc.nc != len(rc.cols) {
|
2011-11-11 16:36:22 +04:00
|
|
|
rc.cols = make([]string, rc.nc)
|
|
|
|
for i := 0; i < rc.nc; i++ {
|
|
|
|
rc.cols[i] = C.GoString(C.sqlite3_column_name(rc.s.s, C.int(i)))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return rc.cols
|
|
|
|
}
|
|
|
|
|
2017-08-30 13:37:57 +03:00
|
|
|
func (rc *SQLiteRows) declTypes() []string {
|
2017-08-30 07:29:47 +03:00
|
|
|
if rc.s.s != nil && rc.decltype == nil {
|
2016-03-06 23:27:17 +03:00
|
|
|
rc.decltype = make([]string, rc.nc)
|
|
|
|
for i := 0; i < rc.nc; i++ {
|
|
|
|
rc.decltype[i] = strings.ToLower(C.GoString(C.sqlite3_column_decltype(rc.s.s, C.int(i))))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return rc.decltype
|
|
|
|
}
|
|
|
|
|
2017-08-30 13:37:57 +03:00
|
|
|
// DeclTypes return column types.
|
|
|
|
func (rc *SQLiteRows) DeclTypes() []string {
|
|
|
|
rc.s.mu.Lock()
|
|
|
|
defer rc.s.mu.Unlock()
|
|
|
|
return rc.declTypes()
|
|
|
|
}
|
|
|
|
|
2019-09-06 01:26:33 +03:00
|
|
|
// Next move cursor to next. Attempts to honor context timeout from QueryContext call.
|
2012-02-20 11:14:49 +04:00
|
|
|
func (rc *SQLiteRows) Next(dest []driver.Value) error {
|
2018-07-18 03:29:43 +03:00
|
|
|
rc.s.mu.Lock()
|
|
|
|
defer rc.s.mu.Unlock()
|
2019-09-06 01:26:33 +03:00
|
|
|
|
2017-08-30 07:29:47 +03:00
|
|
|
if rc.s.closed {
|
|
|
|
return io.EOF
|
|
|
|
}
|
2019-09-06 01:26:33 +03:00
|
|
|
|
|
|
|
if rc.ctx.Done() == nil {
|
|
|
|
return rc.nextSyncLocked(dest)
|
|
|
|
}
|
|
|
|
resultCh := make(chan error)
|
|
|
|
go func() {
|
|
|
|
resultCh <- rc.nextSyncLocked(dest)
|
|
|
|
}()
|
|
|
|
select {
|
2019-12-17 10:07:49 +03:00
|
|
|
case err := <-resultCh:
|
2019-09-06 01:26:33 +03:00
|
|
|
return err
|
|
|
|
case <-rc.ctx.Done():
|
|
|
|
select {
|
|
|
|
case <-resultCh: // no need to interrupt
|
|
|
|
default:
|
|
|
|
// this is still racy and can be no-op if executed between sqlite3_* calls in nextSyncLocked.
|
|
|
|
C.sqlite3_interrupt(rc.s.c.db)
|
|
|
|
<-resultCh // ensure goroutine completed
|
|
|
|
}
|
|
|
|
return rc.ctx.Err()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// nextSyncLocked moves cursor to next; must be called with locked mutex.
|
|
|
|
func (rc *SQLiteRows) nextSyncLocked(dest []driver.Value) error {
|
2018-10-20 18:02:18 +03:00
|
|
|
rv := C._sqlite3_step_internal(rc.s.s)
|
2015-03-21 21:02:03 +03:00
|
|
|
if rv == C.SQLITE_DONE {
|
|
|
|
return io.EOF
|
|
|
|
}
|
|
|
|
if rv != C.SQLITE_ROW {
|
|
|
|
rv = C.sqlite3_reset(rc.s.s)
|
|
|
|
if rv != C.SQLITE_OK {
|
|
|
|
return rc.s.c.lastError()
|
2013-09-09 08:44:24 +04:00
|
|
|
}
|
2015-03-21 21:02:03 +03:00
|
|
|
return nil
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
2012-04-07 08:17:54 +04:00
|
|
|
|
2017-08-30 13:37:57 +03:00
|
|
|
rc.declTypes()
|
2012-04-07 08:17:54 +04:00
|
|
|
|
2011-11-11 16:36:22 +04:00
|
|
|
for i := range dest {
|
2011-11-11 16:38:53 +04:00
|
|
|
switch C.sqlite3_column_type(rc.s.s, C.int(i)) {
|
|
|
|
case C.SQLITE_INTEGER:
|
2012-04-07 08:17:54 +04:00
|
|
|
val := int64(C.sqlite3_column_int64(rc.s.s, C.int(i)))
|
2012-09-11 17:17:09 +04:00
|
|
|
switch rc.decltype[i] {
|
2018-04-17 12:13:35 +03:00
|
|
|
case columnTimestamp, columnDatetime, columnDate:
|
2015-03-05 05:05:58 +03:00
|
|
|
var t time.Time
|
2015-10-10 08:59:25 +03:00
|
|
|
// Assume a millisecond unix timestamp if it's 13 digits -- too
|
|
|
|
// large to be a reasonable timestamp in seconds.
|
|
|
|
if val > 1e12 || val < -1e12 {
|
|
|
|
val *= int64(time.Millisecond) // convert ms to nsec
|
2017-06-30 21:17:04 +03:00
|
|
|
t = time.Unix(0, val)
|
2015-01-02 09:31:46 +03:00
|
|
|
} else {
|
2017-06-30 21:17:04 +03:00
|
|
|
t = time.Unix(val, 0)
|
2015-03-05 05:05:58 +03:00
|
|
|
}
|
2017-06-30 21:17:04 +03:00
|
|
|
t = t.UTC()
|
2015-03-05 05:05:58 +03:00
|
|
|
if rc.s.c.loc != nil {
|
|
|
|
t = t.In(rc.s.c.loc)
|
2015-01-02 09:31:46 +03:00
|
|
|
}
|
2015-03-05 05:05:58 +03:00
|
|
|
dest[i] = t
|
2012-05-25 13:01:03 +04:00
|
|
|
case "boolean":
|
2012-09-11 17:17:09 +04:00
|
|
|
dest[i] = val > 0
|
2012-05-25 13:01:03 +04:00
|
|
|
default:
|
2012-04-07 08:17:54 +04:00
|
|
|
dest[i] = val
|
|
|
|
}
|
2011-11-11 16:38:53 +04:00
|
|
|
case C.SQLITE_FLOAT:
|
|
|
|
dest[i] = float64(C.sqlite3_column_double(rc.s.s, C.int(i)))
|
|
|
|
case C.SQLITE_BLOB:
|
|
|
|
p := C.sqlite3_column_blob(rc.s.s, C.int(i))
|
2013-10-24 17:25:07 +04:00
|
|
|
if p == nil {
|
2018-09-22 21:23:19 +03:00
|
|
|
dest[i] = []byte{}
|
2013-10-24 17:25:07 +04:00
|
|
|
continue
|
|
|
|
}
|
2018-09-22 21:29:00 +03:00
|
|
|
n := C.sqlite3_column_bytes(rc.s.s, C.int(i))
|
|
|
|
dest[i] = C.GoBytes(p, n)
|
2011-11-11 16:38:53 +04:00
|
|
|
case C.SQLITE_NULL:
|
|
|
|
dest[i] = nil
|
|
|
|
case C.SQLITE_TEXT:
|
2012-04-07 08:17:54 +04:00
|
|
|
var err error
|
2014-10-23 21:12:32 +04:00
|
|
|
var timeVal time.Time
|
2015-04-12 14:59:29 +03:00
|
|
|
|
|
|
|
n := int(C.sqlite3_column_bytes(rc.s.s, C.int(i)))
|
|
|
|
s := C.GoStringN((*C.char)(unsafe.Pointer(C.sqlite3_column_text(rc.s.s, C.int(i)))), C.int(n))
|
2012-12-26 05:01:39 +04:00
|
|
|
|
2012-12-30 02:20:27 +04:00
|
|
|
switch rc.decltype[i] {
|
2018-04-17 12:13:35 +03:00
|
|
|
case columnTimestamp, columnDatetime, columnDate:
|
2015-03-05 05:05:58 +03:00
|
|
|
var t time.Time
|
2015-04-15 10:26:27 +03:00
|
|
|
s = strings.TrimSuffix(s, "Z")
|
2015-03-04 16:58:32 +03:00
|
|
|
for _, format := range SQLiteTimestampFormats {
|
2015-03-04 19:17:38 +03:00
|
|
|
if timeVal, err = time.ParseInLocation(format, s, time.UTC); err == nil {
|
2015-03-05 05:05:58 +03:00
|
|
|
t = timeVal
|
2015-03-04 16:58:32 +03:00
|
|
|
break
|
2012-11-04 04:45:58 +04:00
|
|
|
}
|
2012-12-30 04:36:29 +04:00
|
|
|
}
|
|
|
|
if err != nil {
|
2012-12-30 02:20:27 +04:00
|
|
|
// The column is a time value, so return the zero time on parse failure.
|
2015-03-05 05:05:58 +03:00
|
|
|
t = time.Time{}
|
|
|
|
}
|
|
|
|
if rc.s.c.loc != nil {
|
|
|
|
t = t.In(rc.s.c.loc)
|
2012-04-07 08:17:54 +04:00
|
|
|
}
|
2015-03-05 05:05:58 +03:00
|
|
|
dest[i] = t
|
2012-12-26 05:01:39 +04:00
|
|
|
default:
|
2019-01-26 22:54:27 +03:00
|
|
|
dest[i] = s
|
2012-04-07 08:17:54 +04:00
|
|
|
}
|
2011-11-11 16:36:22 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|