connection.h

Go to the documentation of this file.

#ifndef CONNECTION_H
#define CONNECTION_H

#include <memory>
#include <string>

#include "base_exception.h"
#include "config.h"

class Connection {
        public:
                class ConnectionError: public BaseException {
                        public:
                                explicit ConnectionError(const std::string &errorMessage):
                                        BaseException(errorMessage) {}
                };

                class ConnectionClosed: public ConnectionError {
                        public:
                                explicit ConnectionClosed(const std::string &errorMessage):
                                        ConnectionError(errorMessage) {}
                };

        public:
                Connection(const Config &config, time_t timeout);

                virtual ~Connection();

                virtual void close() = 0;

                std::string tryReceiveMessage();

                size_t trySendMessage(const std::string &message);

        private:
                virtual std::string doReceiveMessage(Config::LengthFieldMessageLength messageLength) = 0;

                virtual size_t doSendMessage(const std::string &message) = 0;

                std::string tryReceiveMessageViaBuffer();

                std::string tryReceiveMessageViaLengthField();

                std::string receiveFixedLengthMessage(Config::LengthFieldMessageLength messageLength);

                size_t trySendMessageViaLengthField(const std::string &message);

                void sendMessageAtOnce(const std::string &message);

                // Disable copy ctor and assignment operator since this is
                // a reference object
                Connection(const Connection &);
                Connection &operator=(const Connection &);

        private:
                const Config _config;
                time_t _timeout;
};

#endif // #ifndef CONNECTION_H

// End of file connection.h