RuntimeAssetCacheBucket.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Misc/AssertionMacros.h"
#include "Containers/UnrealString.h"
#include "Containers/Map.h"
#include "Misc/ScopeLock.h"
 
class FCacheEntryMetadata;
 
class FCacheEntryMetadata;
 
class FRuntimeAssetCacheBucket
{
public:
    FRuntimeAssetCacheBucket()
        : Size(DefaultBucketSize)
        , CurrentSize(0)
    { }
 
    FRuntimeAssetCacheBucket(int32 InSize)
        : Size(InSize)
        , CurrentSize(0)
    { }
 
    ~FRuntimeAssetCacheBucket();
 
    int32 GetSize() const
    {
        return Size;
    }
 
    void Reset();
 
    FCacheEntryMetadata* GetMetadata(const FString& Key);
 
    void RemoveMetadataEntry(const FString& Key, bool bBuildFailed = false);
 
    void AddMetadataEntry(const FString& Key, FCacheEntryMetadata* Value, bool bUpdateSize);
 
    int32 GetCurrentSize()
    {
        FScopeLock Lock(&CurrentSizeCriticalSection);
        return CurrentSize;
    }
 
    void AddToCurrentSize(int32 Value)
    {
        FScopeLock Lock(&CurrentSizeCriticalSection);
        CurrentSize += Value;
        check(CurrentSize <= Size);
    }
 
    FCacheEntryMetadata* GetOldestEntry();
 
    static const int32 DefaultBucketSize;
 
private:
    TMap<FString, FCacheEntryMetadata*> CacheMetadata;
 
    FCriticalSection MetadataCriticalSection;
 
    int32 Size;
 
    int32 CurrentSize;
 
    FCriticalSection CurrentSizeCriticalSection;
 
    friend class FRuntimeAssetCacheBucketScopeLock;
};