VehicleLocations.java

package com.schneeloch.bostonbusmap_library.data;

import java.util.Collection;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import com.google.common.base.Objects;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Sets;
import com.schneeloch.bostonbusmap_library.database.Schema;

import javax.xml.transform.Source;

public class VehicleLocations {
    public static class Key implements GroupKey {
		private final Schema.Routes.SourceId transitSourceId;
        private final String routeName;
		private final String id;
		
		public Key(Schema.Routes.SourceId transitSourceId, String routeName, String id) {
			this.transitSourceId = transitSourceId;
            this.routeName = routeName;
			this.id = id;
		}
		
		@Override
		public int hashCode() {
			return Objects.hashCode(transitSourceId, routeName, id);
		}
		
		@Override
		public boolean equals(Object obj) {
			if (obj instanceof Key) {
				Key other = (Key)obj;
				return Objects.equal(this.transitSourceId, other.transitSourceId) &&
                        Objects.equal(this.routeName, other.routeName) &&
						Objects.equal(this.id, other.id);
			}
			return false;
		}
	}
	
	private final ConcurrentHashMap<Key, BusLocation> locations = new ConcurrentHashMap<Key, BusLocation>();

    public void update(Schema.Routes.SourceId transitSourceId, ImmutableSet<String> routeNames, boolean incremental, Map<Key, BusLocation> newItems) {
        Set<VehicleLocations.Key> toRemove = Sets.newHashSet();

        for (VehicleLocations.Key key : locations.keySet()) {
            if (key.transitSourceId == transitSourceId && locations.containsKey(key)) {
                if (!incremental) {
                    if (routeNames.contains(key.routeName)) {
                        toRemove.add(key);
                    }
                }
            }
        }

        for (VehicleLocations.Key key : toRemove) {
            locations.remove(key);
        }

        locations.putAll(newItems);
    }

	public BusLocation get(VehicleLocations.Key id) {
		return locations.get(id);
	}

	public Collection<BusLocation> values() {
		return locations.values();
	}

	public boolean containsKey(Key key) {
		return locations.containsKey(key);
	}

	public Set<VehicleLocations.Key> keySet() {
		return locations.keySet();
	}
}