Merge pull request #3526 from annando/bugfix-index

[friendica.git] / include / dba.php

diff --git a/include/dba.php b/include/dba.php
index b9d7f71e53dbe32e6181b7d06a1a4e3c99d37e51..1f428bf4651b2b43ed0c47dc785761b3b0d1d443 100644 (file)
--- a/include/dba.php
+++ b/include/dba.php
@@ -459,6 +459,27 @@ class dba {
                return $sql;
        }
 
+       /**
+        * @brief beautifies the query - useful for "SHOW PROCESSLIST"
+        *
+        * This is safe when we bind the parameters later.
+        * The parameter values aren't part of the SQL.
+        *
+        * @param string $sql An SQL string without the values
+        * @return string The input SQL string modified if necessary.
+        */
+       public function clean_query($sql) {
+               $search = array("\t", "\n", "\r", "  ");
+               $replace = array(' ', ' ', ' ', ' ');
+               do {
+                       $oldsql = $sql;
+                       $sql = str_replace($search, $replace, $sql);
+               } while ($oldsql != $sql);
+
+               return $sql;
+       }
+
+
        /**
         * @brief Replaces the ? placeholders with the parameters in the $args array
         *
@@ -521,6 +542,7 @@ class dba {
                        logger('Parameter mismatch. Query "'.$sql.'" - Parameters '.print_r($args, true), LOGGER_DEBUG);
                }
 
+               $sql = self::$dbo->clean_query($sql);
                $sql = self::$dbo->any_value_fallback($sql);
 
                if (x($a->config,'system') && x($a->config['system'], 'db_callstack')) {
@@ -708,6 +730,9 @@ class dba {
         * @return int Number of rows
         */
        static public function num_rows($stmt) {
+               if (!is_object($stmt)) {
+                       return 0;
+               }
                switch (self::$dbo->driver) {
                        case 'pdo':
                                return $stmt->rowCount();
@@ -781,6 +806,41 @@ class dba {
                return self::e($sql, $param);
        }
 
+       /**
+        * @brief Locks a table for exclusive write access
+        *
+        * This function can be extended in the future to accept a table array as well.
+        *
+        * @param string $table Table name
+        *
+        * @return boolean was the lock successful?
+        */
+       static public function lock($table) {
+               // See here: https://dev.mysql.com/doc/refman/5.7/en/lock-tables-and-transactions.html
+               self::e("SET autocommit=0");
+               $success = self::e("LOCK TABLES `".self::$dbo->escape($table)."` WRITE");
+               if (!$success) {
+                       self::e("SET autocommit=1");
+               } else {
+                       self::$in_transaction = true;
+               }
+               return $success;
+       }
+
+       /**
+        * @brief Unlocks all locked tables
+        *
+        * @return boolean was the unlock successful?
+        */
+       static public function unlock() {
+               // See here: https://dev.mysql.com/doc/refman/5.7/en/lock-tables-and-transactions.html
+               self::e("COMMIT");
+               $success = self::e("UNLOCK TABLES");
+               self::e("SET autocommit=1");
+               self::$in_transaction = false;
+               return $success;
+       }
+
        /**
         * @brief Starts a transaction
         *
@@ -845,7 +905,7 @@ class dba {
        }
 
        /**
-        * @brief Insert a row into a table
+        * @brief Delete a row from a table
         *
         * @param string $table Table name
         * @param array $param parameter array
@@ -918,7 +978,9 @@ class dba {
 
                if (!$in_process) {
                        // Now we finalize the process
-                       if (!self::$in_transaction) {
+                       $do_transaction = !self::$in_transaction;
+
+                       if ($do_transaction) {
                                self::transaction();
                        }
 
@@ -932,7 +994,7 @@ class dba {
                                        logger(dba::replace_parameters($sql, $command['param']), LOGGER_DATA);
 
                                        if (!self::e($sql, $command['param'])) {
-                                               if (!self::$in_transaction) {
+                                               if ($do_transaction) {
                                                        self::rollback();
                                                }
                                                return false;
@@ -962,7 +1024,7 @@ class dba {
                                                logger(dba::replace_parameters($sql, $field_values), LOGGER_DATA);
 
                                                if (!self::e($sql, $field_values)) {
-                                                       if (!self::$in_transaction) {
+                                                       if ($do_transaction) {
                                                                self::rollback();
                                                        }
                                                        return false;
@@ -970,7 +1032,7 @@ class dba {
                                        }
                                }
                        }
-                       if (!self::$in_transaction) {
+                       if ($do_transaction) {
                                self::commit();
                        }
                        return true;
@@ -1198,6 +1260,7 @@ function q($sql) {
        unset($args[0]);
 
        if ($db && $db->connected) {
+               $sql = $db->clean_query($sql);
                $sql = $db->any_value_fallback($sql);
                $stmt = @vsprintf($sql,$args); // Disabled warnings
                //logger("dba: q: $stmt", LOGGER_ALL);
@@ -1235,6 +1298,7 @@ function qu($sql) {
        unset($args[0]);
 
        if ($db && $db->connected) {
+               $sql = $db->clean_query($sql);
                $sql = $db->any_value_fallback($sql);
                $stmt = @vsprintf($sql,$args); // Disabled warnings
                if ($stmt === false)