Fix the valueType of in memory format

Signed-off-by: Fabio José <fabiojose@gmail.com>
This commit is contained in:
Fabio José 2019-08-30 08:50:22 -03:00
parent ecd250a698
commit 15909cd087
1 changed files with 4 additions and 2 deletions

View File

@ -84,7 +84,8 @@ public class DistributedTracingExtension {
public Format(DistributedTracingExtension extension) {
Objects.requireNonNull(extension);
memory = InMemoryFormat.of(IN_MEMORY_KEY, extension, Object.class);
memory = InMemoryFormat.of(IN_MEMORY_KEY, extension,
DistributedTracingExtension.class);
transport.put("traceparent", extension.getTraceparent());
transport.put("tracestate", extension.getTracestate());
@ -117,7 +118,8 @@ public class DistributedTracingExtension {
dte.setTracestate(tracestate);
InMemoryFormat inMemory =
InMemoryFormat.of("distributedTracing", dte, Object.class);
InMemoryFormat.of("distributedTracing", dte,
DistributedTracingExtension.class);
return Optional.of(
ExtensionFormat.of(inMemory,