Files
UnrealEngineUWP/Engine/Source/Developer/IoStoreUtilities/Internal/ZenStoreHttpClient.h
Per Larsson ddbf9eb5d2 Zen - cook to custom project/oplog ID with -ZenStoreProject=<ProjectId> and -ZenStorePlatform=<OplogId>
#rb none
#rnx
#preflight 19755222

[CL 19804798 by Per Larsson in ue5-main branch]
2022-04-19 02:37:06 -04:00

88 lines
2.7 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "Async/Future.h"
#include "Containers/Array.h"
#include "Containers/StringView.h"
#include "HAL/CriticalSection.h"
#include "HAL/Platform.h"
#include "IO/IoDispatcher.h"
#include "Misc/StringBuilder.h"
#include "Serialization/CompactBinaryPackage.h"
#include "Templates/UniquePtr.h"
#include "ZenServerInterface.h"
class FCbPackage;
class FCbObject;
namespace UE {
namespace Zen {
struct FZenHttpRequestPool;
}
/**
* HTTP protocol implementation of Zen Store client interface
*/
class IOSTOREUTILITIES_API FZenStoreHttpClient
{
public:
FZenStoreHttpClient();
FZenStoreHttpClient(FStringView HostName, uint16 Port);
FZenStoreHttpClient(UE::Zen::FServiceSettings&& InSettings);
~FZenStoreHttpClient();
bool TryCreateProject(FStringView InProjectId, FStringView InOplogId, FStringView ServerRoot,
FStringView EngineRoot, FStringView ProjectRoot);
bool TryCreateOplog(FStringView InProjectId, FStringView InOplogId, bool bFullBuild);
void InitializeReadOnly(FStringView InProjectId, FStringView InOplogId);
bool IsConnected() const;
void StartBuildPass();
TIoStatusOr<uint64> EndBuildPass(FCbPackage OpEntry);
TIoStatusOr<uint64> AppendOp(FCbPackage OpEntry);
TIoStatusOr<uint64> GetChunkSize(const FIoChunkId& Id);
TIoStatusOr<FIoBuffer> ReadChunk(const FIoChunkId& Id, uint64 Offset = 0, uint64 Size = ~0ull);
TIoStatusOr<FIoBuffer> ReadOpLogAttachment(FStringView Id);
#if UE_WITH_ZEN
const TCHAR* GetHostName() const { return ZenService.GetInstance().GetHostName(); }
uint16 GetPort() const { return ZenService.GetInstance().GetPort(); }
const UE::Zen::FZenServiceInstance& GetZenServiceInstance() const { return ZenService.GetInstance(); }
#else // Default to localhost:1337 for platforms where Zen wouldn't be supported yet
const TCHAR* GetHostName() const { return TEXT("localhost"); }
uint16 GetPort() const { return 1337; }
#endif
TFuture<TIoStatusOr<FCbObject>> GetOplog();
TFuture<TIoStatusOr<FCbObject>> GetFiles();
static const UTF8CHAR* FindOrAddAttachmentId(FUtf8StringView AttachmentText);
static const UTF8CHAR* FindAttachmentId(FUtf8StringView AttachmentText);
static FString GetProjectId();
static FString GenerateDefaultProjectId();
private:
TIoStatusOr<FIoBuffer> ReadOpLogUri(FStringBuilderBase& ChunkUri, uint64 Offset = 0, uint64 Size = ~0ull);
static const uint32 PoolEntryCount;
#if UE_WITH_ZEN
UE::Zen::FScopeZenService ZenService;
#endif
TUniquePtr<Zen::FZenHttpRequestPool> RequestPool;
FString OplogPath;
FString OplogNewEntryPath;
FString OplogPrepNewEntryPath;
FString TempDirPath;
uint64 StandaloneThresholdBytes = 1 * 1024 * 1024;
bool bAllowRead = false;
bool bAllowEdit = false;
bool bConnectionSucceeded = false;
};
}