ReplicationStateDescriptorRegistry.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreTypes.h"
#include "Containers/Array.h"
#include "Containers/Map.h"
#include "Templates/RefCounting.h"
#include "UObject/Field.h"
#include "UObject/WeakObjectPtr.h"
#include "UObject/ObjectKey.h"
 
namespace UE::Net
{
    struct FReplicationStateDescriptor;
    namespace Private
    {
        class FReplicationProtocolManager;
    }
}
 
namespace UE::Net::Private
{
 
struct FReplicationStateDescriptorRegistryInitParams
{
    FReplicationProtocolManager* ProtocolManager;
};
 
class FReplicationStateDescriptorRegistry
{
public:
    FReplicationStateDescriptorRegistry();
 
    void Init(const FReplicationStateDescriptorRegistryInitParams& Params);
 
    typedef TArray<TRefCountPtr<const FReplicationStateDescriptor>> FDescriptors;
 
    const FDescriptors* Find(const FFieldVariant& Object, const UObject* ObjectForPruning) const;
 
    const FDescriptors* Find(const FFieldVariant& Object) const { return Find(Object, GetObjectForPruning(Object)); }
 
    void Register(const FFieldVariant& DescriptorKey, const UObject* ObjectForPruning, const FDescriptors& Descriptors);
 
    void Register(const FFieldVariant& DescriptorKey, const FDescriptors& Descriptors) { Register(DescriptorKey, GetObjectForPruning(DescriptorKey), Descriptors); }
 
    void Register(const FFieldVariant& DescriptorKey, const UObject* ObjectForPruning, const TRefCountPtr<const FReplicationStateDescriptor>& Descriptor);
 
    void Register(const FFieldVariant& DescriptorKey, const TRefCountPtr<const FReplicationStateDescriptor>& Descriptor) { Register(DescriptorKey, GetObjectForPruning(DescriptorKey), Descriptor); }
 
    void PruneStaleDescriptors();
 
private:
    static const UObject* GetObjectForPruning(const FFieldVariant& Object);
 
    void InvalidateDescriptors(const FDescriptors& Descriptors) const;
 
    struct FRegisteredDescriptors
    {
        TWeakObjectPtr<const UObject> WeakPtrForPruning;
        FDescriptors Descriptors;
        FObjectKey OwnerKey;
    };
 
    typedef TMap<FFieldVariant, FRegisteredDescriptors> FClassToDescriptorMap;
    FClassToDescriptorMap RegisteredDescriptorsMap;
    FReplicationProtocolManager* ProtocolManager;
};
 
}