data_stream.hpp

//
//  data_stream.hpp
//  skyway
//
//  Created by sandabu on 2022/01/28.
//  Copyright © 2022 NTT Communications. All rights reserved.
//
 
#ifndef SKYWAY_CORE_STREAM_LOCAL_DATA_STREAM_HPP_
#define SKYWAY_CORE_STREAM_LOCAL_DATA_STREAM_HPP_
 
#include <atomic>
#include <mutex>
 
#include "skyway/core/interface/local_stream.hpp"
 
namespace skyway {
namespace core {
namespace stream {
namespace local {
 
using LocalStream = interface::LocalStream;
 
class LocalDataStream : public LocalStream {
public:
    using PublicationId = std::string;
    struct SendingData {
        bool is_binary;
        std::vector<uint8_t> data;
    };
    class InternalListener {
    public:
        virtual bool OnWriteData(const SendingData& buffer,
                                 const PublicationId& publication_id) = 0;
    };
    LocalDataStream();
    void AddInternalListener(const std::string& remote_member_id,
                             const PublicationId& publication_id,
                             InternalListener* listener);
    void RemoveInternalListener(const std::string& remote_member_id,
                                const PublicationId& publication_id);
    bool Write(const std::string& data) const;
    bool Write(const uint8_t* data, size_t length) const;
 
    bool Enable() override;
    bool Disable() override;
 
private:
    std::unordered_map<std::string, std::unordered_map<PublicationId, InternalListener*>>
        listeners_;
    mutable std::mutex listener_mtx_;
    std::atomic<bool> is_enabled_;
};
 
}  // namespace local
}  // namespace stream
}  // namespace core
}  // namespace skyway
 
#endif /* SKYWAY_CORE_STREAM_LOCAL_DATA_STREAM_HPP_ */