Add limit for sql length

Author: drewblin

src/Decorator/JaegerConnectionDecorator.php

@@ -20,9 +20,15 @@ class JaegerConnectionDecorator extends AbstractConnectionDecorator
 {
     private $tracer;
 
-    public function __construct(Connection $connection, TracerInterface $tracer)
+    /**
+     * @var int|null
+     */
+    private $maxSqlLength;
+
+    public function __construct(Connection $connection, TracerInterface $tracer, ?int $maxSqlLength = null)
     {
         $this->tracer = $tracer;
+        $this->maxSqlLength = $maxSqlLength;
         parent::__construct($connection);
     }
 
@@ -215,6 +221,10 @@ public function rollBack()
 
     private function cutLongSql(string $string): string
     {
-        return substr($string, 0, 200);
+        if (null === $this->maxSqlLength) {
+            return $string;
+        }
+
+        return substr($string, 0, $this->maxSqlLength);
     }
 }

src/Wrapper/JaegerConnectionWrapper.php

@@ -23,13 +23,25 @@ class JaegerConnectionWrapper extends Connection
      */
     private $tracer;
 
+    /**
+     * @var int|null
+     */
+    private $maxSqlLength;
+
     public function setTracer(TracerInterface $tracer)
     {
         $this->tracer = $tracer;
 
         return $this;
     }
 
+    public function setMaxSqlLength(?int $maxSqlLength)
+    {
+        $this->maxSqlLength = $maxSqlLength;
+
+        return $this;
+    }
+
     public function connect()
     {
         if ($this->isConnected()) {
@@ -85,6 +97,7 @@ private function wrappedPrepare($sql)
 
         $stmt->setFetchMode($this->defaultFetchMode);
         $stmt->setTracer($this->tracer);
+        $stmt->setMaxSqlLength($this->maxSqlLength);
 
         return $stmt;
     }
@@ -235,6 +248,10 @@ public function rollBack()
 
     private function cutLongSql(string $string): string
     {
-        return substr($string, 0, 200);
+        if (null === $this->maxSqlLength) {
+            return $string;
+        }
+
+        return substr($string, 0, $this->maxSqlLength);
     }
 }

src/Wrapper/JaegerStatementWrapper.php

@@ -16,13 +16,25 @@ class JaegerStatementWrapper extends Statement
      */
     private $tracer;
 
+    /**
+     * @var int|null
+     */
+    private $maxSqlLength;
+
     public function setTracer(TracerInterface $tracer)
     {
         $this->tracer = $tracer;
 
         return $this;
     }
 
+    public function setMaxSqlLength(?int $maxSqlLength)
+    {
+        $this->maxSqlLength = $maxSqlLength;
+
+        return $this;
+    }
+
     public function execute($params = null)
     {
         $span = $this->tracer
@@ -43,6 +55,10 @@ public function execute($params = null)
 
     private function cutLongSql(string $string): string
     {
-        return substr($string, 0, 200);
+        if (null === $this->maxSqlLength) {
+            return $string;
+        }
+
+        return substr($string, 0, $this->maxSqlLength);
     }
 }