PoolCounterClient.php

Go to the documentation of this file.

<?php
namespace MediaWiki\PoolCounter;
 
use MediaWiki\Status\Status;
use PoolCounter;
 
class PoolCounterClient extends PoolCounter {
    private $conn;
 
    private $hostName;
 
    private $manager;
 
    public function setManager( PoolCounterConnectionManager $manager ): void {
        $this->manager = $manager;
    }
 
    public function getConn() {
        if ( !isset( $this->conn ) ) {
            $status = $this->manager->get( $this->key );
            if ( !$status->isOK() ) {
                return $status;
            }
            // @phan-suppress-next-line PhanTypeArraySuspiciousNullable
            $this->conn = $status->value['conn'];
            // @phan-suppress-next-line PhanTypeArraySuspiciousNullable
            $this->hostName = $status->value['hostName'];
 
            // Set the read timeout to be 1.5 times the pool timeout.
            // This allows the server to time out gracefully before we give up on it.
            stream_set_timeout( $this->conn, 0, (int)( $this->timeout * 1e6 * 1.5 ) );
        }
        // TODO: Convert from Status to StatusValue
        return Status::newGood( $this->conn );
    }
 
    public function sendCommand( ...$args ) {
        $args = str_replace( ' ', '%20', $args );
        $cmd = implode( ' ', $args );
        $status = $this->getConn();
        if ( !$status->isOK() ) {
            return $status;
        }
        $conn = $status->value;
        wfDebug( "Sending pool counter command: $cmd\n" );
        if ( fwrite( $conn, "$cmd\n" ) === false ) {
            return Status::newFatal( 'poolcounter-write-error', $this->hostName );
        }
        $response = fgets( $conn );
        if ( $response === false ) {
            return Status::newFatal( 'poolcounter-read-error', $this->hostName );
        }
        $response = rtrim( $response, "\r\n" );
        wfDebug( "Got pool counter response: $response\n" );
        $parts = explode( ' ', $response, 2 );
        $responseType = $parts[0];
        switch ( $responseType ) {
            case 'LOCKED':
                $this->onAcquire();
                break;
            case 'RELEASED':
                $this->onRelease();
                break;
            case 'DONE':
            case 'NOT_LOCKED':
            case 'QUEUE_FULL':
            case 'TIMEOUT':
            case 'LOCK_HELD':
                break;
            case 'ERROR':
            default:
                $parts = explode( ' ', $parts[1], 2 );
                $errorMsg = $parts[1] ?? '(no message given)';
                return Status::newFatal( 'poolcounter-remote-error', $errorMsg, $this->hostName );
        }
        return Status::newGood( constant( "PoolCounter::$responseType" ) );
    }
 
    public function acquireForMe( $timeout = null ) {
        $status = $this->precheckAcquire();
        if ( !$status->isGood() ) {
            return $status;
        }
        return $this->sendCommand( 'ACQ4ME', $this->key, $this->workers, $this->maxqueue,
            $timeout ?? $this->timeout );
    }
 
    public function acquireForAnyone( $timeout = null ) {
        $status = $this->precheckAcquire();
        if ( !$status->isGood() ) {
            return $status;
        }
        return $this->sendCommand( 'ACQ4ANY', $this->key, $this->workers, $this->maxqueue,
            $timeout ?? $this->timeout );
    }
 
    public function release() {
        $status = $this->sendCommand( 'RELEASE' );
 
        if ( $this->conn ) {
            $this->manager->close( $this->conn );
            $this->conn = null;
        }
 
        return $status;
    }
}