Added DbalAutoCommit tag

Author: allflame

src/Decorator/JaegerConnectionDecorator.php

@@ -5,6 +5,7 @@
 
 use Doctrine\DBAL\Cache\QueryCacheProfile;
 use Doctrine\DBAL\Connection;
+use Doctrine\DBAL\Jaeger\Tag\DbalAutoCommitTag;
 use Doctrine\DBAL\Jaeger\Tag\DbalErrorCodeTag;
 use Doctrine\DBAL\Jaeger\Tag\DbalRowNumberTag;
 use Jaeger\Tag\DbInstanceTag;
@@ -30,7 +31,8 @@ public function connect()
             ->start('dbal.connect')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()));
         try {
             parent::connect();
         } catch (\Exception $e) {
@@ -49,6 +51,7 @@ public function prepare($prepareString)
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
             ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()))
             ->addTag(new DbStatementTag($prepareString));
         try {
             return parent::prepare($prepareString);
@@ -68,6 +71,7 @@ public function executeQuery($query, array $params = [], $types = [], QueryCache
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
             ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()))
             ->addTag(new DbStatementTag($query));
         try {
             return parent::executeQuery($query, $params, $types, $qcp);
@@ -87,6 +91,7 @@ public function executeUpdate($query, array $params = [], array $types = [])
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
             ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()))
             ->addTag(new DbStatementTag($query));
         try {
             return parent::executeUpdate($query, $params, $types);
@@ -105,7 +110,8 @@ public function query()
             ->start('dbal.query')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()));
         try {
             return parent::query();
         } catch (\Exception $e) {
@@ -123,7 +129,8 @@ public function exec($statement)
             ->start('dbal.exec')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()));
         try {
             $rows = parent::exec($statement);
             $span->addTag(new DbalRowNumberTag($rows));
@@ -144,7 +151,8 @@ public function beginTransaction()
             ->start('dbal.transaction')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()));
         try {
             parent::beginTransaction();
         } catch (\Exception $e) {
@@ -162,7 +170,8 @@ public function commit()
             ->start('dbal.commit')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()));
         try {
             parent::commit();
         } catch (\Exception $e) {
@@ -180,7 +189,8 @@ public function rollBack()
             ->start('dbal.rollback')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
+            ->addTag(new DbalAutoCommitTag($this->isAutoCommit()));
         try {
             return parent::rollBack();
         } catch (\Exception $e) {

src/Tag/DbalAutoCommitTag.php

@@ -0,0 +1,14 @@
+<?php
+declare(strict_types=1);
+
+namespace Doctrine\DBAL\Jaeger\Tag;
+
+use Jaeger\Tag\BoolTag;
+
+class DbalAutoCommitTag extends BoolTag
+{
+    public function __construct(bool $value)
+    {
+        parent::__construct('dbal.auto_commit', $value);
+    }
+}