JobQueueMemory.php

Go to the documentation of this file.

<?php
class JobQueueMemory extends JobQueue {
    protected static $data = [];

    protected function doBatchPush( array $jobs, $flags ) {
        $unclaimed =& $this->getQueueData( 'unclaimed', [] );

        foreach ( $jobs as $job ) {
            if ( $job->ignoreDuplicates() ) {
                $sha1 = Wikimedia\base_convert(
                    sha1( serialize( $job->getDeduplicationInfo() ) ),
                    16, 36, 31
                );
                if ( !isset( $unclaimed[$sha1] ) ) {
                    $unclaimed[$sha1] = $job;
                }
            } else {
                $unclaimed[] = $job;
            }
        }
    }

    protected function supportedOrders() {
        return [ 'random', 'timestamp', 'fifo' ];
    }

    protected function optimalOrder() {
        return 'fifo';
    }

    protected function doIsEmpty() {
        return ( $this->doGetSize() == 0 );
    }

    protected function doGetSize() {
        $unclaimed = $this->getQueueData( 'unclaimed' );

        return $unclaimed ? count( $unclaimed ) : 0;
    }

    protected function doGetAcquiredCount() {
        $claimed = $this->getQueueData( 'claimed' );

        return $claimed ? count( $claimed ) : 0;
    }

    protected function doPop() {
        if ( $this->doGetSize() == 0 ) {
            return false;
        }

        $unclaimed =& $this->getQueueData( 'unclaimed' );
        $claimed =& $this->getQueueData( 'claimed', [] );

        if ( $this->order === 'random' ) {
            $key = array_rand( $unclaimed );
        } else {
            reset( $unclaimed );
            $key = key( $unclaimed );
        }

        $spec = $unclaimed[$key];
        unset( $unclaimed[$key] );
        $claimed[] = $spec;

        $job = $this->jobFromSpecInternal( $spec );

        end( $claimed );
        $job->metadata['claimId'] = key( $claimed );

        return $job;
    }

    protected function doAck( Job $job ) {
        if ( $this->getAcquiredCount() == 0 ) {
            return;
        }

        $claimed =& $this->getQueueData( 'claimed' );
        unset( $claimed[$job->metadata['claimId']] );
    }

    protected function doDelete() {
        if ( isset( self::$data[$this->type][$this->wiki] ) ) {
            unset( self::$data[$this->type][$this->wiki] );
            if ( !self::$data[$this->type] ) {
                unset( self::$data[$this->type] );
            }
        }
    }

    public function getAllQueuedJobs() {
        $unclaimed = $this->getQueueData( 'unclaimed' );
        if ( !$unclaimed ) {
            return new ArrayIterator( [] );
        }

        return new MappedIterator(
            $unclaimed,
            function ( $value ) {
                $this->jobFromSpecInternal( $value );
            }
        );
    }

    public function getAllAcquiredJobs() {
        $claimed = $this->getQueueData( 'claimed' );
        if ( !$claimed ) {
            return new ArrayIterator( [] );
        }

        return new MappedIterator(
            $claimed,
            function ( $value ) {
                $this->jobFromSpecInternal( $value );
            }
        );
    }

    public function jobFromSpecInternal( IJobSpecification $spec ) {
        return Job::factory( $spec->getType(), $spec->getTitle(), $spec->getParams() );
    }

    private function &getQueueData( $field, $init = null ) {
        if ( !isset( self::$data[$this->type][$this->wiki][$field] ) ) {
            if ( $init !== null ) {
                self::$data[$this->type][$this->wiki][$field] = $init;
            } else {
                return $init;
            }
        }

        return self::$data[$this->type][$this->wiki][$field];
    }
}