InstallerSharedContext.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "HAL/Runnable.h"
#include "Containers/MpscQueue.h"
#include "Interfaces/IBuildInstallerSharedContext.h"
 
namespace BuildPatchServices
{
    class FBuildInstallerThread
        : public IBuildInstallerThread
        , public FRunnable
    {
        enum class EMsg : uint8
        {
            None = 0,
            RunTask = (1 << 0),
            Exit = (1 << 1)
        };
        FRIEND_ENUM_CLASS_FLAGS(EMsg);
 
        struct FMsg
        {
            FMsg() = default;
            FMsg(TUniqueFunction<void()> InTask, EMsg InMsg)
                : Task(MoveTemp(InTask))
                , Msg(InMsg)
            {}
 
            TUniqueFunction<void()> Task;
            EMsg Msg = EMsg::None;
        };
 
    public:
        ~FBuildInstallerThread();
 
        bool StartThread(const TCHAR* DebugName);
 
        // IBuildInstallerThread interface begin
        virtual void RunTask(TUniqueFunction<void()> Task) override;
        // IBuildInstallerThread interface end
 
        // FRunnable interface begin
        virtual uint32 Run() override;
        virtual void Stop() override;
        // FRunnable interface end
 
    private:
        FEvent* DoWorkEvent = nullptr;
        FRunnableThread* Thread = nullptr;
        TMpscQueue<FMsg> MsgQueue;
    };
 
    class FBuildInstallerSharedContext : public IBuildInstallerSharedContext
    {
    public:
        FBuildInstallerSharedContext(const TCHAR* InDebugName) : DebugName(InDebugName) {}
        ~FBuildInstallerSharedContext();
 
        // IBuildInstallerSharedContext interface begin
        virtual IBuildInstallerThread* CreateThread() override;
        virtual void ReleaseThread(IBuildInstallerThread* Thread) override;
        virtual void PreallocateThreads(uint32 NumThreads) override;
        virtual uint32 NumThreadsPerInstaller(bool bUseChunkDBs, bool bHasInstallSource) const override;
        // IBuildInstallerSharedContext interface end
 
    private:
        IBuildInstallerThread* CreateThreadInternal();
 
    private:
        TArray<IBuildInstallerThread*> ThreadFreeList;
        FCriticalSection ThreadFreeListCS;
        const TCHAR* DebugName;
        uint32 ThreadCount = 0;
        bool bWarnOnCreateThread = false;
    };
 
    class FBuildInstallerSharedContextFactory
    {
    public:
        static IBuildInstallerSharedContextRef Create(const TCHAR* DebugName);
    };
}