.NET Core中在SignalR Hub外管理用户连接：如何终止特定用户连接？

核心思路 #

SignalR本身没有提供Hub外直接获取单个连接上下文并调用Abort的API，因此需要通过维护用户-连接ID映射，结合IHubContext或底层生命周期管理器来实现需求。以下是两种可行方案：

方案一：客户端主动断开（推荐，符合SignalR设计模式） #

通过服务端发送指令，让客户端主动终止连接，这种方式更稳定且符合SignalR的通信模型。

1. 实现用户-连接ID映射服务 #

创建一个线程安全的单例服务，用于存储用户与对应连接ID的关联关系：

public class ConnectionMapping<T>
{
    private readonly ConcurrentDictionary<T, HashSet<string>> _connections = new();

    public void Add(T userId, string connectionId)
    {
        _connections.AddOrUpdate(userId, 
            _ => new HashSet<string> { connectionId },
            (_, existing) =>
            {
                lock (existing)
                {
                    existing.Add(connectionId);
                    return existing;
                }
            });
    }

    public IEnumerable<string> GetConnections(T userId)
    {
        _connections.TryGetValue(userId, out var connections);
        return connections ?? Enumerable.Empty<string>();
    }

    public void Remove(T userId, string connectionId)
    {
        if (_connections.TryGetValue(userId, out var connections))
        {
            lock (connections)
            {
                connections.Remove(connectionId);
                if (connections.Count == 0)
                {
                    _connections.TryRemove(userId, out _);
                }
            }
        }
    }
}

在Program.cs中注册为单例：

builder.Services.AddSingleton<ConnectionMapping<string>>(); // 假设用户ID为字符串类型

2. 在Hub中维护映射关系 #

在Hub的连接/断开事件中更新映射表：

public class ChatHub : Hub
{
    private readonly ConnectionMapping<string> _connectionMapping;

    public ChatHub(ConnectionMapping<string> connectionMapping)
    {
        _connectionMapping = connectionMapping;
    }

    public override async Task OnConnectedAsync()
    {
        var userId = Context.User.FindFirst(ClaimTypes.NameIdentifier)?.Value;
        if (!string.IsNullOrEmpty(userId))
        {
            _connectionMapping.Add(userId, Context.ConnectionId);
        }
        await base.OnConnectedAsync();
    }

    public override async Task OnDisconnectedAsync(Exception? exception)
    {
        var userId = Context.User.FindFirst(ClaimTypes.NameIdentifier)?.Value;
        if (!string.IsNullOrEmpty(userId))
        {
            _connectionMapping.Remove(userId, Context.ConnectionId);
        }
        await base.OnDisconnectedAsync(exception);
    }
}

3. 控制器中触发断开指令 #

注入IHubContext和映射服务，向目标用户的所有连接发送断开指令：

[ApiController]
[Route("api/users")]
public class UsersController : ControllerBase
{
    private readonly IHubContext<ChatHub> _hubContext;
    private readonly ConnectionMapping<string> _connectionMapping;

    public UsersController(IHubContext<ChatHub> hubContext, ConnectionMapping<string> connectionMapping)
    {
        _hubContext = hubContext;
        _connectionMapping = connectionMapping;
    }

    [HttpDelete("{userId}")]
    public async Task<IActionResult> RemoveUser(string userId)
    {
        // 先执行用户移除的业务逻辑...

        // 获取该用户的所有连接ID
        var connectionIds = _connectionMapping.GetConnections(userId);
        foreach (var connectionId in connectionIds)
        {
            await _hubContext.Clients.Client(connectionId).SendAsync("ForceDisconnect");
        }

        return Ok();
    }
}

4. 客户端监听并断开 #

在前端代码中监听服务端指令，主动终止连接：

const connection = new signalR.HubConnectionBuilder()
    .withUrl("/chatHub")
    .build();

connection.on("ForceDisconnect", () => {
    connection.stop().then(() => {
        alert("您的账号已被移除，连接已断开");
    });
});

// 启动连接逻辑...

方案二：服务端强制断开（底层API方式） #

如果需要完全在服务端强制断开连接，可以使用SignalR的底层生命周期管理器，但需注意这属于内部API，版本更新可能存在变动。

在控制器中注入IHubLifetimeManager<THub>，直接调用断开方法：

[ApiController]
[Route("api/users")]
public class UsersController : ControllerBase
{
    private readonly IHubLifetimeManager<ChatHub> _lifetimeManager;
    private readonly ConnectionMapping<string> _connectionMapping;

    public UsersController(IHubLifetimeManager<ChatHub> lifetimeManager, ConnectionMapping<string> connectionMapping)
    {
        _lifetimeManager = lifetimeManager;
        _connectionMapping = connectionMapping;
    }

    [HttpDelete("{userId}")]
    public async Task<IActionResult> RemoveUser(string userId)
    {
        // 业务逻辑...

        var connectionIds = _connectionMapping.GetConnections(userId);
        foreach (var connectionId in connectionIds)
        {
            await _lifetimeManager.DisconnectAsync(connectionId, CancellationToken.None);
        }

        return Ok();
    }
}

注意事项 #

• 线程安全：映射表必须使用线程安全集合，HashSet本身非线程安全，需加锁修改。
• 用户ID获取：根据你的认证方式调整Hub中用户ID的获取逻辑（比如从JWT、Cookie Claims中提取）。
• 异常处理：断开连接时需捕获可能的异常（如连接已提前断开）。

内容的提问来源于stack exchange，提问作者Louise Yousre