Пример #1
0
 public static async Task SendStringAsync(BitNoticeMessage message, bool isLocal = false)
 {
     foreach (var user in sockets)
     {
         if (user.Key != message.Receiver)
         {
             continue;
         }
         foreach (var socket in user.Value)
         {
             LogHelper.SaveLog("WebSockets", string.Format("发送消息:{0}:{1}:{2}:{3}", socket.IsRemote, socket.RemoteIp, socket.RemoteTime, JsonConvert.SerializeObject(message)));
             if (socket.IsRemote && !isLocal)
             {//远程通道
                 try
                 {
                     WebClient wc       = new WebClient();
                     byte[]    sendByte = Encoding.Default.GetBytes(JsonConvert.SerializeObject(message));
                     var       url      = string.Format("http://{0}:1899/BitService/SendWebSocketString", socket.RemoteIp);
                     LogHelper.SaveLog("WebSockets", string.Format("发送远程消息:{0}", url));
                     var result    = wc.UploadData(url, sendByte);
                     var strResult = Encoding.Default.GetString(result);
                 }
                 catch (Exception ex) { LogHelper.SaveLog(ex); }
             }
             else if (socket.WebSocket.State == WebSocketState.Open)
             {//本地通道
                 await SendStringAsync(socket.WebSocket, JsonConvert.SerializeObject(message), socket.HttpContext.RequestAborted);
             }
         }
     }
 }
Пример #2
0
        static async Task Acceptor(HttpContext context, Func <Task> n)
        {
            if (!context.WebSockets.IsWebSocketRequest || !SSOClient.IsLogin)
            {
                return;
            }

            CancellationToken ct = context.RequestAborted;
            var currentSocket    = await context.WebSockets.AcceptWebSocketAsync();

            string sid = SSOClient.UserId.ToString();

            if (!sockets.ContainsKey(sid))
            {
                sockets.TryAdd(sid, new List <UserSocket>());
            }
            sockets[sid].Add(new UserSocket()
            {
                HttpContext = context, WebSocket = currentSocket, IsRemote = false
            });
            //LogHelper.SaveLog("WebSockets", "add:sid->" + sid + ",count->" + sockets[sid].Count);

            while (true)
            {
                if (ct.IsCancellationRequested)
                {
                    break;
                }

                string request = await ReceiveStringAsync(currentSocket, ct);

                if (string.IsNullOrEmpty(request))
                {
                    continue;
                }

                BitNoticeMessage msg = JsonConvert.DeserializeObject <BitNoticeMessage>(request);
                msg.Sender = SSOClient.UserId.ToString();
                switch (msg.MessageType)
                {
                case "notice":      //通知消息
                    LogHelper.SaveLog("WebSockets", "接收到通知消息:" + request);
                    break;

                case "chat":        //聊天消息
                    await SendStringAsync(msg);

                    LogHelper.SaveLog("WebSockets", "接收到聊天消息:" + request);
                    break;

                default:
                    continue;
                }
            }
        }