src/applications/search/ngrams/PhabricatorSearchNgrams.php

   1 <?php
   2
   3 abstract class PhabricatorSearchNgrams
   4   extends PhabricatorSearchDAO {
   5
   6   protected $objectID;
   7   protected $ngram;
   8
   9   private $value;
  10   private $ngramEngine;
  11
  12   abstract public function getNgramKey();
  13   abstract public function getColumnName();
  14
  15   final public function setValue($value) {
  16     $this->value = $value;
  17     return $this;
  18   }
  19
  20   final public function getValue() {
  21     return $this->value;
  22   }
  23
  24   protected function getConfiguration() {
  25     return array(
  26       self::CONFIG_TIMESTAMPS => false,
  27       self::CONFIG_COLUMN_SCHEMA => array(
  28         'objectID' => 'uint32',
  29         'ngram' => 'char3',
  30       ),
  31       self::CONFIG_KEY_SCHEMA => array(
  32         'key_ngram' => array(
  33           'columns' => array('ngram', 'objectID'),
  34         ),
  35         'key_object' => array(
  36           'columns' => array('objectID'),
  37         ),
  38       ),
  39     ) + parent::getConfiguration();
  40   }
  41
  42   public function getTableName() {
  43     $application = $this->getApplicationName();
  44     $key = $this->getNgramKey();
  45     return "{$application}_{$key}_ngrams";
  46   }
  47
  48   final public function writeNgram($object_id) {
  49     $ngram_engine = $this->getNgramEngine();
  50     $ngrams = $ngram_engine->getTermNgramsFromString($this->getValue());
  51
  52     $conn_w = $this->establishConnection('w');
  53
  54     $sql = array();
  55     foreach ($ngrams as $ngram) {
  56       $sql[] = qsprintf(
  57         $conn_w,
  58         '(%d, %s)',
  59         $object_id,
  60         $ngram);
  61     }
  62
  63     queryfx(
  64       $conn_w,
  65       'DELETE FROM %T WHERE objectID = %d',
  66       $this->getTableName(),
  67       $object_id);
  68
  69     if ($sql) {
  70       queryfx(
  71         $conn_w,
  72         'INSERT INTO %T (objectID, ngram) VALUES %LQ',
  73         $this->getTableName(),
  74         $sql);
  75     }
  76
  77     return $this;
  78   }
  79
  80   private function getNgramEngine() {
  81     if (!$this->ngramEngine) {
  82       $this->ngramEngine = new PhabricatorSearchNgramEngine();
  83     }
  84
  85     return $this->ngramEngine;
  86   }
  87
  88 }