MasterContext.java

package cgl.iotcloud.core.master;

import cgl.iotcloud.core.Configuration;
import cgl.iotcloud.core.api.thrift.TSensor;
import cgl.iotcloud.core.api.thrift.TSite;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.*;

public class MasterContext {
    private static Logger LOG = LoggerFactory.getLogger(MasterContext.class);

    private Map<String, TSite> sites = new HashMap<String, TSite>();

    private Map<String, List<TSensor>> siteSensors = new HashMap<String, List<TSensor>>();

    private Map<String, List<TSensor>> deactivatedSiteSensors = new HashMap<String, List<TSensor>>();

    private Map<String, TSite> deactivatedSites = new HashMap<String, TSite>();

    private Map conf;

    public MasterContext(Map conf) {
        this.conf = conf;
    }

    public void addSensorSite(TSite site) {
        sites.put(site.getSiteId(), site);
    }

    public TSite getSensorSite(String siteId) {
        return sites.get(siteId);
    }

    public Map<String, TSite> getSensorSites() {
        return sites;
    }

    public boolean addSensor(String site, TSensor details) {
        if (!sites.containsKey(site)) {
            LOG.warn("Adding a sensor to a site not registered siteId: {} sensorName: {}", site, details.getName());
            return false;
        }

        List<TSensor> detailsList = siteSensors.get(site);
        if (detailsList == null) {
            detailsList = new ArrayList<TSensor>();
            siteSensors.put(site, detailsList);
        }

        detailsList.add(details);

        return true;
    }

    public void makeSiteOffline(String site) {
        if (!siteSensors.containsKey(site)) {
            TSite siteDescriptor = sites.get(site);
            deactivatedSites.put(site, siteDescriptor);
            deactivatedSiteSensors.put(site, siteSensors.get(site));

            sites.remove(site);
            siteSensors.remove(site);
        }
    }

    public boolean removeSensor(String site, String id) {
        if (!sites.containsKey(site)) {
            return false;
        }
        List<TSensor> detailsList = siteSensors.get(site);
        if (detailsList == null) {
            LOG.error("Trying to remove sensor which is not registered {}", id);
            return false;
        }
        Iterator<TSensor> itr = detailsList.iterator();
        while (itr.hasNext()) {
            TSensor sensor = itr.next();
            if (sensor.getSensorId().equals(id)) {
                itr.remove();
                return true;
            }
        }
        LOG.error("Trying to remove sensor which is not registered {}", id);
        return false;
    }

    public void removeSite(String site) {
        siteSensors.remove(site);
        sites.remove(site);
    }

    public TSensor getSensor(String siteId, String name) {
        List<TSensor> details = siteSensors.get(siteId);
        if (details != null) {
            for (TSensor detail : details) {
                if (detail.getSensorId().equals(name)) {
                    return detail;
                }
            }
        }
        return null;
    }

    public List<TSensor> getAllSensors() {
        List<TSensor> sensors = new ArrayList<TSensor>();
        for (Map.Entry<String, List<TSensor>> e : siteSensors.entrySet()) {
            sensors.addAll(e.getValue());
        }
        return sensors;
    }

    public List<TSensor> getSensors(String site) {
        List<TSensor> sensors = siteSensors.get(site);
        if (sensors == null) {
            LOG.warn("Trying to access sensors for non existent site {}", site);
            return new ArrayList<TSensor>();
        }
        return new ArrayList<TSensor>(sensors);
    }

    public Map getConf() {
        return conf;
    }

    public String getParentPath() {
        return Configuration.getZkRoot(conf);
    }
}