Addd tags, jaeger connection decorator

Author: allflame

.gitignore

@@ -0,0 +1,5 @@
+.idea/
+/.env
+/vendor/
+/phpunit.xml
+composer.lock

composer.json

@@ -0,0 +1,26 @@
+{
+  "name": "code-tool/doctrine-dbal-jaeger",
+  "require": {
+    "php": "^7.2",
+    "doctrine/dbal": "^2.8",
+    "code-tool/jaeger-client-php": "^2.15"
+  },
+  "require-dev": {
+    "phpunit/phpunit": "^7.2"
+  },
+  "config": {
+    "sort-packages": true
+  },
+  "autoload": {
+    "psr-4": {
+      "Doctrine\\DBAL\\Jaeger\\": "src/"
+    }
+  },
+  "autoload-dev": {
+    "psr-4": {
+      "Doctrine\\DBAL\\Jaeger\\Tests\\": "tests/"
+    }
+  },
+  "prefer-stable": true,
+  "minimum-stability": "dev"
+}

src/Decorator/AbstractConnectionDecorator.php

@@ -0,0 +1,67 @@
+<?php
+declare(strict_types=1);
+
+namespace Doctrine\DBAL\Jaeger\Decorator;
+
+use Doctrine\DBAL\Driver\Connection as DriverConnection;
+use Doctrine\DBAL\ParameterType;
+
+class AbstractConnectionDecorator implements DriverConnection
+{
+    private $connection;
+
+    public function __construct(DriverConnection $connection)
+    {
+        $this->connection = $connection;
+    }
+
+    public function prepare($prepareString)
+    {
+        return $this->connection->prepare($prepareString);
+    }
+
+    public function query()
+    {
+        return $this->connection->query();
+    }
+
+    public function quote($input, $type = ParameterType::STRING)
+    {
+        return $this->connection->quote($input, $type);
+    }
+
+    public function exec($statement)
+    {
+        return $this->connection->exec($statement);
+    }
+
+    public function lastInsertId($name = null)
+    {
+        return $this->connection->lastInsertId($name);
+    }
+
+    public function beginTransaction()
+    {
+        return $this->connection->beginTransaction();
+    }
+
+    public function commit()
+    {
+        return $this->connection->commit();
+    }
+
+    public function rollBack()
+    {
+        return $this->connection->rollBack();
+    }
+
+    public function errorCode()
+    {
+        return $this->connection->errorCode();
+    }
+
+    public function errorInfo()
+    {
+        return $this->connection->errorInfo();
+    }
+}

src/Decorator/JaegerConnectionDecorator.php

@@ -0,0 +1,111 @@
+<?php
+declare(strict_types=1);
+
+namespace Doctrine\DBAL\Jaeger\Decorator;
+
+use Doctrine\DBAL\Driver\Connection as DriverConnection;
+use Doctrine\DBAL\Jaeger\Tag\DbalErrorCodeTag;
+use Doctrine\DBAL\Jaeger\Tag\DbalRowNumberTag;
+use Jaeger\Tag\DbStatementTag;
+use Jaeger\Tag\ErrorTag;
+use Jaeger\Tracer\TracerInterface;
+
+class JaegerConnectionDecorator extends AbstractConnectionDecorator
+{
+    private $tracer;
+
+    public function __construct(DriverConnection $connection, TracerInterface $tracer)
+    {
+        $this->tracer = $tracer;
+        parent::__construct($connection);
+    }
+
+    public function prepare($prepareString)
+    {
+        $span = $this->tracer
+            ->start('dbal.prepare')
+            ->addTag(new DbStatementTag($prepareString));
+        try {
+            return parent::prepare($prepareString);
+        } catch (\Exception $e) {
+            $span->addTag(new DbalErrorCodeTag($e->getCode()))
+                ->addTag(new ErrorTag());
+            throw $e;
+        } finally {
+            $this->tracer->finish($span);
+        }
+    }
+
+    public function query()
+    {
+        $span = $this->tracer->start('dbal.query');
+        try {
+            return parent::query();
+        } catch (\Exception $e) {
+            $span->addTag(new DbalErrorCodeTag($e->getCode()))
+                ->addTag(new ErrorTag());
+            throw $e;
+        } finally {
+            $this->tracer->finish($span);
+        }
+    }
+
+    public function exec($statement)
+    {
+        $span = $this->tracer->start('dbal.exec');
+        try {
+            $rows = parent::exec($statement);
+            $span->addTag(new DbalRowNumberTag($rows));
+
+            return $rows;
+        } catch (\Exception $e) {
+            $span->addTag(new DbalErrorCodeTag($e->getCode()))
+                ->addTag(new ErrorTag());
+            throw $e;
+        } finally {
+            $this->tracer->finish($span);
+        }
+    }
+
+    public function beginTransaction()
+    {
+        $span = $this->tracer->start('dbal.transaction');
+        try {
+            return parent::beginTransaction();
+        } catch (\Exception $e) {
+            $span->addTag(new DbalErrorCodeTag($e->getCode()))
+                ->addTag(new ErrorTag());
+            throw $e;
+        } finally {
+            $this->tracer->finish($span);
+        }
+    }
+
+    public function commit()
+    {
+        $span = $this->tracer->start('dbal.commit');
+        try {
+            return parent::commit();
+        } catch (\Exception $e) {
+            $span->addTag(new DbalErrorCodeTag($e->getCode()))
+                ->addTag(new ErrorTag());
+            throw $e;
+        } finally {
+            $this->tracer->finish($span);
+        }
+    }
+
+    public function rollBack()
+    {
+        $span = $this->tracer->start('dbal.rollback');
+        try {
+            return parent::rollBack();
+        } catch (\Exception $e) {
+            $span->addTag(new DbalErrorCodeTag($e->getCode()))
+                ->addTag(new ErrorTag());
+            throw $e;
+        } finally {
+            $this->tracer->finish($span);
+        }
+    }
+}

src/Tag/DbalErrorCodeTag.php

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

src/Tag/DbalRowNumberTag.php

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