UdpClient.h

Go to the documentation of this file.

#ifndef UDPCLIENT_H
#define UDPCLIENT_H

#include <algorithm>
#include <map>
#include <string>
#include <queue>

#include <boost/asio.hpp>
#include <boost/bind.hpp>
#include <boost/thread.hpp>

#include "Client.h"
#include "Logger.h"
#include "Event.h"
#include "Udp.h"

using boost::asio::ip::udp;
using std::vector;
using std::string;

class UdpClient: public Udp, public Client
{
    public:

        UdpClient(const string &host, int port, boost::asio::io_service & io_service);

        UdpClient(const string &host, int port, boost::asio::io_service & io_service, map<string, string> options);

        virtual ~UdpClient();

        virtual void send(const string & data);

        virtual void send_bytes(const vector<byte> & bytes);

        virtual void shutdown();

        virtual int get_port();

        virtual string get_host();

    protected:

        void init_socket();

        virtual void fire_error_event(const string & message);

        virtual void fire_data_event(const string data, boost::shared_ptr<udp::socket> socket,
                boost::shared_ptr<udp::endpoint> endpoint);

        virtual void close();

    private:

        UdpClient(const UdpClient &other);

        virtual void listen();

        void resolve_handler(const boost::system::error_code &err, udp::resolver::iterator endpoint_iterator);

        void flush();

        boost::shared_ptr<udp::resolver> resolver;

        boost::shared_ptr<udp::socket> socket;

        boost::mutex queue_mtx;

        std::queue<string> msgs_not_sent;

        bool resolved_endpoint;
};

#endif