/*
*
* Copyright (C) 2010 Mattias Runge
*
* 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 2 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, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
*/
#include "CanDaemon.h"
#include <boost/lexical_cast.hpp>
#include <boost/algorithm/string/trim.hpp>
#include "net/Manager.h"
#include "broker/Manager.h"
#include "Message.h"
namespace atom {
namespace can {
CanDaemon::CanDaemon(unsigned int port): broker::Subscriber(port), LOG("can::CanDaemon")
{
try
{
this->server_id_ = net::Manager::Instance()->StartServer(net::TRANSPORT_PROTOCOL_TCP, port);
LOG.Info("Started TCP server on port " + boost::lexical_cast<std::string>(port) + ".");
LOG.Debug("Server id is " + boost::lexical_cast<std::string>(this->server_id_) + ".");
}
catch (std::runtime_error& e)
{
LOG.Error(e.what());
}
}
CanDaemon::~CanDaemon()
{
net::Manager::Instance()->StopServer(this->server_id_);
this->server_id_ = 0;
}
void CanDaemon::SlotOnMessageHandler(broker::Message::Pointer message)
{
if (message->GetType() == broker::Message::CAN_RAW_MESSAGE)
{
std::string* payload_str;
payload_str = static_cast<std::string*>(message->GetPayload().get());
std::string line = *payload_str;
net::Manager::Instance()->SendToAll(this->server_id_, common::Byteset(line.begin(), line.end()));
}
}
void CanDaemon::SlotOnNewDataHandler(net::SocketId id, common::Byteset data)
{
if (std::find(this->clients_.begin(), this->clients_.end(), id) == this->clients_.end())
{
return;
}
std::string str(data.begin(), data.end());
boost::algorithm::trim_right_if(str, boost::is_any_of("\r\n"));
LOG.Debug("Received: \"" + str + "\" from client " + boost::lexical_cast<std::string>(id) + " on server " + boost::lexical_cast<std::string>(this->server_id_));
std::string* payload_str = new std::string(str);
broker::Manager::Instance()->Post(broker::Message::Pointer(new broker::Message(broker::Message::CAN_RAW_MESSAGE, broker::Message::PayloadPointer(payload_str), this)));
//LOG.Info("Sent...");
if (str == "q" || str == "quit")
{
LOG.Debug("Client " + boost::lexical_cast<std::string>(id) + " has requested to be disconnected.");
net::Manager::Instance()->Disconnect(id);
}
}
void CanDaemon::SlotOnNewClientHandler(net::SocketId id, net::SocketId server_id)
{
if (server_id == this->server_id_)
{
this->clients_.insert(id);
}
}
void CanDaemon::SlotOnNewStateHandler(net::SocketId id, net::ClientState client_state)
{
if (std::find(this->clients_.begin(), this->clients_.end(), id) == this->clients_.end())
{
return;
}
if (client_state == net::CLIENT_STATE_DISCONNECTED)
{
LOG.Debug("Client " + boost::lexical_cast<std::string>(id) + " has disconnected.");
this->clients_.erase(id);
}
else if (client_state == net::CLIENT_STATE_CONNECTED)
{
LOG.Debug("Client " + boost::lexical_cast<std::string>(id) + " has connected.");
}
else
{
LOG.Debug("Got state: " + boost::lexical_cast<std::string>((int)client_state));
}
}
}; // namespace can
}; // namespace atom