fix: log and tag Throwable in a span (#3011)

* fix: log and tag Throwable in a span * Update CHANGELOG-2.0.md Co-authored-by: 李铭昕 <[email protected]>
hyperf · Dec 27, 2020 · 0e5ac07 · 0e5ac07
1 parent adbf877
commit 0e5ac07
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 0 deletions.
diff --git a/src/Aspect/HttpClientAspect.php b/src/Aspect/HttpClientAspect.php
@@ -97,6 +97,10 @@ public function process(ProceedingJoinPoint $proceedingJoinPoint)
             if ($result instanceof ResponseInterface) {
                 $span->setTag($this->spanTagManager->get('http_client', 'http.status_code'), $result->getStatusCode());
             }
+        } catch (\Throwable $e) {
+            $span->setTag('error', true);
+            $span->log(['message', $e->getMessage(), 'code' => $e->getCode(), 'stacktrace' => $e->getTraceAsString()]);
+            throw $e;
         } finally {
             $span->finish();
         }

diff --git a/src/Aspect/JsonRpcAspect.php b/src/Aspect/JsonRpcAspect.php
@@ -92,6 +92,13 @@ public function process(ProceedingJoinPoint $proceedingJoinPoint)
         if ($proceedingJoinPoint->methodName === 'send') {
             try {
                 $result = $proceedingJoinPoint->process();
+            } catch (\Throwable $e) {
+                if ($span = CT::get('tracer.span.' . static::class)) {
+                    $span->setTag('error', true);
+                    $span->log(['message', $e->getMessage(), 'code' => $e->getCode(), 'stacktrace' => $e->getTraceAsString()]);
+                    CT::set('tracer.span.' . static::class, $span);
+                }
+                throw $e;
             } finally {
                 /** @var Span $span */
                 if ($span = CT::get('tracer.span.' . static::class)) {

diff --git a/src/Aspect/MethodAspect.php b/src/Aspect/MethodAspect.php
@@ -60,6 +60,10 @@ public function process(ProceedingJoinPoint $proceedingJoinPoint)
         $span = $this->startSpan($key);
         try {
             $result = $proceedingJoinPoint->process();
+        } catch (\Throwable $e) {
+            $span->setTag('error', true);
+            $span->log(['message', $e->getMessage(), 'code' => $e->getCode(), 'stacktrace' => $e->getTraceAsString()]);
+            throw $e;
         } finally {
             $span->finish();
         }

diff --git a/src/Aspect/RedisAspect.php b/src/Aspect/RedisAspect.php
@@ -77,6 +77,10 @@ public function process(ProceedingJoinPoint $proceedingJoinPoint)
         try {
             $result = $proceedingJoinPoint->process();
             $span->setTag($this->spanTagManager->get('redis', 'result'), json_encode($result));
+        } catch (\Throwable $e) {
+            $span->setTag('error', true);
+            $span->log(['message', $e->getMessage(), 'code' => $e->getCode(), 'stacktrace' => $e->getTraceAsString()]);
+            throw $e;
         } finally {
             $span->finish();
         }

diff --git a/src/Aspect/TraceAnnotationAspect.php b/src/Aspect/TraceAnnotationAspect.php
@@ -60,6 +60,10 @@ public function process(ProceedingJoinPoint $proceedingJoinPoint)
         $span->setTag($tag, $source);
         try {
             $result = $proceedingJoinPoint->process();
+        } catch (\Throwable $e) {
+            $span->setTag('error', true);
+            $span->log(['message', $e->getMessage(), 'code' => $e->getCode(), 'stacktrace' => $e->getTraceAsString()]);
+            throw $e;
         } finally {
             $span->finish();
         }