Add PDO_SQLITE_ATTR_TRANSACTION_MODE

Author: stancl

ext/pdo_sqlite/pdo_sqlite.stub.php

@@ -39,6 +39,18 @@ class Sqlite extends \PDO
     /** @cvalue PDO_SQLITE_ATTR_EXPLAIN_STATEMENT */
     public const int ATTR_EXPLAIN_STATEMENT = UNKNOWN;
 
+    /** @cvalue PDO_SQLITE_ATTR_TRANSACTION_MODE */
+    public const int ATTR_TRANSACTION_MODE = UNKNOWN;
+
+    /** @cvalue PDO_SQLITE_ATTR_TRANSACTION_MODE_DEFERRED */
+    public const int ATTR_TRANSACTION_MODE_DEFERRED = UNKNOWN;
+
+    /** @cvalue PDO_SQLITE_ATTR_TRANSACTION_MODE_IMMEDIATE */
+    public const int ATTR_TRANSACTION_MODE_IMMEDIATE = UNKNOWN;
+
+    /** @cvalue PDO_SQLITE_ATTR_TRANSACTION_MODE_EXCLUSIVE */
+    public const int ATTR_TRANSACTION_MODE_EXCLUSIVE = UNKNOWN;
+
 #if SQLITE_VERSION_NUMBER >= 3043000
     public const int EXPLAIN_MODE_PREPARED = 0;
     public const int EXPLAIN_MODE_EXPLAIN = 1;

ext/pdo_sqlite/pdo_sqlite_arginfo.h

@@ -51,6 +51,7 @@ typedef struct {
 	struct pdo_sqlite_func *funcs;
 	struct pdo_sqlite_collation *collations;
 	zend_fcall_info_cache authorizer_fcc;
+	zend_long transaction_mode;
 } pdo_sqlite_db_handle;
 
 typedef struct {
@@ -75,7 +76,11 @@ enum {
 	PDO_SQLITE_ATTR_READONLY_STATEMENT,
 	PDO_SQLITE_ATTR_EXTENDED_RESULT_CODES,
 	PDO_SQLITE_ATTR_BUSY_STATEMENT,
-	PDO_SQLITE_ATTR_EXPLAIN_STATEMENT
+	PDO_SQLITE_ATTR_EXPLAIN_STATEMENT,
+	PDO_SQLITE_ATTR_TRANSACTION_MODE,
+	PDO_SQLITE_ATTR_TRANSACTION_MODE_DEFERRED,
+	PDO_SQLITE_ATTR_TRANSACTION_MODE_IMMEDIATE,
+	PDO_SQLITE_ATTR_TRANSACTION_MODE_EXCLUSIVE
 };
 
 typedef int pdo_sqlite_create_collation_callback(void*, int, const void*, int, const void*);

ext/pdo_sqlite/php_pdo_sqlite_int.h

@@ -255,7 +255,22 @@ static bool sqlite_handle_begin(pdo_dbh_t *dbh)
 {
 	pdo_sqlite_db_handle *H = (pdo_sqlite_db_handle *)dbh->driver_data;
 
-	if (sqlite3_exec(H->db, "BEGIN", NULL, NULL, NULL) != SQLITE_OK) {
+	char *begin_statement;
+	switch (H->transaction_mode) {
+		case 0:
+		case PDO_SQLITE_ATTR_TRANSACTION_MODE_DEFERRED:
+			begin_statement = "BEGIN DEFERRED TRANSACTION";
+			break;
+		case PDO_SQLITE_ATTR_TRANSACTION_MODE_IMMEDIATE:
+			begin_statement = "BEGIN IMMEDIATE TRANSACTION";
+			break;
+		case PDO_SQLITE_ATTR_TRANSACTION_MODE_EXCLUSIVE:
+			begin_statement = "BEGIN EXCLUSIVE TRANSACTION";
+		default:
+			begin_statement = "BEGIN";
+	}
+
+	if (sqlite3_exec(H->db, begin_statement, NULL, NULL, NULL) != SQLITE_OK) {
 		pdo_sqlite_error(dbh);
 		return false;
 	}
@@ -286,11 +301,22 @@ static bool sqlite_handle_rollback(pdo_dbh_t *dbh)
 
 static int pdo_sqlite_get_attribute(pdo_dbh_t *dbh, zend_long attr, zval *return_value)
 {
+	pdo_sqlite_db_handle *H = (pdo_sqlite_db_handle *)dbh->driver_data;
+
 	switch (attr) {
 		case PDO_ATTR_CLIENT_VERSION:
 		case PDO_ATTR_SERVER_VERSION:
 			ZVAL_STRING(return_value, (char *)sqlite3_libversion());
 			break;
+		case PDO_SQLITE_ATTR_TRANSACTION_MODE:
+			zend_long mode;
+			if (H->transaction_mode == 0) {
+				mode = PDO_SQLITE_ATTR_TRANSACTION_MODE_DEFERRED;
+			} else {
+				mode = H->transaction_mode;
+			}
+			ZVAL_LONG(return_value, mode);
+			break;
 
 		default:
 			return 0;
@@ -326,6 +352,18 @@ static bool pdo_sqlite_set_attr(pdo_dbh_t *dbh, zend_long attr, zval *val)
 			}
 			sqlite3_extended_result_codes(H->db, lval);
 			return true;
+		case PDO_SQLITE_ATTR_TRANSACTION_MODE:
+			if (!pdo_get_long_param(&lval, val)) {
+				return false;
+			}
+			if (lval != PDO_SQLITE_ATTR_TRANSACTION_MODE_DEFERRED &&
+				lval != PDO_SQLITE_ATTR_TRANSACTION_MODE_IMMEDIATE &&
+				lval != PDO_SQLITE_ATTR_TRANSACTION_MODE_EXCLUSIVE) {
+				return false;
+			}
+			H->transaction_mode = lval;
+			return true;
+
 	}
 	return false;
 }