feat: 封装了 TCPServer 并通过测试

2025-09-29 09:01:33 +00:00
parent 74eca4a693
commit 38f7801afa
3 changed files with 249 additions and 0 deletions
@@ -0,0 +1,52 @@
+#include <server/tcp_server.hpp>
+
+#include <unistd.h>
+#include <iostream>
+#include <chrono>
+#include <thread>
+#include <string.h>
+
+int main()
+{
+    using namespace ouc_server::server;
+    using namespace ouc_server::ouc_socket;
+
+    TCPServer server;
+    if (!server.start("127.0.0.1", 8080))
+    {
+        std::cerr << "Failed to start server\n";
+        return 1;
+    }
+
+    puts("Start Listening!");
+
+    server.on_connection(
+        [](TCPSocket &client)
+        { std::cout << "New client connected, fd=" << client.get_fd() << "\n"; });
+
+    server.on_message(
+        [&](TCPSocket &client, const std::string &msg)
+        {
+            if (!memcmp("exit", msg.c_str(), 4))
+            {
+                server.remove_fd(client);
+                return;
+            }
+            std::cout << "Received: " << msg;
+
+            std::string ret_str("Echo: " + msg);
+            size_t count = client.send(ret_str.c_str());
+            while (count < ret_str.size())
+                count += client.send(ret_str.c_str() + count);
+        });
+
+    server.on_close(
+        [](TCPSocket &client)
+        { std::cout << "Client disconnected, fd=" << client.get_fd() << "\n"; });
+
+    while (true)
+    {
+        server.loop();
+        std::this_thread::sleep_for(std::chrono::milliseconds(50));
+    }
+}