BacklinkJobUtils.php

Go to the documentation of this file.

<?php
class BacklinkJobUtils {
    public static function partitionBacklinkJob( Job $job, $bSize, $cSize, $opts = array() ) {
        $class = get_class( $job );
        $title = $job->getTitle();
        $params = $job->getParams();

        if ( isset( $params['pages'] ) || empty( $params['recursive'] ) ) {
            $ranges = array(); // sanity; this is a leaf node
            wfWarn( __METHOD__ . " called on {$job->getType()} leaf job (explosive recursion)." );
        } elseif ( isset( $params['range'] ) ) {
            // This is a range job to trigger the insertion of partitioned/title jobs...
            $ranges = $params['range']['subranges'];
            $realBSize = $params['range']['batchSize'];
        } else {
            // This is a base job to trigger the insertion of partitioned jobs...
            $ranges = $title->getBacklinkCache()->partition( $params['table'], $bSize );
            $realBSize = $bSize;
        }

        $extraParams = isset( $opts['params'] ) ? $opts['params'] : array();

        $jobs = array();
        // Combine the first range (of size $bSize) backlinks into leaf jobs
        if ( isset( $ranges[0] ) ) {
            list( $start, $end ) = $ranges[0];
            $titles = $title->getBacklinkCache()->getLinks( $params['table'], $start, $end );
            foreach ( array_chunk( iterator_to_array( $titles ), $cSize ) as $titleBatch ) {
                $pages = array();
                foreach ( $titleBatch as $tl ) {
                    $pages[$tl->getArticleId()] = array( $tl->getNamespace(), $tl->getDBKey() );
                }
                $jobs[] = new $class(
                    $title, // maintain parent job title
                    array( 'pages' => $pages ) + $extraParams
                );
            }
        }
        // Take all of the remaining ranges and build a partition job from it
        if ( isset( $ranges[1] ) ) {
            $jobs[] = new $class(
                $title, // maintain parent job title
                array(
                    'recursive'     => true,
                    'table'         => $params['table'],
                    'range'         => array(
                        'start'     => $ranges[1][0],
                        'end'       => $ranges[count( $ranges ) - 1][1],
                        'batchSize' => $realBSize,
                        'subranges' => array_slice( $ranges, 1 )
                    ),
                ) + $extraParams
            );
        }

        return $jobs;
    }
}