Files
UnrealEngineUWP/Engine/Source/Runtime/StorageServerClient/Private/StorageServerPackageStore.cpp
aurel cordonnier 34f55d3a4a Merge from Release-Engine-Test @ 17946149 to UE5/Main
This represents UE4/Main @17911760, Release-5.0 @17915875 and Dev-PerfTest @17914035

[CL 17949667 by aurel cordonnier in ue5-main branch]
2021-10-27 15:14:40 -04:00

61 lines
2.2 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "StorageServerPackageStore.h"
#include "IO/IoContainerHeader.h"
#include "IO/IoDispatcher.h"
#include "StorageServerConnection.h"
#include "Serialization/MemoryReader.h"
#if !UE_BUILD_SHIPPING
FStorageServerPackageStore::FStorageServerPackageStore(FStorageServerConnection& Connection)
{
FIoChunkId HeaderChunkId = CreateIoChunkId(FIoContainerId::FromName(TEXT("global")).Value(), 0, EIoChunkType::ContainerHeader);
Connection.ReadChunkRequest(HeaderChunkId, 0, uint64(-1), [this](FStorageServerResponse& Response)
{
FIoBuffer Chunk;
if (Response.SerializeChunk(Chunk))
{
FMemoryReaderView Ar(MakeArrayView(reinterpret_cast<const uint8*>(Chunk.Data()), Chunk.DataSize()));
FIoContainerHeader ContainerHeader;
Ar << ContainerHeader;
StoreEntriesData = MoveTemp(ContainerHeader.StoreEntries);
StoreEntriesMap.Reserve(ContainerHeader.PackageCount);
TArrayView<const FFilePackageStoreEntry> StoreEntries(reinterpret_cast<const FFilePackageStoreEntry*>(StoreEntriesData.GetData()), ContainerHeader.PackageCount);
int32 Index = 0;
for (const FFilePackageStoreEntry& StoreEntry : StoreEntries)
{
const FPackageId& PackageId = ContainerHeader.PackageIds[Index];
check(PackageId.IsValid());
StoreEntriesMap.FindOrAdd(PackageId, &StoreEntry);
++Index;
}
}
});
}
bool FStorageServerPackageStore::DoesPackageExist(FPackageId PackageId)
{
return PackageId.IsValid() && StoreEntriesMap.Contains(PackageId);
}
EPackageStoreEntryStatus FStorageServerPackageStore::GetPackageStoreEntry(FPackageId PackageId, FPackageStoreEntry& OutPackageStoreEntry)
{
const FFilePackageStoreEntry* FindEntry = StoreEntriesMap.FindRef(PackageId);
if (FindEntry)
{
OutPackageStoreEntry.ExportInfo.ExportCount = FindEntry->ExportCount;
OutPackageStoreEntry.ExportInfo.ExportBundleCount = FindEntry->ExportBundleCount;
OutPackageStoreEntry.ImportedPackageIds = MakeArrayView(FindEntry->ImportedPackages.Data(), FindEntry->ImportedPackages.Num());
OutPackageStoreEntry.ShaderMapHashes = MakeArrayView(FindEntry->ShaderMapHashes.Data(), FindEntry->ShaderMapHashes.Num());
return EPackageStoreEntryStatus::Ok;
}
else
{
return EPackageStoreEntryStatus::Missing;
}
}
#endif