socket.h

#ifndef B0__SOCKET_H__INCLUDED
#define B0__SOCKET_H__INCLUDED

#include <string>

#include <b0/user_data.h>
#include <b0/logger/interface.h>

#include <boost/function.hpp>
#include <boost/bind.hpp>

namespace b0
{

class Node;


struct SocketPrivate;


class Socket : public logger::LogInterface, public UserData
{
public:
    Socket(Node *node, int type, std::string name, bool managed = true);

    virtual ~Socket();

    void setHasHeader(bool has_header);

    void log(LogLevel level, std::string message) const override;

    virtual void init() = 0;

    virtual void cleanup() = 0;

    virtual void spinOnce();

    void setRemoteAddress(std::string addr);

    std::string getName() const;

    Node & getNode() const;

    bool matchesPattern(const std::string &pattern) const;

private:
    std::unique_ptr<SocketPrivate> private_;

protected:
    Node &node_;

    std::string name_;

    bool has_header_;

    const bool managed_;

    std::string remote_addr_;

public:
    virtual void readRaw(std::string &msg);

    virtual void readRaw(std::string &msg, std::string &type);

    virtual bool poll(long timeout = 0);

    virtual void writeRaw(const std::string &msg, const std::string &type = "");

public:
    void setCompression(std::string algorithm, int level = -1);

private:
    std::string compression_algorithm_;

    int compression_level_;

public:
    int getReadTimeout() const;

    void setReadTimeout(int timeout);

    int getWriteTimeout() const;

    void setWriteTimeout(int timeout);

    int getLingerPeriod() const;

    void setLingerPeriod(int period);

    int getBacklog() const;

    void setBacklog(int backlog);

    bool getImmediate() const;

    void setImmediate(bool immediate);

    bool getConflate() const;

    void setConflate(bool conflate);

    int getReadHWM() const;

    void setReadHWM(int n);

    int getWriteHWM() const;

    void setWriteHWM(int n);

protected:
    void connect(std::string const &addr);

    void disconnect(std::string const &addr);

    void bind(std::string const &addr);

    void unbind(std::string const &addr);

    void setsockopt(int option, const void *optval, size_t optvallen);

    void getsockopt(int option, void *optval, size_t *optvallen) const;

    void setIntOption(int option, int value);

    int getIntOption(int option) const;
};

} // namespace b0

#endif // B0__SOCKET_H__INCLUDED