From fc8d38185b024d3fe2858b46610e586697e57eb6 Mon Sep 17 00:00:00 2001
From: Nikita Salnikov-Tarnovski <gnikem@gmail.com>
Date: Thu, 10 Sep 2020 20:45:19 +0300
Subject: [PATCH] Fix muzzle

---
 .../auto/reactor/ReactorHooksInstrumentation.java      | 10 ++++++++++
 .../webflux/client/WebClientFilterInstrumentation.java |  4 +++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/instrumentation/reactor-3.1/src/main/java/io/opentelemetry/instrumentation/auto/reactor/ReactorHooksInstrumentation.java b/instrumentation/reactor-3.1/src/main/java/io/opentelemetry/instrumentation/auto/reactor/ReactorHooksInstrumentation.java
index db59e2b6b0..51fc687c77 100644
--- a/instrumentation/reactor-3.1/src/main/java/io/opentelemetry/instrumentation/auto/reactor/ReactorHooksInstrumentation.java
+++ b/instrumentation/reactor-3.1/src/main/java/io/opentelemetry/instrumentation/auto/reactor/ReactorHooksInstrumentation.java
@@ -34,6 +34,16 @@ public final class ReactorHooksInstrumentation extends Instrumenter.Default {
     super("reactor-core");
   }
 
+  @Override
+  public String[] helperClassNames() {
+    return new String[] {
+        "io.opentelemetry.instrumentation.reactor.TracingOperator$Lifter",
+        "io.opentelemetry.instrumentation.reactor.TracingOperator",
+        "io.opentelemetry.instrumentation.reactor.TracingSubscriber"
+    };
+  }
+
+
   @Override
   public ElementMatcher<TypeDescription> typeMatcher() {
     return named("reactor.core.publisher.Hooks");
diff --git a/instrumentation/spring/spring-webflux-5.0/auto/src/main/java/io/opentelemetry/instrumentation/auto/spring/webflux/client/WebClientFilterInstrumentation.java b/instrumentation/spring/spring-webflux-5.0/auto/src/main/java/io/opentelemetry/instrumentation/auto/spring/webflux/client/WebClientFilterInstrumentation.java
index 8992adaab8..05440f5228 100644
--- a/instrumentation/spring/spring-webflux-5.0/auto/src/main/java/io/opentelemetry/instrumentation/auto/spring/webflux/client/WebClientFilterInstrumentation.java
+++ b/instrumentation/spring/spring-webflux-5.0/auto/src/main/java/io/opentelemetry/instrumentation/auto/spring/webflux/client/WebClientFilterInstrumentation.java
@@ -48,7 +48,9 @@ public class WebClientFilterInstrumentation extends Instrumenter.Default {
     return new String[] {
       "io.opentelemetry.instrumentation.spring.webflux.client.SpringWebfluxHttpClientTracer",
       "io.opentelemetry.instrumentation.spring.webflux.client.HttpHeadersInjectAdapter",
-      "io.opentelemetry.instrumentation.spring.webflux.client.WebClientTracingFilter"
+      "io.opentelemetry.instrumentation.spring.webflux.client.WebClientTracingFilter",
+      "io.opentelemetry.instrumentation.spring.webflux.client.WebClientTracingFilter$MonoWebClientTrace",
+      "io.opentelemetry.instrumentation.spring.webflux.client.TraceWebClientSubscriber"
     };
   }