Skip to content

Commit fc463a2

Browse files
stevejgordongithub-actions[bot]
authored andcommitted
Prefer SetTag and ensure all methods instrument in the same way (#7632)
1 parent 1ff7739 commit fc463a2

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

src/Elastic.Clients.Elasticsearch/Client/ElasticsearchClient.cs

+10-5
Original file line numberDiff line numberDiff line change
@@ -138,10 +138,15 @@ internal TResponse DoRequest<TRequest, TResponse, TRequestParameters>(
138138

139139
using (var activity = _activitySource.StartActivity($"Elasticsearch: {request.HttpMethod} {urlTemplate}", ActivityKind.Client))
140140
{
141-
activity?.AddTag("db.system", "elasticsearch");
142-
response = _transport.Request<TResponse>(request.HttpMethod, resolvedUrl, postData, parameters);
141+
activity?.SetTag("db.system", "elasticsearch");
142+
activity?.SetCustomProperty("elastic.transport.client", true);
143+
144+
response = _transport.Request<TResponse>(request.HttpMethod, resolvedUrl, postData, request.RequestParameters);
145+
146+
if (response.ApiCallDetails.RequestBodyInBytes is not null)
147+
activity?.SetTag("db.statement", System.Text.Encoding.UTF8.GetString(response.ApiCallDetails.RequestBodyInBytes));
143148
}
144-
149+
145150
PostRequestProductCheck<TRequest, TResponse>(request, response);
146151

147152
if (_productCheckStatus == ProductCheckStatus.Failed)
@@ -206,13 +211,13 @@ internal TResponse DoRequest<TRequest, TResponse, TRequestParameters>(
206211

207212
using (var activity = _activitySource.StartActivity($"Elasticsearch: {request.HttpMethod} {urlTemplate}", ActivityKind.Client))
208213
{
209-
activity?.AddTag("db.system", "elasticsearch");
214+
activity?.SetTag("db.system", "elasticsearch");
210215
activity?.SetCustomProperty("elastic.transport.client", true);
211216

212217
response = _transport.Request<TResponse>(request.HttpMethod, resolvedUrl, postData, request.RequestParameters);
213218

214219
if (response.ApiCallDetails.RequestBodyInBytes is not null)
215-
activity?.AddTag("db.statement", System.Text.Encoding.UTF8.GetString(response.ApiCallDetails.RequestBodyInBytes));
220+
activity?.SetTag("db.statement", System.Text.Encoding.UTF8.GetString(response.ApiCallDetails.RequestBodyInBytes));
216221
}
217222

218223
PostRequestProductCheck<TRequest, TResponse>(request, response);

0 commit comments

Comments
 (0)