feat(renderer): Add clear method to RendererWorkingSet for producer cleanup

This commit is contained in:
extio-de
2025-12-23 14:38:51 +01:00
parent 411d4c1db2
commit 5282d1eb2f
3 changed files with 77 additions and 10 deletions

View File

@@ -18,6 +18,8 @@ public interface RendererWorkingSet {
Map<String, RenderingBo> commit(Class<? extends AbstractClientModule> producer, boolean clone);
void clear(Class<? extends AbstractClientModule> producer);
void getLiveSet(List<RenderingBo> combinedLiveSet);
}

View File

@@ -74,21 +74,21 @@ public class RendererWorkingSetImpl implements RendererWorkingSet {
final var previousWorkMap = previousWork.get();
if (previousWorkMap != null) {
final var releasedBoClasses = this.obtainSetFromPool();
for (final RenderingBo renderingBO : previousWorkMap.values()) {
if (releasedBoClasses.add(renderingBO.getClass())) {
renderingBO.closeStatic();
}
this.rendererBoPool.release(renderingBO);
}
this.returnMapToPool(previousWorkMap);
this.returnSetToPool(releasedBoClasses);
this.releaseBoClasses(previousWorkMap);
}
return rendererWork.next();
}
@Override
public void clear(final Class<? extends AbstractClientModule> producer) {
final RendererWork rendererWork = this.workingSet.remove(producer);
if (rendererWork != null) {
this.releaseBoClasses(rendererWork.live());
this.releaseBoClasses(rendererWork.next());
}
}
@Override
public void getLiveSet(final List<RenderingBo> combinedLiveSet) {
this.workingSet.values().forEach(rendererWork -> combinedLiveSet.addAll(rendererWork.live().values()));
@@ -98,6 +98,23 @@ public class RendererWorkingSetImpl implements RendererWorkingSet {
return this.workingSet.computeIfAbsent(producer, k -> new RendererWork(this.obtainMapFromPool(), this.obtainMapFromPool()));
}
private void releaseBoClasses(final Map<String, RenderingBo> work) {
if (work == null || work.isEmpty()) {
return;
}
final var releasedBoClasses = this.obtainSetFromPool();
for (final RenderingBo renderingBO : work.values()) {
if (releasedBoClasses.add(renderingBO.getClass())) {
renderingBO.closeStatic();
}
this.rendererBoPool.release(renderingBO);
}
this.returnMapToPool(work);
this.returnSetToPool(releasedBoClasses);
}
private Map<String, RenderingBo> obtainMapFromPool() {
var map = this.mapsPool.poll();
if (map == null) {

View File

@@ -146,6 +146,54 @@ public class RendererWorkingSetImplTest {
assertTrue(live.isEmpty());
}
@Test
void clearRemovesProducerAndReleasesWork() {
final var bo1 = new TestBoA();
bo1.setId("bo1");
final var bo2 = new TestBoB();
bo2.setId("bo2");
this.workingSet.add(TestModuleA.class, bo1);
this.workingSet.commit(TestModuleA.class, false);
this.workingSet.add(TestModuleA.class, bo2);
// Before clear
final List<RenderingBo> liveBefore = new ArrayList<>();
this.workingSet.getLiveSet(liveBefore);
assertTrue(liveBefore.contains(bo1));
assertNotNull(this.workingSet.get(TestModuleA.class, "bo2"));
this.workingSet.clear(TestModuleA.class);
// After clear
final List<RenderingBo> liveAfter = new ArrayList<>();
this.workingSet.getLiveSet(liveAfter);
assertTrue(liveAfter.isEmpty());
final var uncommittedAfter = this.workingSet.getUncommittedWork(TestModuleA.class);
assertTrue(uncommittedAfter.isEmpty());
verify(this.renderingBoPool, times(1)).release(bo1);
verify(this.renderingBoPool, times(1)).release(bo2);
assertEquals(1, bo1.closeStaticCalls());
assertEquals(1, bo2.closeStaticCalls());
}
@Test
void clearOnUnknownProducerDoesNothing() {
final RenderingBo bo1 = mock(RenderingBo.class);
when(bo1.getId()).thenReturn("bo1");
this.workingSet.add(TestModuleA.class, bo1);
this.workingSet.commit(TestModuleA.class, false);
this.workingSet.clear(TestModuleC.class);
final List<RenderingBo> live = new ArrayList<>();
this.workingSet.getLiveSet(live);
assertEquals(1, live.size());
assertTrue(live.contains(bo1));
}
@Test
void secondCommitReleasesPreviousLiveWorkAndCallsCloseStaticOncePerClass() {
final var bo1 = new TestBoA();