infobox manager: make threadsafe

Currently this gets called from multiple threads, mostly client and EDT,
causing unsafe acces to infoBoxes.
This commit is contained in:
Adam
2020-05-24 15:22:25 -04:00
parent cb1fc598ba
commit ba9ffb1d60
3 changed files with 114 additions and 37 deletions

View File

@@ -26,6 +26,7 @@ package net.runelite.client.ui.overlay.infobox;
import java.awt.Color; import java.awt.Color;
import java.awt.image.BufferedImage; import java.awt.image.BufferedImage;
import javax.annotation.Nonnull;
import lombok.AccessLevel; import lombok.AccessLevel;
import lombok.Getter; import lombok.Getter;
import lombok.Setter; import lombok.Setter;
@@ -33,6 +34,7 @@ import net.runelite.client.plugins.Plugin;
public abstract class InfoBox public abstract class InfoBox
{ {
@Nonnull
@Getter(AccessLevel.PACKAGE) @Getter(AccessLevel.PACKAGE)
private final Plugin plugin; private final Plugin plugin;
@@ -52,7 +54,7 @@ public abstract class InfoBox
@Setter @Setter
private String tooltip; private String tooltip;
public InfoBox(BufferedImage image, Plugin plugin) public InfoBox(BufferedImage image, @Nonnull Plugin plugin)
{ {
this.plugin = plugin; this.plugin = plugin;
setImage(image); setImage(image);

View File

@@ -28,10 +28,9 @@ import com.google.common.base.Preconditions;
import com.google.common.collect.ComparisonChain; import com.google.common.collect.ComparisonChain;
import java.awt.Graphics; import java.awt.Graphics;
import java.awt.image.BufferedImage; import java.awt.image.BufferedImage;
import java.util.ArrayList;
import java.util.Collections; import java.util.Collections;
import java.util.Iterator;
import java.util.List; import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.function.Predicate; import java.util.function.Predicate;
import javax.inject.Inject; import javax.inject.Inject;
import javax.inject.Singleton; import javax.inject.Singleton;
@@ -45,7 +44,7 @@ import net.runelite.client.util.AsyncBufferedImage;
@Slf4j @Slf4j
public class InfoBoxManager public class InfoBoxManager
{ {
private final List<InfoBox> infoBoxes = new ArrayList<>(); private final List<InfoBox> infoBoxes = new CopyOnWriteArrayList<>();
private final RuneLiteConfig runeLiteConfig; private final RuneLiteConfig runeLiteConfig;
@Inject @Inject
@@ -69,8 +68,19 @@ public class InfoBoxManager
log.debug("Adding InfoBox {}", infoBox); log.debug("Adding InfoBox {}", infoBox);
updateInfoBoxImage(infoBox); updateInfoBoxImage(infoBox);
infoBoxes.add(infoBox);
refreshInfoBoxes(); synchronized (this)
{
int idx = Collections.binarySearch(infoBoxes, infoBox, (b1, b2) -> ComparisonChain
.start()
.compare(b1.getPriority(), b2.getPriority())
.compare(b1.getPlugin().getName(), b2.getPlugin().getName())
.result());
if (idx < 0)
{
infoBoxes.add(-idx - 1, infoBox);
}
}
BufferedImage image = infoBox.getImage(); BufferedImage image = infoBox.getImage();
@@ -81,21 +91,19 @@ public class InfoBoxManager
} }
} }
public void removeInfoBox(InfoBox infoBox) public synchronized void removeInfoBox(InfoBox infoBox)
{ {
if (infoBoxes.remove(infoBox)) if (infoBoxes.remove(infoBox))
{ {
log.debug("Removed InfoBox {}", infoBox); log.debug("Removed InfoBox {}", infoBox);
refreshInfoBoxes();
} }
} }
public void removeIf(Predicate<InfoBox> filter) public synchronized void removeIf(Predicate<InfoBox> filter)
{ {
if (infoBoxes.removeIf(filter)) if (infoBoxes.removeIf(filter))
{ {
log.debug("Removed InfoBoxes for filter {}", filter); log.debug("Removed InfoBoxes for filter {}", filter);
refreshInfoBoxes();
} }
} }
@@ -104,25 +112,9 @@ public class InfoBoxManager
return Collections.unmodifiableList(infoBoxes); return Collections.unmodifiableList(infoBoxes);
} }
public void cull() public synchronized void cull()
{ {
boolean culled = false; infoBoxes.removeIf(InfoBox::cull);
for (Iterator<InfoBox> it = infoBoxes.iterator(); it.hasNext();)
{
InfoBox box = it.next();
if (box.cull())
{
log.debug("Culling InfoBox {}", box);
it.remove();
culled = true;
}
}
if (culled)
{
refreshInfoBoxes();
}
} }
public void updateInfoBoxImage(final InfoBox infoBox) public void updateInfoBoxImage(final InfoBox infoBox)
@@ -161,13 +153,4 @@ public class InfoBoxManager
infoBox.setScaledImage(resultImage); infoBox.setScaledImage(resultImage);
} }
private void refreshInfoBoxes()
{
infoBoxes.sort((b1, b2) -> ComparisonChain
.start()
.compare(b1.getPriority(), b2.getPriority())
.compare(b1.getPlugin().getName(), b2.getPlugin().getName())
.result());
}
} }

View File

@@ -0,0 +1,92 @@
/*
* Copyright (c) 2020, Adam <Adam@sigterm.info>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package net.runelite.client.ui.overlay.infobox;
import com.google.inject.Guice;
import com.google.inject.testing.fieldbinder.Bind;
import com.google.inject.testing.fieldbinder.BoundFieldModule;
import java.awt.Color;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import javax.inject.Inject;
import net.runelite.client.config.RuneLiteConfig;
import net.runelite.client.plugins.Plugin;
import static org.junit.Assert.assertEquals;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import static org.mockito.Mockito.mock;
import org.mockito.junit.MockitoJUnitRunner;
@RunWith(MockitoJUnitRunner.class)
public class InfoBoxManagerTest
{
@Inject
private InfoBoxManager infoBoxManager;
@Mock
@Bind
private RuneLiteConfig runeLiteConfig;
@Before
public void before()
{
Guice.createInjector(BoundFieldModule.of(this)).injectMembers(this);
}
private static class TestInfobox extends InfoBox
{
public TestInfobox(InfoBoxPriority infoBoxPriority)
{
super(null, mock(Plugin.class));
setPriority(infoBoxPriority);
}
@Override
public String getText()
{
return null;
}
@Override
public Color getTextColor()
{
return null;
}
}
@Test
public void testSorting()
{
infoBoxManager.addInfoBox(new TestInfobox(InfoBoxPriority.LOW));
infoBoxManager.addInfoBox(new TestInfobox(InfoBoxPriority.HIGH));
infoBoxManager.addInfoBox(new TestInfobox(InfoBoxPriority.MED));
List<InfoBoxPriority> order = infoBoxManager.getInfoBoxes().stream().map(InfoBox::getPriority).collect(Collectors.toList());
assertEquals(Arrays.asList(InfoBoxPriority.HIGH, InfoBoxPriority.MED, InfoBoxPriority.LOW), order);
}
}