NetBlobAssembler.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
#include "Iris/ReplicationSystem/NetBlob/PartialNetBlob.h"
#include "Iris/Serialization/NetBitStreamWriter.h"
 
class USequentialPartialNetBlobHandlerConfig;
 
namespace UE::Net
{
 
struct FNetBlobAssemblerInitParams
{
    const USequentialPartialNetBlobHandlerConfig* PartialNetBlobHandlerConfig = nullptr;
};
 
class FNetBlobAssembler
{
public:
    IRISCORE_API FNetBlobAssembler();
 
    IRISCORE_API void Init(const FNetBlobAssemblerInitParams& InitParams);
 
    IRISCORE_API void AddPartialNetBlob(FNetSerializationContext& Context, FNetRefHandle RefHandle, const TRefCountPtr<FPartialNetBlob>& PartialNetBlob);
 
    bool IsReadyToAssemble() const { return bIsReadyToAssemble; }
 
    bool IsSequenceBroken() const { return bIsBrokenSequence; }
 
    IRISCORE_API TRefCountPtr<FNetBlob> Assemble(FNetSerializationContext& Context);
 
private:
    FNetBlobCreationInfo NetBlobCreationInfo;
    TArray<uint32> Payload;
    FNetBitStreamWriter BitWriter;
    FNetRefHandle RefHandle;
    const USequentialPartialNetBlobHandlerConfig* PartialNetBlobHandlerConfig = nullptr;
    uint32 NextPartIndex = 0;
    // Initialized to an arbitrary number. If we receive the first part of a sequence it doesn't matter. If the next received blob isn't the first part it's a broken sequence.
    uint32 NextSequenceSumber = 0U;
    // The sequence number of the last part expected to be received before being ready to assemble.
    uint32 LastPartSequenceNumber = ~0U;
    uint32 FirstPayloadBitCount = 0;
    bool bIsReadyToAssemble = false;
    bool bIsBrokenSequence = false;
    bool bIsProcessingReliable = false;
};
 
}