TcpClient.h

Go to the documentation of this file.

/*
 * File:   TcpClient.h
 * Author: jtedesco
 *
 * Created on May 26, 2011, 12:32 PM
 */

#ifndef TCPCLIENT_H
#define TCPCLIENT_H

#include <boost/lexical_cast.hpp>
#include <boost/thread.hpp>

#include <string.h>

#include "Client.h"
#include "Logger.h"
#include "Tcp.h"

using boost::asio::ip::tcp;

class TcpClient: public Tcp, public Client
{

    public:

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

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

        virtual ~TcpClient();

        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();

        FB_JSAPI_EVENT(disconnect, 1, (const string &));

        FB_JSAPI_EVENT(connect, 0, ());

    protected:

        virtual void fire_error_event(const string & message);

        virtual void fire_disconnect_event(const string & message);

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

        virtual void close();

    private:

        TcpClient(const TcpClient &other);

        void init();

        void init_socket();

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

        void connect_handler(const boost::system::error_code & error_code, tcp::resolver::iterator endpoint_iterator);

        void listen();

        void flush();

        boost::shared_ptr<tcp::socket> connection;

        boost::shared_ptr<tcp::resolver> resolver;

        bool connected;

        boost::mutex connected_mutex;

        queue<string> data_queue;

        boost::mutex data_queue_mutex;
};

#endif  /* TCPCLIENT_H */