DBSiteStore.php

Go to the documentation of this file.

<?php
 
use Wikimedia\Rdbms\ILoadBalancer;
 
class DBSiteStore implements SiteStore {
 
    protected $sites = null;
 
    private $dbLoadBalancer;
 
    public function __construct( ILoadBalancer $dbLoadBalancer ) {
        $this->dbLoadBalancer = $dbLoadBalancer;
    }
 
    public function getSites() {
        $this->loadSites();
 
        return $this->sites;
    }
 
    protected function loadSites() {
        $this->sites = new SiteList();
 
        $dbr = $this->dbLoadBalancer->getConnectionRef( DB_REPLICA );
 
        $res = $dbr->select(
            'sites',
            [
                'site_id',
                'site_global_key',
                'site_type',
                'site_group',
                'site_source',
                'site_language',
                'site_protocol',
                'site_domain',
                'site_data',
                'site_forward',
                'site_config',
            ],
            '',
            __METHOD__,
            [ 'ORDER BY' => 'site_global_key' ]
        );
 
        foreach ( $res as $row ) {
            $site = Site::newForType( $row->site_type );
            $site->setGlobalId( $row->site_global_key );
            $site->setInternalId( (int)$row->site_id );
            $site->setForward( (bool)$row->site_forward );
            $site->setGroup( $row->site_group );
            $site->setLanguageCode( $row->site_language === ''
                ? null
                : $row->site_language
            );
            $site->setSource( $row->site_source );
            $site->setExtraData( unserialize( $row->site_data ) );
            $site->setExtraConfig( unserialize( $row->site_config ) );
            $this->sites[] = $site;
        }
 
        // Batch load the local site identifiers.
        $ids = $dbr->select(
            'site_identifiers',
            [
                'si_site',
                'si_type',
                'si_key',
            ],
            [],
            __METHOD__
        );
 
        foreach ( $ids as $id ) {
            if ( $this->sites->hasInternalId( $id->si_site ) ) {
                $site = $this->sites->getSiteByInternalId( $id->si_site );
                $site->addLocalId( $id->si_type, $id->si_key );
                $this->sites->setSite( $site );
            }
        }
    }
 
    public function getSite( $globalId ) {
        if ( $this->sites === null ) {
            $this->sites = $this->getSites();
        }
 
        return $this->sites->hasSite( $globalId ) ? $this->sites->getSite( $globalId ) : null;
    }
 
    public function saveSite( Site $site ) {
        return $this->saveSites( [ $site ] );
    }
 
    public function saveSites( array $sites ) {
        if ( empty( $sites ) ) {
            return true;
        }
 
        $dbw = $this->dbLoadBalancer->getConnectionRef( DB_PRIMARY );
 
        $dbw->startAtomic( __METHOD__ );
 
        $success = true;
 
        $internalIds = [];
        $localIds = [];
 
        foreach ( $sites as $site ) {
            if ( $site->getInternalId() !== null ) {
                $internalIds[] = $site->getInternalId();
            }
 
            $fields = [
                // Site data
                'site_global_key' => $site->getGlobalId(), // TODO: check not null
                'site_type' => $site->getType(),
                'site_group' => $site->getGroup(),
                'site_source' => $site->getSource(),
                'site_language' => $site->getLanguageCode() ?? '',
                'site_protocol' => $site->getProtocol(),
                'site_domain' => strrev( $site->getDomain() ) . '.',
                'site_data' => serialize( $site->getExtraData() ),
 
                // Site config
                'site_forward' => $site->shouldForward() ? 1 : 0,
                'site_config' => serialize( $site->getExtraConfig() ),
            ];
 
            $rowId = $site->getInternalId();
            if ( $rowId !== null ) {
                $success = $dbw->update(
                    'sites', $fields, [ 'site_id' => $rowId ], __METHOD__
                ) && $success;
            } else {
                $success = $dbw->insert( 'sites', $fields, __METHOD__ ) && $success;
                $rowId = $dbw->insertId();
            }
 
            foreach ( $site->getLocalIds() as $idType => $ids ) {
                foreach ( $ids as $id ) {
                    $localIds[] = [ $rowId, $idType, $id ];
                }
            }
        }
 
        if ( $internalIds !== [] ) {
            $dbw->delete(
                'site_identifiers',
                [ 'si_site' => $internalIds ],
                __METHOD__
            );
        }
 
        foreach ( $localIds as $localId ) {
            $dbw->insert(
                'site_identifiers',
                [
                    'si_site' => $localId[0],
                    'si_type' => $localId[1],
                    'si_key' => $localId[2],
                ],
                __METHOD__
            );
        }
 
        $dbw->endAtomic( __METHOD__ );
 
        $this->reset();
 
        return $success;
    }
 
    public function reset() {
        $this->sites = null;
    }
 
    public function clear() {
        $dbw = $this->dbLoadBalancer->getConnectionRef( DB_PRIMARY );
 
        $dbw->startAtomic( __METHOD__ );
        $ok = $dbw->delete( 'sites', '*', __METHOD__ );
        $ok = $dbw->delete( 'site_identifiers', '*', __METHOD__ ) && $ok;
        $dbw->endAtomic( __METHOD__ );
 
        $this->reset();
 
        return $ok;
    }
 
}