energy_storage/src/app/Station.cpp

#include "Station.h"


Station::Station(int id) : id(id)
{
}

void Station::addDevice(int deviceId, std::shared_ptr<Device> device)
{
    mapDevice_[deviceId] = device;
}

std::shared_ptr<Device> Station::getDevice(int deviceId)
{
    auto iter = mapDevice_.find(deviceId);
    if (iter!=mapDevice_.end())
    {
        return iter->second;
    }
    return nullptr;
}