client.h

Go to the documentation of this file.

#ifndef CPPSERVER_NANOMSG_CLIENT_H
#define CPPSERVER_NANOMSG_CLIENT_H

#include "socket.h"

#include "threads/thread.h"

#include <atomic>
#include <thread>

namespace CppServer {
namespace Nanomsg {


class Client
{
public:

    explicit Client(Domain domain, Protocol protocol, const std::string& address, bool threading = true);
    Client(const Client&) = delete;
    Client(Client&&) = default;
    virtual ~Client();

    Client& operator=(const Client&) = delete;
    Client& operator=(Client&&) = default;

    Socket& socket() noexcept { return _socket; }

    bool IsConnected() const noexcept { return _socket.IsOpened() && _socket.IsConnected() && _connected; }


    bool Connect();

    bool Disconnect();

    bool Reconnect();


    size_t Send(const void* buffer, size_t size);

    size_t Send(const std::string& text) { return Send(text.data(), text.size()); }

    size_t Send(const Message& message) { return Send(message.buffer(), message.size()); }


    size_t TrySend(const void* buffer, size_t size);

    size_t TrySend(const std::string& text) { return TrySend(text.data(), text.size()); }

    size_t TrySend(const Message& message) { return TrySend(message.buffer(), message.size()); }


    size_t Receive(Message& message);


    size_t TryReceive(Message& message);

protected:

    virtual void onThreadInitialize() {}

    virtual void onThreadCleanup() {}

    virtual void onConnected() {}
    virtual void onDisconnected() {}

    virtual void onIdle() { CppCommon::Thread::Yield(); }


    virtual void onReceived(Message& message) {}


    virtual void onError(int error, const std::string& message) {}

private:
    // Nanomsg endpoint address
    std::string _address;
    // Nanomsg socket
    Socket _socket;
    std::atomic<bool> _connected;
    // Nanomsg client thread
    bool _threading;
    std::thread _thread;
    std::atomic<bool> _joining;
    std::atomic<bool> _receiving;

    void ClientLoop();
};

} // namespace Nanomsg
} // namespace CppServer

#endif // CPPSERVER_NANOMSG_CLIENT_H