Merge pull request #1 from andrew-demb/allow-disable-auto-subscribe

Allow to disable auto register subscriber on bundle boot

Author: allflame

src/JaegerMongoDbBundle.php

@@ -11,7 +11,10 @@ class JaegerMongoDbBundle extends Bundle
     public function boot()
     {
         parent::boot();
-        \MongoDB\Driver\Monitoring\addSubscriber($this->container->get('jaeger.mongodb.query.time.collector'));
+
+        if ($this->container->getParameter('jaeger.mongodb.auto_subscribe')) {
+            \MongoDB\Driver\Monitoring\addSubscriber($this->container->get('jaeger.mongodb.query.time.collector'));
+        }
     }
 
 

src/Resources/DependencyInjection/Configuration.php

@@ -0,0 +1,32 @@
+<?php
+declare(strict_types=1);
+
+namespace Jaeger\MongoDb\Symfony\Resources\DependencyInjection;
+
+use Symfony\Component\Config\Definition\Builder\TreeBuilder;
+use Symfony\Component\Config\Definition\ConfigurationInterface;
+
+class Configuration implements ConfigurationInterface
+{
+    public function getConfigTreeBuilder()
+    {
+        $treeBuilder = new TreeBuilder('jaeger_mongodb');
+        if (method_exists($treeBuilder, 'getRootNode')) {
+            $rootNode = $treeBuilder->getRootNode();
+        } else {
+            $rootNode = $treeBuilder->root('jaeger_mongodb');
+        }
+
+        // @formatter:off
+        $rootNode
+            ->children()
+                ->booleanNode('auto_subscribe')
+                    ->info('Register collector on bundle boot')
+                    ->defaultTrue()
+                ->end()
+            ->end();
+        // @formatter:on
+
+        return $treeBuilder;
+    }
+}

src/Resources/DependencyInjection/JaegerMongoDbExtension.php

@@ -18,5 +18,15 @@ public function load(array $configs, ContainerBuilder $container)
             new FileLocator(__DIR__ . '/../config')
         );
         $loader->load('services.yml');
+
+        $configuration = $this->getConfiguration($configs, $container);
+        $config = $this->processConfiguration($configuration, $configs);
+
+        $container->setParameter('jaeger.mongodb.auto_subscribe', $config['auto_subscribe']);
+    }
+
+    public function getAlias(): string
+    {
+        return 'jaeger_mongodb';
     }
 }