|
7 | 7 | import datadog.trace.bootstrap.instrumentation.decorator.HttpClientDecorator;
|
8 | 8 | import java.net.URI;
|
9 | 9 | import java.net.URISyntaxException;
|
| 10 | +import java.util.ArrayList; |
| 11 | +import java.util.List; |
| 12 | +import java.util.Map; |
| 13 | +import org.jboss.netty.handler.codec.http.HttpHeaders; |
10 | 14 | import org.jboss.netty.handler.codec.http.HttpRequest;
|
11 | 15 | import org.jboss.netty.handler.codec.http.HttpResponse;
|
12 | 16 |
|
@@ -57,11 +61,23 @@ protected int status(final HttpResponse httpResponse) {
|
57 | 61 |
|
58 | 62 | @Override
|
59 | 63 | protected String getRequestHeader(HttpRequest request, String headerName) {
|
60 |
| - return request.headers().get(headerName); |
| 64 | + // return request.headers().get(headerName); |
| 65 | + HttpHeaders headers = request.headers(); |
| 66 | + List<String> values = new ArrayList<>(); |
| 67 | + for (Map.Entry<String, String> header : headers) { |
| 68 | + values.add(header.getValue()); |
| 69 | + } |
| 70 | + return String.join(", ", values); |
61 | 71 | }
|
62 | 72 |
|
63 | 73 | @Override
|
64 | 74 | protected String getResponseHeader(HttpResponse response, String headerName) {
|
65 |
| - return response.headers().get(headerName); |
| 75 | + // return response.headers().get(headerName); |
| 76 | + HttpHeaders headers = response.headers(); |
| 77 | + List<String> values = new ArrayList<>(); |
| 78 | + for (Map.Entry<String, String> header : headers) { |
| 79 | + values.add(header.getValue()); |
| 80 | + } |
| 81 | + return String.join(", ", values); |
66 | 82 | }
|
67 | 83 | }
|
0 commit comments