From e02bbc03819ea6dfc61212b7dbd772bc0809ae7a Mon Sep 17 00:00:00 2001
From: Gert-Jan Timmer <gjr.timmer@gmail.com>
Date: Tue, 29 May 2018 13:13:38 +0200
Subject: ADD: PRAGMA journal_mode

---
 sqlite3.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'sqlite3.go')

diff --git a/sqlite3.go b/sqlite3.go
index 4b79aac..3855882 100644
--- a/sqlite3.go
+++ b/sqlite3.go
@@ -838,6 +838,10 @@ func errorString(err Error) string {
 //     This pragma enables or disables the enforcement of CHECK constraints.
 //     The default setting is off, meaning that CHECK constraints are enforced by default.
 //
+//   _journal=MODE
+//     Set journal mode for the databases associated with the current connection.
+//     https://www.sqlite.org/pragma.html#pragma_journal_mode
+//
 //   _recursive_triggers=Boolean | _rt=Boolean
 //     Enable or disable recursive triggers.
 //
@@ -863,6 +867,7 @@ func (d *SQLiteDriver) Open(dsn string) (driver.Conn, error) {
 	caseSensitiveLike := -1
 	deferForeignKeys := -1
 	ignoreCheckConstraints := -1
+	journalMode := "DELETE"
 	foreignKeys := -1
 	recursiveTriggers := -1
 
@@ -1022,6 +1027,19 @@ func (d *SQLiteDriver) Open(dsn string) (driver.Conn, error) {
 			}
 		}
 
+		// Journal Mode (_journal)
+		//
+		// https://www.sqlite.org/pragma.html#pragma_journal_mode
+		//
+		if val := params.Get("_journal"); val != "" {
+			switch strings.ToUpper(val) {
+			case "DELETE", "TRUNCATE", "PERSIST", "MEMORY", "WAL", "OFF":
+				journalMode = strings.ToUpper(val)
+			default:
+				return nil, fmt.Errorf("Invalid _journal: %v, expecting value of 'DELETE TRUNCATE PERSIST MEMORY WAL OFF'", val)
+			}
+		}
+
 		// Recursive Triggers (_recursive_triggers)
 		//
 		// https://www.sqlite.org/pragma.html#pragma_recursive_triggers
@@ -1118,6 +1136,13 @@ func (d *SQLiteDriver) Open(dsn string) (driver.Conn, error) {
 		}
 	}
 
+	// Journal Mode
+	// Because default Journal Mode is DELETE this PRAGMA can always be executed.
+	if err := exec(fmt.Sprintf("PRAGMA journal_mode = %s;", journalMode)); err != nil {
+		C.sqlite3_close_v2(db)
+		return nil, err
+	}
+
 	// Recursive Triggers
 	if recursiveTriggers > -1 {
 		if err := exec(fmt.Sprintf("PRAGMA recursive_triggers = %d;", recursiveTriggers)); err != nil {
-- 
cgit v1.2.3