From 617d0bb4f68dd20a786140cb2494513d1e00fe98 Mon Sep 17 00:00:00 2001
From: Andrew Stanton-Nurse <andrew@stanton-nurse.com>
Date: Thu, 19 Apr 2018 11:04:39 -0700
Subject: [PATCH] fix websocket sample (#2096)

---
 samples/WebSocketSample/Program.cs | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/samples/WebSocketSample/Program.cs b/samples/WebSocketSample/Program.cs
index ef4afc3772..e4b204af48 100644
--- a/samples/WebSocketSample/Program.cs
+++ b/samples/WebSocketSample/Program.cs
@@ -11,15 +11,24 @@ namespace WebSocketSample
 {
     public class Program
     {
-        public static void Main(string[] args)
+        public static async Task<int> Main(string[] args)
         {
-            RunWebSockets().GetAwaiter().GetResult();
+            if (args.Length < 1)
+            {
+                Console.Error.WriteLine("Usage: WebSocketSample <URL>");
+                Console.Error.WriteLine("");
+                Console.Error.WriteLine("To connect to an ASP.NET Connection Handler, use 'ws://example.com/path/to/hub' or 'wss://example.com/path/to/hub' (for HTTPS)");
+                return 1;
+            }
+
+            await RunWebSockets(args[0]);
+            return 0;
         }
 
-        private static async Task RunWebSockets()
+        private static async Task RunWebSockets(string url)
         {
             var ws = new ClientWebSocket();
-            await ws.ConnectAsync(new Uri("ws://localhost:5000/chat/ws"), CancellationToken.None);
+            await ws.ConnectAsync(new Uri(url), CancellationToken.None);
 
             Console.WriteLine("Connected");