Changeset 1316 for cpc/trunk/project/lib/model/solr/SolrListener.php

Timestamp:

Sep 4, 2010, 8:54:34 PM (10 years ago)

Author:

teymour

Message:

Un vrai fichier de commands (protégé)

File:

• cpc/trunk/project/lib/model/solr/SolrListener.php (modified) (4 diffs)

cpc/trunk/project/lib/model/solr/SolrListener.php

@@ -1 +1 @@
 <?php
-
-
-class SolrConnector extends sfLogger
-{
-  private $solr = NULL;
-  private $_options = NULL;
-
-  public static function getFileCommands() {
-    return sfConfig::get('sf_log_dir').'/solr/commands.log';
-  }
-
-  protected function doLog($message, $priority)
-  {
-    error_log(sprintf('%s (%s)', $message, sfLogger::getPriorityName($priority)));
-  }
-
-  public function __construct( $listener_options = NULL)
-  {
-    $host = sfConfig::get('app_solr_host', 'localhost');
-    $port = sfConfig::get('app_solr_port', '8983');
-    $url = sfConfig::get('app_solr_url', '/solr');
-    $this->solr = new Apache_Solr_Service($host, $port, $url);
-    
-    if(!$this->solr->ping()) {
-      throw new Exception('Search is not available right now.');
-//      $this->doLog('SolrConnector: Arg, Search is not available right now', sfLogger::ERR);
-    }
-    
-    $this->_options = $listener_options;
-
-    return $this->solr;
-  }
-  
-
-  public function updateFromCommands() {
-    echo "UpdateFromCommands\n";
-    echo file_exists(self::getFileCommands())."\n";
-    if (!file_exists(self::getFileCommands().'.lock') && file_exists(self::getFileCommands())) {
-      echo "rename \n";
-      rename(self::getFileCommands(), self::getFileCommands().'.lock');
-    }
-    if (!file_exists(self::getFileCommands().'.lock')) {
-      echo "No lock\n";
-      return ;
-    }
-    foreach(file(self::getFileCommands().'.lock') as $line) {
-      if (preg_match('/\] (UPDATE|REMOVE): (.+)/', $line, $matches)) {
-        echo $matches[1]." ".$matches[2]."\n";
-        $obj = json_decode($matches[2]);
-        if ($matches[1] == 'UPDATE') {
-          $this->updateLuceneRecord($obj);
-        }else{
-          $this->solr->deleteById($obj->id);
-        }
-      }
-    }
-    unlink(self::getFileCommands().'.lock');
-  }
-
-
-  public function deleteLuceneRecord($id)
-  {
-    if($this->solr->deleteById($id) )
-      return $this->solr->commit();
-    return false;
-  }
-
-  public function updateLuceneRecord($obj)
-  {
-     $document = new Apache_Solr_Document(); 
-     $document->addField('id', $obj->id); 
-     $document->addField('object_id', $obj->object_id); 
-     $document->addField('object_name', $obj->object_name); 
-     if (isset($obj->wordcount))
-       $document->addField('wordcount', $obj->wordcount); 
-     if (isset($obj->title))
-       $document->addField('title', $obj->title->content, $obj->title->weight); 
-     if (isset($obj->description))
-       $document->addField('description', $obj->description->content, $obj->description->weight); 
-     if (isset($obj->date))
-       $document->addField('date', $obj->date->content, $obj->date->weight); 
-     $this->solr->addDocument($document);
-     $this->solr->commit();
-  }
-
-  public function deleteAll() {
-    $this->solr->deleteByQuery('*:*');
-    $this->solr->commit();
-  }
-
-  public function search($queryString, $params = array(), $offset = 0, $maxHits = 0) {
-    if($maxHits == 0)
-        $maxHits = sfConfig::get('app_solr_max_hits', 256);
-    $response = $this->solr->search($queryString, $offset, $maxHits, $params);
-    return unserialize($response->getRawResponse());
-  }
-  
-}
 
 class SolrListener extends Doctrine_Record_Listener
@@ -141 +43 @@
     }
 
-    private function sendCommand($status, $json) {
-      self::getFileCommand()->log($status.': '.json_encode($json));
-    }
-    
-  private function get_and_strip($obj, $field) {
-    $f = $obj->get($field);
-    if ($f) {
+    private function get_and_strip($obj, $field) {
+      $f = $obj->get($field);
+      if ($f) {
       if (get_class($f) && ! $f->id)
         return ;
@@ -172 +70 @@
   }
   
-    // Réindexation après une création / modification
-    public function postSave(Doctrine_Event $event)
-    {
-      $obj = $event->getInvoker();
-      //      $this->getSolrConnector()->updateLuceneRecord($obj);
-
-      if ($t = $this->_options['index_if'] && $t && $obj->get($t))
-        return ;
-      
-      $json = array();
-      $json['id'] = $this->getLuceneObjId($obj);
-      $json['object_id'] =  $obj->getId();
-      $json['object_name'] = get_class($obj);
-      
-
+  // Réindexation après une création / modification
+  public function postSave(Doctrine_Event $event)
+  {
+    $obj = $event->getInvoker();
+    
+    if ($t = $this->_options['index_if'] && $t && $obj->get($t))
+      return ;
+    
+    $json = array();
+    $json['id'] = $this->getLuceneObjId($obj);
+    $json['object_id'] =  $obj->getId();
+    $json['object_name'] = get_class($obj);
+    
     if ($t = $this->_options['description']) {
       $content = $this->getObjFieldsValue($obj, $t);
       $wordcount = str_word_count($content);
     }
-
+    
     if (isset($this->_options['extra_weight'])) 
       $extra_weight = $this->_options['extra_weight'];
@@ -248 +144 @@
     
     $json['tags']['weight'] = $extra_weight;
-
     
-    $this->sendCommand('UPDATE', $json);
-    }
-    
-    // Désindexation après une suppression
-    public function postDelete(Doctrine_Event $event)
-    {
-      $obj = $event->getInvoker();
-      //      $this->getSolrConnector()->deleteLuceneRecord($obj);
-      $json = new stdClass();
-      $json->id = $this->getLuceneObjId($obj);
-      $this->sendCommand('DELETE', $json);
-    }
-    
+    SolrCommands::addCommand('UPDATE', $json);
+  }
+  
+  // Désindexation après une suppression
+  public function postDelete(Doctrine_Event $event)
+  {
+    $obj = $event->getInvoker();
+    $json = new stdClass();
+    $json->id = $this->getLuceneObjId($obj);
+    SolrCommands::addCommand('DELETE', $json);
+  }
+  
 }