| | | 1 | | using Microsoft.AspNetCore.SignalR; |
| | | 2 | | |
| | | 3 | | namespace Syki.Back.Hubs; |
| | | 4 | | |
| | | 5 | | [AuthBearer] |
| | | 6 | | public class SykiHub : Hub |
| | | 7 | | { |
| | | 8 | | public override Task OnConnectedAsync() |
| | | 9 | | { |
| | 0 | 10 | | var userId = Context.User.Id; |
| | | 11 | | |
| | 0 | 12 | | if (SykiHubUsersStore.Users.ContainsKey(userId)) |
| | | 13 | | { |
| | 0 | 14 | | SykiHubUsersStore.Users[userId].Add(Context.ConnectionId); |
| | | 15 | | } |
| | | 16 | | else |
| | | 17 | | { |
| | 0 | 18 | | SykiHubUsersStore.Users.TryAdd(userId, [Context.ConnectionId]); |
| | | 19 | | } |
| | | 20 | | |
| | 0 | 21 | | return base.OnConnectedAsync(); |
| | | 22 | | } |
| | | 23 | | |
| | | 24 | | public override Task OnDisconnectedAsync(Exception? exception) |
| | | 25 | | { |
| | 0 | 26 | | var userId = Context.User.Id; |
| | | 27 | | |
| | 0 | 28 | | if (SykiHubUsersStore.Users.ContainsKey(userId)) |
| | | 29 | | { |
| | 0 | 30 | | SykiHubUsersStore.Users[userId].Remove(Context.ConnectionId); |
| | 0 | 31 | | if (SykiHubUsersStore.Users[userId].Count == 0) |
| | | 32 | | { |
| | 0 | 33 | | SykiHubUsersStore.Users.Remove(userId, out _); |
| | | 34 | | } |
| | | 35 | | } |
| | | 36 | | |
| | 0 | 37 | | return base.OnDisconnectedAsync(exception); |
| | | 38 | | } |
| | | 39 | | } |