Add retry fields to the current span

vtex · Oct 10, 2023 · 73084b9 · 73084b9
1 parent ecb4e28
commit 73084b9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/HttpClient/middlewares/request/setupAxios/interceptors/exponentialBackoff.ts b/src/HttpClient/middlewares/request/setupAxios/interceptors/exponentialBackoff.ts
@@ -3,7 +3,7 @@ import { HttpLogEvents } from '../../../../../tracing/LogEvents'
 import { HttpRetryLogFields } from '../../../../../tracing/LogFields'
 import { isAbortedOrNetworkErrorOrRouterTimeout } from '../../../../../utils/retry'
 import { RequestConfig } from '../../../../typings'
-import { TraceableRequestConfig } from '../../../tracing'
+import { CustomHttpTags } from '../../../../../tracing/Tags'
 
 function fixConfig(axiosInstance: AxiosInstance, config: RequestConfig) {
   if (axiosInstance.defaults.httpAgent === config.httpAgent) {
@@ -66,6 +66,10 @@ const onResponseError = (http: AxiosInstance) => (error: any) => {
     config.transformRequest = [data => data]
 
     config.tracing?.rootSpan?.log({ event: HttpLogEvents.SETUP_REQUEST_RETRY, [HttpRetryLogFields.RETRY_NUMBER]: config.retryCount, [HttpRetryLogFields.RETRY_IN]: delay })
+    config.tracing?.rootSpan?.addTags({
+      [CustomHttpTags.HTTP_RETRY_COUNT]: config.retryCount,
+      [CustomHttpTags.HTTP_RETRY_ERROR_CODE]: error.code
+    })
 
     return new Promise(resolve => setTimeout(() => resolve(http(config)), delay))
   }

diff --git a/src/tracing/Tags.ts b/src/tracing/Tags.ts
@@ -69,6 +69,9 @@ export const enum CustomHttpTags {
   HTTP_MEMOIZATION_CACHE_RESULT = 'http.cache.memoization',
   HTTP_DISK_CACHE_RESULT = 'http.cache.disk',
   HTTP_ROUTER_CACHE_RESULT = 'http.cache.router',
+
+  HTTP_RETRY_ERROR_CODE = 'http.retry.error.code',
+  HTTP_RETRY_COUNT = 'http.retry.count'
 }
 
 export const UserlandTags = {