SocketCollection.php

<?php declare(strict_types=1);

namespace hollodotme\FastCGI\Sockets;

use Countable;
use Exception;
use hollodotme\FastCGI\Exceptions\ReadFailedException;
use hollodotme\FastCGI\Exceptions\WriteFailedException;
use hollodotme\FastCGI\Interfaces\ConfiguresSocketConnection;
use hollodotme\FastCGI\Interfaces\EncodesNameValuePair;
use hollodotme\FastCGI\Interfaces\EncodesPacket;
use function array_search;
use function count;

final class SocketCollection implements Countable
{
	/** @var Socket[] */
	private array $sockets = [];

	/**
	 * @param ConfiguresSocketConnection $connection
	 * @param EncodesPacket              $packetEncoder
	 * @param EncodesNameValuePair       $nameValuePairEncoder
	 *
	 * @return Socket
	 * @throws Exception
	 * @throws WriteFailedException
	 */
	public function new(
		ConfiguresSocketConnection $connection,
		EncodesPacket $packetEncoder,
		EncodesNameValuePair $nameValuePairEncoder
	) : Socket
	{
		for ( $i = 0; $i < 10; $i++ )
		{
			$socketId = SocketId::new();

			if ( $this->exists( $socketId->getValue() ) )
			{
				continue;
			}

			$this->sockets[ $socketId->getValue() ] = new Socket(
				$socketId,
				$connection,
				$packetEncoder,
				$nameValuePairEncoder
			);

			return $this->sockets[ $socketId->getValue() ];
		}

		throw new WriteFailedException( 'Could not allocate a new socket ID' );
	}

	/**
	 * @param int $socketId
	 *
	 * @return Socket
	 * @throws ReadFailedException
	 */
	public function getById( int $socketId ) : Socket
	{
		$this->guardSocketExists( $socketId );

		return $this->sockets[ $socketId ];
	}

	/**
	 * @param array<int, resource> $resources
	 *
	 * @return array<int>
	 * @throws ReadFailedException
	 */
	public function getSocketIdsByResources( array $resources ) : array
	{
		$socketIds = [];

		foreach ( $resources as $resource )
		{
			$socketIds[] = $this->getByResource( $resource )->getId();
		}

		return $socketIds;
	}

	/**
	 * @param resource $resource
	 *
	 * @return Socket
	 * @throws ReadFailedException
	 */
	public function getByResource( $resource ) : Socket
	{
		$socketId = array_search( $resource, $this->collectResources(), true );

		if ( false === $socketId )
		{
			throw new ReadFailedException( 'Socket not found for resource' );
		}

		return $this->sockets[ $socketId ];
	}

	/**
	 * @param int $socketId
	 *
	 * @throws ReadFailedException
	 */
	private function guardSocketExists( int $socketId ) : void
	{
		if ( !$this->exists( $socketId ) )
		{
			throw new ReadFailedException( 'Socket not found for socket ID: ' . $socketId );
		}
	}

	private function exists( int $socketId ) : bool
	{
		return isset( $this->sockets[ $socketId ] );
	}

	public function remove( int $socketId ) : void
	{
		unset( $this->sockets[ $socketId ] );
	}

	public function getIdleSocket( ConfiguresSocketConnection $connection ) : ?Socket
	{
		if ( $this->isEmpty() )
		{
			return null;
		}

		foreach ( $this->sockets as $socket )
		{
			if ( !$socket->usesConnection( $connection ) )
			{
				continue;
			}

			if ( !$socket->isIdle() )
			{
				continue;
			}

			if ( !$socket->isUsable() )
			{
				$this->remove( $socket->getId() );
				continue;
			}

			return $socket;
		}

		return null;
	}

	public function hasBusySockets() : bool
	{
		if ( $this->isEmpty() )
		{
			return false;
		}

		foreach ( $this->sockets as $socket )
		{
			if ( $socket->isBusy() )
			{
				return true;
			}
		}

		return false;
	}

	/**
	 * @return array<int, resource>
	 */
	public function collectResources() : array
	{
		$resources = [];

		foreach ( $this->sockets as $socket )
		{
			$socket->collectResource( $resources );
		}

		return $resources;
	}

	public function count() : int
	{
		return count( $this->sockets );
	}

	public function isEmpty() : bool
	{
		return [] === $this->sockets;
	}

    /** @return array<int> */
    public function hasResponses() : array
    {
        if ( $this->isEmpty() )
        {
            return [];
        }

        foreach ( $this->sockets as $socket )
        {
            if ( $socket->hasResponse() )
            {
                $socketIds[] = $socket->getId();
            }
        }

        return $socketIds ?? [];
    }
}