DatabaseMessageIndex.php

<?php
declare ( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageLoading;
 
use MediaWiki\MediaWikiServices;
use Wikimedia\Rdbms\ILoadBalancer;
 
class DatabaseMessageIndex extends MessageIndexStore {
    private ?array $index = null;
    private ILoadBalancer $loadBalancer;
 
    public function __construct() {
        $this->loadBalancer = MediaWikiServices::getInstance()->getDBLoadBalancer();
    }
 
    public function retrieve( bool $readLatest = false ): array {
        if ( $this->index !== null && !$readLatest ) {
            return $this->index;
        }
 
        $dbr = $this->loadBalancer->getConnection( $readLatest ? DB_PRIMARY : DB_REPLICA );
        $res = $dbr->newSelectQueryBuilder()
            ->select( '*' )
            ->from( 'translate_messageindex' )
            ->caller( __METHOD__ )
            ->fetchResultSet();
        $this->index = [];
        foreach ( $res as $row ) {
            $this->index[$row->tmi_key] = $this->unserialize( $row->tmi_value );
        }
 
        return $this->index;
    }
 
    public function get( string $key ) {
        $dbr = $this->loadBalancer->getConnection( DB_REPLICA );
        $value = $dbr->newSelectQueryBuilder()
            ->select( 'tmi_value' )
            ->from( 'translate_messageindex' )
            ->where( [ 'tmi_key' => $key ] )
            ->caller( __METHOD__ )
            ->fetchField();
 
        return is_string( $value ) ? $this->unserialize( $value ) : null;
    }
 
    public function store( array $array, array $diff ): void {
        $updates = [];
 
        foreach ( [ $diff['add'], $diff['mod'] ] as $changes ) {
            foreach ( $changes as $key => $data ) {
                [ , $new ] = $data;
                $updates[] = [
                    'tmi_key' => $key,
                    'tmi_value' => $this->serialize( $new ),
                ];
            }
        }
 
        $index = [ 'tmi_key' ];
        $deletions = array_keys( $diff['del'] );
 
        $dbw = $this->loadBalancer->getConnection( DB_PRIMARY );
        $dbw->startAtomic( __METHOD__ );
 
        if ( $updates !== [] ) {
            $dbw->replace( 'translate_messageindex', [ $index ], $updates, __METHOD__ );
        }
 
        if ( $deletions !== [] ) {
            $dbw->delete( 'translate_messageindex', [ 'tmi_key' => $deletions ], __METHOD__ );
        }
 
        $dbw->endAtomic( __METHOD__ );
 
        $this->index = $array;
    }
}