Spring/spring-pulsar
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fully clear reactive sender cache on Lifecycle stop

Browse Source
This commit is contained in:
Chris Bono
2023-10-15 21:52:23 -05:00
parent a2646d9a68
commit df00542990
2 changed files with 34 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gradle/aggregate-jacoco-report.gradle
View File

@@ -12,6 +12,8 @@ project.afterEvaluate {
tasks.create(name: 'aggregateJacocoTestReport', type: JacocoReport) {
dependsOn ':spring-pulsar-cache-provider:compileJava', ':spring-pulsar-cache-provider-caffeine:compileJava'
group = 'verification'
description = 'Generates aggregate code coverage report for all projects test tasks'

32
spring-pulsar-reactive/src/main/java/org/springframework/pulsar/reactive/core/DefaultReactivePulsarSenderFactory.java
View File

@@ -35,6 +35,7 @@ import org.springframework.pulsar.core.DefaultTopicResolver;
import org.springframework.pulsar.core.TopicResolver;
import org.springframework.util.Assert;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ReflectionUtils;
/**
* Default implementation of {@link ReactivePulsarSenderFactory}.
@@ -172,13 +173,44 @@ public final class DefaultReactivePulsarSenderFactory<T>
@Override
public void doStop() {
try {
this.reflectivelyClearCache();
this.reactiveMessageSenderCache.close();
}
catch (Exception e) {
throw new RuntimeException(e);
}
}
/**
* Workaround to reflectively clear the underlying producer cache.
*
* TODO: Remove once this is supported in the Reactive client.
*/
private void reflectivelyClearCache() {
// reactiveMessageSenderCache
// (org.apache.pulsar.reactive.client.internal.adapter.ProducerCache)
var cacheProviderField = ReflectionUtils.findField(this.reactiveMessageSenderCache.getClass(), "cacheProvider");
ReflectionUtils.makeAccessible(cacheProviderField);
// org.apache.pulsar.reactive.client.producercache.CaffeineShadedProducerCacheProvider
var cacheProvider = ReflectionUtils.getField(cacheProviderField, this.reactiveMessageSenderCache);
// org.apache.pulsar.reactive.shade.com.github.benmanes.caffeine.cache.BoundedLocalCache$BoundedLocalAsyncCache
var cacheField = ReflectionUtils.findField(cacheProvider.getClass(), "cache");
ReflectionUtils.makeAccessible(cacheField);
var cache = ReflectionUtils.getField(cacheField, cacheProvider);
// org.apache.pulsar.reactive.shade.com.github.benmanes.caffeine.cache.SSLMSAW
var actualCacheField = ReflectionUtils.findField(cache.getClass(), "cache");
ReflectionUtils.makeAccessible(actualCacheField);
var actualCache = ReflectionUtils.getField(actualCacheField, cache);
var clearMethod = ReflectionUtils.findMethod(actualCache.getClass(), "clear");
ReflectionUtils.makeAccessible(clearMethod);
ReflectionUtils.invokeMethod(clearMethod, actualCache);
}
/**
* Builder for {@link DefaultReactivePulsarSenderFactory}.
*