src/applications/search/engineextension/PhabricatorFulltextIndexEngineExtension.php

   1 <?php
   2
   3 final class PhabricatorFulltextIndexEngineExtension
   4   extends PhabricatorIndexEngineExtension {
   5
   6   const EXTENSIONKEY = 'fulltext';
   7
   8   private $configurationVersion;
   9
  10   public function getExtensionName() {
  11     return pht('Fulltext Engine');
  12   }
  13
  14   public function getIndexVersion($object) {
  15     $version = array();
  16
  17     // When "cluster.search" is reconfigured, new indexes which don't have any
  18     // data yet may have been added. We err on the side of caution and assume
  19     // that every document may need to be reindexed.
  20     $version[] = $this->getConfigurationVersion();
  21
  22     if ($object instanceof PhabricatorApplicationTransactionInterface) {
  23       // If this is a normal object with transactions, we only need to
  24       // reindex it if there are new transactions (or comment edits).
  25       $version[] = $this->getTransactionVersion($object);
  26       $version[] = $this->getCommentVersion($object);
  27     }
  28
  29     if (!$version) {
  30       return null;
  31     }
  32
  33     return implode(':', $version);
  34   }
  35
  36   public function shouldIndexObject($object) {
  37     return ($object instanceof PhabricatorFulltextInterface);
  38   }
  39
  40   public function indexObject(
  41     PhabricatorIndexEngine $engine,
  42     $object) {
  43
  44     $engine = $object->newFulltextEngine();
  45     if (!$engine) {
  46       return;
  47     }
  48
  49     $engine->setObject($object);
  50
  51     $engine->buildFulltextIndexes();
  52   }
  53
  54   private function getTransactionVersion($object) {
  55     $xaction = $object->getApplicationTransactionTemplate();
  56
  57     $xaction_row = queryfx_one(
  58       $xaction->establishConnection('r'),
  59       'SELECT id FROM %T WHERE objectPHID = %s
  60         ORDER BY id DESC LIMIT 1',
  61       $xaction->getTableName(),
  62       $object->getPHID());
  63     if (!$xaction_row) {
  64       return 'none';
  65     }
  66
  67     return $xaction_row['id'];
  68   }
  69
  70   private function getCommentVersion($object) {
  71     $xaction = $object->getApplicationTransactionTemplate();
  72
  73     $comment = $xaction->getApplicationTransactionCommentObject();
  74     if (!$comment) {
  75       return 'none';
  76     }
  77
  78     $comment_row = queryfx_one(
  79       $comment->establishConnection('r'),
  80       'SELECT c.id FROM %T x JOIN %T c
  81         ON x.phid = c.transactionPHID
  82         WHERE x.objectPHID = %s
  83         ORDER BY c.id DESC LIMIT 1',
  84       $xaction->getTableName(),
  85       $comment->getTableName(),
  86       $object->getPHID());
  87     if (!$comment_row) {
  88       return 'none';
  89     }
  90
  91     return $comment_row['id'];
  92   }
  93
  94   private function getConfigurationVersion() {
  95     if ($this->configurationVersion === null) {
  96       $this->configurationVersion = $this->newConfigurationVersion();
  97     }
  98     return $this->configurationVersion;
  99   }
 100
 101   private function newConfigurationVersion() {
 102     $raw = array(
 103       'services' => PhabricatorEnv::getEnvConfig('cluster.search'),
 104     );
 105
 106     $json = phutil_json_encode($raw);
 107
 108     return PhabricatorHash::digestForIndex($json);
 109   }
 110
 111
 112 }