netty-reactor instrumentation leaks Token on cancellation

[chringwer]

Description

After upgrading the NR java agent to 8.12.0 we observed much higher GC pressure than before. Investigating some heap dumps revealed an issue with the reactor-netty instrumentation module which seems to not handle cancellations properly in some cases so that tokens are not expired timely.

In particular FluxMapFuseable_Instrumentation needs to expire the token when cancelled which is a pretty common scenario for example when doing .flatMap(...).next(). LambdaSubscriber_Instrumentation and LambdaMonoSubscriber_Instrumentation should handle dispose() and expire their tokens accordingly, I guess.

For us, patching FluxMapFuseable_Instrumentation with a cancel() implementation resolved the memory issues:

    @Weave(type = MatchType.ExactClass, originalName = "reactor.core.publisher.FluxMapFuseable$MapFuseableSubscriber")
    static final class MapFuseableSubscriber_Instrumentation<T, R> {
     ...
        public void cancel() {
            if (token != null) {
                token.linkAndExpire();
                token = null;
            }
            Weaver.callOriginal();
        }
     ...
    }

Expected Behavior

Cancelling an instrumented publisher should not leak any objects

Troubleshooting or NR Diag results

Steps to Reproduce

Your Environment

Java 17, NR agent 8.12.0

Additional context

[workato-integration[bot]]

https://new-relic.atlassian.net/browse/NR-288379