Fix for eclipse-paho#19 - deleting chart series

fitrailham · Nov 10, 2017 · defb1d5 · defb1d5
1 parent cbb3850
commit defb1d5
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/spy-common-ui/src/main/java/pl/baczkowicz/spy/ui/controllers/LineChartPaneController.java b/spy-common-ui/src/main/java/pl/baczkowicz/spy/ui/controllers/LineChartPaneController.java
@@ -753,6 +753,8 @@ private void removeSeries()
 		if (itemToDelete != null)
 		{
 			seriesTable.getItems().remove(itemToDelete);
+			allSeries.remove(itemToDelete);
+			unpopulateTopicToSeries(itemToDelete);
 		}
 		else
 		{
@@ -1247,6 +1249,14 @@ private void populateTopicToSeries(final ChartSeriesProperties series)
 		topicToSeries.get(series.getTopic()).add(series);
 	}
 
+	private void unpopulateTopicToSeries(final ChartSeriesProperties series)
+	{
+		if (topicToSeries.get(series.getTopic()) != null)
+		{
+			topicToSeries.remove(series.getTopic());
+		}		
+	}
+
 //	public void setSubscription(MqttSubscription subscription)
 //	{
 //		this.subscription = subscription;