MessageAddressBook.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 "Misc/ScopeLock.h"
 
class FMessageAddressBook
{
public:
 
    FMessageAddressBook()
    {
        CriticalSection = new FCriticalSection();
    }
 
    ~FMessageAddressBook()
    {
        delete CriticalSection;
    }
 
public:
 
    void Add(const FMessageAddress& Address, const FGuid& NodeId)
    {
        FScopeLock Lock(CriticalSection);
 
        Entries.FindOrAdd(Address) = NodeId;
    }
 
    void Clear()
    {
        FScopeLock Lock(CriticalSection);
 
        Entries.Reset();
    }
 
    bool Contains(const FMessageAddress& Address)
    {
        FScopeLock Lock(CriticalSection);
 
        return Entries.Contains(Address);
    }
 
    TArray<FGuid> GetNodesFor(const TArray<FMessageAddress>& Addresses)
    {
        TArray<FGuid> FoundNodes;
 
        FScopeLock Lock(CriticalSection);
 
        for (const auto& Address : Addresses)
        {
            FGuid* NodeId = Entries.Find(Address);
 
            if (NodeId != nullptr)
            {
                FoundNodes.AddUnique(*NodeId);
            }
        }
 
        return FoundNodes;
    }
 
    void RemoveAll(TArray<FMessageAddress>& OutRemovedAddresses)
    {
        OutRemovedAddresses.Reset();
 
        FScopeLock Lock(CriticalSection);
 
        Entries.GenerateKeyArray(OutRemovedAddresses);
        Entries.Reset();
    }
 
    void RemoveNode(const FGuid& NodeId, TArray<FMessageAddress>& OutRemovedAddresses)
    {
        OutRemovedAddresses.Reset();
 
        FScopeLock Lock(CriticalSection);
 
        for (const auto& EntryPair : Entries)
        {
            if (EntryPair.Value == NodeId)
            {
                OutRemovedAddresses.Add(EntryPair.Key);
            }
        }
 
        for (const auto& Address : OutRemovedAddresses)
        {
            Entries.Remove(Address);
        }
    }
 
private:
 
    FCriticalSection* CriticalSection;
 
    TMap<FMessageAddress, FGuid> Entries;
};