wikimedia/mediawiki-core

<?php

namespace MediaWiki\User\TempUser;

use Wikimedia\Rdbms\ILoadBalancer;

/**
 * A serial provider which allocates IDs from the local database, or from a
 * shared database if $wgSharedDB is used. It is "local" in the sense that it
 * uses the same DB connection as the local wiki.
 *
 * @since 1.39
 */
class LocalSerialProvider extends DBSerialProvider {
    private ILoadBalancer $lb;

    /**
     * @param array $config
     *   - numShards (int, default 1): A small integer. This can be set to a
     *     value greater than 1 to avoid acquiring a global lock when
     *     allocating IDs, at the expense of making the IDs be non-monotonic.
     * @param ILoadBalancer $lb
     */
    public function __construct( $config, ILoadBalancer $lb ) {
        parent::__construct( $config );
        $this->lb = $lb;
    }

    protected function getDB() {
        return $this->lb->getConnection(
            DB_PRIMARY,
            [],
            false,
            // So that startAtomic() will start a commit, reducing lock time.
            // Without this flag, the transaction will be open until the start
            // of request shutdown. This could be omitted to reduce the
            // connection overhead, with numShards tuned upwards to compensate.
            ILoadBalancer::CONN_TRX_AUTOCOMMIT
        );
    }

    protected function getTableName() {
        return 'user_autocreate_serial';
    }
}