Storage.java

import java.util.HashMap;
import java.util.List;
import java.util.concurrent.locks.ReentrantLock;

public class Storage {
    private HashMap<String,Task> tasks;
    private HashMap<String,DeviceStats> devices;
    private ReentrantLock l;

    Storage (HashMap<String,Task> tasks) {
        this.tasks = new HashMap<String,Task>();
        this.devices = new HashMap<String,DeviceStats>();
        for (Task t : tasks.values()) {
            this.tasks.put(t.getTaskId(),t);
            Device d = t.getDevice();
            if (d == null)
                return ;
            String dId = d.getId();
            if (!this.devices.containsKey(dId)) {
                DeviceStats ds = new DeviceStats(dId);
                ds.addTask(t);
                this.devices.put(dId, ds);
            }
            else {
                this.devices.get(dId).addTask(t);
            }
        }
    }

    public void updateStorage(String device, Metrics metrics) {
        DeviceStats ds = this.devices.get(device);
        if (ds != null) {
            ds.updateDeviceStats(metrics);
        }
    }

    public List<Task> getDeviceTasks(String dId) {
        List<Task> t = null;
        DeviceStats d = this.devices.get(dId);
        if (d != null) t = d.getTasks();
        return t;
    }

    public String getInfo() {
        StringBuilder sb = new StringBuilder();
        sb.append("Tarefas enviadas:\n");
        sb.append(getTasksList());
        sb.append("\n");
        sb.append("Métricas recebidas:\n");
        for (DeviceStats d : this.devices.values()) {
            sb.append(d.toString());
            sb.append("\n");
        }
        sb.append("");
        return sb.toString();
    }

    public String getTasksInfo(){
        StringBuilder sb = new StringBuilder();
        for (Task t : this.tasks.values()) {
            sb.append(t.toString());
            sb.append("\n");
        }
        return sb.toString();
    }

    public String getTasksList(){
        StringBuilder sb = new StringBuilder();
        for (Task t : this.tasks.values()) {
            sb.append(t.getTaskId());
            sb.append("\n");
        }
        return sb.toString();
    }

    public String getDeviceTasksInfo(String device){
        StringBuilder sb = new StringBuilder();
        for (Task t : this.tasks.values()) {
            if(t.getDevice().getId().equals(device)){
                sb.append(t.toString());
                sb.append("\n");
            }
        }
        return sb.toString();
    }

    public String getDeviceMetrics (String device){
        StringBuilder sb = new StringBuilder();
        sb.append(devices.get(device).toString());
        return sb.toString();
    }

    public String [] getDeviceArray(){
        String [] res = new String[devices.size()];
        int i = 0;
        for (String s : devices.keySet()){
            res[i] = s;
            i++;
        }
        return res;
    }
}