From 218051303f1b07af4ed9362ab7df1be07704c821 Mon Sep 17 00:00:00 2001
From: Maksim Kotlyar <kotlyar.maksim@gmail.com>
Date: Tue, 2 Jan 2018 20:09:15 +0200
Subject: [PATCH] Add ability to change used context.

---
 DependencyInjection/Configuration.php            | 1 +
 DependencyInjection/EnqueueElasticaExtension.php | 4 +++-
 Resources/config/services.yml                    | 4 ++--
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/DependencyInjection/Configuration.php b/DependencyInjection/Configuration.php
index c4e1a5e..c7f693c 100644
--- a/DependencyInjection/Configuration.php
+++ b/DependencyInjection/Configuration.php
@@ -16,6 +16,7 @@ public function getConfigTreeBuilder()
         $rootNode = $tb->root('enqueue_elastica');
         $rootNode
             ->children()
+                ->scalarNode('context')->isRequired()->defaultValue('enqueue.transport.context')->cannotBeEmpty()->end()
                 ->arrayNode('doctrine')
                     ->children()
                         ->arrayNode('queue_listeners')
diff --git a/DependencyInjection/EnqueueElasticaExtension.php b/DependencyInjection/EnqueueElasticaExtension.php
index f23213d..3e32626 100644
--- a/DependencyInjection/EnqueueElasticaExtension.php
+++ b/DependencyInjection/EnqueueElasticaExtension.php
@@ -21,6 +21,8 @@ public function load(array $configs, ContainerBuilder $container)
         $loader = new YamlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));
         $loader->load('services.yml');
 
+        $container->setAlias('enqueue_elastica.context', $config['context']);
+
         if (false == empty($config['doctrine']['queue_listeners'])) {
             foreach ($config['doctrine']['queue_listeners'] as $listenerConfig) {
                 $listenerId = sprintf(
@@ -31,7 +33,7 @@ public function load(array $configs, ContainerBuilder $container)
 
                 $container->register($listenerId, SyncIndexWithObjectChangeListener::class)
                     ->setPublic(true)
-                    ->addArgument(new Reference('enqueue.transport.context'))
+                    ->addArgument(new Reference('enqueue_elastica.context'))
                     ->addArgument($listenerConfig['model_class'])
                     ->addArgument($listenerConfig)
                     ->addTag('doctrine.event_subscriber', ['connection' => $listenerConfig['connection']])
diff --git a/Resources/config/services.yml b/Resources/config/services.yml
index a807ca4..e900cd3 100644
--- a/Resources/config/services.yml
+++ b/Resources/config/services.yml
@@ -22,7 +22,7 @@ services:
       class: 'Enqueue\ElasticaBundle\Persister\Listener\PurgePopulateQueueListener'
       public: true
       arguments:
-          - '@enqueue.transport.context'
+          - '@enqueue_elastica.context'
       tags:
           -  { name: 'kernel.event_subscriber' }
 
@@ -30,7 +30,7 @@ services:
       class: 'Enqueue\ElasticaBundle\Persister\QueuePagerPersister'
       public: true
       arguments:
-          - '@enqueue.transport.context'
+          - '@enqueue_elastica.context'
           - '@fos_elastica.persister_registry'
           - '@event_dispatcher'
       tags: