3 * PHP memory-backed job queue code.
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 * http://www.gnu.org/copyleft/gpl.html
24 * Class to handle job queues stored in PHP memory for testing
26 * JobQueueGroup does not remember every queue instance, so statically track it here
31 class JobQueueMemory
extends JobQueue
{
33 protected static $data = [];
35 public function __construct( array $params ) {
36 parent
::__construct( $params );
38 $this->dupCache
= new HashBagOStuff();
42 * @see JobQueue::doBatchPush
44 * @param IJobSpecification[] $jobs
47 protected function doBatchPush( array $jobs, $flags ) {
48 $unclaimed =& $this->getQueueData( 'unclaimed', [] );
50 foreach ( $jobs as $job ) {
51 if ( $job->ignoreDuplicates() ) {
52 $sha1 = sha1( serialize( $job->getDeduplicationInfo() ) );
53 if ( !isset( $unclaimed[$sha1] ) ) {
54 $unclaimed[$sha1] = $job;
63 * @see JobQueue::supportedOrders
67 protected function supportedOrders() {
68 return [ 'random', 'timestamp', 'fifo' ];
72 * @see JobQueue::optimalOrder
76 protected function optimalOrder() {
81 * @see JobQueue::doIsEmpty
85 protected function doIsEmpty() {
86 return ( $this->doGetSize() == 0 );
90 * @see JobQueue::doGetSize
94 protected function doGetSize() {
95 $unclaimed = $this->getQueueData( 'unclaimed' );
97 return $unclaimed ?
count( $unclaimed ) : 0;
101 * @see JobQueue::doGetAcquiredCount
105 protected function doGetAcquiredCount() {
106 $claimed = $this->getQueueData( 'claimed' );
108 return $claimed ?
count( $claimed ) : 0;
112 * @see JobQueue::doPop
114 * @return RunnableJob|bool
116 protected function doPop() {
117 if ( $this->doGetSize() == 0 ) {
121 $unclaimed =& $this->getQueueData( 'unclaimed' );
122 $claimed =& $this->getQueueData( 'claimed', [] );
124 if ( $this->order
=== 'random' ) {
125 $key = array_rand( $unclaimed );
128 $key = key( $unclaimed );
131 $spec = $unclaimed[$key];
132 unset( $unclaimed[$key] );
135 $job = $this->jobFromSpecInternal( $spec );
138 $job->setMetadata( 'claimId', key( $claimed ) );
144 * @see JobQueue::doAck
146 * @param RunnableJob $job
148 protected function doAck( RunnableJob
$job ) {
149 if ( $this->getAcquiredCount() == 0 ) {
153 $claimed =& $this->getQueueData( 'claimed' );
154 unset( $claimed[$job->getMetadata( 'claimId' )] );
158 * @see JobQueue::doDelete
160 protected function doDelete() {
161 if ( isset( self
::$data[$this->type
][$this->domain
] ) ) {
162 unset( self
::$data[$this->type
][$this->domain
] );
163 if ( !self
::$data[$this->type
] ) {
164 unset( self
::$data[$this->type
] );
170 * @see JobQueue::getAllQueuedJobs
172 * @return Iterator of Job objects.
174 public function getAllQueuedJobs() {
175 $unclaimed = $this->getQueueData( 'unclaimed' );
177 return new ArrayIterator( [] );
180 return new MappedIterator(
182 function ( $value ) {
183 return $this->jobFromSpecInternal( $value );
189 * @see JobQueue::getAllAcquiredJobs
191 * @return Iterator of Job objects.
193 public function getAllAcquiredJobs() {
194 $claimed = $this->getQueueData( 'claimed' );
196 return new ArrayIterator( [] );
199 return new MappedIterator(
201 function ( $value ) {
202 return $this->jobFromSpecInternal( $value );
208 * @param IJobSpecification $spec
209 * @return RunnableJob
211 public function jobFromSpecInternal( IJobSpecification
$spec ) {
212 return $this->factoryJob( $spec->getType(), $spec->getParams() );
216 * @param string $field
221 private function &getQueueData( $field, $init = null ) {
222 if ( !isset( self
::$data[$this->type
][$this->domain
][$field] ) ) {
223 if ( $init !== null ) {
224 self
::$data[$this->type
][$this->domain
][$field] = $init;
230 return self
::$data[$this->type
][$this->domain
][$field];