message_json.h

Go to the documentation of this file.

#pragma once
 
#include "librats/util/rats_export.h"
#include "librats/node/peer_network.h"
#include "librats/peer/peer.h"
#include "librats/peer/peer_id.h"
#include "librats/core/bytes.h"
#include "librats/util/json.h"
 
#include <functional>
#include <mutex>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace librats {
 
class RATS_API MessageJson final : public Subsystem {
public:
    using Handler      = std::function<void(const PeerId& from, const librats::Json& data)>;
    using SendCallback = std::function<void(bool ok, const std::string& error)>;
 
    void on(const std::string& type, Handler handler);
 
    void once(const std::string& type, Handler handler);
 
    void off(const std::string& type);
 
    void send(const std::string& type, const librats::Json& data, SendCallback cb = nullptr);
 
    void send(const PeerId& to, const std::string& type, const librats::Json& data,
              SendCallback cb = nullptr);
 
    // Subsystem — no background thread; purely event-driven.
    void attach(NodeContext& ctx) override;
    void start() override {}
    void stop() override {}
 
private:
    void on_typed(const PeerId& from, ByteView payload);
    static Bytes encode(const std::string& type, const librats::Json& data);
 
    struct Entry {
        Handler handler;
        bool    once;
    };
 
    PeerNetwork* network_ = nullptr;
    mutable std::mutex mutex_;
    std::unordered_map<std::string, std::vector<Entry>> handlers_;
};
 
} // namespace librats