2010-11-21 22:48:11 +00:00
parent 01018862f7
commit 952af43b97
23 changed files with 3314 additions and 0 deletions
@@ -0,0 +1,100 @@
+<?php
+/**
+ * The SocketPool class that will handle all the sockets.
+ * Manages a pool of socket ressources with socket_select()
+ * @author jpt
+ * @package Socket
+ */
+class Socket_SocketPool {
+
+	/**
+	 * Pool that contains socket ressources.
+	 * @var array
+	 */
+	protected $sockets;
+
+	/**
+	 * Creates sockets.
+	 * @return void
+	 */
+	function __construct() {
+		$this->sockets = array();
+	}
+
+	/**
+	 * Closes all connections.
+	 * @return void
+	 */
+	function __destruct() {
+		unset($this->sockets);
+	}
+
+	/**
+	 * Returns the amount of active sockets.
+	 * @return int
+	 */
+	public function countSockets() {
+		return count($this->sockets);
+	}
+
+	/**
+	 * Adds a socket to the pool
+	 * @param ressource $add_socket
+	 * @return void
+	 */
+	public function addSocket($add_socket) {
+		array_push($this->sockets, $add_socket);
+	}
+
+	/**
+	 * Removes the given socket from the pool.
+	 * The socket will be shutdown.
+	 * @param ressource $remove_socket
+	 * @return void
+	 */
+	public function removeSocket($remove_socket) {
+		foreach($this->sockets AS $key=>$socket) {
+			if($socket !== $remove_socket) {
+				continue;
+			} else {
+				unset($socket);
+				unset($this->sockets[$key]);
+				break;
+			}
+		}
+	}
+
+	/**
+	 * Creates a new TcpSocket and adds it to the pool.
+	 * @throws Exception_SocketException
+	 * @param boolean $IPv6 will determine whether the socket uses IPv4 or IPv6.
+	 * @return ressource
+	 */
+	public function createTcpSocket($IPv6 = FALSE) {
+		$Domain = ($IPv6) ? AF_INET6 : AF_INET;
+		$socket = socket_create($Domain, SOCK_STREAM, SOL_TCP);
+		socket_set_block($socket);
+		if($socket === FALSE) throw new Exception_SocketException("socket_create() failed!", 1290273709);
+		$this->addSocket($socket);
+		return $socket;
+	}
+
+	/**
+	 * Returns an array of sockets one can read from.
+	 * @param array $read
+	 * @param array $write
+	 * @param array $except
+	 * @param int $timeout
+	 * @throws Exception_SocketException
+	 * @return array
+	 */
+	public function select($read, $write, $except, $timeout = NULL) {
+		$n = socket_select($read, $write, $except, $timeout);
+		if($n === FALSE) throw new Exception_SocketException("socket_select() failed!", 1290273693);
+		if($n === 0) return array("read" => array(), "write" => array(), "except" => array());
+		return array("read" => $read, "write" => $write, "except" => $except);
+	}
+
+
+}
+?>