NetBlobHandler.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
#include "CoreTypes.h"
#include "Iris/ReplicationSystem/NetBlob/NetBlob.h"
#include "Templates/RefCounting.h"
#include "NetBlobHandler.generated.h"
 
namespace UE::Net::Private
{
    class FNetBlobHandlerManager;
}
 
namespace UE::Net
{
 
IRISCORE_API extern const FName GNetError_UnsupportedNetBlob;
 
}
 
class INetBlobReceiver
{
protected:
    using FNetBlobCreationInfo = UE::Net::FNetBlobCreationInfo;
    using FNetBlob = UE::Net::FNetBlob;
 
public:
    virtual TRefCountPtr<UE::Net::FNetBlob> CreateNetBlob(const FNetBlobCreationInfo&) const = 0;
    virtual void OnNetBlobReceived(UE::Net::FNetSerializationContext& Context, const TRefCountPtr<FNetBlob>&) = 0;
 
};
 
UCLASS(transient, MinimalApi, Abstract)
class UNetBlobHandler : public UObject, public INetBlobReceiver
{
    GENERATED_BODY()
 
public:
    virtual ~UNetBlobHandler();
 
    TRefCountPtr<FNetBlob> CreateNetBlob(UE::Net::ENetBlobFlags Flags) const;
 
    UE::Net::FNetBlobType GetNetBlobType() const { return NetBlobType; }
 
    IRISCORE_API virtual void AddConnection(uint32 ConnectionId);
 
    IRISCORE_API virtual void RemoveConnection(uint32 ConnectionId);
 
protected:
    UNetBlobHandler();  
 
private:
    // INetBlobReceiver
 
    virtual TRefCountPtr<FNetBlob> CreateNetBlob(const FNetBlobCreationInfo&) const override PURE_VIRTUAL(CreateNetBlob, return nullptr;);
 
    virtual void OnNetBlobReceived(UE::Net::FNetSerializationContext& Context, const TRefCountPtr<FNetBlob>&) override PURE_VIRTUAL(OnNetBlobReceived,);
 
private:
    friend class UE::Net::Private::FNetBlobHandlerManager;
 
    UE::Net::FNetBlobType NetBlobType;
};
 
inline TRefCountPtr<UE::Net::FNetBlob> UNetBlobHandler::CreateNetBlob(UE::Net::ENetBlobFlags Flags) const
{
    FNetBlobCreationInfo CreationInfo;
    CreationInfo.Flags = Flags;
    CreationInfo.Type = GetNetBlobType();
    return CreateNetBlob(CreationInfo);
}