open-telemetry
/
opentelemetry-java-instrumentation
mirror of https://github.com/open-telemetry/opentelemetry-java-instrumentation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Refactored spymemcached (#67) 

* Refactored spymemcached

* Fixed formatting issues

* Fixed formatting in an accidentally changed file

Co-authored-by: Trask Stalnaker <trask.stalnaker@gmail.com>
This commit is contained in:
Pontus Rydin 2020-01-16 22:36:32 -05:00 committed by Trask Stalnaker
parent 1d474353a5
commit 736c70ffe0
6 changed files with 26 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
java-agent/instrumentation/spymemcached-2.12/src/main/java/io/opentelemetry/auto/instrumentation/spymemcached/BulkGetCompletionListener.java
View File

@ -1,6 +1,6 @@
package io.opentelemetry.auto.instrumentation.spymemcached;
import io.opentelemetry.auto.instrumentation.api.AgentSpan;
import io.opentelemetry.trace.Span;
import java.util.concurrent.ExecutionException;
import net.spy.memcached.MemcachedConnection;
import net.spy.memcached.internal.BulkGetFuture;
@ -18,7 +18,7 @@ public class BulkGetCompletionListener extends CompletionListener<BulkGetFuture<
}
@Override
protected void processResult(final AgentSpan span, final BulkGetFuture<?> future)
protected void processResult(final Span span, final BulkGetFuture<?> future)
throws ExecutionException, InterruptedException {
/*
Note: for now we do not have an affective way of representing results of bulk operations,

25
java-agent/instrumentation/spymemcached-2.12/src/main/java/io/opentelemetry/auto/instrumentation/spymemcached/CompletionListener.java
View File

@ -1,11 +1,10 @@
package io.opentelemetry.auto.instrumentation.spymemcached;
import static io.opentelemetry.auto.instrumentation.api.AgentTracer.activateSpan;
import static io.opentelemetry.auto.instrumentation.api.AgentTracer.startSpan;
import static io.opentelemetry.auto.instrumentation.spymemcached.MemcacheClientDecorator.DECORATE;
import static io.opentelemetry.auto.instrumentation.spymemcached.MemcacheClientDecorator.TRACER;
import io.opentelemetry.auto.instrumentation.api.AgentScope;
import io.opentelemetry.auto.instrumentation.api.AgentSpan;
import io.opentelemetry.context.Scope;
import io.opentelemetry.trace.Span;
import java.util.concurrent.CancellationException;
import java.util.concurrent.ExecutionException;
import lombok.extern.slf4j.Slf4j;
@ -26,12 +25,12 @@ public abstract class CompletionListener<T> {
static final String MISS = "miss";
private final MemcachedConnection connection;
private final AgentSpan span;
private final Span span;
public CompletionListener(final MemcachedConnection connection, final String methodName) {
this.connection = connection;
span = startSpan(OPERATION_NAME);
try (final AgentScope scope = activateSpan(span, false)) {
span = TRACER.spanBuilder(OPERATION_NAME).startSpan();
try (final Scope scope = TRACER.withSpan(span)) {
DECORATE.afterStart(span);
DECORATE.onConnection(span, connection);
DECORATE.onOperation(span, methodName);
@ -39,7 +38,7 @@ public abstract class CompletionListener<T> {
}
protected void closeAsyncSpan(final T future) {
try (final AgentScope scope = activateSpan(span, false)) {
try (final Scope scope = TRACER.withSpan(span)) {
try {
processResult(span, future);
} catch (final CancellationException e) {
@ -61,23 +60,23 @@ public abstract class CompletionListener<T> {
DECORATE.onError(span, e);
} finally {
DECORATE.beforeFinish(span);
span.finish();
span.end();
}
}
}
protected void closeSyncSpan(final Throwable thrown) {
try (final AgentScope scope = activateSpan(span, false)) {
try (final Scope scope = TRACER.withSpan(span)) {
DECORATE.onError(span, thrown);
DECORATE.beforeFinish(span);
span.finish();
span.end();
}
}
protected abstract void processResult(AgentSpan span, T future)
protected abstract void processResult(Span span, T future)
throws ExecutionException, InterruptedException;
protected void setResultTag(final AgentSpan span, final boolean hit) {
protected void setResultTag(final Span span, final boolean hit) {
span.setAttribute(MEMCACHED_RESULT, hit ? HIT : MISS);
}
}

4
java-agent/instrumentation/spymemcached-2.12/src/main/java/io/opentelemetry/auto/instrumentation/spymemcached/GetCompletionListener.java
View File

@ -1,6 +1,6 @@
package io.opentelemetry.auto.instrumentation.spymemcached;
import io.opentelemetry.auto.instrumentation.api.AgentSpan;
import io.opentelemetry.trace.Span;
import java.util.concurrent.ExecutionException;
import net.spy.memcached.MemcachedConnection;
import net.spy.memcached.internal.GetFuture;
@ -17,7 +17,7 @@ public class GetCompletionListener extends CompletionListener<GetFuture<?>>
}
@Override
protected void processResult(final AgentSpan span, final GetFuture<?> future)
protected void processResult(final Span span, final GetFuture<?> future)
throws ExecutionException, InterruptedException {
final Object result = future.get();
setResultTag(span, result != null);

8
java-agent/instrumentation/spymemcached-2.12/src/main/java/io/opentelemetry/auto/instrumentation/spymemcached/MemcacheClientDecorator.java
View File

@ -1,14 +1,18 @@
package io.opentelemetry.auto.instrumentation.spymemcached;
import io.opentelemetry.OpenTelemetry;
import io.opentelemetry.auto.api.MoreTags;
import io.opentelemetry.auto.api.SpanTypes;
import io.opentelemetry.auto.decorator.DatabaseClientDecorator;
import io.opentelemetry.auto.instrumentation.api.AgentSpan;
import io.opentelemetry.trace.Span;
import io.opentelemetry.trace.Tracer;
import net.spy.memcached.MemcachedConnection;
public class MemcacheClientDecorator extends DatabaseClientDecorator<MemcachedConnection> {
public static final MemcacheClientDecorator DECORATE = new MemcacheClientDecorator();
public static final Tracer TRACER = OpenTelemetry.getTracerFactory().get("io.opentelemetry.auto");
@Override
protected String[] instrumentationNames() {
return new String[] {"spymemcached"};
@ -44,7 +48,7 @@ public class MemcacheClientDecorator extends DatabaseClientDecorator<MemcachedCo
return null;
}
public AgentSpan onOperation(final AgentSpan span, final String methodName) {
public Span onOperation(final Span span, final String methodName) {
final char[] chars =
methodName

4
java-agent/instrumentation/spymemcached-2.12/src/main/java/io/opentelemetry/auto/instrumentation/spymemcached/OperationCompletionListener.java
View File

@ -1,6 +1,6 @@
package io.opentelemetry.auto.instrumentation.spymemcached;
import io.opentelemetry.auto.instrumentation.api.AgentSpan;
import io.opentelemetry.trace.Span;
import java.util.concurrent.ExecutionException;
import net.spy.memcached.MemcachedConnection;
import net.spy.memcached.internal.OperationFuture;
@ -19,7 +19,7 @@ public class OperationCompletionListener
}
@Override
protected void processResult(final AgentSpan span, final OperationFuture<? extends Object> future)
protected void processResult(final Span span, final OperationFuture<? extends Object> future)
throws ExecutionException, InterruptedException {
future.get();
}

4
java-agent/instrumentation/spymemcached-2.12/src/main/java/io/opentelemetry/auto/instrumentation/spymemcached/SyncCompletionListener.java
View File

@ -1,6 +1,6 @@
package io.opentelemetry.auto.instrumentation.spymemcached;
import io.opentelemetry.auto.instrumentation.api.AgentSpan;
import io.opentelemetry.trace.Span;
import java.util.concurrent.ExecutionException;
import lombok.extern.slf4j.Slf4j;
import net.spy.memcached.MemcachedConnection;
@ -12,7 +12,7 @@ public class SyncCompletionListener extends CompletionListener<Void> {
}
@Override
protected void processResult(final AgentSpan span, final Void future)
protected void processResult(final Span span, final Void future)
throws ExecutionException, InterruptedException {
log.error("processResult was called on SyncCompletionListener. This should never happen. ");
}