ProtoDelimSerializer.hpp

Go to the documentation of this file.

#pragma once
 
#include <google/protobuf/io/gzip_stream.h>
#include <google/protobuf/util/delimited_message_util.h>
#include <srs/converters/ProtoSerializerBase.hpp>
 
namespace srs::process
{
    const auto protobuf_delim_deserializer_converter = [](const proto::Data& proto_data,
                                                          std::string& output_data) -> int
    {
        namespace protobuf = google::protobuf;
        namespace io = protobuf::io;
        auto output_stream = io::StringOutputStream{ &output_data };
 
        if constexpr (common::PROTOBUF_ENABLE_GZIP)
        {
            auto option = io::GzipOutputStream::Options{};
            option.compression_level = common::GZIP_DEFAULT_COMPRESSION_LEVEL;
            auto gzip_output = io::GzipOutputStream{ &output_stream, option };
            protobuf::util::SerializeDelimitedToZeroCopyStream(proto_data, &gzip_output);
            gzip_output.Flush();
        }
        else
        {
            protobuf::util::SerializeDelimitedToZeroCopyStream(proto_data, &output_stream);
        }
        return 0;
    };
 
    class ProtoDelimSerializer : public ProtoSerializerBase<decltype(protobuf_delim_deserializer_converter)>
    {
      public:
        explicit ProtoDelimSerializer(asio::thread_pool& thread_pool)
            : ProtoSerializerBase{ thread_pool, "ProtoDelimSerializer", protobuf_delim_deserializer_converter }
        {
        }
        static constexpr auto ConverterOption = std::array{ proto_frame };
    };
} // namespace srs