Transfer & connect timeouts, in seconds & milliseconds

WordPress · Jan 23, 2014 · 0c6a797 · 0c6a797
1 parent 541e6d0
commit 0c6a797
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 5 deletions.
diff --git a/examples/timeout.php b/examples/timeout.php
@@ -0,0 +1,17 @@
+<?php
+
+// First, include Requests
+include('../library/Requests.php');
+
+// Next, make sure Requests can load internal classes
+Requests::register_autoloader();
+
+// Define a timeout of 2.5 seconds
+$options = array(
+	'timeout' => 2.5,
+);
+
+// Now let's make a request to a page that will delay its response by 3 seconds
+$request = Requests::get('http://httpbin.org/delay/3', array(), $options);
+
+// An exception will be thrown, stating a timeout of the request !
diff --git a/library/Requests.php b/library/Requests.php
@@ -253,7 +253,9 @@ public static function patch($url, $headers, $data = array(), $options = array()
 	 * options:
 	 *
 	 * - `timeout`: How long should we wait for a response?
-	 *    (integer, seconds, default: 10)
+	 *    (float, seconds with a millisecond precision, default: 10, example: 0.01)
+	 * - `connect_timeout`: How long should we wait while trying to connect?
+	 *    (float, seconds with a millisecond precision, default: 10, example: 0.01)
 	 * - `useragent`: Useragent to send to the server
 	 *    (string, default: php-requests/$version)
 	 * - `follow_redirects`: Should we follow 3xx redirects?
@@ -443,6 +445,7 @@ public static function request_multiple($requests, $options = array()) {
 	protected static function get_default_options($multirequest = false) {
 		$defaults = array(
 			'timeout' => 10,
+			'connect_timeout' => 10,
 			'useragent' => 'php-requests/' . self::VERSION,
 			'redirected' => 0,
 			'redirects' => 10,

diff --git a/library/Requests/Transport/cURL.php b/library/Requests/Transport/cURL.php
@@ -249,9 +249,17 @@ protected function setup_handle($url, $headers, $data, $options) {
 				break;
 		}
 
+		if( is_int($options['timeout']) or version_compare($this->version, '7.16.2', '<') ) {
+			curl_setopt($this->fp, CURLOPT_TIMEOUT, ceil($options['timeout']));
+		} else {
+			curl_setopt($this->fp, CURLOPT_TIMEOUT_MS, round($options['timeout'] * 1000) );
+		}
+		if( is_int($options['connect_timeout'])  or version_compare($this->version, '7.16.2', '<') ) {
+			curl_setopt($this->fp, CURLOPT_CONNECTTIMEOUT, ceil($options['connect_timeout']));
+		} else {
+			curl_setopt($this->fp, CURLOPT_CONNECTTIMEOUT_MS, round($options['connect_timeout'] * 1000));
+		}
 		curl_setopt($this->fp, CURLOPT_URL, $url);
-		curl_setopt($this->fp, CURLOPT_TIMEOUT, $options['timeout']);
-		curl_setopt($this->fp, CURLOPT_CONNECTTIMEOUT, $options['timeout']);
 		curl_setopt($this->fp, CURLOPT_REFERER, $url);
 		curl_setopt($this->fp, CURLOPT_USERAGENT, $options['useragent']);
 		curl_setopt($this->fp, CURLOPT_HTTPHEADER, $headers);

diff --git a/library/Requests/Transport/fsockopen.php b/library/Requests/Transport/fsockopen.php
@@ -99,7 +99,7 @@ public function request($url, $headers = array(), $data = array(), $options = ar
 
 		$options['hooks']->dispatch('fsockopen.remote_socket', array(&$remote_socket));
 
-		$fp = stream_socket_client($remote_socket, $errno, $errstr, $options['timeout'], STREAM_CLIENT_CONNECT, $context);
+		$fp = stream_socket_client($remote_socket, $errno, $errstr, ceil($options['connect_timeout']), STREAM_CLIENT_CONNECT, $context);
 
 		restore_error_handler();
 
@@ -198,7 +198,10 @@ public function request($url, $headers = array(), $data = array(), $options = ar
 			$options['hooks']->dispatch('fsockopen.after_request', array(&$fake_headers));
 			return '';
 		}
-		stream_set_timeout($fp, $options['timeout']);
+
+        $timeout_sec = (int) floor( $options['timeout'] );
+		$timeout_msec = $timeout_sec == $options['timeout'] ? 0 : 1000000 * $options['timeout'] % 1000000;
+		stream_set_timeout( $fp, $timeout_sec, $timeout_msec );
 
 		$this->info = stream_get_meta_data($fp);