RuntimeAssetCache.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "UObject/NameTypes.h"
#include "HAL/ThreadSafeCounter.h"
#include "Stats/Stats.h"
#include "RuntimeAssetCacheInterface.h"
#include "RuntimeAssetCacheAsyncWorker.h"
 
class FRuntimeAssetCacheBucket;
class IRuntimeAssetCacheBuilder;
 
DECLARE_FLOAT_ACCUMULATOR_STAT_EXTERN(TEXT("RAC ASync Wait Time"), STAT_RAC_ASyncWaitTime, STATGROUP_RAC, );
 
class IRuntimeAssetCacheBuilder;
class IRuntimeAssetCacheBackend;
class FCacheEntryMetadata;
class FRuntimeAssetCacheBucket;
 
class FRuntimeAssetCache final : public FRuntimeAssetCacheInterface
{
public:
    virtual FVoidPtrParam GetSynchronous(IRuntimeAssetCacheBuilder* CacheBuilder) override;
    virtual int32 GetAsynchronous(IRuntimeAssetCacheBuilder* CacheBuilder, const FOnRuntimeAssetCacheAsyncComplete& OnCompletionDelegate) override;
    virtual int32 GetAsynchronous(IRuntimeAssetCacheBuilder* CacheBuilder) override;
    virtual int32 GetCacheSize(FName Bucket) const override;
    virtual bool ClearCache() override;
    virtual bool ClearCache(FName Bucket) override;
    virtual void WaitAsynchronousCompletion(int32 Handle) override;
    virtual FVoidPtrParam GetAsynchronousResults(int32 Handle) override;
    virtual bool PollAsynchronousCompletion(int32 Handle) override;
    virtual void AddToAsyncCompletionCounter(int32 Addend) override;
    virtual void Tick() override;
public:
    FRuntimeAssetCache();
 
    ~FRuntimeAssetCache();
 
private:
 
    int32 GetNextHandle()
    {
        return CurrentAsyncTaskHandle.Increment();
    }
 
    TMap<FName, FRuntimeAssetCacheBucket*> Buckets;
 
    TMap<int32, FAsyncTask<FRuntimeAssetCacheAsyncWorker>*> PendingTasks;
 
    FCriticalSection SynchronizationObject;
 
    FThreadSafeCounter CurrentAsyncTaskHandle;
 
    FThreadSafeCounter PendingTasksCounter;
};