From 69f9d9c6bfc2fbdad11c32773e5bd10058b3c3e1 Mon Sep 17 00:00:00 2001
From: Kris Wallsmith <kris.wallsmith@gmail.com>
Date: Sun, 22 Aug 2010 19:16:19 -0400
Subject: [PATCH] [DoctrineMongoDBBundle] added logger and data collector for
 WDT

---
 .../DoctrineMongoDBDataCollector.php          | 40 +++++++++++++++++
 .../DoctrineMongoDBExtension.php              |  1 +
 .../Logger/DoctrineMongoDBLogger.php          | 43 +++++++++++++++++++
 .../Resources/config/mongodb.xml              | 11 +++++
 4 files changed, 95 insertions(+)
 create mode 100644 src/Symfony/Bundle/DoctrineMongoDBBundle/DataCollector/DoctrineMongoDBDataCollector.php
 create mode 100644 src/Symfony/Bundle/DoctrineMongoDBBundle/Logger/DoctrineMongoDBLogger.php

diff --git a/src/Symfony/Bundle/DoctrineMongoDBBundle/DataCollector/DoctrineMongoDBDataCollector.php b/src/Symfony/Bundle/DoctrineMongoDBBundle/DataCollector/DoctrineMongoDBDataCollector.php
new file mode 100644
index 000000000000..620efb0ea99e
--- /dev/null
+++ b/src/Symfony/Bundle/DoctrineMongoDBBundle/DataCollector/DoctrineMongoDBDataCollector.php
@@ -0,0 +1,40 @@
+<?php
+
+namespace Symfony\Bundle\DoctrineMongoDBBundle\DataCollector;
+
+use Symfony\Component\HttpKernel\Profiler\DataCollector\DataCollector;
+use Symfony\Bundle\DoctrineMongoDBBundle\Logger\DoctrineMongoDBLogger;
+
+/**
+ * Data collector for the Doctrine MongoDB ODM.
+ * 
+ * @author Kris Wallsmith <kris.wallsmith@symfony-project.com>
+ */
+class DoctrineMongoDBDataCollector extends DataCollector
+{
+    protected $logger;
+
+    public function __construct(DoctrineMongoDBLogger $logger)
+    {
+        $this->logger = $logger;
+    }
+
+    public function collect()
+    {
+        $this->data['nb_queries'] = $this->logger->getNbQueries();
+    }
+
+    public function getSummary()
+    {
+        $color = $this->data['nb_queries'] < 10 ? '#2d2' : '#d22';
+
+        return sprintf('<img style="margin-left: 10px; vertical-align: middle" alt="" src="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAGXRFWHRTb2Z0d2FyZQBBZG9iZSBJbWFnZVJlYWR5ccllPAAAAeJJREFUeNqM009r1EAUAPD3kkm22bapthSlq/EPUg9FUMSLoCAVvPUm4smr4Ico+BW89SCe6tVDj2oVLwVRS6HqTcUqwe7W7SbZbGbmzTiTvei2ML6QDAwzP968l0GtNfwdS/eu1KMSBHEytbhf9HbLdrGFiPX8i9VP/6z3YCTKvITidwETyeTN5NbZNTzFHhVZGWc/MsjTfHT5QUBKCcFMOD97be6JRhgLj0fXxy9PrXi+h6jRDfCcQ3xx+iE0/ZbgAogrCE5Gd8JzzftaKjfgHw0uBUm0JCoOpAmUUqBIgwEe4LgfO4HwdLSoQoykJJBEYGuspQaMgwU227jhBHQDFwjMZmW+Wg0BPVyJR9hVJwAT3rQkCVJbxpzZ1M0ew9ZPVWrG3YVMDKTZSDYDZY+gzYuWAs2VdHfhW3/bFq/OoEYMYB8zUkd8dgMf8zWxV+XkKRA1YNL3EGhf9uSX/ksnQCl/X7zqPCZmAVkDyDzgH3qrtCe2nACGCMV6Z7n3pv2chT6wgEG1nW1UG91lDP7jT0TfA3NxurtPv96dfOu/Pp/Oves++36buPxl2zoabHSiGlTDS9Uv24PNbOXMheRE61hrpxE14LA4APjMr8eoOQY/03RTaLVTDkrggh8K/BFgAGj/AWtED/mBAAAAAElFTkSuQmCC" />
+            <span style="color: %s">%d</span>
+        ', $color, $this->data['nb_queries']);
+    }
+
+    public function getName()
+    {
+        return 'mongodb';
+    }
+}
diff --git a/src/Symfony/Bundle/DoctrineMongoDBBundle/DependencyInjection/DoctrineMongoDBExtension.php b/src/Symfony/Bundle/DoctrineMongoDBBundle/DependencyInjection/DoctrineMongoDBExtension.php
index 0247739554e8..1ded0d2c1501 100755
--- a/src/Symfony/Bundle/DoctrineMongoDBBundle/DependencyInjection/DoctrineMongoDBExtension.php
+++ b/src/Symfony/Bundle/DoctrineMongoDBBundle/DependencyInjection/DoctrineMongoDBExtension.php
@@ -129,6 +129,7 @@ protected function loadDocumentManager(array $documentManager, ContainerBuilder
             'setProxyNamespace' => $container->getParameter('doctrine.odm.mongodb.proxy_namespace'),
             'setAutoGenerateProxyClasses' => $container->getParameter('doctrine.odm.mongodb.auto_generate_proxy_classes'),
             'setDefaultDB' => $defaultDatabase,
+            'setLoggerCallable' => array(new Reference('doctrine.odm.mongodb.logger'), 'logQuery'),
         );
         foreach ($methods as $method => $arg) {
             $odmConfigDef->addMethodCall($method, array($arg));
diff --git a/src/Symfony/Bundle/DoctrineMongoDBBundle/Logger/DoctrineMongoDBLogger.php b/src/Symfony/Bundle/DoctrineMongoDBBundle/Logger/DoctrineMongoDBLogger.php
new file mode 100644
index 000000000000..4216a5ae7dbb
--- /dev/null
+++ b/src/Symfony/Bundle/DoctrineMongoDBBundle/Logger/DoctrineMongoDBLogger.php
@@ -0,0 +1,43 @@
+<?php
+
+namespace Symfony\Bundle\DoctrineMongoDBBundle\Logger;
+
+use Symfony\Component\HttpKernel\Log\LoggerInterface;
+use Symfony\Component\Yaml\Yaml;
+
+/**
+ * Logger for the Doctrine MongoDB ODM.
+ * 
+ * @author Kris Wallsmith <kris.wallsmith@symfony-project.com>
+ */
+class DoctrineMongoDBLogger
+{
+    protected $logger;
+    protected $nbQueries;
+
+    public function __construct(LoggerInterface $logger = null)
+    {
+        $this->logger = $logger;
+        $this->nbQueries = 0;
+    }
+
+    public function logQuery($query)
+    {
+        ++$this->nbQueries;
+
+        if (null !== $this->logger) {
+            switch (key($query)) {
+                case 'batchInsert':
+                    $this->logger->info(Yaml::dump(array('data' => '[omitted]') + $query, 0));
+                    break;
+                default:
+                    $this->logger->info(Yaml::dump($query, 0));
+            }
+        }
+    }
+
+    public function getNbQueries()
+    {
+        return $this->nbQueries;
+    }
+}
diff --git a/src/Symfony/Bundle/DoctrineMongoDBBundle/Resources/config/mongodb.xml b/src/Symfony/Bundle/DoctrineMongoDBBundle/Resources/config/mongodb.xml
index 091a6f816882..fa41a2ce5075 100755
--- a/src/Symfony/Bundle/DoctrineMongoDBBundle/Resources/config/mongodb.xml
+++ b/src/Symfony/Bundle/DoctrineMongoDBBundle/Resources/config/mongodb.xml
@@ -12,6 +12,8 @@
     <parameter key="doctrine.odm.mongodb.connection_class">Doctrine\ODM\MongoDB\Mongo</parameter>
     <parameter key="doctrine.odm.mongodb.configuration_class">Doctrine\ODM\MongoDB\Configuration</parameter>
     <parameter key="doctrine.odm.mongodb.document_manager_class">Doctrine\ODM\MongoDB\DocumentManager</parameter>
+    <parameter key="doctrine.odm.mongodb.logger_class">Symfony\Bundle\DoctrineMongoDBBundle\Logger\DoctrineMongoDBLogger</parameter>
+    <parameter key="doctrine.odm.mongodb.data_collector_class">Symfony\Bundle\DoctrineMongoDBBundle\DataCollector\DoctrineMongoDBDataCollector</parameter>
 
     <!-- proxies -->
     <parameter key="doctrine.odm.mongodb.proxy_namespace">Proxies</parameter>
@@ -61,5 +63,14 @@
 
     <!-- cache -->
     <service id="doctrine.odm.mongodb.cache.array" class="%doctrine.odm.mongodb.cache.array_class%" />
+
+    <!-- logger -->
+    <service id="doctrine.odm.mongodb.logger" class="%doctrine.odm.mongodb.logger_class%">
+      <argument type="service" id="logger" on-invalid="null" />
+     </service>
+    <service id="doctrine.odm.mongodb.data_collector" class="%doctrine.odm.mongodb.data_collector_class%">
+      <tag name="data_collector" />
+      <argument type="service" id="doctrine.odm.mongodb.logger" />
+    </service>
   </services>
 </container>