MessageHandlers.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "IMessageContext.h"
#include "IMessageHandler.h"
 
 
template<typename HandlerType>
class TRawMessageCatchall
    : public IMessageHandler
{
public:
 
    typedef void (HandlerType::*FuncType)(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>&);
 
public:
 
    TRawMessageCatchall(HandlerType* InHandler, FuncType InFunc)
        : Handler(InHandler)
        , Func(InFunc)
    {
        check(InHandler != nullptr);
    }
 
    ~TRawMessageCatchall() { }
 
public:
 
    //~ IMessageHandler interface
    
    virtual void HandleMessage(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context) override
    {
        (Handler->*Func)(Context);
    }
    
private:
 
    HandlerType* Handler;
 
    FuncType Func;
};
 
 
template<typename MessageType, typename HandlerType>
class TRawMessageHandler
    : public IMessageHandler
{
public:
 
    typedef void (HandlerType::*FuncType)(const MessageType&, const TSharedRef<IMessageContext, ESPMode::ThreadSafe>&);
 
public:
 
    TRawMessageHandler(HandlerType* InHandler, FuncType InFunc)
        : Handler(InHandler)
        , Func(InFunc)
    {
        check(InHandler != nullptr);
    }
 
    ~TRawMessageHandler() { }
 
public:
 
    //~ IMessageHandler interface
    
    virtual void HandleMessage(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context) override
    {
        UStruct* Struct = MessageType::StaticStruct();
        if (Struct && Context->GetMessageTypePathName() == Struct->GetStructPathName())
        {
            (Handler->*Func)(*static_cast<const MessageType*>(Context->GetMessage()), Context);
        }   
    }
    
private:
 
    HandlerType* Handler;
 
    FuncType Func;
};
 
 
class FFunctionMessageCatchall
    : public IMessageHandler
{
public:
 
    typedef TFunction<void(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>&)> FuncType;
 
public:
 
    FFunctionMessageCatchall(FuncType InFunc)
        : Func(MoveTemp(InFunc))
    { }
 
    ~FFunctionMessageCatchall() { }
 
public:
 
    //~ IMessageHandler interface
    
    virtual void HandleMessage(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context) override
    {
        Func(Context);
    }
    
private:
 
    FuncType Func;
};
 
 
template<typename MessageType>
class TFunctionMessageHandler
    : public IMessageHandler
{
public:
 
    typedef TFunction<void(const MessageType&, const TSharedRef<IMessageContext, ESPMode::ThreadSafe>&)> FuncType;
 
public:
 
    TFunctionMessageHandler(FuncType InFunc)
        : Func(MoveTemp(InFunc))
    { }
 
    ~TFunctionMessageHandler() { }
 
public:
 
    //~ IMessageHandler interface
    
    virtual void HandleMessage(const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context) override
    {
        if (Context->GetMessageTypePathName() == MessageType::StaticStruct()->GetStructPathName())
        {
            Func(*static_cast<const MessageType*>(Context->GetMessage()), Context);
        }   
    }
    
private:
 
    FuncType Func;
};