server.h

Go to the documentation of this file.

#ifndef CPPSERVER_NANOMSG_SERVER_H
#define CPPSERVER_NANOMSG_SERVER_H

#include "socket.h"

#include "threads/thread.h"

#include <atomic>
#include <thread>

namespace CppServer {
namespace Nanomsg {


class Server
{
public:

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

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

    Socket& socket() noexcept { return _socket; }

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


    bool Start();

    bool Stop();

    bool Restart();


    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 onStarted() {}
    virtual void onStopped() {}

    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 server thread
    bool _threading;
    std::thread _thread;
    std::atomic<bool> _joining;
    std::atomic<bool> _receiving;

    void ServerLoop();
};

} // namespace Nanomsg
} // namespace CppServer

#endif // CPPSERVER_NANOMSG_SERVER_H