diff options
Diffstat (limited to 'www/wiki/includes/cache/localisation/LCStoreDB.php')
-rw-r--r-- | www/wiki/includes/cache/localisation/LCStoreDB.php | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/www/wiki/includes/cache/localisation/LCStoreDB.php b/www/wiki/includes/cache/localisation/LCStoreDB.php new file mode 100644 index 00000000..c57145c0 --- /dev/null +++ b/www/wiki/includes/cache/localisation/LCStoreDB.php @@ -0,0 +1,117 @@ +<?php +/** + * 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 + * + * @file + */ + +use Wikimedia\Rdbms\IDatabase; +use Wikimedia\Rdbms\DBQueryError; + +/** + * LCStore implementation which uses the standard DB functions to store data. + * This will work on any MediaWiki installation. + */ +class LCStoreDB implements LCStore { + + /** @var string */ + private $currentLang; + /** @var bool */ + private $writesDone = false; + /** @var IDatabase */ + private $dbw; + /** @var array */ + private $batch = []; + /** @var bool */ + private $readOnly = false; + + public function get( $code, $key ) { + if ( $this->writesDone && $this->dbw ) { + $db = $this->dbw; // see the changes in finishWrite() + } else { + $db = wfGetDB( DB_REPLICA ); + } + + $value = $db->selectField( + 'l10n_cache', + 'lc_value', + [ 'lc_lang' => $code, 'lc_key' => $key ], + __METHOD__ + ); + + return ( $value !== false ) ? unserialize( $db->decodeBlob( $value ) ) : null; + } + + public function startWrite( $code ) { + if ( $this->readOnly ) { + return; + } elseif ( !$code ) { + throw new MWException( __METHOD__ . ": Invalid language \"$code\"" ); + } + + $this->dbw = wfGetDB( DB_MASTER ); + $this->readOnly = $this->dbw->isReadOnly(); + + $this->currentLang = $code; + $this->batch = []; + } + + public function finishWrite() { + if ( $this->readOnly ) { + return; + } elseif ( is_null( $this->currentLang ) ) { + throw new MWException( __CLASS__ . ': must call startWrite() before finishWrite()' ); + } + + $this->dbw->startAtomic( __METHOD__ ); + try { + $this->dbw->delete( + 'l10n_cache', + [ 'lc_lang' => $this->currentLang ], + __METHOD__ + ); + foreach ( array_chunk( $this->batch, 500 ) as $rows ) { + $this->dbw->insert( 'l10n_cache', $rows, __METHOD__ ); + } + $this->writesDone = true; + } catch ( DBQueryError $e ) { + if ( $this->dbw->wasReadOnlyError() ) { + $this->readOnly = true; // just avoid site down time + } else { + throw $e; + } + } + $this->dbw->endAtomic( __METHOD__ ); + + $this->currentLang = null; + $this->batch = []; + } + + public function set( $key, $value ) { + if ( $this->readOnly ) { + return; + } elseif ( is_null( $this->currentLang ) ) { + throw new MWException( __CLASS__ . ': must call startWrite() before set()' ); + } + + $this->batch[] = [ + 'lc_lang' => $this->currentLang, + 'lc_key' => $key, + 'lc_value' => $this->dbw->encodeBlob( serialize( $value ) ) + ]; + } + +} |