Initial copy of Microsoft.AspNetCore.SignalR.StackExchangeRedis

Author: ascott18

IntelliTect.SignalR.SqlServer.sln

@@ -0,0 +1,39 @@
+
+Microsoft Visual Studio Solution File, Format Version 12.00
+# Visual Studio Version 16
+VisualStudioVersion = 16.0.31424.327
+MinimumVisualStudioVersion = 10.0.40219.1
+Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "demo", "demo", "{62241700-24B6-4D13-A183-2502D628ED43}"
+EndProject
+Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "src", "src", "{21874FF3-BE97-4F3B-AD63-EC347FF757FF}"
+EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "IntelliTect.SignalR.SqlServer", "src\IntelliTect.SignalR.SqlServer\IntelliTect.SignalR.SqlServer.csproj", "{62E576F6-E0F9-43AC-848E-97E7E93E4EF0}"
+EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "DemoServer", "demo\DemoServer\DemoServer.csproj", "{FACCC87C-37E3-4095-9997-75EE3C54CD49}"
+EndProject
+Global
+	GlobalSection(SolutionConfigurationPlatforms) = preSolution
+		Debug|Any CPU = Debug|Any CPU
+		Release|Any CPU = Release|Any CPU
+	EndGlobalSection
+	GlobalSection(ProjectConfigurationPlatforms) = postSolution
+		{62E576F6-E0F9-43AC-848E-97E7E93E4EF0}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{62E576F6-E0F9-43AC-848E-97E7E93E4EF0}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{62E576F6-E0F9-43AC-848E-97E7E93E4EF0}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{62E576F6-E0F9-43AC-848E-97E7E93E4EF0}.Release|Any CPU.Build.0 = Release|Any CPU
+		{FACCC87C-37E3-4095-9997-75EE3C54CD49}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{FACCC87C-37E3-4095-9997-75EE3C54CD49}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{FACCC87C-37E3-4095-9997-75EE3C54CD49}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{FACCC87C-37E3-4095-9997-75EE3C54CD49}.Release|Any CPU.Build.0 = Release|Any CPU
+	EndGlobalSection
+	GlobalSection(SolutionProperties) = preSolution
+		HideSolutionNode = FALSE
+	EndGlobalSection
+	GlobalSection(NestedProjects) = preSolution
+		{62E576F6-E0F9-43AC-848E-97E7E93E4EF0} = {21874FF3-BE97-4F3B-AD63-EC347FF757FF}
+		{FACCC87C-37E3-4095-9997-75EE3C54CD49} = {62241700-24B6-4D13-A183-2502D628ED43}
+	EndGlobalSection
+	GlobalSection(ExtensibilityGlobals) = postSolution
+		SolutionGuid = {41DEFC27-E139-4A35-989F-DA872E4B9F0A}
+	EndGlobalSection
+EndGlobal

src/IntelliTect.SignalR.SqlServer/IntelliTect.SignalR.SqlServer.csproj

@@ -0,0 +1,15 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+  <PropertyGroup>
+    <TargetFramework>net5.0</TargetFramework>
+  </PropertyGroup>
+
+  <ItemGroup>
+    <PackageReference Include="Microsoft.Data.SqlClient" Version="3.0.0" />
+    <PackageReference Include="Microsoft.Orleans.Clustering.AdoNet" Version="3.5.0" />
+    <PackageReference Include="Microsoft.Orleans.OrleansRuntime" Version="3.5.0" />
+    <PackageReference Include="Microsoft.Orleans.Persistence.AdoNet" Version="3.5.0" />
+    <PackageReference Include="SignalR.Orleans" Version="2.0.1" />
+  </ItemGroup>
+
+</Project>

src/IntelliTect.SignalR.SqlServer/Internal/AckHandler.cs

@@ -0,0 +1,99 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+using System;
+using System.Collections.Concurrent;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.Internal;
+
+namespace Microsoft.AspNetCore.SignalR.StackExchangeRedis.Internal
+{
+    internal class AckHandler : IDisposable
+    {
+        private readonly ConcurrentDictionary<int, AckInfo> _acks = new ConcurrentDictionary<int, AckInfo>();
+        private readonly Timer _timer;
+        private readonly TimeSpan _ackThreshold = TimeSpan.FromSeconds(30);
+        private readonly TimeSpan _ackInterval = TimeSpan.FromSeconds(5);
+        private readonly object _lock = new object();
+        private bool _disposed;
+
+        public AckHandler()
+        {
+            _timer = NonCapturingTimer.Create(state => ((AckHandler)state!).CheckAcks(), state: this, dueTime: _ackInterval, period: _ackInterval);
+        }
+
+        public Task CreateAck(int id)
+        {
+            lock (_lock)
+            {
+                if (_disposed)
+                {
+                    return Task.CompletedTask;
+                }
+
+                return _acks.GetOrAdd(id, _ => new AckInfo()).Tcs.Task;
+            }
+        }
+
+        public void TriggerAck(int id)
+        {
+            if (_acks.TryRemove(id, out var ack))
+            {
+                ack.Tcs.TrySetResult();
+            }
+        }
+
+        private void CheckAcks()
+        {
+            if (_disposed)
+            {
+                return;
+            }
+
+            var utcNow = DateTime.UtcNow;
+
+            foreach (var pair in _acks)
+            {
+                var elapsed = utcNow - pair.Value.Created;
+                if (elapsed > _ackThreshold)
+                {
+                    if (_acks.TryRemove(pair.Key, out var ack))
+                    {
+                        ack.Tcs.TrySetCanceled();
+                    }
+                }
+            }
+        }
+
+        public void Dispose()
+        {
+            lock (_lock)
+            {
+                _disposed = true;
+
+                _timer.Dispose();
+
+                foreach (var pair in _acks)
+                {
+                    if (_acks.TryRemove(pair.Key, out var ack))
+                    {
+                        ack.Tcs.TrySetCanceled();
+                    }
+                }
+            }
+        }
+
+        private class AckInfo
+        {
+            public TaskCompletionSource Tcs { get; private set; }
+            public DateTime Created { get; private set; }
+
+            public AckInfo()
+            {
+                Created = DateTime.UtcNow;
+                Tcs = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
+            }
+        }
+    }
+}

src/IntelliTect.SignalR.SqlServer/Internal/DefaultHubMessageSerializer.cs

@@ -0,0 +1,40 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using Microsoft.AspNetCore.SignalR.Protocol;
+
+namespace Microsoft.AspNetCore.SignalR.Internal
+{
+    internal class DefaultHubMessageSerializer
+    {
+        private readonly List<IHubProtocol> _hubProtocols;
+
+        public DefaultHubMessageSerializer(IHubProtocolResolver hubProtocolResolver, IList<string>? globalSupportedProtocols, IList<string>? hubSupportedProtocols)
+        {
+            var supportedProtocols = hubSupportedProtocols ?? globalSupportedProtocols ?? Array.Empty<string>();
+            _hubProtocols = new List<IHubProtocol>(supportedProtocols.Count);
+            foreach (var protocolName in supportedProtocols)
+            {
+                var protocol = hubProtocolResolver.GetProtocol(protocolName, (supportedProtocols as IReadOnlyList<string>) ?? supportedProtocols.ToList());
+                if (protocol != null)
+                {
+                    _hubProtocols.Add(protocol);
+                }
+            }
+        }
+
+        public IReadOnlyList<SerializedMessage> SerializeMessage(HubMessage message)
+        {
+            var list = new List<SerializedMessage>(_hubProtocols.Count);
+            foreach (var protocol in _hubProtocols)
+            {
+                list.Add(new SerializedMessage(protocol.Name, protocol.GetMessageBytes(message)));
+            }
+
+            return list;
+        }
+    }
+}

src/IntelliTect.SignalR.SqlServer/Internal/GroupAction.cs

@@ -0,0 +1,15 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+namespace Microsoft.AspNetCore.SignalR.StackExchangeRedis.Internal
+{
+    // The size of the enum is defined by the protocol. Do not change it. If you need more than 255 items,
+    // add an additional enum.
+    internal enum GroupAction : byte
+    {
+        // These numbers are used by the protocol, do not change them and always use explicit assignment
+        // when adding new items to this enum. 0 is intentionally omitted
+        Add = 1,
+        Remove = 2,
+    }
+}

src/IntelliTect.SignalR.SqlServer/Internal/RedisChannels.cs

@@ -0,0 +1,75 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+using System.Runtime.CompilerServices;
+
+namespace Microsoft.AspNetCore.SignalR.StackExchangeRedis.Internal
+{
+    internal class RedisChannels
+    {
+        private readonly string _prefix;
+
+        /// <summary>
+        /// Gets the name of the channel for sending to all connections.
+        /// </summary>
+        /// <remarks>
+        /// The payload on this channel is <see cref="RedisInvocation"/> objects containing
+        /// invocations to be sent to all connections
+        /// </remarks>
+        public string All { get; }
+
+        /// <summary>
+        /// Gets the name of the internal channel for group management messages.
+        /// </summary>
+        public string GroupManagement { get; }
+
+        public RedisChannels(string prefix)
+        {
+            _prefix = prefix;
+
+            All = prefix + ":all";
+            GroupManagement = prefix + ":internal:groups";
+        }
+
+        /// <summary>
+        /// Gets the name of the channel for sending a message to a specific connection.
+        /// </summary>
+        /// <param name="connectionId">The ID of the connection to get the channel for.</param>
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public string Connection(string connectionId)
+        {
+            return _prefix + ":connection:" + connectionId;
+        }
+
+        /// <summary>
+        /// Gets the name of the channel for sending a message to a named group of connections.
+        /// </summary>
+        /// <param name="groupName">The name of the group to get the channel for.</param>
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public string Group(string groupName)
+        {
+            return _prefix + ":group:" + groupName;
+        }
+
+        /// <summary>
+        /// Gets the name of the channel for sending a message to all collections associated with a user.
+        /// </summary>
+        /// <param name="userId">The ID of the user to get the channel for.</param>
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public string User(string userId)
+        {
+            return _prefix + ":user:" + userId;
+        }
+
+        /// <summary>
+        /// Gets the name of the acknowledgement channel for the specified server.
+        /// </summary>
+        /// <param name="serverName">The name of the server to get the acknowledgement channel for.</param>
+        /// <returns></returns>
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public string Ack(string serverName)
+        {
+            return _prefix + ":internal:ack:" + serverName;
+        }
+    }
+}

src/IntelliTect.SignalR.SqlServer/Internal/RedisGroupCommand.cs

@@ -0,0 +1,42 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+namespace Microsoft.AspNetCore.SignalR.StackExchangeRedis.Internal
+{
+    internal readonly struct RedisGroupCommand
+    {
+        /// <summary>
+        /// Gets the ID of the group command.
+        /// </summary>
+        public int Id { get; }
+
+        /// <summary>
+        /// Gets the name of the server that sent the command.
+        /// </summary>
+        public string ServerName { get; }
+
+        /// <summary>
+        /// Gets the action to be performed on the group.
+        /// </summary>
+        public GroupAction Action { get; }
+
+        /// <summary>
+        /// Gets the group on which the action is performed.
+        /// </summary>
+        public string GroupName { get; }
+
+        /// <summary>
+        /// Gets the ID of the connection to be added or removed from the group.
+        /// </summary>
+        public string ConnectionId { get; }
+
+        public RedisGroupCommand(int id, string serverName, GroupAction action, string groupName, string connectionId)
+        {
+            Id = id;
+            ServerName = serverName;
+            Action = action;
+            GroupName = groupName;
+            ConnectionId = connectionId;
+        }
+    }
+}

src/IntelliTect.SignalR.SqlServer/Internal/RedisInvocation.cs

@@ -0,0 +1,28 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+using System.Collections.Generic;
+using Microsoft.AspNetCore.SignalR.Protocol;
+
+namespace Microsoft.AspNetCore.SignalR.StackExchangeRedis.Internal
+{
+    internal readonly struct RedisInvocation
+    {
+        /// <summary>
+        /// Gets a list of connections that should be excluded from this invocation.
+        /// May be null to indicate that no connections are to be excluded.
+        /// </summary>
+        public IReadOnlyList<string>? ExcludedConnectionIds { get; }
+
+        /// <summary>
+        /// Gets the message serialization cache containing serialized payloads for the message.
+        /// </summary>
+        public SerializedHubMessage Message { get; }
+
+        public RedisInvocation(SerializedHubMessage message, IReadOnlyList<string>? excludedConnectionIds)
+        {
+            Message = message;
+            ExcludedConnectionIds = excludedConnectionIds;
+        }
+    }
+}