From ddbc5525b374e4a8102c028d7d2fbf6567f7fb5d Mon Sep 17 00:00:00 2001
From: Sergei Malafeev <sergeymalafeev@gmail.com>
Date: Mon, 28 Oct 2019 12:59:21 +0800
Subject: [PATCH] #15 add more tags for transport client

Signed-off-by: Sergei Malafeev <sergeymalafeev@gmail.com>
---
 .../contrib/elasticsearch5/TracingPreBuiltTransportClient.java  | 2 ++
 .../contrib/elasticsearch6/TracingPreBuiltTransportClient.java  | 2 ++
 .../contrib/elasticsearch7/TracingPreBuiltTransportClient.java  | 2 ++
 3 files changed, 6 insertions(+)

diff --git a/opentracing-elasticsearch5-client/src/main/java/io/opentracing/contrib/elasticsearch5/TracingPreBuiltTransportClient.java b/opentracing-elasticsearch5-client/src/main/java/io/opentracing/contrib/elasticsearch5/TracingPreBuiltTransportClient.java
index 66992da..e48a2f3 100644
--- a/opentracing-elasticsearch5-client/src/main/java/io/opentracing/contrib/elasticsearch5/TracingPreBuiltTransportClient.java
+++ b/opentracing-elasticsearch5-client/src/main/java/io/opentracing/contrib/elasticsearch5/TracingPreBuiltTransportClient.java
@@ -84,6 +84,8 @@ protected <Request extends ActionRequest, Response extends ActionResponse, Reque
       Action<Request, Response, RequestBuilder> action, Request request,
       ActionListener<Response> listener) {
     Tracer.SpanBuilder spanBuilder = tracer.buildSpan(request.getClass().getSimpleName())
+        .withTag("action", action.name())
+        .withTag("description", request.getDescription())
         .withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_CLIENT);
 
     Span span = spanBuilder.start();
diff --git a/opentracing-elasticsearch6-client/src/main/java/io/opentracing/contrib/elasticsearch6/TracingPreBuiltTransportClient.java b/opentracing-elasticsearch6-client/src/main/java/io/opentracing/contrib/elasticsearch6/TracingPreBuiltTransportClient.java
index d8ced97..0455ad5 100644
--- a/opentracing-elasticsearch6-client/src/main/java/io/opentracing/contrib/elasticsearch6/TracingPreBuiltTransportClient.java
+++ b/opentracing-elasticsearch6-client/src/main/java/io/opentracing/contrib/elasticsearch6/TracingPreBuiltTransportClient.java
@@ -84,6 +84,8 @@ protected <Request extends ActionRequest, Response extends ActionResponse, Reque
       Action<Request, Response, RequestBuilder> action, Request request,
       ActionListener<Response> listener) {
     Tracer.SpanBuilder spanBuilder = tracer.buildSpan(request.getClass().getSimpleName())
+        .withTag("action", action.name())
+        .withTag("description", request.getDescription())
         .withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_CLIENT);
 
     Span span = spanBuilder.start();
diff --git a/opentracing-elasticsearch7-client/src/main/java/io/opentracing/contrib/elasticsearch7/TracingPreBuiltTransportClient.java b/opentracing-elasticsearch7-client/src/main/java/io/opentracing/contrib/elasticsearch7/TracingPreBuiltTransportClient.java
index 0e70b5b..1385110 100644
--- a/opentracing-elasticsearch7-client/src/main/java/io/opentracing/contrib/elasticsearch7/TracingPreBuiltTransportClient.java
+++ b/opentracing-elasticsearch7-client/src/main/java/io/opentracing/contrib/elasticsearch7/TracingPreBuiltTransportClient.java
@@ -82,6 +82,8 @@ public TracingPreBuiltTransportClient(Settings settings,
   protected <Request extends ActionRequest, Response extends ActionResponse> void doExecute(
       ActionType<Response> action, Request request, ActionListener<Response> listener) {
     Tracer.SpanBuilder spanBuilder = tracer.buildSpan(request.getClass().getSimpleName())
+        .withTag("action", action.name())
+        .withTag("description", request.getDescription())
         .withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_CLIENT);
 
     Span span = spanBuilder.start();