src/applications/search/garbagecollector/PhabricatorSearchFerretNgramGarbageCollector.php

   1 <?php
   2
   3 final class PhabricatorSearchFerretNgramGarbageCollector
   4   extends PhabricatorGarbageCollector {
   5
   6   const COLLECTORCONST = 'search.ferret.ngram';
   7
   8   public function getCollectorName() {
   9     return pht('Ferret Engine Ngrams');
  10   }
  11
  12   public function hasAutomaticPolicy() {
  13     return true;
  14   }
  15
  16   protected function collectGarbage() {
  17     $all_objects = id(new PhutilClassMapQuery())
  18       ->setAncestorClass('PhabricatorFerretInterface')
  19       ->execute();
  20
  21     $did_collect = false;
  22     foreach ($all_objects as $object) {
  23       $engine = $object->newFerretEngine();
  24       $conn = $object->establishConnection('w');
  25
  26       $ngram_row = queryfx_one(
  27         $conn,
  28         'SELECT ngram FROM %T WHERE needsCollection = 1 LIMIT 1',
  29         $engine->getCommonNgramsTableName());
  30       if (!$ngram_row) {
  31         continue;
  32       }
  33
  34       $ngram = $ngram_row['ngram'];
  35
  36       queryfx(
  37         $conn,
  38         'DELETE FROM %T WHERE ngram = %s',
  39         $engine->getNgramsTableName(),
  40         $ngram);
  41
  42       queryfx(
  43         $conn,
  44         'UPDATE %T SET needsCollection = 0 WHERE ngram = %s',
  45         $engine->getCommonNgramsTableName(),
  46         $ngram);
  47
  48       $did_collect = true;
  49       break;
  50     }
  51
  52     return $did_collect;
  53   }
  54
  55 }