private async Task Handle(WebsocketClient webSocket)
{
WebsocketClientCollection.Add(webSocket);
logger.Info($"Websocket client added.");
WebSocketReceiveResult result = null;
do
{
var buffer = new byte[1024 * 1];
//等待接收消息
result = await webSocket.WebSocket.ReceiveAsync(new ArraySegment <byte>(buffer), CancellationToken.None);
if (result.MessageType == WebSocketMessageType.Text && !result.CloseStatus.HasValue)
{
var msgString = Encoding.UTF8.GetString(buffer);
logger.Info($"Websocket client ReceiveAsync message {msgString}.");
var message = JsonConvert.DeserializeObject <Message>(msgString);
message.SendClientId = message.SendClientId;
MessageRoute(message);
}
}while (!result.CloseStatus.HasValue);
WebsocketClientCollection.Remove(webSocket);
logger.Info($"Websocket client closed.");
}
private void MessageRoute(Message message)
{
var client = WebsocketClientCollection.Get(CurrentUser.UserId);
switch (message.Action)
{
case "join":
var joinJson = new
{
Msg = message.Msg,
Sender = message.SendClientId,
RoomId = client.RoomNo,
MyClientId = "",
MessageTime = DateTime.Now.ToString("yyyy-MM-dd hh:mm:ss"),
NickName = "",
Action = message.Action
};
client.RoomNo = message.Msg;
logger.Info($"Websocket client {message.SendClientId} join room {client.RoomNo}.");
break;
case "send_to_room":
client.RoomNo = "8888";
if (string.IsNullOrEmpty(client.RoomNo))
{
break;
}
var clients = WebsocketClientCollection.GetRoomClients(client.RoomNo);
var msgJson = new
{
Msg = message.Msg,
Sender = CurrentUser.UserId,
RoomId = client.RoomNo,
MessageTime = DateTime.Now.ToString("yyyy-MM-dd hh:mm:ss"),
NickName = CurrentUser.UserName,
Action = message.Action
};
clients.ForEach(c =>
{
c.SendMessageAsync(JsonConvert.SerializeObject(msgJson));
});
logger.Info($"Websocket client {message.SendClientId} send message {message.Msg} to room {client.RoomNo}");
break;
case "leave":
var roomNo = client.RoomNo;
client.RoomNo = "";
client.SendMessageAsync($"{message.NickName} leave room {roomNo} success .");
logger.Info($"Websocket client {message.SendClientId} leave room {roomNo}");
break;
default:
break;
}
}