src/Components/Server/src/Circuits/CircuitHandleRegistry.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

namespace Microsoft.AspNetCore.Components.Server.Circuits;

internal sealed class CircuitHandleRegistry : ICircuitHandleRegistry
{
    public CircuitHandle GetCircuitHandle(IDictionary<object, object?> circuitHandles, object circuitKey)
    {
        if (circuitHandles.TryGetValue(circuitKey, out var circuitHandle))
        {
            return (CircuitHandle)circuitHandle;
        }

        return null; ;
    }

    public CircuitHost GetCircuit(IDictionary<object, object?> circuitHandles, object circuitKey)
    {
        if (circuitHandles.TryGetValue(circuitKey, out var circuitHandle))
        {
            return ((CircuitHandle)circuitHandle).CircuitHost;
        }

        return null;
    }

    public void SetCircuit(IDictionary<object, object?> circuitHandles, object circuitKey, CircuitHost circuitHost)
    {
        circuitHandles[circuitKey] = circuitHost?.Handle;
    }
}