ircbot/Classes/Socket/SocketPool.php

<?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
	 */
	public function __construct() {
		$this->sockets = array();
	}

	/**
	 * Closes all connections.
	 * @return void
	 */
	public 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);
	}


}
?>