zhaolei
8 days ago 921de2254ff5712a44ed8575ee8efe34252f6603
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
using Bootstrap.DataAccess;
using Prow.Web.SignalR;
using Microsoft.AspNetCore.SignalR;
using System;
using System.Data.Common;
using System.Threading;
using Task = System.Threading.Tasks.Task;
 
namespace Bootstrap.Admin
{
    /// <summary>
    /// SignalR 操作扩展类
    /// </summary>
    public static class SignalRExtensions
    {
        /// <summary>
        /// 推送异常信息到客户端方法扩展
        /// </summary>
        /// <param name="context"></param>
        /// <param name="ex"></param>
        /// <param name="token"></param>
        /// <returns></returns>
        public static async Task SendMessageBody(this IHubContext<SignalRHub> context, Exception ex, CancellationToken token = default)
        {
            var category = "App";
            if (ex.GetType().IsSubclassOf(typeof(DbException))) category = "DB";
            await context.SendMessageBody(new MessageBody() { Category = category, Message = ex.Message }, token);
        }
 
        /// <summary>
        /// 推送 MessageBody 到客户端方法扩展
        /// </summary>
        /// <param name="context"></param>
        /// <param name="messageBody"></param>
        /// <param name="token"></param>
        /// <returns></returns>
        public static Task SendMessageBody(this IHubContext<SignalRHub> context, MessageBody messageBody, CancellationToken token = default) => context.Clients.All.SendAsync("rev", messageBody, token);
 
        /// <summary>
        /// 推送任务消息到客户端扩展
        /// </summary>
        /// <param name="context"></param>
        /// <param name="args"></param>
        /// <returns></returns>
        public static Task SendTaskLog(this IHubContext<TaskLogHub> context, string args) => context.Clients.All.SendAsync("rev", args);
    }
}