Fix Netty addListener instrumentation (#10254)
This commit is contained in:
parent
e28f0dfcf2
commit
9e208c8574
|
@ -43,7 +43,7 @@ public final class FutureListenerWrappers {
|
||||||
};
|
};
|
||||||
|
|
||||||
public static boolean shouldWrap(GenericFutureListener<? extends Future<?>> listener) {
|
public static boolean shouldWrap(GenericFutureListener<? extends Future<?>> listener) {
|
||||||
return shouldWrap.get(listener.getClass());
|
return listener != null && shouldWrap.get(listener.getClass());
|
||||||
}
|
}
|
||||||
|
|
||||||
@SuppressWarnings("unchecked")
|
@SuppressWarnings("unchecked")
|
||||||
|
|
|
@ -82,6 +82,8 @@ public class NettyFutureInstrumentation implements TypeInstrumentation {
|
||||||
for (int i = 0; i < listeners.length; ++i) {
|
for (int i = 0; i < listeners.length; ++i) {
|
||||||
if (FutureListenerWrappers.shouldWrap(listeners[i])) {
|
if (FutureListenerWrappers.shouldWrap(listeners[i])) {
|
||||||
wrappedListeners[i] = FutureListenerWrappers.wrap(context, listeners[i]);
|
wrappedListeners[i] = FutureListenerWrappers.wrap(context, listeners[i]);
|
||||||
|
} else {
|
||||||
|
wrappedListeners[i] = listeners[i];
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
listeners = wrappedListeners;
|
listeners = wrappedListeners;
|
||||||
|
|
Loading…
Reference in New Issue