MessageBridgeBuilder.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "IMessageContext.h"
#include "IMessageBridge.h"
#include "IMessageBus.h"
#include "IMessageTransport.h"
#include "IMessagingModule.h"
 
 
class FMessageBridgeBuilder
{
public:
 
    FMessageBridgeBuilder()
        : Address(FMessageAddress::NewAddress())
        , BusPtr(IMessagingModule::Get().GetDefaultBus())
        , Disabled(false)
        , Transport(nullptr)
    { }
 
    FMessageBridgeBuilder(const TSharedRef<IMessageBus, ESPMode::ThreadSafe>& Bus)
        : Address(FMessageAddress::NewAddress())
        , BusPtr(Bus)
        , Disabled(false)
        , Transport(nullptr)
    { }
 
public:
 
    FMessageBridgeBuilder& ThatIsDisabled()
    {
        Disabled = true;
 
        return *this;
    }
 
    FMessageBridgeBuilder& UsingTransport(const TSharedRef<IMessageTransport, ESPMode::ThreadSafe>& InTransport)
    {
        Transport = InTransport;
 
        return *this;
    }
 
    FMessageBridgeBuilder& WithAddress(const FMessageAddress& InAddress)
    {
        Address = InAddress;
 
        return *this;
    }
 
public:
 
    TSharedPtr<IMessageBridge, ESPMode::ThreadSafe> Build()
    {
        TSharedPtr<IMessageBridge, ESPMode::ThreadSafe> Bridge;
 
        check(Transport.IsValid());
 
        auto Bus = BusPtr.Pin();
 
        if (Bus.IsValid())
        {
            Bridge = IMessagingModule::Get().CreateBridge(Address, Bus.ToSharedRef(), Transport.ToSharedRef());
 
            if (Bridge.IsValid())
            {
                if (Disabled)
                {
                    Bridge->Disable();
                }
                else
                {
                    Bridge->Enable();
                }
            }
        }
 
        return Bridge;
    }
 
    operator TSharedPtr<IMessageBridge, ESPMode::ThreadSafe>()
    {
        return Build();
    }
 
private:
 
    FMessageAddress Address;
 
    TWeakPtr<IMessageBus, ESPMode::ThreadSafe> BusPtr;
 
    bool Disabled;
 
    TSharedPtr<IMessageTransport, ESPMode::ThreadSafe> Transport;
};