SharedConnectionFilterStatus.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "HAL/Platform.h"
#include "Iris/IrisConstants.h"
#include "Iris/ReplicationSystem/ReplicationView.h" // for UE_IRIS_INLINE_VIEWS_PER_CONNECTION
#include "Iris/ReplicationSystem/Filtering/NetObjectFilter.h"
#include "Net/Core/Connection/ConnectionHandle.h"
#include "Containers/Map.h"
#include "Containers/Set.h"
 
namespace UE::Net
{
 
class FSharedConnectionFilterStatus
{
public:
    IRISCORE_API bool SetFilterStatus(FConnectionHandle ConnectionHandle, ENetFilterStatus FilterStatus);
    
    ENetFilterStatus GetFilterStatus() const;
    
    IRISCORE_API void RemoveConnection(FConnectionHandle ConnectionHandle);
 
    uint32 GetParentConnectionId() const;
 
private:
    // Only keep track of connections which allow replication as the replication status is Disallow by default.
    TSet<uint32, DefaultKeyFuncs<uint32>, TInlineSetAllocator<UE_IRIS_INLINE_VIEWS_PER_CONNECTION>> AllowConnections;
    uint32 ParentConnectionId = InvalidConnectionId;
};
 
class FSharedConnectionFilterStatusCollection
{
public:
    IRISCORE_API void SetFilterStatus(FConnectionHandle ConnectionHandle, ENetFilterStatus FilterStatus);
 
    IRISCORE_API ENetFilterStatus GetFilterStatus(uint32 ParentConnectionId) const;
 
    IRISCORE_API void RemoveConnection(FConnectionHandle ConnectionHandle);
 
private:
    FSharedConnectionFilterStatus* FindSharedConnectionFilterStatus(uint32 ParentConnectionId);
    const FSharedConnectionFilterStatus* FindSharedConnectionFilterStatus(uint32 ParentConnectionId) const;
    FSharedConnectionFilterStatus& FindOrAddSharedConnectionFilterStatus(uint32 ParentConnectionId);
 
    TMap<uint32, FSharedConnectionFilterStatus> ParentToFilterStatus;
};
 
inline ENetFilterStatus FSharedConnectionFilterStatus::GetFilterStatus() const
{
    return AllowConnections.IsEmpty() ? ENetFilterStatus::Disallow : ENetFilterStatus::Allow;
}
 
inline uint32 FSharedConnectionFilterStatus::GetParentConnectionId() const
{
    return ParentConnectionId;
}
 
}