add CPU affinity

Author: freshthinking

include/slick_socket/tcp_client.h

@@ -17,6 +17,7 @@ struct TCPClientConfig
     uint16_t server_port = 5000;
     int receive_buffer_size = 4096;
     std::chrono::milliseconds connection_timeout{30000};
+    int cpu_affinity = -1;  // -1 means no affinity, otherwise specify CPU core index
 };
 
 template<typename DerivedT, typename LoggerT = ConsoleLogger>

include/slick_socket/tcp_client_unix.h

@@ -10,6 +10,7 @@
 #include <fcntl.h>
 #include <csignal>
 #include <cstring>
+#include <pthread.h>
 
 namespace slick_socket
 {
@@ -145,6 +146,26 @@ inline void TCPClientBase<DerivedT, LoggerT>::client_loop()
 {
     logger_.logInfo("Client loop started");
 
+    // Set CPU affinity if specified
+    if (config_.cpu_affinity >= 0)
+    {
+        cpu_set_t cpuset;
+        CPU_ZERO(&cpuset);
+        CPU_SET(config_.cpu_affinity, &cpuset);
+        
+        pthread_t thread = pthread_self();
+        int result = pthread_setaffinity_np(thread, sizeof(cpu_set_t), &cpuset);
+        if (result != 0)
+        {
+            logger_.logWarning("Failed to set CPU affinity to core {}: {}", 
+                             config_.cpu_affinity, std::strerror(result));
+        }
+        else
+        {
+            logger_.logInfo("Client thread pinned to CPU core {}", config_.cpu_affinity);
+        }
+    }
+
     // Connection established - handle server communication
     std::vector<uint8_t> buffer(config_.receive_buffer_size);
 

include/slick_socket/tcp_client_win32.h

@@ -4,6 +4,7 @@
 
 #include "tcp_client.h"
 #include <ws2tcpip.h>
+#include <windows.h>
 
 #pragma comment(lib, "ws2_32.lib")
 namespace slick_socket
@@ -142,6 +143,23 @@ inline void TCPClientBase<DerivedT, LoggerT>::client_loop()
 {
     logger_.logDebug("Client loop started");
 
+    // Set CPU affinity if specified
+    if (config_.cpu_affinity >= 0)
+    {
+        DWORD_PTR mask = 1ULL << config_.cpu_affinity;
+        HANDLE thread = GetCurrentThread();
+        DWORD_PTR result = SetThreadAffinityMask(thread, mask);
+        if (result == 0)
+        {
+            logger_.logWarning("Failed to set CPU affinity to core {}: error {}", 
+                             config_.cpu_affinity, GetLastError());
+        }
+        else
+        {
+            logger_.logInfo("Client thread pinned to CPU core {}", config_.cpu_affinity);
+        }
+    }
+
     // Connection established - handle server communication
     std::vector<uint8_t> buffer(config_.receive_buffer_size);
 

include/slick_socket/tcp_server.h

@@ -24,6 +24,7 @@ struct TCPServerConfig
     bool reuse_address = true;
     int receive_buffer_size = 4096;
     std::chrono::milliseconds connection_timeout{30000};
+    int cpu_affinity = -1;  // -1 means no affinity, otherwise specify CPU core index
 };
 
 template<typename DrivedT, typename LoggerT = ConsoleLogger>

include/slick_socket/tcp_server_unix.h

@@ -12,6 +12,7 @@
 #include <csignal>
 #include <algorithm>
 #include <cstring>
+#include <pthread.h>
 
 namespace slick_socket
 {
@@ -194,6 +195,26 @@ inline void TCPServerBase<DrivedT, LoggerT>::disconnect_client(int client_id)
 template<typename DrivedT, typename LoggerT>
 void TCPServerBase<DrivedT, LoggerT>::server_loop()
 {
+    // Set CPU affinity if specified
+    if (config_.cpu_affinity >= 0)
+    {
+        cpu_set_t cpuset;
+        CPU_ZERO(&cpuset);
+        CPU_SET(config_.cpu_affinity, &cpuset);
+        
+        pthread_t thread = pthread_self();
+        int result = pthread_setaffinity_np(thread, sizeof(cpu_set_t), &cpuset);
+        if (result != 0)
+        {
+            logger_.logWarning("Failed to set CPU affinity to core {}: {}", 
+                             config_.cpu_affinity, std::strerror(result));
+        }
+        else
+        {
+            logger_.logInfo("Server thread pinned to CPU core {}", config_.cpu_affinity);
+        }
+    }
+
     fd_set read_fds;
     struct timeval timeout;
     std::vector<uint8_t> buffer(config_.receive_buffer_size);

include/slick_socket/tcp_server_win32.h

@@ -4,6 +4,7 @@
 
 #include "tcp_server.h"
 #include <ws2tcpip.h>
+#include <windows.h>
 #include <queue>
 #include <algorithm>
 
@@ -194,6 +195,23 @@ inline void TCPServerBase<DrivedT, LoggerT>::disconnect_client(int client_id)
 template<typename DrivedT, typename LoggerT>
 void TCPServerBase<DrivedT, LoggerT>::server_loop()
 {
+    // Set CPU affinity if specified
+    if (config_.cpu_affinity >= 0)
+    {
+        DWORD_PTR mask = 1ULL << config_.cpu_affinity;
+        HANDLE thread = GetCurrentThread();
+        DWORD_PTR result = SetThreadAffinityMask(thread, mask);
+        if (result == 0)
+        {
+            logger_.logWarning("Failed to set CPU affinity to core {}: error {}", 
+                             config_.cpu_affinity, GetLastError());
+        }
+        else
+        {
+            logger_.logInfo("Server thread pinned to CPU core {}", config_.cpu_affinity);
+        }
+    }
+
     fd_set read_fds;
     struct timeval timeout;
     std::vector<uint8_t> buffer(config_.receive_buffer_size);