DepositorySensorStatusServerResource.java

/*
 * This file is part of WattDepot.
 *
 *  Copyright (C) 2015  Cam Moore
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.wattdepot.server.http.api;

import org.restlet.data.Status;
import org.restlet.resource.ResourceException;
import org.wattdepot.common.domainmodel.CollectorProcessDefinition;
import org.wattdepot.common.domainmodel.Labels;
import org.wattdepot.common.domainmodel.Measurement;
import org.wattdepot.common.domainmodel.Sensor;
import org.wattdepot.common.domainmodel.SensorGroup;
import org.wattdepot.common.domainmodel.SensorStatus;
import org.wattdepot.common.domainmodel.SensorStatusEnum;
import org.wattdepot.common.domainmodel.SensorStatusList;
import org.wattdepot.common.exception.IdNotFoundException;
import org.wattdepot.common.exception.MisMatchedOwnerException;
import org.wattdepot.common.http.api.DepositorySensorStatusResource;
import org.wattdepot.common.util.DateConvert;
import org.wattdepot.common.util.tstamp.Tstamp;

import javax.xml.datatype.XMLGregorianCalendar;
import java.util.Date;
import java.util.List;
import java.util.logging.Level;

/**
 * DepositorySensorStatusServerResource - Returns the SensorStatus(es) for the Sensor or SensorGroup.
 *
 * @author Cam Moore
 */
public class DepositorySensorStatusServerResource extends WattDepotServerResource implements DepositorySensorStatusResource {
  private String depositoryId;
  private String sensorId;

  /*
   * (non-Javadoc)
   *
   * @see org.restlet.resource.Resource#doInit()
   */
  @Override
  protected void doInit() throws ResourceException {
    super.doInit();
    this.sensorId = getQuery().getValues(Labels.SENSOR);
    this.depositoryId = getAttribute(Labels.DEPOSITORY_ID);
  }

  @Override
  public SensorStatusList retrieve() {
    getLogger().log(
        Level.INFO,
        "GET /wattdepot/{" + orgId + "}/" + Labels.DEPOSITORY + "/{" + depositoryId + "}/"
            + Labels.SENSOR_STATUS + "/?" + Labels.SENSOR + "={" + sensorId + "}");
    if (isInRole(orgId)) {
      SensorStatusList list = new SensorStatusList();
      Integer numDays = 7;
      try {
        XMLGregorianCalendar now = Tstamp.makeTimestamp();
        XMLGregorianCalendar weekAgo = Tstamp.incrementDays(now, -1 * numDays);
        Date endDate = DateConvert.convertXMLCal(now);
        Date startDate = DateConvert.convertXMLCal(weekAgo);
        depot.getDepository(depositoryId, orgId, true);
        Sensor sensor = depot.getSensor(sensorId, orgId, false);
        if (sensor != null) {
          SensorStatus status = new SensorStatus();
          status.setSensorId(sensorId);
          status.setTimestamp(now);
          String cpdName = sensorId + "-" + depositoryId;
          CollectorProcessDefinition cpd = depot.getCollectorProcessDefinition(cpdName, orgId, true);
          Long numExpectedMeasurements = 0l;
          if (cpd != null) {
            numExpectedMeasurements = Math.round(60.0 / cpd.getPollingInterval() * 60.0 * 24 * numDays);
          }
          List<Measurement> measurementList = depot.getMeasurements(depositoryId, orgId, sensorId, startDate, endDate, false);
          int count = measurementList.size();
          if (count > 0.8 * numExpectedMeasurements) {
            status.setStatus(SensorStatusEnum.GREEN);
          }
          else if (count > 0.5 * numExpectedMeasurements) {
            status.setStatus(SensorStatusEnum.YELLOW);
          }
          else if (count > 0) {
            status.setStatus(SensorStatusEnum.RED);
          }
          else {
            status.setStatus(SensorStatusEnum.BLACK);
          }
          list.getStatuses().add(status);
          return list;
        }
        else {
          SensorGroup group = depot.getSensorGroup(sensorId, orgId, false);
          if (group != null) {
            for (String s : group.getSensors()) {
              sensor = depot.getSensor(s, orgId, false);
              if (sensor != null) {
                SensorStatus status = new SensorStatus();
                status.setSensorId(s);
                status.setTimestamp(now);
                String cpdName = s + "-" + depositoryId;
                CollectorProcessDefinition cpd = depot.getCollectorProcessDefinition(cpdName, orgId, false);
                Long numExpectedMeasurements = 60l * 24 * numDays; // one measurement a minute.
                if (cpd != null) {
                  numExpectedMeasurements = Math.round(60.0 / cpd.getPollingInterval() * 60.0 * 24 * numDays);
                }
                List<Measurement> measurementList = depot.getMeasurements(depositoryId, orgId, s, startDate, endDate, false);
                int count = measurementList.size();
                if (count > 0.8 * numExpectedMeasurements) {
                  status.setStatus(SensorStatusEnum.GREEN);
                }
                else if (count > 0.5 * numExpectedMeasurements) {
                  status.setStatus(SensorStatusEnum.YELLOW);
                }
                else if (count > 0) {
                  status.setStatus(SensorStatusEnum.RED);
                }
                else {
                  status.setStatus(SensorStatusEnum.BLACK);
                }
                list.getStatuses().add(status);
              }
            }
            return list;
          }
        }
      }
      catch (IdNotFoundException e) {
        setStatus(Status.CLIENT_ERROR_BAD_REQUEST, depositoryId + " is not a Depository id.");
        return null;
      }
      catch (MisMatchedOwnerException e) {
        setStatus(Status.CLIENT_ERROR_BAD_REQUEST, depositoryId + " is not in organization " + orgId + ".");
        return null;
      }

      return list;
    }
    return null;
  }
}