com.iobeam.api.resource.Import Maven / Gradle / Ivy

Go to download
package com.iobeam.api.resource;

import org.json.JSONArray;
import org.json.JSONObject;

import java.io.Serializable;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

/**
 * A resource representing an import request resource for IOBeam.
 */
@Deprecated
public class Import implements Serializable {

    private String deviceId;
    private final long projectId;

    /**
     * DataSet is a convenience name for a `HashSet` containing `DataPoint`s.
     */
    @Deprecated
    static final class DataSet extends HashSet {

        public DataSet() {
            super();
        }

        public DataSet(Set pts) {
            super(pts);
        }
    }

    @Deprecated
    static final class DataStore extends HashMap {

        public DataStore() {
            super();
        }

        public DataStore(Map> store) {
            super(convert(store));
        }

        private static Map convert(Map> store) {
            Map ret = new HashMap();
            for (String k : store.keySet()) {
                ret.put(k, new DataSet(store.get(k)));
            }
            return ret;
        }
    }

    private final DataStore store = new DataStore();

    /**
     * Creates a new Import object for a particular device and project.
     *
     * @param deviceId  Device ID this data is for
     * @param projectId Project ID this data is for
     */
    public Import(String deviceId, long projectId) {
        this.deviceId = deviceId;
        this.projectId = projectId;
    }

    public void setDeviceId(String deviceId) {
        this.deviceId = deviceId;
    }

    public String getDeviceId() {
        return deviceId;
    }

    public long getProjectId() {
        return projectId;
    }

    public long getTotalSize() {
        int total = 0;
        for (DataSet set : store.values()) {
            total += set.size();
        }
        return total;
    }

    /**
     * Gets a mapping from series to an unordered set of DataPoints.
     *
     * @return Mapping from a series name to unordered set of data.
     */
    public Map> getSeries() {
        return new HashMap>(store);
    }

    /**
     * Gets the unordered set of data associated with a series.
     *
     * @param label The name of the series to get
     * @return Set of unordered data for that series.
     */
    public DataSet getDataSet(String label) {
        return store.get(label);
    }

    /**
     * Adds a new data point to a particular series.
     *
     * @param series    The series this data point should be added to. If it doesn't exist, it will
     *                  be created.
     * @param dataPoint Data to be added.
     */
    public void addDataPoint(String series, DataPoint dataPoint) {
        DataSet set = store.get(series);
        if (set == null) {
            set = new DataSet();
            store.put(series, set);
        }
        set.add(dataPoint);
    }

    /**
     * Adds a set of `DataPoint`s to a series.
     *
     * @param series     The series the DataPoints should be added to. If the series doesn't exist,
     *                   it will be created.
     * @param dataPoints Data to be added.
     */
    public void addDataPointSet(String series, Set dataPoints) {
        DataSet set = store.get(series);
        if (set == null) {
            set = new DataSet(dataPoints);
            store.put(series, set);
        } else {
            set.addAll(dataPoints);
        }
    }

    /**
     * Adds a set of `DataPoint`s to a series.
     *
     * @param series  The series to add the contents of DataSet to. If the series doesn't exist, it
     *                will be created.
     * @param dataSet Data to be added.
     */
    public void addDataSet(String series, DataSet dataSet) {
        addDataPointSet(series, dataSet);
    }

    /**
     * Custom serialization method for this object to JSON.
     *
     * @param out In-out parameter that maps JSON keys to their values.
     * @return A JSONObject representing this object.
     */
    public JSONObject serialize(Map out) {
        JSONObject json = new JSONObject();
        json.put("device_id", deviceId);
        json.put("project_id", projectId);
        out.put("device_id", deviceId);
        out.put("project_id", projectId);
        JSONArray sourcesArr = new JSONArray();
        List sourceNames = new ArrayList(store.keySet());
        Collections.sort(sourceNames);
        for (String k : sourceNames) {
            JSONObject temp = new JSONObject();
            temp.put("name", k);
            JSONArray dataArr = new JSONArray();
            for (DataPoint d : store.get(k)) {
                dataArr.put(d.toJson());
            }
            temp.put("data", dataArr);
            sourcesArr.put(temp);
        }
        json.put("sources", sourcesArr);
        out.put("sources", sourcesArr);
        return json;
    }

    public static Import fromJson(final JSONObject json) throws ParseException {
        return new Import(json.getString("device_id"),
                          json.getLong("project_id"));
    }

    @Override
    public String toString() {
        return "Import{" +
               "deviceId=" + deviceId +
               ", projectId='" + projectId + "'" +
               '}';
    }
}