mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/CategoryTree
synced 2024-11-27 09:43:06 +00:00
9d98cc1ae6
Replace the dynamic property on SpecialTrackingCategories object, this was used to store preloaded Category objects. Dynamic properties are deprecated in php8.2 The preload and storage is now done with the service. Adjust doc to LinkTarget (Id0cc2ca) Bug: T324897 Change-Id: I891ad79bc357d32585ef4d9206d398c5a75222aa
100 lines
2.9 KiB
PHP
100 lines
2.9 KiB
PHP
<?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
|
|
*/
|
|
|
|
namespace MediaWiki\Extension\CategoryTree;
|
|
|
|
use Category;
|
|
use MediaWiki\Linker\LinkTarget;
|
|
use Wikimedia\Rdbms\ILoadBalancer;
|
|
|
|
/**
|
|
* Caches Category::class objects
|
|
*/
|
|
class CategoryCache {
|
|
/** @var (?Category)[] Keys are category database names, values are either a Category object or null */
|
|
private $cache = [];
|
|
|
|
/** @var ILoadBalancer */
|
|
private $loadBalancer;
|
|
|
|
/**
|
|
* @param ILoadBalancer $loadBalancer
|
|
*/
|
|
public function __construct(
|
|
ILoadBalancer $loadBalancer
|
|
) {
|
|
$this->loadBalancer = $loadBalancer;
|
|
}
|
|
|
|
/**
|
|
* Get a preloaded Category object or null when the Category does not exists. Loaded the Category on demand,
|
|
* if not in cache, use self::doQuery when requesting a high number of category
|
|
* @param LinkTarget $categoryTarget
|
|
* @return ?Category
|
|
*/
|
|
public function getCategory( LinkTarget $categoryTarget ): ?Category {
|
|
if ( $categoryTarget->getNamespace() !== NS_CATEGORY ) {
|
|
return null;
|
|
}
|
|
$categoryDbKey = $categoryTarget->getDBkey();
|
|
|
|
if ( !array_key_exists( $categoryDbKey, $this->cache ) ) {
|
|
$this->doQuery( [ $categoryTarget ] );
|
|
}
|
|
|
|
return $this->cache[$categoryDbKey];
|
|
}
|
|
|
|
/**
|
|
* Preloads category counts in this cache
|
|
* @param LinkTarget[] $linkTargets
|
|
*/
|
|
public function doQuery( array $linkTargets ): void {
|
|
$categoryDbKeys = [];
|
|
foreach ( $linkTargets as $linkTarget ) {
|
|
if ( $linkTarget->getNamespace() !== NS_CATEGORY ) {
|
|
continue;
|
|
}
|
|
$categoryDbKey = $linkTarget->getDBkey();
|
|
if ( !array_key_exists( $categoryDbKey, $this->cache ) ) {
|
|
$categoryDbKeys[] = $categoryDbKey;
|
|
// To cache db misses, also avoid duplicates in the db query
|
|
$this->cache[$categoryDbKey] = null;
|
|
}
|
|
}
|
|
if ( $categoryDbKeys === [] ) {
|
|
return;
|
|
}
|
|
|
|
$rows = $this->loadBalancer->getConnection( ILoadBalancer::DB_REPLICA )
|
|
->newSelectQueryBuilder()
|
|
->select( [ 'cat_id', 'cat_title', 'cat_pages', 'cat_subcats', 'cat_files' ] )
|
|
->from( 'category' )
|
|
->where( [ 'cat_title' => $categoryDbKeys ] )
|
|
->caller( __METHOD__ )
|
|
->fetchResultSet();
|
|
|
|
foreach ( $rows as $row ) {
|
|
$this->cache[$row->cat_title] = Category::newFromRow( $row );
|
|
}
|
|
}
|
|
|
|
}
|