Fix bug

Author: f0cii

moxt/libc.cpp

@@ -7,6 +7,16 @@
 #define ANKERL_NANOBENCH_IMPLEMENT
 #include <nanobench.h>
 
+#ifdef __linux
+#include <sys/types.h>
+#include <unistd.h>
+#elif _WIN32
+#include <windows.h>
+#elif __APPLE__
+#include <sys/types.h>
+#include <unistd.h>
+#endif
+
 SEQ_FUNC int64_t seq_voidptr_to_int(void *p) {
     return reinterpret_cast<int64_t>(p);
 }
@@ -59,4 +69,15 @@ SEQ_FUNC void seq_register_signal_handler(int signum, SignalHandler handler) {
         perror("sigaction");
         exit(EXIT_FAILURE);
     }
+}
+
+SEQ_FUNC int64_t seq_thread_id() {
+#ifdef __linux
+    pid_t threadId = gettid();
+#elif _WIN32
+    DWORD threadId = GetCurrentThreadId();
+#elif __APPLE__
+    pid_t threadId = gettid();
+#endif
+    return threadId;
 }

moxt/libc.hpp

@@ -3,6 +3,7 @@
 
 #include "common.hpp"
 #include <snmalloc/snmalloc.h>
+#include <thread>
 
 using namespace snmalloc;
 
@@ -54,5 +55,7 @@ SEQ_FUNC std::atomic<int64_t> *seq_atomic_int64_new(int64_t i);
 SEQ_FUNC void seq_atomic_int64_free(std::atomic<int64_t> *p);
 SEQ_FUNC int64_t seq_atomic_int64_load(std::atomic<int64_t> *p);
 SEQ_FUNC void seq_atomic_int64_store(std::atomic<int64_t> *p, int64_t i);
+// »ñÈ¡Ïß³Ìid
+SEQ_FUNC int64_t seq_thread_id();
 
 #endif

moxt/libmisc.cpp

@@ -29,9 +29,7 @@
 #include "moxt/utils/cast.hpp"
 #include <openssl/err.h>
 #include <openssl/opensslv.h>
-// #include <snmalloc/override/new.cc>
 #include "moxt/utils/snowflake.hpp"
-#include <atomic>
 #include <parallel_hashmap/phmap.h>
 #include <snmalloc/snmalloc.h>
 

moxt/utils/snowflake.cpp

@@ -41,14 +41,15 @@ uint32_t ipToWorkerId(const std::string &ipAddress) {
             ++it;
         }
     }
-    return workerId;
+    // Ensure workerId is within range [0, 1023]
+    return workerId % 1024;
 }
 
 bool Snowflake::initWorkerId() {
     const std::string &ip = getLocalIp();
-    logi("local ip: {}", ip);
+    // logi("local ip: {}", ip);
     workerId_ = ipToWorkerId(ip);
-    logi("WorkerId init success: {}", workerId_);
+    // logi("WorkerId init success: {}", workerId_);
     return true;
 }