Added connect method

Author: allflame

src/Decorator/JaegerConnectionDecorator.php

@@ -18,19 +18,37 @@ class JaegerConnectionDecorator extends AbstractConnectionDecorator
 {
     private $tracer;
 
-   public function __construct(Connection $connection, TracerInterface $tracer)
-   {
-       $this->tracer = $tracer;
-       parent::__construct($connection);
-   }
+    public function __construct(Connection $connection, TracerInterface $tracer)
+    {
+        $this->tracer = $tracer;
+        parent::__construct($connection);
+    }
+
+    public function connect()
+    {
+        $span = $this->tracer
+            ->start('dbal.connect')
+            ->addTag(new DbInstanceTag($this->getDatabase()))
+            ->addTag(new DbUser($this->getUsername()))
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
+        try {
+            parent::connect();
+        } catch (\Exception $e) {
+            $span->addTag(new DbalErrorCodeTag($e->getCode()))
+                ->addTag(new ErrorTag());
+            throw $e;
+        } finally {
+            $this->tracer->finish($span);
+        }
+    }
 
     public function prepare($prepareString)
     {
         $span = $this->tracer
             ->start('dbal.prepare')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType('sql'))
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
             ->addTag(new DbStatementTag($prepareString));
         try {
             return parent::prepare($prepareString);
@@ -49,7 +67,7 @@ public function executeQuery($query, array $params = [], $types = [], QueryCache
             ->start('dbal.execute')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType('sql'))
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
             ->addTag(new DbStatementTag($query));
         try {
             return parent::executeQuery($query, $params, $types, $qcp);
@@ -68,7 +86,7 @@ public function executeUpdate($query, array $params = [], array $types = [])
             ->start('dbal.execute')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType('sql'))
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()))
             ->addTag(new DbStatementTag($query));
         try {
             return parent::executeUpdate($query, $params, $types);
@@ -81,15 +99,13 @@ public function executeUpdate($query, array $params = [], array $types = [])
         }
     }
 
-
-
     public function query()
     {
         $span = $this->tracer
             ->start('dbal.query')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType('sql'));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
         try {
             return parent::query();
         } catch (\Exception $e) {
@@ -107,7 +123,7 @@ public function exec($statement)
             ->start('dbal.exec')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType('sql'));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
         try {
             $rows = parent::exec($statement);
             $span->addTag(new DbalRowNumberTag($rows));
@@ -128,7 +144,7 @@ public function beginTransaction()
             ->start('dbal.transaction')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType('sql'));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
         try {
             parent::beginTransaction();
         } catch (\Exception $e) {
@@ -146,7 +162,7 @@ public function commit()
             ->start('dbal.commit')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType('sql'));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
         try {
             parent::commit();
         } catch (\Exception $e) {
@@ -164,7 +180,7 @@ public function rollBack()
             ->start('dbal.rollback')
             ->addTag(new DbInstanceTag($this->getDatabase()))
             ->addTag(new DbUser($this->getUsername()))
-            ->addTag(new DbType('sql'));
+            ->addTag(new DbType($this->getDatabasePlatform()->getName()));
         try {
             return parent::rollBack();
         } catch (\Exception $e) {