IntraTransport.h

Go to the documentation of this file.

 
#include "RobotRaconteur/RobotRaconteurNode.h"
#include <boost/shared_array.hpp>
 
#pragma once
 
namespace RobotRaconteur
{
class ROBOTRACONTEUR_CORE_API IntraTransportConnection;

 
class ROBOTRACONTEUR_CORE_API IntraTransport : public Transport, public RR_ENABLE_SHARED_FROM_THIS<IntraTransport>
{
    friend class IntraTransportConnection;
 
  private:
    static std::list<RR_WEAK_PTR<IntraTransport> > peer_transports;
    static boost::mutex peer_transports_lock;
 
    bool transportopen;
    bool is_server;
 
  public:
    RR_UNORDERED_MAP<uint32_t, RR_SHARED_PTR<ITransportConnection> > TransportConnections;
    boost::mutex TransportConnections_lock;

    IntraTransport(const RR_SHARED_PTR<RobotRaconteurNode>& node = RobotRaconteurNode::sp());
 
    RR_OVIRTUAL ~IntraTransport() RR_OVERRIDE;
 
    RR_OVIRTUAL bool IsServer() const RR_OVERRIDE;
 
    RR_OVIRTUAL bool IsClient() const RR_OVERRIDE;
 
    RR_OVIRTUAL std::string GetUrlSchemeString() const RR_OVERRIDE;
 
    RR_OVIRTUAL std::vector<std::string> GetServerListenUrls() RR_OVERRIDE;
 
    RR_OVIRTUAL void SendMessage(const RR_INTRUSIVE_PTR<Message>& m) RR_OVERRIDE;
 
    RR_OVIRTUAL void AsyncSendMessage(
        const RR_INTRUSIVE_PTR<Message>& m,
        const boost::function<void(const RR_SHARED_PTR<RobotRaconteurException>&)>& handler) RR_OVERRIDE;
 
    RR_OVIRTUAL void AsyncCreateTransportConnection(
        boost::string_ref url, const RR_SHARED_PTR<Endpoint>& e,
        boost::function<void(const RR_SHARED_PTR<ITransportConnection>&,
                             const RR_SHARED_PTR<RobotRaconteurException>&)>& handler) RR_OVERRIDE;
 
    RR_OVIRTUAL RR_SHARED_PTR<ITransportConnection> CreateTransportConnection(
        boost::string_ref url, const RR_SHARED_PTR<Endpoint>& e) RR_OVERRIDE;
 
    RR_OVIRTUAL void CloseTransportConnection(const RR_SHARED_PTR<Endpoint>& e) RR_OVERRIDE;

    virtual void StartServer();

    virtual void StartClient();
 
  protected:
    virtual void CloseTransportConnection_timed(const boost::system::error_code& err, const RR_SHARED_PTR<Endpoint>& e,
                                                const RR_SHARED_PTR<void>& timer);
 
  public:
    RR_OVIRTUAL bool CanConnectService(boost::string_ref url) RR_OVERRIDE;
 
    RR_OVIRTUAL void Close() RR_OVERRIDE;
 
    RR_OVIRTUAL void CheckConnection(uint32_t endpoint) RR_OVERRIDE;
 
    RR_OVIRTUAL void PeriodicCleanupTask() RR_OVERRIDE;
 
    RR_OVIRTUAL uint32_t TransportCapability(boost::string_ref name) RR_OVERRIDE;
 
    RR_OVIRTUAL void MessageReceived(const RR_INTRUSIVE_PTR<Message>& m) RR_OVERRIDE;
 
    RR_OVIRTUAL void AsyncGetDetectedNodes(
        const std::vector<std::string>& schemes,
        const boost::function<void(const RR_SHARED_PTR<std::vector<NodeDiscoveryInfo> >&)>& handler,
        int32_t timeout = RR_TIMEOUT_INFINITE) RR_OVERRIDE;
 
    template <typename T, typename F>
    boost::signals2::connection AddCloseListener(const RR_SHARED_PTR<T>& t, const F& f)
    {
        boost::mutex::scoped_lock lock(closed_lock);
        if (closed)
        {
            lock.unlock();
            boost::bind(f, t)();
            return boost::signals2::connection();
        }
 
        return close_signal.connect(boost::signals2::signal<void()>::slot_type(boost::bind(f, t.get())).track(t));
    }
 
    bool TryGetNodeInfo(NodeID& node_id, std::string& node_name, std::string& service_nonce);
 
    RR_OVIRTUAL void LocalNodeServicesChanged() RR_OVERRIDE;
 
    void SendNodeDiscovery();
 
    void DiscoverAllNodes();
 
  protected:
    virtual void register_transport(const RR_SHARED_PTR<ITransportConnection>& connection);
    virtual void erase_transport(const RR_SHARED_PTR<ITransportConnection>& connection);
 
    void NodeDetected(const NodeDiscoveryInfo& info);
 
    boost::mutex init_lock;
    bool is_init;
 
    void Init();
 
    bool closed;
    boost::mutex closed_lock;
    boost::signals2::signal<void()> close_signal;
};
 
#ifndef ROBOTRACONTEUR_NO_CXX11_TEMPLATE_ALIASES
using IntraTransportPtr = RR_SHARED_PTR<IntraTransport>;
#endif
 
} // namespace RobotRaconteur