MessageDispatchTask.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Stats/Stats.h"
#include "IMessageContext.h"
#include "IMessageBusListener.h"
#include "Bus/MessageTracer.h"
 
class IMessageReceiver;
 
class FMessageDispatchTask
{
public:
 
    FMessageDispatchTask(
        ENamedThreads::Type InThread,
        TSharedRef<IMessageContext, ESPMode::ThreadSafe> InContext,
        TWeakPtr<IMessageReceiver, ESPMode::ThreadSafe> InRecipient,
        TSharedPtr<FMessageTracer, ESPMode::ThreadSafe> InTracer);
 
public:
 
    void DoTask(ENamedThreads::Type CurrentThread, const FGraphEventRef& MyCompletionGraphEvent);
    
    ENamedThreads::Type GetDesiredThread()
    {
        return Thread;
    }
 
    TStatId GetStatId() const;
 
    static ESubsequentsMode::Type GetSubsequentsMode() 
    { 
        return ESubsequentsMode::TrackSubsequents; 
    }
 
private:
 
    TSharedRef<IMessageContext, ESPMode::ThreadSafe> Context;
 
    TWeakPtr<IMessageReceiver, ESPMode::ThreadSafe> RecipientPtr;
 
    ENamedThreads::Type Thread;
 
    TWeakPtr<FMessageTracer, ESPMode::ThreadSafe> TracerPtr;
};
 
 
class FBusNotificationDispatchTask
{
public:
 
    FBusNotificationDispatchTask(
        ENamedThreads::Type InThread,
        TWeakPtr<IBusListener, ESPMode::ThreadSafe> InListener,
        FMessageAddress InAddress,
        EMessageBusNotification InNotification)
        : Thread(InThread)
        , ListenerPtr(InListener)
        , Address(InAddress)
        , Notification(InNotification)
    {}
 
public:
 
    void DoTask(ENamedThreads::Type CurrentThread, const FGraphEventRef& MyCompletionGraphEvent);
 
    ENamedThreads::Type GetDesiredThread()
    {
        return Thread;
    }
 
    TStatId GetStatId() const;
 
    static ESubsequentsMode::Type GetSubsequentsMode()
    {
        return ESubsequentsMode::TrackSubsequents;
    }
 
private:
    ENamedThreads::Type Thread;
 
    TWeakPtr<IBusListener, ESPMode::ThreadSafe> ListenerPtr;
 
    FMessageAddress Address;
 
    EMessageBusNotification Notification;
};