Skip to content

Commit

Permalink
rename to NettyConnectionInstrumentationFlag
Browse files Browse the repository at this point in the history
  • Loading branch information
Mateusz Rzeszutek committed Aug 1, 2023
1 parent 1c78bf8 commit 25eb375
Show file tree
Hide file tree
Showing 6 changed files with 20 additions and 19 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -34,16 +34,16 @@ public final class NettyClientInstrumenterFactory {

private final OpenTelemetry openTelemetry;
private final String instrumentationName;
private final NettyInstrumentationFlag connectionTelemetryState;
private final NettyInstrumentationFlag sslTelemetryState;
private final NettyConnectionInstrumentationFlag connectionTelemetryState;
private final NettyConnectionInstrumentationFlag sslTelemetryState;
private final Map<String, String> peerServiceMapping;
private final boolean emitExperimentalHttpClientMetrics;

public NettyClientInstrumenterFactory(
OpenTelemetry openTelemetry,
String instrumentationName,
NettyInstrumentationFlag connectionTelemetryState,
NettyInstrumentationFlag sslTelemetryState,
NettyConnectionInstrumentationFlag connectionTelemetryState,
NettyConnectionInstrumentationFlag sslTelemetryState,
Map<String, String> peerServiceMapping,
boolean emitExperimentalHttpClientMetrics) {
this.openTelemetry = openTelemetry;
Expand Down Expand Up @@ -83,12 +83,12 @@ public Instrumenter<HttpRequestAndChannel, HttpResponse> createHttpInstrumenter(
}

public NettyConnectionInstrumenter createConnectionInstrumenter() {
if (connectionTelemetryState == NettyInstrumentationFlag.DISABLED) {
if (connectionTelemetryState == NettyConnectionInstrumentationFlag.DISABLED) {
return NoopConnectionInstrumenter.INSTANCE;
}

boolean connectionTelemetryFullyEnabled =
connectionTelemetryState == NettyInstrumentationFlag.ENABLED;
connectionTelemetryState == NettyConnectionInstrumentationFlag.ENABLED;

Instrumenter<NettyConnectionRequest, Channel> instrumenter =
Instrumenter.<NettyConnectionRequest, Channel>builder(
Expand All @@ -109,11 +109,12 @@ public NettyConnectionInstrumenter createConnectionInstrumenter() {
}

public NettySslInstrumenter createSslInstrumenter() {
if (sslTelemetryState == NettyInstrumentationFlag.DISABLED) {
if (sslTelemetryState == NettyConnectionInstrumentationFlag.DISABLED) {
return NoopSslInstrumenter.INSTANCE;
}

boolean sslTelemetryFullyEnabled = sslTelemetryState == NettyInstrumentationFlag.ENABLED;
boolean sslTelemetryFullyEnabled =
sslTelemetryState == NettyConnectionInstrumentationFlag.ENABLED;
NettySslNetAttributesGetter netAttributesGetter = new NettySslNetAttributesGetter();
Instrumenter<NettySslRequest, Void> instrumenter =
Instrumenter.<NettySslRequest, Void>builder(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,12 @@
* This class is internal and is hence not for public use. Its APIs are unstable and can change at
* any time.
*/
public enum NettyInstrumentationFlag {
public enum NettyConnectionInstrumentationFlag {
ENABLED,
ERROR_ONLY,
DISABLED;

public static NettyInstrumentationFlag enabledOrErrorOnly(boolean b) {
public static NettyConnectionInstrumentationFlag enabledOrErrorOnly(boolean b) {
return b ? ENABLED : ERROR_ONLY;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

package io.opentelemetry.javaagent.instrumentation.netty.v4_0.client;

import static io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyInstrumentationFlag.enabledOrErrorOnly;
import static io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyConnectionInstrumentationFlag.enabledOrErrorOnly;

import io.netty.handler.codec.http.HttpResponse;
import io.opentelemetry.api.GlobalOpenTelemetry;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

package io.opentelemetry.javaagent.instrumentation.netty.v4_1;

import static io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyInstrumentationFlag.enabledOrErrorOnly;
import static io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyConnectionInstrumentationFlag.enabledOrErrorOnly;

import io.netty.handler.codec.http.HttpResponse;
import io.opentelemetry.api.GlobalOpenTelemetry;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
import io.opentelemetry.instrumentation.api.instrumenter.http.HttpClientAttributesExtractorBuilder;
import io.opentelemetry.instrumentation.netty.v4.common.HttpRequestAndChannel;
import io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyClientInstrumenterFactory;
import io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyInstrumentationFlag;
import io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyConnectionInstrumentationFlag;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
Expand Down Expand Up @@ -112,8 +112,8 @@ public NettyClientTelemetry build() {
new NettyClientInstrumenterFactory(
openTelemetry,
"io.opentelemetry.netty-4.1",
NettyInstrumentationFlag.DISABLED,
NettyInstrumentationFlag.DISABLED,
NettyConnectionInstrumentationFlag.DISABLED,
NettyConnectionInstrumentationFlag.DISABLED,
Collections.emptyMap(),
emitExperimentalHttpClientMetrics)
.createHttpInstrumenter(extractorConfigurer, additionalAttributesExtractors));
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,8 @@
import io.opentelemetry.instrumentation.api.instrumenter.http.HttpSpanStatusExtractor;
import io.opentelemetry.instrumentation.api.instrumenter.net.PeerServiceAttributesExtractor;
import io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyClientInstrumenterFactory;
import io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyConnectionInstrumentationFlag;
import io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyConnectionInstrumenter;
import io.opentelemetry.instrumentation.netty.v4.common.internal.client.NettyInstrumentationFlag;
import io.opentelemetry.javaagent.bootstrap.internal.CommonConfig;
import io.opentelemetry.javaagent.bootstrap.internal.DeprecatedConfigProperties;
import io.opentelemetry.javaagent.bootstrap.internal.InstrumentationConfig;
Expand Down Expand Up @@ -72,9 +72,9 @@ public final class ReactorNettySingletons {
GlobalOpenTelemetry.get(),
INSTRUMENTATION_NAME,
connectionTelemetryEnabled
? NettyInstrumentationFlag.ENABLED
: NettyInstrumentationFlag.DISABLED,
NettyInstrumentationFlag.DISABLED,
? NettyConnectionInstrumentationFlag.ENABLED
: NettyConnectionInstrumentationFlag.DISABLED,
NettyConnectionInstrumentationFlag.DISABLED,
CommonConfig.get().getPeerServiceMapping(),
CommonConfig.get().shouldEmitExperimentalHttpClientMetrics());
CONNECTION_INSTRUMENTER = instrumenterFactory.createConnectionInstrumenter();
Expand Down

0 comments on commit 25eb375

Please sign in to comment.