SquidPurgeClientPool.php

Go to the documentation of this file.

<?php
class SquidPurgeClientPool {
    protected $clients = [];

    protected $timeout = 5;

    function __construct( $options = [] ) {
        if ( isset( $options['timeout'] ) ) {
            $this->timeout = $options['timeout'];
        }
    }

    public function addClient( $client ) {
        $this->clients[] = $client;
    }

    public function run() {
        $done = false;
        $startTime = microtime( true );
        while ( !$done ) {
            $readSockets = $writeSockets = [];
            foreach ( $this->clients as $clientIndex => $client ) {
                $sockets = $client->getReadSocketsForSelect();
                foreach ( $sockets as $i => $socket ) {
                    $readSockets["$clientIndex/$i"] = $socket;
                }
                $sockets = $client->getWriteSocketsForSelect();
                foreach ( $sockets as $i => $socket ) {
                    $writeSockets["$clientIndex/$i"] = $socket;
                }
            }
            if ( !count( $readSockets ) && !count( $writeSockets ) ) {
                break;
            }
            $exceptSockets = null;
            $timeout = min( $startTime + $this->timeout - microtime( true ), 1 );
            MediaWiki\suppressWarnings();
            $numReady = socket_select( $readSockets, $writeSockets, $exceptSockets, $timeout );
            MediaWiki\restoreWarnings();
            if ( $numReady === false ) {
                wfDebugLog( 'squid', __METHOD__ . ': Error in stream_select: ' .
                    socket_strerror( socket_last_error() ) . "\n" );
                break;
            }
            // Check for timeout, use 1% tolerance since we aimed at having socket_select()
            // exit at precisely the overall timeout
            if ( microtime( true ) - $startTime > $this->timeout * 0.99 ) {
                wfDebugLog( 'squid', __CLASS__ . ": timeout ({$this->timeout}s)\n" );
                break;
            } elseif ( !$numReady ) {
                continue;
            }

            foreach ( $readSockets as $key => $socket ) {
                list( $clientIndex, ) = explode( '/', $key );
                $client = $this->clients[$clientIndex];
                $client->doReads();
            }
            foreach ( $writeSockets as $key => $socket ) {
                list( $clientIndex, ) = explode( '/', $key );
                $client = $this->clients[$clientIndex];
                $client->doWrites();
            }

            $done = true;
            foreach ( $this->clients as $client ) {
                if ( !$client->isIdle() ) {
                    $done = false;
                }
            }
        }
        foreach ( $this->clients as $client ) {
            $client->close();
        }
    }
}