diff options
Diffstat (limited to 'it_dbi.class')
| -rw-r--r-- | it_dbi.class | 25 |
1 files changed, 23 insertions, 2 deletions
diff --git a/it_dbi.class b/it_dbi.class index 6a2b369..786fb90 100644 --- a/it_dbi.class +++ b/it_dbi.class @@ -44,6 +44,8 @@ class it_dbi implements Iterator '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 + 'interruptible_queries' => false, # make queries interruptible by php signal handlers + 'timeout' => null, # timeout for queries ); var $_key; # Key of currently loaded record or null (public readonly) @@ -1126,9 +1128,28 @@ function _tables($p) { return (array)$result; } +function __query($query, $p) +{ + if ($p['timeout'] || $this->_p['interruptible_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 +1159,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); } } |