runelite-client: add item manager for caching item prices

runelite-client/src/main/java/net/runelite/client/ItemManager.java

@@ -0,0 +1,127 @@
+/*
+ * Copyright (c) 2017, 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;
+
+import com.google.common.cache.CacheBuilder;
+import com.google.common.cache.CacheLoader;
+import com.google.common.cache.LoadingCache;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.common.util.concurrent.ListeningExecutorService;
+import com.google.common.util.concurrent.MoreExecutors;
+import java.io.IOException;
+import java.util.concurrent.TimeUnit;
+import net.runelite.http.api.item.ItemClient;
+import net.runelite.http.api.item.ItemPrice;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class ItemManager
+{
+	private static class ItemLoader extends CacheLoader<Integer, ItemPrice>
+	{
+		private static final Logger logger = LoggerFactory.getLogger(ItemLoader.class);
+
+		private final ListeningExecutorService executorService;
+		private final ItemClient client = new ItemClient();
+
+		ItemLoader(RuneLite runelite)
+		{
+			executorService = MoreExecutors.listeningDecorator(runelite.getExecutor());
+		}
+
+		@Override
+		public ItemPrice load(Integer key) throws Exception
+		{
+			// guava's Cache doesn't support null values
+			return EMPTY;
+		}
+
+		@Override
+		public ListenableFuture<ItemPrice> reload(Integer key, ItemPrice oldValue)
+		{
+			logger.debug("Submitting lookup for item {}", key);
+
+			return executorService.submit(() -> fetch(key));
+		}
+
+		private ItemPrice fetch(Integer key)
+		{
+			try
+			{
+				return client.lookupItemPrice(key);
+			}
+			catch (IOException ex)
+			{
+				logger.warn("unable to look up item!", ex);
+				return null;
+			}
+		}
+	};
+
+	private final LoadingCache<Integer, ItemPrice> itemPrices;
+	private static final ItemPrice EMPTY = new ItemPrice();
+
+	public ItemManager(RuneLite runelite)
+	{
+		itemPrices = CacheBuilder.newBuilder()
+			.concurrencyLevel(4) // from runelite's ScheduledExecutorService
+			.maximumSize(512L)
+			.expireAfterAccess(1, TimeUnit.HOURS)
+			.build(new ItemLoader(runelite));
+	}
+
+	public ItemPrice get(int itemId)
+	{
+		ItemPrice itemPrice = itemPrices.getIfPresent(itemId);
+		if (itemPrice != null && itemPrice != EMPTY)
+		{
+			return itemPrice;
+		}
+
+		itemPrices.refresh(itemId);
+		return null;
+	}
+
+	/**
+	 * Convert a quantity to stack size
+	 *
+	 * @param quantity
+	 * @return
+	 */
+	public static String quantityToStackSize(int quantity)
+	{
+		if (quantity >= 10_000_000)
+		{
+			return quantity / 1_000_000 + "M";
+		}
+
+		if (quantity >= 100_000)
+		{
+			return quantity / 1_000 + "K";
+		}
+
+		return "" + quantity;
+	}
+}

runelite-client/src/main/java/net/runelite/client/RuneLite.java

@@ -79,6 +79,7 @@ public class RuneLite
 
 	private AccountSession accountSession;
 	private final ConfigManager configManager = new ConfigManager(eventBus);
+	private final ItemManager itemManager = new ItemManager(this);
 
 	static
 	{
@@ -325,13 +326,18 @@ public class RuneLite
 		return trayIcon;
 	}
 
+	public AccountSession getAccountSession()
+	{
+		return accountSession;
+	}
+
 	public ConfigManager getConfigManager()
 	{
 		return configManager;
 	}
 
-	public AccountSession getAccountSession()
+	public ItemManager getItemManager()
 	{
-		return accountSession;
+		return itemManager;
 	}
 }