Merge pull request #3 from OpenTagOS/feature/prepared_execute_tracing

Prepared statement execute call tracing.

Author: allflame

src/Wrapper/JaegerConnectionWrapper.php

@@ -65,7 +65,7 @@ public function prepare($prepareString)
             ->addTag(new DbStatementTag($prepareString))
             ->addTag(new DbalNestingLevelTag($this->getTransactionNestingLevel()));
         try {
-            return parent::prepare($prepareString);
+            return $this->wrappedPrepare($prepareString);
         } catch (\Exception $e) {
             $span->addTag(new DbalErrorCodeTag($e->getCode()))
                 ->addTag(new ErrorTag());
@@ -75,6 +75,20 @@ public function prepare($prepareString)
         }
     }
 
+    private function wrappedPrepare($sql)
+    {
+        try {
+            $stmt = new JaegerStatementWrapper($sql, $this);
+        } catch (\Throwable $e) {
+            $this->handleExceptionDuringQuery($e, $sql);
+        }
+
+        $stmt->setFetchMode($this->defaultFetchMode);
+        $stmt->setTracer($this->tracer);
+
+        return $stmt;
+    }
+
     public function executeQuery($query, array $params = [], $types = [], QueryCacheProfile $qcp = null)
     {
         $span = $this->tracer

src/Wrapper/JaegerStatementWrapper.php

@@ -0,0 +1,43 @@
+<?php
+declare(strict_types=1);
+
+namespace Doctrine\DBAL\Jaeger\Wrapper;
+
+use Doctrine\DBAL\Jaeger\Tag\DbalErrorCodeTag;
+use Doctrine\DBAL\Statement;
+use Jaeger\Tag\DbStatementTag;
+use Jaeger\Tag\ErrorTag;
+use Jaeger\Tracer\TracerInterface;
+
+class JaegerStatementWrapper extends Statement
+{
+    /**
+     * @var TracerInterface $tracer
+     */
+    private $tracer;
+
+    public function setTracer(TracerInterface $tracer)
+    {
+        $this->tracer = $tracer;
+
+        return $this;
+    }
+
+    public function execute($params = null)
+    {
+        $span = $this->tracer
+            ->start('dbal.prepare.execute')
+            ->addTag(new DbStatementTag($this->sql));
+
+        try {
+            return parent::execute($params);
+        } catch (\Exception $e) {
+            $span->addTag(new DbalErrorCodeTag($e->getCode()))
+                ->addTag(new ErrorTag());
+
+            throw $e;
+        } finally {
+            $this->tracer->finish($span);
+        }
+    }
+}