MessageBridge.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "IMessageBridge.h"
#include "IMessageContext.h"
#include "IMessageReceiver.h"
#include "IMessageSender.h"
#include "IMessageTransportHandler.h"
#include "Misc/Guid.h"
#include "Templates/SharedPointer.h"
 
#include "Bridge/MessageAddressBook.h"
 
class IMessageBus;
class IMessageSubscription;
class IMessageTransport;
 
 
class FMessageBridge
    : public TSharedFromThis<FMessageBridge, ESPMode::ThreadSafe>
    , public IMessageBridge
    , public IMessageReceiver
    , public IMessageSender
    , protected IMessageTransportHandler
{
public:
 
    FMessageBridge(
        const FMessageAddress InAddress,
        const TSharedRef<IMessageBus, ESPMode::ThreadSafe>& InBus,
        const TSharedRef<IMessageTransport, ESPMode::ThreadSafe>& InTransport
    );
 
    virtual ~FMessageBridge();
 
public:
 
    //~ IMessageBridge interface
 
    virtual void Disable() override;
    virtual void Enable() override;
    virtual bool IsEnabled() const override;
    virtual FGuid LookupAddress(const FMessageAddress& InAddress);
 
public:
 
    //~ IMessageReceiver interface
 
    virtual FName GetDebugName() const override;
    virtual const FGuid& GetRecipientId() const override;
    virtual ENamedThreads::Type GetRecipientThread() const override;
    virtual bool IsLocal() const override;
    virtual void ReceiveMessage(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context) override;
 
public:
 
    //~ IMessageSender interface
 
    virtual FMessageAddress GetSenderAddress() override;
    virtual void NotifyMessageError(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context, const FString& Error) override;
 
protected:
 
    //~ IMessageTransportHandler interface
 
    virtual void DiscoverTransportNode(const FGuid& NodeId) override;
    virtual void ForgetTransportNode(const FGuid& NodeId) override;
    virtual void ReceiveTransportMessage(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context, const FGuid& NodeId) override;
 
private:
 
    void HandleMessageBusShutdown();
 
private:
 
    FMessageAddress Address;
 
    FMessageAddressBook AddressBook;
 
    TSharedPtr<IMessageBus, ESPMode::ThreadSafe> Bus;
 
    bool Enabled;
 
    const FGuid Id;
 
    TSharedPtr<IMessageSubscription, ESPMode::ThreadSafe> MessageSubscription;
 
    TSharedPtr<IMessageTransport, ESPMode::ThreadSafe> Transport;
};