added events for manager factory

Author: Simonas Šerlinskas

Event/Events.php

@@ -30,4 +30,15 @@ final class Events
      * The POST_COMMIT event occurs after committing queries to ES
      */
     const POST_COMMIT = 'es.post_commit';
+
+    /**
+     * The PRE_MANAGER_CREATE event occurs before manager is created, right after client is initiated.
+     *  You can modify anything in the core elasticsearch-php client by this event.
+     */
+    const PRE_MANAGER_CREATE = 'es.pre_manager_create';
+
+    /**
+     * The POST_MANAGER_CREATE event occurs after manager is created.
+     */
+    const POST_MANAGER_CREATE = 'es.post_manager_create';
 }

Event/PostCreateManagerEvent.php

@@ -0,0 +1,49 @@
+<?php
+
+/*
+ * This file is part of the ONGR package.
+ *
+ * (c) NFQ Technologies UAB <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace ONGR\ElasticsearchBundle\Event;
+
+use ONGR\ElasticsearchBundle\Service\Manager;
+use Symfony\Component\EventDispatcher\Event;
+
+class PostCreateManagerEvent extends Event
+{
+    /**
+     * @var Manager
+     */
+    private $manager;
+
+    /**
+     * PostCreateManagerEvent constructor.
+     *
+     * @param Manager $manager
+     */
+    public function __construct(Manager $manager)
+    {
+        $this->manager = $manager;
+    }
+
+    /**
+     * @return Manager
+     */
+    public function getManager()
+    {
+        return $this->manager;
+    }
+
+    /**
+     * @param Manager $manager
+     */
+    public function setManager($manager)
+    {
+        $this->manager = $manager;
+    }
+}

Event/PreCreateManagerEvent.php

@@ -0,0 +1,72 @@
+<?php
+
+/*
+ * This file is part of the ONGR package.
+ *
+ * (c) NFQ Technologies UAB <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace ONGR\ElasticsearchBundle\Event;
+
+use Elasticsearch\ClientBuilder;
+use Symfony\Component\EventDispatcher\Event;
+
+class PreCreateManagerEvent extends Event
+{
+    /**
+     * @var ClientBuilder
+     */
+    private $client;
+
+    /**
+     * @var array
+     */
+    private $indexSettings;
+
+    /**
+     * CreateManagerEvent constructor.
+     *
+     * @param PreClientBuilder $client
+     * @param $indexSettings array
+     */
+    public function __construct(ClientBuilder $client, &$indexSettings)
+    {
+        $this->client = $client;
+        $this->indexSettings = $indexSettings;
+    }
+
+    /**
+     * @return ClientBuilder
+     */
+    public function getClient()
+    {
+        return $this->client;
+    }
+
+    /**
+     * @param ClientBuilder $client
+     */
+    public function setClient($client)
+    {
+        $this->client = $client;
+    }
+
+    /**
+     * @return array
+     */
+    public function getIndexSettings()
+    {
+        return $this->indexSettings;
+    }
+
+    /**
+     * @param array $indexSettings
+     */
+    public function setIndexSettings($indexSettings)
+    {
+        $this->indexSettings = $indexSettings;
+    }
+}

Exception/BulkWithErrorsException.php

@@ -34,4 +34,4 @@ public function getResponse()
     {
         return $this->response;
     }
-}
+}

Service/ManagerFactory.php

@@ -12,6 +12,9 @@
 namespace ONGR\ElasticsearchBundle\Service;
 
 use Elasticsearch\ClientBuilder;
+use ONGR\ElasticsearchBundle\Event\Events;
+use ONGR\ElasticsearchBundle\Event\PostCreateManagerEvent;
+use ONGR\ElasticsearchBundle\Event\PreCreateManagerEvent;
 use ONGR\ElasticsearchBundle\Mapping\MetadataCollector;
 use ONGR\ElasticsearchBundle\Result\Converter;
 use Psr\Log\LoggerInterface;
@@ -130,6 +133,12 @@ public function createManager($managerName, $connection, $analysis, $managerConf
             ),
         ];
 
+        $this->eventDispatcher &&
+            $this->eventDispatcher->dispatch(
+                Events::POST_MANAGER_CREATE,
+                new PreCreateManagerEvent($client, $indexSettings)
+            );
+
         $manager = new Manager(
             $managerName,
             $managerConfig,
@@ -147,6 +156,9 @@ public function createManager($managerName, $connection, $analysis, $managerConf
         $manager->setEventDispatcher($this->eventDispatcher);
         $manager->setBulkCommitSize($managerConfig['bulk_size']);
 
+        $this->eventDispatcher &&
+            $this->eventDispatcher->dispatch(Events::POST_MANAGER_CREATE, new PostCreateManagerEvent($manager));
+
         return $manager;
     }
 }