SessionManager.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Misc/Guid.h"
#include "IMessageContext.h"
#include "IMessageBus.h"
#include "MessageEndpoint.h"
#include "ISessionInstanceInfo.h"
#include "ISessionInfo.h"
#include "SessionLogMessage.h"
#include "ISessionManager.h"
#include "Containers/Ticker.h"
 
class FSessionInfo;
struct FEngineServicePong;
struct FSessionServicePong;
 
class FSessionManager
    : public TSharedFromThis<FSessionManager>
    , public ISessionManager
{
public:
 
    FSessionManager(const TSharedRef<IMessageBus, ESPMode::ThreadSafe>& InMessageBus);
    
    ~FSessionManager();
 
public:
 
    //~ ISessionManager interface
 
    virtual void AddOwner(const FString& InOwner) override;
    virtual const TArray<TSharedPtr<ISessionInstanceInfo>>& GetSelectedInstances() const override;
    virtual const TSharedPtr<ISessionInfo>& GetSelectedSession() const override;
    virtual void GetSessions(TArray<TSharedPtr<ISessionInfo>>& OutSessions) const override;
    virtual bool IsInstanceSelected(const TSharedRef<ISessionInstanceInfo>& Instance) const override;
 
    DECLARE_DERIVED_EVENT(FSessionManager, ISessionManager::FCanSelectSessionEvent, FCanSelectSessionEvent)
    virtual FCanSelectSessionEvent& OnCanSelectSession() override
    {
        return CanSelectSessionDelegate;
    }
 
    DECLARE_DERIVED_EVENT(FSessionManager, ISessionManager::FInstanceSelectionChangedEvent, FInstanceSelectionChangedEvent)
    virtual FInstanceSelectionChangedEvent& OnInstanceSelectionChanged() override
    {
        return InstanceSelectionChangedDelegate;
    }
 
    DECLARE_DERIVED_EVENT(FSessionManager, ISessionManager::FLogReceivedEvent, FLogReceivedEvent)
    virtual FLogReceivedEvent& OnLogReceived() override
    {
        return LogReceivedEvent;
    }
 
    DECLARE_DERIVED_EVENT(FSessionManager, ISessionManager::FSelectedSessionChangedEvent, FSelectedSessionChangedEvent)
    virtual FSelectedSessionChangedEvent& OnSelectedSessionChanged() override
    {
        return SelectedSessionChangedEvent;
    }
 
    virtual FSimpleMulticastDelegate& OnSessionsUpdated() override;
    virtual FSimpleMulticastDelegate& OnSessionInstanceUpdated() override;
    virtual void RemoveOwner(const FString& InOwner) override;
    virtual bool SelectSession(const TSharedPtr<ISessionInfo>& Session) override;
    virtual bool SetInstanceSelected(const TSharedRef<ISessionInstanceInfo>& Instance, bool Selected) override;
 
protected:
 
    void FindExpiredSessions(const FDateTime& Now);
 
    bool IsValidOwner(const FString& Owner);
 
    void RefreshSessions();
 
    void SendPing();
 
private:
 
    void HandleEnginePongMessage(const FEngineServicePong& Message, const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context);
 
    void HandleLogReceived(const TSharedRef<ISessionInfo>& Session, const TSharedRef<ISessionInstanceInfo>& Instance, const TSharedRef<FSessionLogMessage>& Message);
 
    void HandleSessionPongMessage(const FSessionServicePong& Message, const TSharedRef<IMessageContext, ESPMode::ThreadSafe>& Context);
 
    bool HandleTicker(float DeltaTime);
 
    virtual TSharedPtr<ISessionInstanceInfo> GetInstance(const FGuid& Id) const override;
 
private:
 
    FGuid AutomationControllerAddress;
 
    FDateTime LastPingTime;
 
    TWeakPtr<IMessageBus, ESPMode::ThreadSafe> MessageBusPtr;
 
    TSharedPtr<FMessageEndpoint, ESPMode::ThreadSafe> MessageEndpoint;
 
    TArray<TSharedPtr<ISessionInstanceInfo>> SelectedInstances;
 
    TSharedPtr<ISessionInfo> SelectedSession;
 
    TMap<FGuid, TSharedPtr<FSessionInfo>> Sessions;
 
private:
 
    FCanSelectSessionEvent CanSelectSessionDelegate;
 
    FInstanceSelectionChangedEvent InstanceSelectionChangedDelegate;
 
    TArray<FString> FilteredOwners;
 
    FLogReceivedEvent LogReceivedEvent;
 
    FSelectedSessionChangedEvent SelectedSessionChangedEvent;
 
    FSimpleMulticastDelegate SessionsUpdatedDelegate;
 
    FSimpleMulticastDelegate SessionInstanceUpdatedDelegate;
 
    FTSTicker::FDelegateHandle TickDelegateHandle;
};