DirectHubGc.cxx

Go to the documentation of this file.

 
#include "utils/DirectHub.hxx"
 
class DirectHubGcSegmenter : public MessageSegmenter
{
public:
    DirectHubGcSegmenter()
    {
        clear();
    }
 
    ssize_t segment_message(const void *d, size_t size) override
    {
        if (!size)
        {
            return 0; // nothing to do.
        }
        const char *data = static_cast<const char *>(d);
        if (packetLen_ == 0)
        {
            // beginning of packet.
            isGcPacket_ = data[0] == ':';
        }
        size_t ofs = 0;
        if (isGcPacket_)
        {
            // looking for terminating ;
            while ((ofs < size) && (data[ofs] != ';'))
            {
                ++ofs;
            }
            if (ofs < size)
            {
                // found the terminating ;
                ++ofs;
                // append any garbage we still have.
                while ((ofs < size) && (data[ofs] != ':'))
                {
                    ++ofs;
                }
                packetLen_ += ofs;
                return packetLen_;
            }
            else
            {
                // ran out of payload without finding terminating ;
                packetLen_ += size;
                return 0;
            }
        }
        else
        {
            // Looking for starting ':'
            while ((ofs < size) && (data[ofs] != ':'))
            {
                ++ofs;
            }
            packetLen_ += ofs;
            if (ofs < size)
            {
                // found it
                return packetLen_;
            }
            else
            {
                return 0;
            }
        }
    }
 
    void clear() override
    {
        isGcPacket_ = false;
        packetLen_ = 0;
    }
 
private:
    uint32_t isGcPacket_ : 1;
 
    uint32_t packetLen_ : 30;
};
 
MessageSegmenter *create_gc_message_segmenter()
{
    return new DirectHubGcSegmenter();
}
 
class DirectHubTrivialSegmenter : public MessageSegmenter
{
public:
    ssize_t segment_message(const void *d, size_t size) override
    {
        total_ += size;
        LOG(VERBOSE, "segment %zu total %zu", size, total_);
        return size;
    }
 
    void clear() override
    {
    }
 
    size_t total_ {0};
};
 
MessageSegmenter *create_trivial_message_segmenter()
{
    return new DirectHubTrivialSegmenter();
}