shortmessageconnection.h

// Copyright (c) 2019 Internetstiftelsen
// Written by Göran Andersson <initgoran@gmail.com>
 
#pragma once
 
#include "socketconnection.h"
 
class ShortMessageConnection : public SocketConnection {
public:
    ShortMessageConnection(const std::string &label, Task *owner,
                         const std::string &hostname, uint16_t port);
 
    ShortMessageConnection(const std::string &label, Task *owner, int fd,
                           const char *ip = "unknown", uint16_t port = 0);
 
    PollState connected() override;
 
    PollState readData(char *buf, size_t len) override;
 
    void sendMessage(const std::string &msg);
 
private:
    // Message we're currently receiving, or empty.
    std::string msg;
 
    // Bytes left for the above message to be complete,
    // or 0 if we're not currently receiving a message
    size_t bytes_left = 0;
 
    bool reading_header = true;
};