implement timeout and cancelable_queries

1 files changed, 22 insertions, 2 deletions

diff --git a/it_dbi.class b/it_dbi.class
index 6d798a6..86c5e6d 100644
--- a/it_dbi.class
+++ b/it_dbi.class
@@ -43,6 +43,7 @@ class it_dbi
 		'throttle_writes' => 0, # sleep for 'throttle_writes' multiplied by the execution time after every write
 		'unbuffered' => false, # use MYSQLI_USE_RESULT (WARNING this is not at all equivalent to normal it_dbi WARNING)
 		'ignored_warnings' => "", # regex of additional mysql warnings numbers to ignore
+		'cancelable_queries' => false,
 	);
 
 	var $_key;         # Key of currently loaded record or null (public readonly)
@@ -1126,9 +1127,28 @@ function _tables($p) {
 	return (array)$result;
 }
 
+function __query($query, $p)
+{
+	if ($p['timeout'] || $this->_p['cancelable_queries'])
+	{
+		$starttime = microtime(true);
+		mysqli_query($this->_link, $query, ($p['unbuffered'] ? MYSQLI_USE_RESULT : MYSQLI_STORE_RESULT) | MYSQLI_ASYNC);
+		do {
+			$read = $error = $reject = [$this->_link];
+			if ($p['timeout'] && (microtime(true) - $starttime) > $p['timeout'])
+				return false;
+		} while (!mysqli_poll($read, $error, $reject, 1, 0));
+		return mysqli_reap_async_query($this->_link);
+	}
+	else
+	{
+		return mysqli_query($this->_link, $query, $p['unbuffered'] ? MYSQLI_USE_RESULT : MYSQLI_STORE_RESULT);
+	}
+}
+
 function _query($query, $p)
 {
-	if (!($result = mysqli_query($this->_link, $query, $p['unbuffered'] ? MYSQLI_USE_RESULT : MYSQLI_STORE_RESULT)) && $p['safety'])
+	if (!($result = $this->__query($query, $p)) && $p['safety'])
 	{
 		$errno = mysqli_errno($this->_link);
 		if (($p['safety'] < 2) && ($errno == 1062)) # Duplicate entry
@@ -1138,7 +1158,7 @@ function _query($query, $p)
 		{
 			it::log('sqllog', "it_dbi(): reconnecting mysqli_connect {$p['server']}, {$p['db']}");
 			$this->_connect(array('reconnect' => true));
-			$result = mysqli_query($this->_link, $query, $p['unbuffered'] ? MYSQLI_USE_RESULT : MYSQLI_STORE_RESULT);
+			$result = $this->__query($query, $p);
 		}
 	}