src/Extern/ExternElasticService.php line 63

Open in your IDE?
  1. <?php
  3. namespace App\Extern;
  5. use Elastica\Client;
  6. use Elastica\Exception\NotFoundException;
  7. use Elastica\Index;
  8. use Elastica\Search;
  9. use Psr\Log\LoggerInterface;
  10. use RuntimeException;
  12. class ExternElasticService
  13. {
  14.     protected $client;
  15.     protected $search;
  17.     /** @var Index $index */
  18.     protected $index;
  19.     /** @var LoggerInterface $logger */
  20.     private $logger;
  22.     public function __construct($host, $port, $transport, LoggerInterface $logger)
  23.     {
  24.         $this->logger = $logger;
  25.         $this->client = new Client(
  26.             [
  27.                 'servers' => [
  28.                     [
  29.                         'host' => $host,
  30.                         'port' => $port,
  31.                         'transport' => $transport
  32.                     ]
  33.                 ],
  34.             ]
  35.         );
  37.         $this->search = new Search($this->client);
  38.     }
  40.     public function getCount()
  41.     {
  42.         return $this->search->count();
  43.     }
  45.     public function getClient()
  46.     {
  47.         return $this->client;
  48.     }
  50.     public function getSearch()
  51.     {
  52.         return $this->search;
  53.     }
  55.     public function getIndex()
  56.     {
  57.         return $this->index;
  58.     }
  60.     public function addAndSetIndex(string $index)
  61.     {
  62.         try {
  63.             $this->search->addIndex($index);
  64.             $this->index = $this->client->getIndex($index);
  65.         } catch (RuntimeException $exception) {
  66.             $this->logger->alert(
  67.                 "impossible to add index " . $index . ": " . $exception->getMessage(),
  68.                 ['exception' => $exception]
  69.             );
  70.         }
  71.     }
  73.     public function getDocument($id)
  74.     {
  75.         $doc = null;
  76.         try {
  77.             $doc = $this->index->getDocument($id)->getData();
  78.             $doc["_id"] = $id;
  79.         } catch (NotFoundException $exception) {
  80.             $doc = null;
  81.         } catch (RuntimeException $exception) {
  82.             $this->logger->alert(
  83.                 "impossible to get document: " . $exception->getMessage(),
  84.                 ['exception' => $exception]
  85.             );
  86.         }
  87.         return $doc;
  88.     }
  90.     public function updateDocument($id, $data)
  91.     {
  92.         $upd = null;
  93.         try {
  94.             $doc = $this->index->getDocument($id);
  95.             $doc->setData(array_merge($doc->getData(), (array)json_decode($data)));
  96.             $upd = $this->index->updateDocument($doc);
  97.         } catch (RuntimeException $exception) {
  98.             $this->logger->alert(
  99.                 "impossible to update document: " . $exception->getMessage(),
  100.                 ['exception' => $exception]
  101.             );
  102.             $upd = null;
  103.         }
  105.         return $upd;
  106.     }
  108.     public function addDocument($id, $data)
  109.     {
  110.         try {
  111.             $doc = $this->index->createDocument($id, $data);
  112.             return $this->index->addDocument($doc);
  113.         } catch (RuntimeException $exception) {
  114.             $this->logger->alert(
  115.                 "impossible to delete document: " . $exception->getMessage(),
  116.                 ['exception' => $exception]
  117.             );
  118.         }
  119.         return null;
  120.     }
  122.     public function deleteDocument($id)
  123.     {
  124.         $del = null;
  125.         try {
  126.             $doc = $this->index->getDocument($id);
  127.             $del = $this->index->deleteDocument($doc);
  128.         } catch (NotFoundException $exception) {
  129.             $del = null;
  130.         } catch (RuntimeException $exception) {
  131.             $this->logger->alert(
  132.                 "impossible to delete document: " . $exception->getMessage(),
  133.                 ['exception' => $exception]
  134.             );
  135.         }
  136.         return $del;
  137.     }
  138. }