AuthHub.cs 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. using Microsoft.AspNetCore.Authorization;
  2. using Microsoft.AspNetCore.Cors;
  3. using Microsoft.AspNetCore.SignalR;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Linq;
  7. using System.Threading.Tasks;
  8. using HubCallerContextDict = System.Collections.Concurrent.ConcurrentDictionary<string, Microsoft.AspNetCore.SignalR.HubCallerContext>;
  9. namespace WK_KLI_LAPS_COMMONTOOL_Service.Hub
  10. {
  11. public class Userhub
  12. {
  13. public string Userid { get; set; }
  14. public HashSet<string> ConnectionIds { get; set; }
  15. }
  16. [EnableCors("CorsPolicy")]
  17. [AllowAnonymous]
  18. public class AuthHub : Microsoft.AspNetCore.SignalR.Hub
  19. {
  20. public static HubCallerContextDict _HubDict = new HubCallerContextDict();
  21. [Authorize]
  22. public async Task SendMessage()
  23. {
  24. await Clients.All.SendAsync("Logoutclient");
  25. }
  26. public override Task OnConnectedAsync()
  27. {
  28. //if (this.Context.User.Identity.IsAuthenticated)
  29. // {
  30. // var _userId = this.Context.User.Identity.Name;
  31. var _userId = Context.GetHttpContext().Request.Query["userid"];
  32. if (_HubDict.TryGetValue(_userId, out HubCallerContext _clientHub))
  33. {
  34. if (
  35. !object.Equals(_clientHub.ConnectionId, this.Context.ConnectionId)
  36. )
  37. {
  38. _HubDict.TryRemove(_userId, out _clientHub);
  39. _clientHub.Abort();
  40. }
  41. }
  42. _HubDict.TryAdd(_userId, this.Context);
  43. // }
  44. return base.OnConnectedAsync();
  45. }
  46. public override Task OnDisconnectedAsync(Exception exception)
  47. {
  48. if (this.Context.User.Identity.IsAuthenticated)
  49. {
  50. var _userId = this.Context.User.Identity.Name;
  51. if (_HubDict.TryGetValue(_userId, out HubCallerContext _clientHub))
  52. {
  53. if (object.Equals(this.Context.ConnectionId, _clientHub.ConnectionId))
  54. {
  55. _HubDict.TryRemove(_userId, out _clientHub);
  56. _clientHub.Abort();
  57. }
  58. }
  59. }
  60. return base.OnDisconnectedAsync(exception);
  61. }
  62. public string GetUser(string userid)
  63. {
  64. string status = "no";
  65. if (_HubDict.TryGetValue(userid, out HubCallerContext _clientHub))
  66. {
  67. status = "yes";
  68. }
  69. return status;
  70. }
  71. public string GetConnectionId(string userid)
  72. {
  73. string clientid = "";
  74. if (_HubDict.TryGetValue(userid, out HubCallerContext _clientHub))
  75. {
  76. clientid = _clientHub.ConnectionId;
  77. }
  78. return clientid;
  79. }
  80. public void logout(string to)
  81. {
  82. _HubDict.TryRemove(to, out HubCallerContext _clientHub);
  83. }
  84. }
  85. }