✨ TestClient, but wait, what are we actually need?

Signed-off-by: 秋雨落 <i@rain.cx>
2025-05-22 02:43:00 +00:00 · 2023-09-13 16:12:27 +08:00
parent c43b48398e
commit 185b231e7d
10 changed files with 320 additions and 8 deletions
@@ -0,0 +1,156 @@
+using System.Formats.Cbor;
+using System.Net.WebSockets;
+
+var webSocket = new ClientWebSocket();
+var cborWriter = new CborWriter();
+
+while (true)
+{
+    try
+    {
+        await Process();
+    }
+    catch (Exception ex)
+    {
+        Console.WriteLine(ex);
+    }
+}
+
+async Task Receive()
+{
+    var buffer = new byte[1024 * 4];
+    var result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
+
+    while (!result.CloseStatus.HasValue)
+    {
+        var reader = new CborReader(new ReadOnlyMemory<byte>(buffer, 0, result.Count));
+        Console.WriteLine(reader.BytesRemaining);
+        
+        result = await webSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
+    }
+}
+
+async Task Process()
+{
+    Console.Write("> ");
+    var operation = Console.ReadLine();
+
+    if (operation == "")
+    {
+        return;
+    }
+
+    if (operation == "connect")
+    {
+        var uri = Console.ReadLine();
+        await webSocket.ConnectAsync(new Uri(uri!), CancellationToken.None);
+        Receive();
+    }
+    else if (operation == "message")
+    {
+        cborWriter.Reset();
+        cborWriter.WriteStartArray(null);
+    }
+    else if (operation == "send")
+    {
+        cborWriter.WriteEndArray();
+        var bytes = cborWriter.Encode();
+        await webSocket.SendAsync(bytes, WebSocketMessageType.Binary, true, CancellationToken.None);
+    }
+    else if (operation == "bye")
+    {
+        await webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, "", CancellationToken.None);
+        webSocket = new ClientWebSocket();
+    }
+    else if (operation == "help")
+    {
+        Console.WriteLine("Every space should use enter.");
+        Console.WriteLine("Connect to a server: connect <uri>");
+        Console.WriteLine("New message: message");
+        Console.WriteLine("Send message: send");
+        Console.WriteLine("Close websocket: bye");
+        Console.WriteLine("Insert data: data <type> <value>");
+    }
+    else if (operation == "data")
+    {
+        var type = Console.ReadLine();
+        var valueStr = Console.ReadLine();
+        
+        if (type == "string")
+        {
+            cborWriter.WriteTextString(valueStr!);
+        }
+        else if (type == "int")
+        {
+            var result = int.TryParse(valueStr, out var value);
+            while (!result)
+            {
+                valueStr = Console.ReadLine();
+                
+                result = int.TryParse(valueStr, out value);
+            }
+            cborWriter.WriteInt32(value);
+        }
+        else if (type == "long")
+        {
+            var result = long.TryParse(valueStr, out var value);
+            while (!result)
+            {
+                valueStr = Console.ReadLine();
+                
+                result = long.TryParse(valueStr, out value);
+            }
+            cborWriter.WriteInt64(value);
+        }
+        else if (type == "bool")
+        {
+            var result = bool.TryParse(valueStr, out var value);
+            while (!result)
+            {
+                valueStr = Console.ReadLine();
+                
+                result = bool.TryParse(valueStr, out value);
+            }
+            cborWriter.WriteBoolean(value);
+        }
+        else if (type == "float")
+        {
+            var result = float.TryParse(valueStr, out var value);
+            while (!result)
+            {
+                valueStr = Console.ReadLine();
+                
+                result = float.TryParse(valueStr, out value);
+            }
+            cborWriter.WriteSingle(value);
+        }
+        else if (type == "double")
+        {
+            var result = double.TryParse(valueStr, out var value);
+            while (!result)
+            {
+                valueStr = Console.ReadLine();
+                
+                result = double.TryParse(valueStr, out value);
+            }
+            cborWriter.WriteDouble(value);
+        }
+        else if (type == "decimal")
+        {
+            var result = decimal.TryParse(valueStr, out var value);
+            while (!result)
+            {
+                valueStr = Console.ReadLine();
+                
+                result = decimal.TryParse(valueStr, out value);
+            }
+            cborWriter.WriteDecimal(value);
+        }
+        else if (type == "null")
+        {
+            cborWriter.WriteNull();
+        }
+    }
+
+    Console.WriteLine("Done!");
+}