protobuf_sink.h

#pragma once
#include <fstream>
#include <g3log/logmessage.hpp>
#include <iostream>
 
#include "google/protobuf/any.pb.h"
#include "software/logger/custom_logging_levels.h"
#include "software/logger/proto_logger.h"
#include "software/networking/unix/threaded_unix_sender.h"
 
static const std::string TYPE_DELIMITER = "!!!";
 
 
using UnixSenderMap =
    std::unordered_map<std::string, std::unique_ptr<ThreadedUnixSender>>;
 
 
class ProtobufSink
{
   public:
    ProtobufSink(std::string runtime_dir,
                 const std::shared_ptr<ProtoLogger>& proto_logger);
 
    /*
     * Send the protobuf to /tmp/tbots/(protobuf typename)
     *
     * @param log_entry The entry to log
     */
    void sendProtobuf(g3::LogMessageMover log_entry);
 
   private:
    UnixSenderMap unix_senders_;
    std::string runtime_dir_;
    std::shared_ptr<ProtoLogger> proto_logger;
};
 
/*
 * Serialize a protobuf and pack it into a msg
 *
 * @param os The output stream
 * @param message The message to serialize
 * @return The output stream containing the base64 encoded serialized Any protobuf
 */
std::ostream& operator<<(std::ostream& os, const google::protobuf::Message& message);