DBSerialProvider.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\User\TempUser;
 
use Wikimedia\Rdbms\IDatabase;
 
abstract class DBSerialProvider implements SerialProvider {
    private $numShards;
 
    public function __construct( $config ) {
        $this->numShards = $config['numShards'] ?? 1;
    }
 
    public function acquireIndex(): int {
        if ( $this->numShards ) {
            $shard = mt_rand( 0, $this->numShards - 1 );
        } else {
            $shard = 0;
        }
 
        $dbw = $this->getDB();
        $table = $this->getTableName();
        $dbw->startAtomic( __METHOD__ );
        $dbw->upsert(
            $table,
            [
                'uas_shard' => $shard,
                'uas_value' => 1,
            ],
            [ [ 'uas_shard' ] ],
            [ 'uas_value=uas_value+1' ],
            __METHOD__
        );
        $value = $dbw->newSelectQueryBuilder()
            ->select( 'uas_value' )
            ->from( $table )
            ->where( [ 'uas_shard' => $shard ] )
            ->caller( __METHOD__ )
            ->fetchField();
        $dbw->endAtomic( __METHOD__ );
        return $value * $this->numShards + $shard;
    }
 
    abstract protected function getDB();
 
    abstract protected function getTableName();
}