CDBMessageIndex.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageLoading;
 
use Cdb\Reader;
use Cdb\Writer;
use MediaWiki\Extension\Translate\Utilities\Utilities;
 
class CDBMessageIndex extends MessageIndexStore {
    private ?array $index = null;
    private ?Reader $reader = null;
    private const FILENAME = 'translate_messageindex.cdb';
 
    public function retrieve( bool $readLatest = false ): array {
        $reader = $this->getReader();
        if ( $this->index !== null ) {
            return $this->index;
        }
 
        $this->index = [];
        foreach ( $this->getKeys() as $key ) {
            $this->index[$key] = $this->unserialize( $reader->get( $key ) );
        }
 
        return $this->index;
    }
 
    public function getKeys(): array {
        $reader = $this->getReader();
        $keys = [];
        $key = $reader->firstkey();
        while ( $key !== false ) {
            $keys[] = $key;
            $key = $reader->nextkey();
        }
 
        return $keys;
    }
 
    public function get( string $key ) {
        $reader = $this->getReader();
        // We might have the full cache loaded
        if ( $this->index !== null ) {
            return $this->index[$key] ?? null;
        }
 
        $value = $reader->get( $key );
        return is_string( $value ) ? $this->unserialize( $value ) : null;
    }
 
    public function store( array $array, array $diff ): void {
        $this->reader = null;
 
        $file = Utilities::cacheFile( self::FILENAME );
        $cache = Writer::open( $file );
 
        foreach ( $array as $key => $value ) {
            $value = $this->serialize( $value );
            $cache->set( $key, $value );
        }
 
        $cache->close();
 
        $this->index = $array;
    }
 
    private function getReader() {
        if ( $this->reader ) {
            return $this->reader;
        }
 
        $file = Utilities::cacheFile( self::FILENAME );
        if ( !file_exists( $file ) ) {
            // Create an empty index
            $this->store( [], [] );
        }
 
        $this->reader = Reader::open( $file );
        return $this->reader;
    }
}