Make DB connection wrapping configurable and transparent

Author: rootpd

README.md

@@ -30,8 +30,10 @@ sentry:
         - email
     session_sections: # optional, list of session sections to track
         - mySection
-    priority_mapping:
+    priority_mapping: # optional, mapping of custom log levels to ones that's Sentry aware of
         mypriority: warning
+
+    db_tracing: true # optional, defaults to "false", wraps DB connection so it adds span for every query
 ```
 
 ### Priority-Severity mapping

src/DI/SentryExtension.php

@@ -11,6 +11,7 @@
 use Nette\Schema\Expect;
 use Nette\Schema\Schema;
 use Rootpd\NetteSentry\ApplicationMonitor;
+use Rootpd\NetteSentry\DatabaseConnection;
 use Rootpd\NetteSentry\SentryLogger;
 use Sentry\Integration\EnvironmentIntegration;
 use Sentry\Integration\TransactionIntegration;
@@ -55,6 +56,13 @@ public function loadConfiguration()
                 $logger->addSetup('setProfilesSampleRate', [$this->config->profiles_sample_rate]);
             }
 
+            if ($this->config->db_tracing) {
+                /** @var ServiceDefinition $dbDefinition */
+                $dbDefinition = $this->getContainerBuilder()->getDefinition('database.default.connection');
+                $dbFactory = $dbDefinition->getFactory();
+                $dbDefinition->setFactory(DatabaseConnection::class, $dbFactory->arguments);
+            }
+
             $integrations[] = new Statement(TransactionIntegration::class);
             $integrations[] = new Statement(EnvironmentIntegration::class);
         }
@@ -71,13 +79,17 @@ public function loadConfiguration()
     public function getConfigSchema(): Schema
     {
         return Expect::structure([
+            // sentry configs
             'dsn' => Expect::string()->dynamic(),
             'environment' => Expect::string('local')->dynamic(),
             'user_fields' => Expect::listOf(Expect::string())->default([]),
             'session_sections' => Expect::listOf(Expect::string())->default([]),
             'priority_mapping' => Expect::arrayOf(Expect::string(), Expect::string())->default([]),
             'traces_sample_rate' => Expect::float()->dynamic(),
             'profiles_sample_rate' => Expect::float()->dynamic(),
+
+            // extension configs
+            'db_tracing' => Expect::bool(false)->dynamic(),
         ]);
     }