-
Notifications
You must be signed in to change notification settings - Fork 45
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
da8aa21
commit c6c0bc4
Showing
2 changed files
with
183 additions
and
99 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,148 @@ | ||
| <?php | ||
|
|
||
| class CountQuery { | ||
| public const MODE_REDIRECT = 'redirect'; | ||
| public const MODE_LINK = 'link'; | ||
| public const MODE_TRANSCLUSION = 'transclusion'; | ||
|
|
||
| public const SINGLE_NS = 1; | ||
| public const NO_FROM_NS = 2; | ||
| public const NO_LINK_TARGET = 4; | ||
|
|
||
| private $fromNamespaces; | ||
| private $db; | ||
| private $title; | ||
|
|
||
| public function __construct($fromNamespaces, $db, $title) { | ||
| $this->fromNamespaces = $fromNamespaces; | ||
| $this->db = $db; | ||
| $this->title = $title; | ||
| } | ||
|
|
||
| private function createCond( | ||
| $prefix, | ||
| $titleSQL, | ||
| $namespaceSQL, | ||
| $flags, | ||
| $joins = [], | ||
| $wheres = [] | ||
| ) { | ||
| $hasFromNS = ~$flags & CountQuery::NO_FROM_NS; | ||
| $usesLinkTarget = ~$flags & CountQuery::NO_LINK_TARGET; | ||
| $hasNS = $usesLinkTarget || (~$flags & CountQuery::SINGLE_NS); | ||
|
|
||
| if ($this->fromNamespaces !== '' && $hasFromNS) { | ||
| array_push($wheres, "{$prefix}_from_namespace IN ({$this->fromNamespaces})"); | ||
| } | ||
|
|
||
| if ($this->fromNamespaces !== '' && !$hasFromNS) { | ||
| array_push( | ||
| $joins, | ||
| <<<SQL | ||
| JOIN page AS source ON | ||
| source.page_id = {$prefix}_from | ||
| AND source.page_namespace IN ({$this->fromNamespaces}) | ||
| SQL | ||
| ); | ||
| } | ||
|
|
||
| $linkInfoPrefix = $usesLinkTarget ? 'lt' : $prefix; | ||
| $titleColumn = $linkInfoPrefix . '_' . ($hasNS ? 'title' : 'to'); | ||
|
|
||
| array_push($wheres, "$titleColumn = $titleSQL"); | ||
|
|
||
| if ($hasNS) { | ||
| array_push($wheres, "{$linkInfoPrefix}_namespace = $namespaceSQL"); | ||
| } | ||
|
|
||
| if ($usesLinkTarget) { | ||
| array_push($joins, "JOIN linktarget ON {$prefix}_target_id = lt_id"); | ||
| } | ||
|
|
||
| return implode(' ', $joins) . " WHERE " . implode(' AND ', $wheres); | ||
| } | ||
|
|
||
| private function createDirectCond($prefix, $flags) { | ||
| return $this->createCond( | ||
| $prefix, | ||
| $this->db->quote($this->title->getDBKey()), | ||
| $this->title->getNamespaceId(), | ||
| $flags | ||
| ); | ||
| } | ||
|
|
||
| private function createIndirectCond($table, $prefix, $flags) { | ||
| $joins = [ | ||
| 'JOIN page AS target ON target.page_id = rd_from', | ||
| "JOIN $table" | ||
| ]; | ||
|
|
||
| $wheres = [ | ||
| "rd_title = {$this->db->quote($this->title->getDBKey())}", | ||
| "rd_namespace = {$this->title->getNamespaceId()}", | ||
| "(rd_interwiki IS NULL OR rd_interwiki = {$this->db->quote('')})" | ||
| ]; | ||
|
|
||
| return $this->createCond( | ||
| $prefix, | ||
| 'target.page_title', | ||
| 'target.page_namespace', | ||
| $flags, | ||
| $joins, | ||
| $wheres | ||
| ); | ||
| } | ||
|
|
||
| private function createQuery($table, $prefix, $mode, $flags) { | ||
| return match ($mode) { | ||
| self::MODE_REDIRECT => <<<SQL | ||
| SELECT COUNT(rd_from) FROM $table | ||
| {$this->createDirectCond($prefix, $flags)} | ||
| AND ({$prefix}_interwiki is NULL or {$prefix}_interwiki = {$this->db->quote('')}) | ||
| SQL, | ||
| // Transclusions of a redirect that follow the redirect are also added as a transclusion of the redirect target. | ||
| // There is no way to differentiate from a page with a indirect link and a page with a indirect and a direct link | ||
| // in this case, only the indirect link is recorded. Pages can also transclude a page with a redirect without | ||
| // following the redirect, so a valid indirect link must have an associated direct link. | ||
| self::MODE_TRANSCLUSION => <<<SQL | ||
| SELECT | ||
| COUNT({$prefix}_from), | ||
| COUNT({$prefix}_from) - COUNT(indirect_link), | ||
| COUNT(indirect_link) | ||
| FROM $table | ||
| LEFT JOIN ( | ||
| SELECT DISTINCT {$prefix}_from AS indirect_link | ||
| FROM redirect | ||
| {$this->createIndirectCond($table, $prefix, $flags)} | ||
| ) AS temp ON {$prefix}_from = indirect_link | ||
| {$this->createDirectCond($prefix, $flags)} | ||
| SQL, | ||
| self::MODE_LINK => <<<SQL | ||
| SELECT | ||
| COUNT(DISTINCT COALESCE(direct_link, indirect_link)), | ||
| COUNT(direct_link), | ||
| COUNT(indirect_link) | ||
| FROM ( | ||
| SELECT {$prefix}_from AS direct_link, NULL AS indirect_link | ||
| FROM $table | ||
| {$this->createDirectCond($prefix, $flags)} | ||
| UNION ALL | ||
| SELECT DISTINCT NULL AS direct_link, {$prefix}_from AS indirect_link | ||
| FROM redirect | ||
| {$this->createIndirectCond($table, $prefix, $flags)} | ||
| ) AS temp | ||
| SQL | ||
| }; | ||
| } | ||
|
|
||
| public function runQuery($table, $prefix, $mode, $flags = 0) { | ||
| $query = $this->createQuery($table, $prefix, $mode, $flags); | ||
| $res = $this->db->query($query)->fetch(); | ||
|
|
||
| return $mode == self::MODE_REDIRECT ? (int) $res[0] : [ | ||
| 'all' => (int) $res[0], | ||
| 'direct' => (int) $res[1], | ||
| 'indirect' => (int) $res[2] | ||
| ]; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters