public HubCallerContext TryGetHubCallerContext(ScopedClientKey scopedClientKey)
{
var oneKey = scopedClientKey.ToOneKey();
ScopedContexts.TryGetValue(oneKey, out var theOne);
return(theOne);
}
public ScopedConnection GetScopedConnectionByScopedClientKey(ScopedClientKey scopedClientKey)
{
if (scopedClientKey == null)
{
return(null);
}
Scopes.TryGetValue(scopedClientKey.ScopeGroupId, out var connDic);
var theOne = connDic?.Values.SingleOrDefault(x => x.ClientId.Equals(scopedClientKey.ClientId, StringComparison.OrdinalIgnoreCase));
return(theOne);
}
public Task UpdateScopedConnectionBagsOutSideHub(OnUpdateBagsHubContextEvent theEvent)
{
if (theEvent == null)
{
throw new ArgumentNullException(nameof(theEvent));
}
var hubContext = theEvent.Context;
if (hubContext == null)
{
throw new ArgumentNullException(nameof(theEvent.Context));
}
if (theEvent.Bags == null || theEvent.Bags.Count == 0)
{
return(Task.CompletedTask);
}
var scopedClientKey = ScopedClientKey.Create().WithScopeGroupId(theEvent.ScopeGroupId).WithClientId(theEvent.ClientId);
var hubCallerContext = TryGetHubCallerContext(scopedClientKey);
if (hubCallerContext == null)
{
return(Task.CompletedTask);
}
var bags = theEvent.Bags;
var connectionId = hubCallerContext.ConnectionId;
var conn = _repository.GetScopedConnection(connectionId);
if (conn == null)
{
return(Task.FromResult(0));
}
foreach (var bag in bags)
{
conn.Bags[bag.Key] = bag.Value;
}
_repository.AddOrUpdateScopedConnection(conn);
var scopedConnections = _repository.GetScopedConnections(conn.ScopeGroupId);
var connections = scopedConnections.OrderBy(x => x.CreateAt).ToList();
var clientProxy = hubContext.Clients.Group(conn.ScopeGroupId);
return(clientProxy.SendAsync(ScopedConst.ForClient.ScopedConnectionsUpdated(), connections));
}
public async Task OnConnected(OnConnectedEvent theEvent)
{
if (theEvent?.RaiseHub == null)
{
return;
}
var hub = theEvent.RaiseHub;
var conn = new ScopedConnection();
var connectionId = hub.Context.ConnectionId;
conn.ConnectionId = connectionId;
var now = DateHelper.Instance.GetDateNow();
conn.CreateAt = now;
conn.LastUpdateAt = now;
conn.ScopeGroupId = hub.TryGetQueryParameterValue(nameof(conn.ScopeGroupId), string.Empty);
conn.ClientId = hub.TryGetQueryParameterValue(nameof(conn.ClientId), string.Empty);
conn.UpdateDesc();
_repository.AddOrUpdateScopedConnection(conn);
if (!string.IsNullOrWhiteSpace(conn.ClientId))
{
//scoped clients with same clientId, old should be kicked off
var scopedClientKey = ScopedClientKey.Create().WithClientId(conn.ClientId).WithScopeGroupId(conn.ScopeGroupId);
var oneKey = scopedClientKey.ToOneKey();
if (ScopedContexts.TryGetValue(oneKey, out var oldClientHub))
{
await KickSameScopedClient(hub, oldClientHub, scopedClientKey).ConfigureAwait(false);
}
ScopedContexts[oneKey] = hub.Context;
}
await hub.Groups.AddToGroupAsync(conn.ConnectionId, conn.ScopeGroupId).ConfigureAwait(false);
var scopedConnections = _repository
.GetScopedConnections(conn.ScopeGroupId)
.OrderBy(x => x.CreateAt).ToList();
var clientProxy = hub.Clients.Group(conn.ScopeGroupId);
await clientProxy.SendAsync(ScopedConst.ForClient.ScopedConnectionsUpdated(), scopedConnections).ConfigureAwait(false);
}
private async Task KickSameScopedClient(Hub hub, HubCallerContext oldClientHub, ScopedClientKey scopedClientKey)
{
if (oldClientHub == null)
{
return;
}
//already exist, should kick!
var oldConnectionId = oldClientHub.ConnectionId;
var theConn = _repository.GetScopedConnection(oldConnectionId);
if (theConn != null)
{
var clientProxy = hub.Clients.Client(oldConnectionId);
if (clientProxy != null)
{
var scopedConnections = _repository.GetScopedConnections(scopedClientKey.ScopeGroupId);
var connections = scopedConnections.OrderBy(x => x.CreateAt).ToList();
var theOne = connections.SingleOrDefault(x => x.ConnectionId.Equals(oldConnectionId));
if (theOne != null)
{
var oneKey = scopedClientKey.ToOneKey();
var message = string.Format("{0} is kicked by another same scoped client!", oneKey);
theOne.Desc = message;
}
await clientProxy.SendAsync(ScopedConst.ForClient.ScopedConnectionsUpdated(), connections).ConfigureAwait(false);
}
}
_repository.RemoveScopedConnection(oldConnectionId);
oldClientHub.Abort();
}
