diff options
author | Pierre Schmitz <pierre@archlinux.de> | 2013-08-12 09:28:15 +0200 |
---|---|---|
committer | Pierre Schmitz <pierre@archlinux.de> | 2013-08-12 09:28:15 +0200 |
commit | 08aa4418c30cfc18ccc69a0f0f9cb9e17be6c196 (patch) | |
tree | 577a29fb579188d16003a209ce2a2e9c5b0aa2bd /includes/site/SiteSQLStore.php | |
parent | cacc939b34e315b85e2d72997811eb6677996cc1 (diff) |
Update to MediaWiki 1.21.1
Diffstat (limited to 'includes/site/SiteSQLStore.php')
-rw-r--r-- | includes/site/SiteSQLStore.php | 491 |
1 files changed, 491 insertions, 0 deletions
diff --git a/includes/site/SiteSQLStore.php b/includes/site/SiteSQLStore.php new file mode 100644 index 00000000..41238055 --- /dev/null +++ b/includes/site/SiteSQLStore.php @@ -0,0 +1,491 @@ +<?php + +/** + * Represents the site configuration of a wiki. + * Holds a list of sites (ie SiteList) and takes care + * of retrieving and caching site information when appropriate. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * http://www.gnu.org/copyleft/gpl.html + * + * @since 1.21 + * + * @file + * @ingroup Site + * + * @license GNU GPL v2+ + * @author Jeroen De Dauw < jeroendedauw@gmail.com > + */ +class SiteSQLStore implements SiteStore { + + /** + * @since 1.21 + * + * @var SiteList|null + */ + protected $sites = null; + + /** + * @var ORMTable + */ + protected $sitesTable; + + /** + * @var string|null + */ + private $cacheKey = null; + + /** + * @var int + */ + private $cacheTimeout = 3600; + + /** + * @since 1.21 + * + * @param ORMTable|null $sitesTable + * + * @return SiteStore + */ + public static function newInstance( ORMTable $sitesTable = null ) { + return new static( $sitesTable ); + } + + /** + * Constructor. + * + * @since 1.21 + * + * @param ORMTable|null $sitesTable + */ + protected function __construct( ORMTable $sitesTable = null ) { + if ( $sitesTable === null ) { + $sitesTable = $this->newSitesTable(); + } + + $this->sitesTable = $sitesTable; + } + + /** + * Constructs a cache key to use for caching the list of sites. + * + * This includes the concrete class name of the site list as well as a version identifier + * for the list's serialization, to avoid problems when unserializing site lists serialized + * by an older version, e.g. when reading from a cache. + * + * The cache key also includes information about where the sites were loaded from, e.g. + * the name of a database table. + * + * @see SiteList::getSerialVersionId + * + * @return String The cache key. + */ + protected function getCacheKey() { + wfProfileIn( __METHOD__ ); + + if ( $this->cacheKey === null ) { + $type = 'SiteList#' . SiteList::getSerialVersionId(); + $source = $this->sitesTable->getName(); + + if ( $this->sitesTable->getTargetWiki() !== false ) { + $source = $this->sitesTable->getTargetWiki() . '.' . $source; + } + + $this->cacheKey = wfMemcKey( "$source/$type" ); + } + + wfProfileOut( __METHOD__ ); + return $this->cacheKey; + } + + /** + * @see SiteStore::getSites + * + * @since 1.21 + * + * @param string $source either 'cache' or 'recache' + * + * @return SiteList + */ + public function getSites( $source = 'cache' ) { + wfProfileIn( __METHOD__ ); + + if ( $source === 'cache' ) { + if ( $this->sites === null ) { + $cache = wfGetMainCache(); + $sites = $cache->get( $this->getCacheKey() ); + + if ( is_object( $sites ) ) { + $this->sites = $sites; + } else { + $this->loadSites(); + } + } + } + else { + $this->loadSites(); + } + + wfProfileOut( __METHOD__ ); + return $this->sites; + } + + /** + * Returns a new Site object constructed from the provided ORMRow. + * + * @since 1.21 + * + * @param ORMRow $siteRow + * + * @return Site + */ + protected function siteFromRow( ORMRow $siteRow ) { + wfProfileIn( __METHOD__ ); + + $site = Site::newForType( $siteRow->getField( 'type', Site::TYPE_UNKNOWN ) ); + + $site->setGlobalId( $siteRow->getField( 'global_key' ) ); + + $site->setInternalId( $siteRow->getField( 'id' ) ); + + if ( $siteRow->hasField( 'forward' ) ) { + $site->setForward( $siteRow->getField( 'forward' ) ); + } + + if ( $siteRow->hasField( 'group' ) ) { + $site->setGroup( $siteRow->getField( 'group' ) ); + } + + if ( $siteRow->hasField( 'language' ) ) { + $site->setLanguageCode( $siteRow->getField( 'language' ) === '' ? null : $siteRow->getField( 'language' ) ); + } + + if ( $siteRow->hasField( 'source' ) ) { + $site->setSource( $siteRow->getField( 'source' ) ); + } + + if ( $siteRow->hasField( 'data' ) ) { + $site->setExtraData( $siteRow->getField( 'data' ) ); + } + + if ( $siteRow->hasField( 'config' ) ) { + $site->setExtraConfig( $siteRow->getField( 'config' ) ); + } + + wfProfileOut( __METHOD__ ); + return $site; + } + + /** + * Fetches the site from the database and loads them into the sites field. + * + * @since 1.21 + */ + protected function loadSites() { + wfProfileIn( __METHOD__ ); + + $this->sites = new SiteList(); + + foreach ( $this->sitesTable->select() as $siteRow ) { + $this->sites[] = $this->siteFromRow( $siteRow ); + } + + // Batch load the local site identifiers. + $ids = wfGetDB( $this->sitesTable->getReadDb() )->select( + 'site_identifiers', + array( + 'si_site', + 'si_type', + 'si_key', + ), + array(), + __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 ); + } + } + + $cache = wfGetMainCache(); + $cache->set( $this->getCacheKey(), $this->sites, $this->cacheTimeout ); + + wfProfileOut( __METHOD__ ); + } + + /** + * @see SiteStore::getSite + * + * @since 1.21 + * + * @param string $globalId + * @param string $source + * + * @return Site|null + */ + public function getSite( $globalId, $source = 'cache' ) { + wfProfileIn( __METHOD__ ); + + $sites = $this->getSites( $source ); + + wfProfileOut( __METHOD__ ); + return $sites->hasSite( $globalId ) ? $sites->getSite( $globalId ) : null; + } + + /** + * @see SiteStore::saveSite + * + * @since 1.21 + * + * @param Site $site + * + * @return boolean Success indicator + */ + public function saveSite( Site $site ) { + return $this->saveSites( array( $site ) ); + } + + /** + * @see SiteStore::saveSites + * + * @since 1.21 + * + * @param Site[] $sites + * + * @return boolean Success indicator + */ + public function saveSites( array $sites ) { + wfProfileIn( __METHOD__ ); + + if ( empty( $sites ) ) { + wfProfileOut( __METHOD__ ); + return true; + } + + $dbw = $this->sitesTable->getWriteDbConnection(); + + $trx = $dbw->trxLevel(); + + if ( $trx == 0 ) { + $dbw->begin( __METHOD__ ); + } + + $success = true; + + $internalIds = array(); + $localIds = array(); + + foreach ( $sites as $site ) { + $fields = array( + // Site data + 'global_key' => $site->getGlobalId(), // TODO: check not null + 'type' => $site->getType(), + 'group' => $site->getGroup(), + 'source' => $site->getSource(), + 'language' => $site->getLanguageCode() === null ? '' : $site->getLanguageCode(), + 'protocol' => $site->getProtocol(), + 'domain' => strrev( $site->getDomain() ) . '.', + 'data' => $site->getExtraData(), + + // Site config + 'forward' => $site->shouldForward(), + 'config' => $site->getExtraConfig(), + ); + + if ( $site->getInternalId() !== null ) { + $fields['id'] = $site->getInternalId(); + $internalIds[] = $site->getInternalId(); + } + + $siteRow = new ORMRow( $this->sitesTable, $fields ); + $success = $siteRow->save( __METHOD__ ) && $success; + + foreach ( $site->getLocalIds() as $idType => $ids ) { + foreach ( $ids as $id ) { + $localIds[] = array( $siteRow->getId(), $idType, $id ); + } + } + } + + if ( $internalIds !== array() ) { + $dbw->delete( + 'site_identifiers', + array( 'si_site' => $internalIds ), + __METHOD__ + ); + } + + foreach ( $localIds as $localId ) { + $dbw->insert( + 'site_identifiers', + array( + 'si_site' => $localId[0], + 'si_type' => $localId[1], + 'si_key' => $localId[2], + ), + __METHOD__ + ); + } + + if ( $trx == 0 ) { + $dbw->commit( __METHOD__ ); + } + + // purge cache + $this->reset(); + + wfProfileOut( __METHOD__ ); + return $success; + } + + /** + * Purges the internal and external cache of the site list, forcing the list + * of sites to be re-read from the database. + * + * @since 1.21 + */ + public function reset() { + wfProfileIn( __METHOD__ ); + // purge cache + $cache = wfGetMainCache(); + $cache->delete( $this->getCacheKey() ); + $this->sites = null; + + wfProfileOut( __METHOD__ ); + } + + /** + * Clears the list of sites stored in the database. + * + * @see SiteStore::clear() + * + * @return bool success + */ + public function clear() { + wfProfileIn( __METHOD__ ); + $dbw = $this->sitesTable->getWriteDbConnection(); + + $trx = $dbw->trxLevel(); + + if ( $trx == 0 ) { + $dbw->begin( __METHOD__ ); + } + + $ok = $dbw->delete( 'sites', '*', __METHOD__ ); + $ok = $dbw->delete( 'site_identifiers', '*', __METHOD__ ) && $ok; + + if ( $trx == 0 ) { + $dbw->commit( __METHOD__ ); + } + + $this->reset(); + + wfProfileOut( __METHOD__ ); + return $ok; + } + + /** + * @since 1.21 + * + * @return ORMTable + */ + protected function newSitesTable() { + return new ORMTable( + 'sites', + array( + 'id' => 'id', + + // Site data + 'global_key' => 'str', + 'type' => 'str', + 'group' => 'str', + 'source' => 'str', + 'language' => 'str', + 'protocol' => 'str', + 'domain' => 'str', + 'data' => 'array', + + // Site config + 'forward' => 'bool', + 'config' => 'array', + ), + array( + 'type' => Site::TYPE_UNKNOWN, + 'group' => Site::GROUP_NONE, + 'source' => Site::SOURCE_LOCAL, + 'data' => array(), + + 'forward' => false, + 'config' => array(), + 'language' => '', + ), + 'ORMRow', + 'site_' + ); + } + +} + +/** + * @deprecated + */ +class Sites extends SiteSQLStore { + + /** + * Factory for creating new site objects. + * + * @since 1.21 + * @deprecated + * + * @param string|boolean false $globalId + * + * @return Site + */ + public static function newSite( $globalId = false ) { + $site = new Site(); + + if ( $globalId !== false ) { + $site->setGlobalId( $globalId ); + } + + return $site; + } + + /** + * @deprecated + * @return SiteStore + */ + public static function singleton() { + static $singleton; + + if ( $singleton === null ) { + $singleton = new static(); + } + + return $singleton; + } + + /** + * @deprecated + * @return SiteList + */ + public function getSiteGroup( $group ) { + return $this->getSites()->getGroup( $group ); + } + +} |