{"id":673,"date":"2019-09-26T19:01:25","date_gmt":"2019-09-26T10:01:25","guid":{"rendered":"http:\/\/csharp.ihavenomoney.co.kr\/?p=673"},"modified":"2019-09-26T19:01:25","modified_gmt":"2019-09-26T10:01:25","slug":"web-sockets-5","status":"publish","type":"post","link":"https:\/\/csharp.ihavenomoney.co.kr\/?p=673","title":{"rendered":"web-sockets-5"},"content":{"rendered":"\n
using Microsoft.AspNetCore.Hosting; \/\/IWebHostBuilder\r\nusing Microsoft.AspNetCore.Builder; \/\/IApplicationBuilder\r\nusing Microsoft.AspNetCore.Http; \/\/WriteAsync\r\nusing Microsoft.Extensions.DependencyInjection;\r\nusing Microsoft.Extensions.Logging;\r\nusing Microsoft.AspNetCore;\r\nusing System;\r\nusing System.Threading;\r\nusing System.Net.WebSockets;\r\nusing System.Text;\r\nusing System.IO;\r\nusing System.Collections.Concurrent;\r\nusing System.Linq;\r\nusing System.Threading.Tasks;\r\nusing System.Collections.Generic;\r\n\r\nnamespace WebApplication1\r\n{\r\n    public class ConnectionManager\r\n    {\r\n        ConcurrentDictionary<string, (WebSocket socket, string nickname)> _sockets = new ConcurrentDictionary<string, (WebSocket, string)>();\r\n\r\n        public string AddSocket(WebSocket socket)\r\n        {\r\n            var id = Guid.NewGuid().ToString();\r\n\r\n            if (!_sockets.TryAdd(id, (socket, string.Empty)))\r\n                throw new Exception($\"Problem in adding socket with Id {id}\");\r\n\r\n            return id;\r\n        }\r\n\r\n        public bool SetNickName(string id, string nickname)\r\n        {\r\n            if (_sockets.TryGetValue(id, out var x))\r\n            {\r\n                _sockets[id] = (x.socket, nickname);\r\n                return true;\r\n            }\r\n            else\r\n                return false;\r\n        }\r\n\r\n        public (bool, string) GetNickNameById(string id)\r\n        {\r\n            if (_sockets.TryGetValue(id, out var x))\r\n                return (true, x.nickname);\r\n            else\r\n                return (false, null);\r\n        }\r\n\r\n        public (bool, WebSocket socket) GetByNick(string nickname)\r\n        {\r\n            var found = _sockets.Where(x => x.Value.nickname.Equals(nickname, StringComparison.CurrentCultureIgnoreCase)).Take(1).ToList();\r\n\r\n            if (found.Count == 0)\r\n                return (false, null);\r\n            else\r\n                return (true, found[0].Value.socket);\r\n        }\r\n\r\n        public bool RemoveSocket(string id) => _sockets.TryRemove(id, out (WebSocket, string) _);\r\n\r\n        public List<(WebSocket socket, string id, string nickname)> Other(string id) =>\r\n            _sockets.Where(x => x.Key != id)\r\n            .Select(x => (x.Value.socket, x.Key, x.Value.nickname))\r\n            .ToList();\r\n    }\r\n\r\n    public enum CommandType\r\n    {\r\n        List,\r\n        Send,\r\n        Nick,\r\n        Quit\r\n    }\r\n\r\n    public class Command\r\n    {\r\n        public CommandType Type { get; set; }\r\n\r\n        public (string, string, string) Data { get; set; }\r\n    }\r\n\r\n    public class CommandHandler\r\n    {\r\n        public (bool, Command) Parse(string cmd)\r\n        {\r\n            try\r\n            {\r\n                if (cmd.StartsWith(\"#\"))\r\n                {\r\n                    var segment = cmd.Split(new[] { ' ' });\r\n\r\n                    if (segment.Length > 0)\r\n                    {\r\n                        switch (segment[0])\r\n                        {\r\n                            case \"#list\": return (true, new Command { Type = CommandType.List, Data = (\"\", \"\", \"\") });\r\n                            case \"#quit\": return (true, new Command { Type = CommandType.Quit, Data = (\"\", \"\", \"\") });\r\n                            case \"#nick\": return (true, new Command { Type = CommandType.Nick, Data = (segment[1], \"\", \"\") });\r\n                            case \"#talk\": return (true, new Command { Type = CommandType.Send, Data = (segment[1], string.Join(\" \", segment.Skip(2)), \"\") });\r\n                            default: return (false, null);\r\n                        }\r\n                    }\r\n                }\r\n\r\n                return (false, null);\r\n            }\r\n            catch\r\n            {\r\n                return (false, null);\r\n            }\r\n        }\r\n    }\r\n\r\n    public class Startup\r\n    {\r\n        async Task ReceiveAsync(ConnectionManager cm, ILogger log, WebSocket socket, string socketId, Func<ConnectionManager, string, Task> responseHandlerAsync)\r\n        {\r\n            var bufferSize = new byte[4]; \/\/This is especially small just to exercise the code that handles data that is larger than buffer\r\n            var receiveBuffer = new ArraySegment<byte>(bufferSize);\r\n            WebSocketReceiveResult result;\r\n\r\n            while (socket.State == WebSocketState.Open)\r\n            {\r\n                using (var ms = new MemoryStream())\r\n                {\r\n                    do\r\n                    {\r\n                        result = await socket.ReceiveAsync(receiveBuffer, CancellationToken.None);\r\n\r\n                        if (result.MessageType == WebSocketMessageType.Close)\r\n                        {\r\n                            log.LogDebug($\"Socket Id {socketId} : Receive closing message.\");\r\n                            var removalStatus = cm.RemoveSocket(socketId);\r\n                            log.LogDebug($\"Socket Id {socketId} removal status {removalStatus}.\");\r\n                            break;\r\n                        }\r\n\r\n                        if (result.MessageType != WebSocketMessageType.Text)\r\n                            throw new Exception(\"Unexpected Message\");\r\n\r\n                        ms.Write(receiveBuffer.Array, receiveBuffer.Offset, result.Count);\r\n                    }\r\n                    while (!result.EndOfMessage && !result.CloseStatus.HasValue);\r\n\r\n                    if (result.MessageType == WebSocketMessageType.Text)\r\n                    {\r\n                        ms.Seek(0, SeekOrigin.Begin);\r\n\r\n                        string clientRequest = string.Empty;\r\n                        using (var reader = new StreamReader(ms, Encoding.UTF8))\r\n                        {\r\n                            clientRequest = reader.ReadToEnd();\r\n                        }\r\n\r\n                        log.LogDebug($\"Socket Id {socketId} : Receive: {clientRequest}\");\r\n\r\n                        await responseHandlerAsync(cm, clientRequest);\r\n                    }\r\n\r\n                    if (result.CloseStatus.HasValue)\r\n                        break;\r\n                }\r\n            }\r\n        }\r\n\r\n        public ArraySegment<byte> Reply(string content) => new ArraySegment<byte>(Encoding.UTF8.GetBytes(content));\r\n\r\n        public void Configure(IApplicationBuilder app, IHostingEnvironment env, ILoggerFactory logger)\r\n        {\r\n           \/\/ logger.AddConsole((str, level) => !str.Contains(\"Microsoft.AspNetCore\") && level >= LogLevel.Trace);\r\n\r\n            var log = logger.CreateLogger(\"\");\r\n\r\n            app.UseWebSockets();\r\n\r\n            var cm = new ConnectionManager();\r\n\r\n            app.Use(async (context, next) =>\r\n            {\r\n                if (!context.WebSockets.IsWebSocketRequest)\r\n                {\r\n                    await next();\r\n                    return;\r\n                }\r\n\r\n                var socket = await context.WebSockets.AcceptWebSocketAsync();\r\n                var socketId = cm.AddSocket(socket);\r\n\r\n                var cmdHandler = new CommandHandler();\r\n                await ReceiveAsync(cm, log, socket, socketId, async (connectionManager, clientRequest) =>\r\n                {\r\n                    var (isOK, cmd) = cmdHandler.Parse(clientRequest);\r\n\r\n                    if (isOK)\r\n                    {\r\n                        switch (cmd.Type)\r\n                        {\r\n                            case CommandType.List:\r\n                                {\r\n                                    var others = connectionManager.Other(socketId).Select(x => string.IsNullOrWhiteSpace(x.nickname) ? \"NoNick\" : x.nickname).ToList();\r\n                                    if (others.Count > 0)\r\n                                        await socket.SendAsync(Reply(string.Join(\",\", others)), WebSocketMessageType.Text, true, CancellationToken.None);\r\n                                    else\r\n                                        await socket.SendAsync(Reply(\"No other user on this channel\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n\r\n                                    break;\r\n                                }\r\n\r\n                            case CommandType.Nick:\r\n                                {\r\n                                    var isOk = connectionManager.SetNickName(socketId, cmd.Data.Item1);\r\n                                    if (isOK)\r\n                                        await socket.SendAsync(Reply($\"Nickname now {cmd.Data.Item1}\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n                                    else\r\n                                        await socket.SendAsync(Reply($\"#nick fails\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n\r\n                                    break;\r\n                                }\r\n\r\n                            case CommandType.Send:\r\n                                {\r\n                                    var (isFound, sck) = connectionManager.GetByNick(cmd.Data.Item1);\r\n                                    if (isFound)\r\n                                    {\r\n                                        var (isOk, sender) = connectionManager.GetNickNameById(socketId);\r\n                                        if (isOK)\r\n                                            await sck.SendAsync(Reply($\"From {sender}: {cmd.Data.Item2}\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n                                        else\r\n                                            await sck.SendAsync(Reply($\"From Unknown: {cmd.Data.Item2}\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n\r\n                                        await socket.SendAsync(Reply($\"Message sent to {cmd.Data.Item1}\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n                                    }\r\n                                    else\r\n                                        await socket.SendAsync(Reply($\"{cmd.Data.Item1} not found\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n\r\n                                    break;\r\n                                }\r\n\r\n                            case CommandType.Quit:\r\n                                {\r\n                                    connectionManager.RemoveSocket(socketId);\r\n                                    await socket.SendAsync(Reply(\"Quitting chat\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n                                    await socket.CloseAsync(WebSocketCloseStatus.NormalClosure, \"\", CancellationToken.None);\r\n\r\n                                    break;\r\n                                }\r\n\r\n                            default:\r\n                                {\r\n                                    await socket.SendAsync(Reply(\"Command not understood\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n\r\n                                    break;\r\n                                }\r\n                        }\r\n                    }\r\n                    else\r\n                        await socket.SendAsync(Reply(\"Command not understood\"), WebSocketMessageType.Text, true, CancellationToken.None);\r\n                });\r\n\r\n                if (socket.State != WebSocketState.Open)\r\n                    log.LogDebug($\"Socket Id {socketId} with status {socket.State}\");\r\n            });\r\n\r\n            app.Run(async context =>\r\n            {\r\n                context.Response.Headers.Add(\"content-type\", \"text\/html\");\r\n                await context.Response.WriteAsync(@\"\r\n<html>                \r\n    <head>\r\n        <script src=\"\"https:\/\/code.jquery.com\/jquery-3.2.1.min.js\"\" integrity=\"\"sha256-hwg4gsxgFZhOsEEamdOYGBf13FyQuiTwlAQgxVSNgt4=\"\" crossorigin=\"\"anonymous\"\"><\/script>\r\n    <\/head>\r\n    <body>\r\n        <h1>Web Socket (please open this page at 2 tabs at least)<\/h1>\r\n        <p>Commands<p>\r\n        <ul>\r\n            <li>#list<\/li> \r\n            <li>#nick <i>nickname<\/i><\/li>\r\n            <li>#talk <i>nickname<\/i> <i>text<\/i><\/li>\r\n            <li>#quit<\/li>\r\n        <\/ul>\r\n        <input type=\"\"text\"\" length=\"\"50\"\" id=\"\"msg\"\" value=\"\"#nick anne\"\"\/> \r\n        <button type=\"\"button\"\" id=\"\"send\"\">Send<\/button>\r\n        <button type=\"\"button\"\" id=\"\"close\"\">Close<\/button>\r\n        <br\/>\r\n        <ul id=\"\"responses\"\"><\/ul>\r\n        <script>\r\n            $(function(){\r\n                var url = \"\"wss:\/\/localhost:44323\/\"\";\r\n                var socket = new WebSocket(url);\r\n                var send = $(\"\"#send\"\");\r\n                var close = $(\"\"#close\"\");\r\n                var msg = $(\"\"#msg\"\");\r\n                var responses = $(\"\"#responses\"\");\r\n\r\n                socket.onopen = function(e){\r\n                    responses.append('<li>Socket opened<\/li>');\r\n                    send.click(function(){\r\n                        if (socket.readyState !== WebSocket.OPEN){\r\n                            alert('Socket is closed. Cannot send message.');\r\n                            return;\r\n                        }\r\n                        socket.send(msg.val()); \r\n                    });\r\n                };\r\n\r\n                close.click(function(){\r\n                    if (socket.readyState !== WebSocket.OPEN){\r\n                        alert('You cannot close this connection because it is already closed');\r\n                        return;\r\n                    }\r\n                    socket.close();    \r\n                });\r\n\r\n                socket.onmessage = function(e){\r\n                    var response = e.data;\r\n                    responses.append('<BR\/>');\r\n                  responses.append(response.trim());\r\n\r\n                    \/\/alert(response.trim());\r\n\r\n                };\r\n\r\n                socket.onclose = function(e){\r\n                    responses.append('<li>Socket closed<\/li>');\r\n                };\r\n            });\r\n        <\/script>\r\n    <\/body>\r\n<\/html>\");\r\n            });\r\n        }\r\n    }\r\n\r\n    public class Program\r\n    {\r\n        public static void Main(string[] args)\r\n        {\r\n            CreateWebHostBuilder(args).Build().Run();\r\n        }\r\n\r\n        static IWebHostBuilder CreateWebHostBuilder(string[] args) =>\r\n            WebHost.CreateDefaultBuilder(args)\r\n                .UseStartup<Startup>()\r\n                .UseEnvironment(\"Development\");\r\n    }\r\n}<\/pre>\n
\uacb0\uacfc :
\nWeb Socket (please open this page at 2 tabs at least)<\/p>\n
Commands<\/p>\n
\u2022#list
\n\u2022#nick nickname
\n\u2022#talk nickname text
\n\u2022#quit
\n[     ] Send Close <\/p>\n
\u2022Socket opened
\nNickname now nickname2
\nMessage sent to nickname1
\nFrom nickname1: text
\nMessage sent to nickname
\nnickname1,nickname
\nMessage sent to nickname1
\n\u2022Socket closed<\/p>\n","protected":false},"excerpt":{"rendered":"
using Microsoft.AspNetCore.Hosting; \/\/IWebHostBuilder using Microsoft.AspNetCore.Builder; \/\/IApplicationBuilder using Microsoft.AspNetCore.Http; \/\/WriteAsync using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.Logging; using Microsoft.AspNetCore; using System; using System.Threading; using System.Net.WebSockets; using System.Text; using System.IO; using System.Collections.Concurrent; using System.Linq; using System.Threading.Tasks; using System.Collections.Generic; namespace WebApplication1 { public class ConnectionManager { ConcurrentDictionary<string, (WebSocket socket, string nickname)> _sockets = new ConcurrentDictionary<string, (WebSocket, string)>(); public string AddSocket(WebSocket socket)\u2026 Read More »<\/a><\/span><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[108],"tags":[],"class_list":["post-673","post","type-post","status-publish","format-standard","hentry","category-net-core"],"jetpack_featured_media_url":"","_links":{"self":[{"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=\/wp\/v2\/posts\/673","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=673"}],"version-history":[{"count":1,"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=\/wp\/v2\/posts\/673\/revisions"}],"predecessor-version":[{"id":674,"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=\/wp\/v2\/posts\/673\/revisions\/674"}],"wp:attachment":[{"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=673"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=673"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/csharp.ihavenomoney.co.kr\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=673"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}