Merge pull request #1 from Local-Connectivity-Lab/zhennan_fix_linux

Fix build on Linux and various refactoring

Author: johnnzhou

Package.swift

@@ -29,6 +29,7 @@ let package = Package(
                 .product(name: "NIOConcurrencyHelpers", package: "swift-nio"),
                 .product(name: "NIOWebSocket", package: "swift-nio"),
                 .product(name: "NIOHTTP1", package: "swift-nio"),
+                .product(name: "NIOFoundationCompat", package: "swift-nio"),
                 .product(name: "NIOSSL", package: "swift-nio-ssl"),
                 .product(
                     name: "NIOTransportServices",
@@ -43,6 +44,14 @@ let package = Package(
             name: "LCLWebSocketTests",
             dependencies: ["LCLWebSocket"]
         ),
+        .executableTarget(
+            name: "AutobahnClient",
+            dependencies: ["LCLWebSocket"]
+        ),
+        .executableTarget(
+            name: "AutobahnServer",
+            dependencies: ["LCLWebSocket"]
+        ),
         .executableTarget(name: "Client", dependencies: ["LCLWebSocket"]),
         .executableTarget(name: "Server", dependencies: ["LCLWebSocket"]),
     ]

Sources/AutobahnClient/AutobahnClient.swift

@@ -0,0 +1,75 @@
+//
+// This source file is part of the LCL open source project
+//
+// Copyright (c) 2021-2024 Local Connectivity Lab and the project authors
+// Licensed under Apache License v2.0
+//
+// See LICENSE for license information
+// See CONTRIBUTORS for the list of project authors
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+
+import Foundation
+import LCLWebSocket
+import NIOCore
+import NIOPosix
+
+@main
+struct TestWebSocketClient {
+
+    static let config = LCLWebSocket.Configuration(
+        maxFrameSize: 1 << 16,
+        autoPingConfiguration: .disabled,
+        leftoverBytesStrategy: .forwardBytes
+    )
+
+    static let serverAddress = "127.0.0.1"
+    static let serverPort = 9001
+    static let agentName = "LCLWebSocketClient"
+
+    public static func main() throws {
+
+        let elg = MultiThreadedEventLoopGroup(numberOfThreads: 1)
+        var client = LCLWebSocket.client(on: elg)
+        let totalTestCount = elg.any().makePromise(of: Int.self)
+
+        client.onText { websocket, text in
+            guard let total = Int(text) else {
+                fatalError()
+            }
+            totalTestCount.succeed(total)
+        }
+
+        try client.connect(to: "ws://\(Self.serverAddress):\(Self.serverPort)/getCaseCount", configuration: Self.config)
+            .wait()
+
+        let total = try totalTestCount.futureResult.wait()
+        print("Running total tests: \(total)")
+
+        for i in 1...total {
+            var client = LCLWebSocket.client(on: elg)
+            client.onText { ws, text in
+                ws.send(.init(string: text), opcode: .text)
+            }
+            client.onBinary { ws, binary in
+                ws.send(binary, opcode: .binary)
+            }
+            try client.connect(
+                to: "ws://\(Self.serverAddress):\(Self.serverPort)/runCase?case=\(i)&agent=\(Self.agentName)",
+                configuration: Self.config
+            ).wait()
+        }
+
+        let closeClient = LCLWebSocket.client()
+        do {
+            try closeClient.connect(
+                to: "ws://\(Self.serverAddress):\(Self.serverPort)/updateReports?agent=\(Self.agentName)",
+                configuration: Self.config
+            ).wait()
+        } catch {
+            print("Error closing client: \(error)")
+        }
+
+    }
+}

Sources/AutobahnServer/AutohahnServer.swift

@@ -0,0 +1,54 @@
+//
+// This source file is part of the LCL open source project
+//
+// Copyright (c) 2021-2024 Local Connectivity Lab and the project authors
+// Licensed under Apache License v2.0
+//
+// See LICENSE for license information
+// See CONTRIBUTORS for the list of project authors
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+
+import Foundation
+import LCLWebSocket
+import NIOCore
+import NIOPosix
+
+@main
+struct AutohahnServer {
+
+    static let config = LCLWebSocket.Configuration(
+        maxFrameSize: 1 << 16,
+        autoPingConfiguration: .disabled,
+        leftoverBytesStrategy: .forwardBytes
+    )
+
+    static let serverAddress = "127.0.0.1"
+    static let serverPort = 9000
+
+    static func main() throws {
+
+        let args = CommandLine.arguments
+        var port: Int = Self.serverPort
+        precondition(args.count == 1 || args.count == 3, "Usage: \(args[0]) [--port <port>]")
+        let portCmdIndex = args.firstIndex(where: { $0 == "--port" })
+        if let portCmdIndex = portCmdIndex {
+            let portIndex = portCmdIndex + 1
+            port = Int(args[portIndex]) ?? port
+        }
+
+        let elg = MultiThreadedEventLoopGroup(numberOfThreads: 4)
+        var server = LCLWebSocket.server(on: elg)
+
+        server.onBinary { ws, buffer in
+            ws.send(buffer, opcode: .binary)
+        }
+
+        server.onText { ws, text in
+            ws.send(.init(string: text), opcode: .text)
+        }
+
+        try server.listen(host: Self.serverAddress, port: port, configuration: Self.config).wait()
+    }
+}