IdentityServer4 + SignalR Core +RabbitMQ 构建web即时通讯（三）
后台服务用户与认证
新建一个空的.net core web项目Demo.Chat，端口配置为5001，安装以下nuget包
1.IdentityServer4.AccessTokenValidation，IdentityServer4客户端认证所用；
2.Microsoft.AspNetCore.SignalR
3.RabbitMQ.Client
添加appsettings.json
{
"RabbitMQ": {
"Host": "192.168.1.107",
"User": "admin",
"Password": "123123"
},
"Authentication": {
"Authority": "http://localhost:5000"
}
}
这里我们新增两个Dto类，一个消息传输类MsgDto，一个用户数据类UserDto
public class MsgDto
{
public UserDto FromUser { get; set; }
public UserDto ToUser { get; set; }
public string Content { get; set; }
public DateTime SendTime { get; set; }
}
public class UserDto
{
// signalr当前的连接id
public string ConnectionId { get; set; }
public Guid Id { get; set; }
public string UserName { get; set; }
public string EMail { get; set; }
public string Avatar { get; set; }
}
当用户认证通过后，从Identity返回的token中我们已经返回了用户的基础信息了，那这里我们如何获取呢？很简单在上下文的User中Claims属性里面，所以这里我们增加一个扩展方法来转换为UserDto
public static UserDto GetUser(this ClaimsPrincipal claimsPrincipal)
{
return new UserDto
{
Id = new Guid(claimsPrincipal.Claims.FirstOrDefault(r => r.Type == "sub").Value),
EMail = claimsPrincipal.Claims.FirstOrDefault(r => r.Type == "email").Value,
UserName = claimsPrincipal.Claims.FirstOrDefault(r => r.Type == "username").Value,
Avatar = claimsPrincipal.Claims.FirstOrDefault(r => r.Type == "avatar").Value,
};
}
既然是在线聊天那必须得存储当前所有的在线用户对吧？新建一个OnlineUsers类，这里我们就不用数据库了，Demo嘛，里面就3个用户，嘿嘿。当然你完全可以自由发挥使用其他redis，mongo什么什么的。
public class OnlineUsers
{
/// <summary>
/// 用户id作为key
/// </summary>
private static ConcurrentDictionary<Guid, UserDto> onlineUsers { get; } = new ConcurrentDictionary<Guid, UserDto>(); public void AddOrUpdateUser(UserDto user)
{
onlineUsers.AddOrUpdate(user.Id, user, (id, r) => user);
} public List<UserDto> Get()
{
return onlineUsers.Values.ToList();
} public UserDto Get(Guid userId)
{
onlineUsers.TryGetValue(userId, out UserDto user);
return user;
} public void Remove(Guid userId)
{
if (onlineUsers.ContainsKey(userId))
onlineUsers.TryRemove(userId, out UserDto user);
}
}
后台服务RabbitMQ消息处理
RabbitMQ消息队列相关的知识这里我也不再赘述，园子里面很多，大家自行研究，RabbitMQ大概有2个种模式：生产消费者模式和发布/订阅模式，生产消费者模式即消息只能被使用一次，比如一个商品生产出来你只能卖给一个消费者对吧，发布/订阅即只要订阅了都会收到该消息。这里我们用到的是生产消费者模式，参考官方文档。
消息发送和收到消息的处理，这里我们分为2个类单独处理，MsgSender和MsgHandler。
MsgSender:当用户发送了一条消息，后端收到后就将消息添加到消息队列，MsgHandler：一直处于运行状态，当收到队列的消息时，开始处理消息，调用SignalR的方法，发送消息到客户端，RabbitMQ的连接配置在appsettings.json中，注入IConfiguration获取
MsgSender
public class MsgSender
{
public MsgSender(IConfiguration configuration)
{
factory = new ConnectionFactory();
factory.HostName = configuration.GetValue<string>("RabbitMQ:Host");
factory.UserName = configuration.GetValue<string>("RabbitMQ:User");
factory.Password = configuration.GetValue<string>("RabbitMQ:Password");
}
ConnectionFactory factory; public void Send(MsgDto msg)
{
using (var connection = factory.CreateConnection())
{
using (var channel = connection.CreateModel())
{
channel.QueueDeclare("chat_queue", false, false, false, null);//创建一个名称为hello的消息队列
var body = Encoding.UTF8.GetBytes(JsonConvert.SerializeObject(msg));
channel.BasicPublish("", "chat_queue", null, body); //开始传递
}
}
}
}
MsgHandler，需要注入IHubContext接口，用于发送消息到客户端,ps:在Hub类中,可以通过Clients直接发送消息到客户端,在其他类里面可以使用这个接口,获取到Clients。
public class MsgHandler : IDisposable
{
public MsgHandler(IConfiguration configuration, IHubContext<MessageHub> hubContext)
{
factory = new ConnectionFactory();
factory.HostName = configuration.GetValue<string>("RabbitMQ:Host");
factory.UserName = configuration.GetValue<string>("RabbitMQ:User");
factory.Password = configuration.GetValue<string>("RabbitMQ:Password");
this.hubContext = hubContext;
connection = factory.CreateConnection();
channel = connection.CreateModel(); }
ConnectionFactory factory;
// 注入SignalR的消息处理器上下文，用以发送消息到客户端
IHubContext<MessageHub> hubContext;
IConnection connection;
IModel channel;
public void BeginHandleMsg()
{
channel.QueueDeclare("chat_queue", false, false, false, null);
var consumer = new EventingBasicConsumer(channel);
channel.BasicConsume("chat_queue", false, consumer);
consumer.Received += (model, arg) =>
{
var body = arg.Body;
var message = Encoding.UTF8.GetString(body);
var msg = JsonConvert.DeserializeObject<MsgDto>(message);
// 通过消息处理器上下文发送消息到客户端
hubContext.Clients?.Client(msg.ToUser.ConnectionId)
?.SendAsync("Receive", msg); channel.BasicAck(arg.DeliveryTag, false);
};
} public void Dispose()
{
channel?.Dispose();
connection?.Dispose();
}
}
后台服务SignalR消息处理器
关于SignalR，官方文档
SignalR的核心就是继承自Hub消息处理类，这个类中所有的public 方法都可以给客户端调用。我们的聊天室比较简陋，只需要一个Send方法给客户端就够了，是吧？当然服务端需要2个主动发送消息到客户端的方法，1.当有用户登录时通知所有客户端刷新在线用户列表，2.有什么错误的时候发送错误消息给客户端，比如我们不允许离线发送，用户发了条消息给一个不在线的用户。
另外当用户登录和离开时需要在OnlineUsers中进行注册和注销。
MessageHub，我们的聊天室必须登录，所以加上Authorize特性。
[Authorize]
public class MessageHub : Hub
{
MsgSender msgSender;
MsgHandler msgQueueHandler;
OnlineUsers onlineUsers;
public MessageHub(MsgSender msgSender, MsgHandler msgQueueHandler, OnlineUsers onlineUsers)
{
this.msgSender = msgSender;
this.msgQueueHandler = msgQueueHandler;
this.onlineUsers = onlineUsers;
} public async Task Send(string toUserId, string message)
{
string timestamp = DateTime.Now.ToShortTimeString();
var toUser = onlineUsers.Get(new Guid(toUserId));
if (toUser == null)
{
await SendErrorAsync("用户已离线");
return;
}
var fromUser = Context.User.GetUser();
msgSender.Send(new Dtos.MsgDto
{
Content = message,
FromUser = fromUser,
SendTime = DateTime.Now,
ToUser = toUser
});
} /// <summary>
/// 当有用户登录时 添加在线用户，并设置用户的ConnectionId
/// </summary>
/// <returns></returns>
public override async Task OnConnectedAsync()
{
await base.OnConnectedAsync();
var user = Context.User.GetUser();
if (user == null)
{
await SendErrorAsync("您没有登录");
return;
}
user.ConnectionId = Context.ConnectionId;
onlineUsers.AddOrUpdateUser(user);
await SendUserInfo();
await RefreshUsersAsync();
} /// <summary>
/// 当有用户离开时，注销用户登录
/// </summary>
/// <param name="exception"></param>
/// <returns></returns>
public override async Task OnDisconnectedAsync(Exception exception)
{
//disconnection
await base.OnDisconnectedAsync(exception);
var userId = Context.User?.GetUser()?.Id;
if (userId.HasValue)
onlineUsers.Remove(userId.Value);
await RefreshUsersAsync();
} private async Task RefreshUsersAsync()
{
var users = onlineUsers.Get().Where(r => r.Id != Context.User.GetUser().Id).ToList();
// 发送给所有的在线客户端，通知刷新在线用户
await Clients.All.SendAsync("Refresh", users);
} private async Task SendErrorAsync(string errorMsg)
{
// 发送错误消息给调用者
await Clients.Caller.SendAsync("Error", errorMsg);
} }
这里就冒出来另外一个新的问题了，SignalR使用的是websocket，据我了解到的是没有header头这个东西的，而jwt token默认是通过header中Authorization信息进行认证的。那这个授权又如何实现呢？想办法咯，既然header传不进来，那直接url传进来总可以吧。
后台服务：服务注册与认证授权
好了，我们先将需要的服务先配置下。
AddIdentityServerAuthentication实际上是AddJwtBearer的扩展，你要喜欢也可以用AddJwtBearer配置，由IdentityServer4.AccessTokenValidation提供，配置认证Authority为http://localshot:5000(Demo.Identity配置的端口号为5000，appsetting.json中配置),ApiName和Secret与Identity端配置的ApiResource一致。
public void ConfigureServices(IServiceCollection services)
{
// 注册消息处理器 消息发送器，在线用户类
services.AddSingleton<MsgHandler>()
.AddSingleton<MsgSender>()
.AddSingleton<OnlineUsers>(); // 增加认证服务
services.AddAuthentication(r =>
{
r.DefaultScheme = "JwtBearer";
})
// 增加jwt认证
.AddIdentityServerAuthentication("JwtBearer", r =>
{
// 配置认证服务器
r.Authority = Configuration.GetValue<string>("Authentication:Authority");
// 配置无需验证https
r.RequireHttpsMetadata = false;
// 配置 当前资源服务器的名称
r.ApiName = "chatapi";
// 配置 当前资源服务器的连接密码
r.ApiSecret = "";
r.SaveToken = true;
}); // 跨域
services.AddCors(r =>
{
r.AddPolicy("all", policy =>
{
policy
.AllowAnyOrigin()
.AllowAnyHeader()
.AllowAnyMethod()
.AllowCredentials()
;
});
});
// 增加授权服务
services.AddAuthorization();
// 增加SignalR 服务
services.AddSignalR();
}
刚刚提到SignalR认证的问题，具体如何实现呢？这里也有2种方式，1.使用中间件在认证之前从url中获取token并添加到header中；2.r.MapHub<MessageHub>("/msg")，可以配置在参数中添加自定义的IAuthorizeData接口，可以自己实现获取token验证，我觉得比较麻烦，这里我们使用第一种方式。
添加中间件，这个中间件一定要在UseAuthentication之前：
// signalr jwt认证 token添加
app.Use(async (context, next) =>
{
// 这里从url中获取token参数，实际应用请实际考虑，加一些过滤条件
if (context.Request.Query.TryGetValue("token", out var token))
{
// 从url中拿到header，再添加到header中，一定要在UseAuthentication之前
context.Request.Headers.Add("Authorization", $"Bearer {token}");
}
await next.Invoke();
});
好了，还有一个问题，前面写的MsgHandler什么时候开始处理消息？Dispose什么时候调用？这里我们使用IApplicationLifetime接口，该接口提供了应用的整个生命周期事件处理。在应用启动的时候我们注册消息处理，应用结束时Dispose。
// 应用启动时开始处理消息
applicationLifetime.ApplicationStarted.Register(msgHandler.BeginHandleMsg);
// 应用退出时,释放资源
applicationLifetime.ApplicationStopping.Register(msgHandler.Dispose);
完整的Configure代码：
public void Configure(
IApplicationBuilder app,
IHostingEnvironment env,
MsgHandler msgHandler,
IApplicationLifetime applicationLifetime)
{
if (env.IsDevelopment())
{
app.UseDeveloperExceptionPage();
}
app.UseStaticFiles();
app.UseCors("all"); app.UseAuthentication();
// 使用SignalR 并添加MessageHub类的消息处理器
app.UseSignalR(r =>
{
r.MapHub<MessageHub>("/msg");
}); // 应用启动时开始处理消息
applicationLifetime.ApplicationStarted.Register(msgHandler.BeginHandleMsg);
// 应用退出时,释放资源
applicationLifetime.ApplicationStopping.Register(msgHandler.Dispose);
}
另外用户登录后需要展示用户信息，邮件地址啊头像什么的，这里我们也有2种方式，1是消息处理器中，当用户连接后主动发送消息给用户；2是建一个Api接口，当然放在消息处理器中会显得更纯洁，web项目里面没有一个controller，这里我们使用第一种方式。
在MessageHub中添加方法，在OnConnectedAsync方法中调用
private async Task SendUserInfo()
{
await Clients.Caller.SendAsync("UserInfo", Context.User.GetUser());
}
聊天室web前端
官方提供了js库，可以用npm安装，npm install @aspnet/signalr。
这个前端嘛，我就不花大功夫去做得漂亮高大上了，暂时就把代码直接丢在Demo.chat里面吧,2个页面，登录页login，聊天室页面chat。
关于前端就不啰嗦了，再啰嗦就是关公面前耍大刀了，什么angular,vue,老夫写代码统统jquery。其他的大家自己发挥了。
login.html
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8" />
<title>登录聊天室</title>
<script src="https://cdn.bootcss.com/jquery/3.3.1/jquery.min.js"></script>
</head>
<body>
<div>
<fieldset>
<legend>登录聊天室</legend>
<div>
<input type="text" name="uername" id="username" value="" />
</div>
<div>
<input type="password" name="password" id="password" value="" />
</div>
<div>
<button id="login" type="button">登录</button>
</div>
</fieldset>
</div>
<script type="text/javascript">
$(function () {
var identityUrl = 'http://localhost:5000/connect/token'; $('#login').click(function () { $.post(identityUrl, {
client_id: 'chat_client',
grant_type: 'password',
scope: 'openid chatapi profile offline_access',
username: $('#username').val(),
password: $('#password').val()
}, function (result) {
if (result && result.access_token) {
sessionStorage['token'] = result.access_token;
window.location = "http://localhost:5001/chat.html";
}
}, 'json');
});
});
</script>
</body>
</html>
chat.html
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8" />
<title></title>
<script src="https://cdn.bootcss.com/jquery/3.3.1/jquery.min.js"></script>
<script src="/lib/signalr.js"></script>
</head>
<body>
<div>
<div>
<input type="hidden" id="userId" value="" />
<p><label>UserName:</label><i id="userName"></i></p>
<p><label>EMail:</label><i id="email"></i></p>
<p><label>Avatar:</label><img id="avatar" style="width:48px; height:48px; border-radius:50%; overflow:hidden;" /> </p>
</div>
<div style="width:700px;height:500px;border:1px solid red;">
<ul id="msgList"></ul>
</div>
<div>
<select id="users"></select>
</div>
<div>
<textarea id="msgSendContent" placeholder="请输入发送消息" cols="" rows=""></textarea>
<br />
<button id="send" type="button">发送</button>
</div> </div> <script type="text/javascript">
$(function () {
var token = sessionStorage['token']; if (!token) {
alert('请先登录!');
window.location = 'http://localhost:5001/login.html';
return;
}
function timeFormat(time) {
time = new Date(time)
return time.toLocaleDateString() + ' ' + time.toLocaleTimeString();
} var connection = new signalR.HubConnectionBuilder()
.withUrl("/msg?token=" + token)
.configureLogging(signalR.LogLevel.Information)
.build(); connection.on('Receive', function (msg) {
var $ul = $('#msgList');
var $li = $('<li>' + msg.fromUser.userName + '[' + timeFormat(msg.sendTime) + '] : ' + msg.content + '</li>');
$ul.append($li);
}); connection.on('UserInfo', function (userInfo) {
$('#userName').text(userInfo.userName);
$('#email').text(userInfo.eMail);
$('#avatar').attr('src', userInfo.avatar);
$('#userId').val(userInfo.id);
}); connection.on('Refresh', function (users) {
$('#users').empty();
users.forEach(function (user) {
if (user.id != $('#userId').val())
$('#users').append('<option value="' + user.id + '">' + user.userName + '</option>');
});;
}); connection.on('Error', function (err) {
alert(err);
}); connection.start().catch(err => console.error(err.toString()));
$('#send').click(function () {
var msg = $('#msgSendContent').val();
var toUerId = $('#users').val();
connection.invoke('Send', toUerId, msg).catch(err => console.error(err));
var $ul = $('#msgList');
var $li = $('<li>我[' + timeFormat(new Date()) + '] : ' + msg + '</li>');
$ul.append($li);
});
});
</script>
</body>
</html>
好了，代码就写完了，同时运行Demo.Identity和Demo.Chat。打开2个浏览器：http://localhost:5001/login.html。
输入用户名密码登录；
发送个消息试试：
是不是很简陋？嘿嘿
好了，到处为止。其他不完善的地方,自己动手,丰衣足食,如离线消息,token自动刷新等等.
