SERVER-52822 Introduced ThreadContext class

2 files changed, 36 insertions, 0 deletions

diff --git a/src/mongo/platform/process_id.cpp b/src/mongo/platform/process_id.cpp
index 48bc0354316..cff6af7e2db 100644
--- a/src/mongo/platform/process_id.cpp
+++ b/src/mongo/platform/process_id.cpp
@@ -31,11 +31,21 @@
 
 #include "mongo/platform/process_id.h"
 
+#ifndef _WIN32
+#include <pthread.h>
+#endif
+
+#if defined(__linux__)
+#include <sys/syscall.h>
+#include <sys/types.h>
+#endif
+
 #include <iostream>
 #include <limits>
 #include <sstream>
 
 #include "mongo/base/static_assert.h"
+#include "mongo/util/assert_util.h"
 
 namespace mongo {
 
@@ -51,12 +61,33 @@ inline NativeProcessId getCurrentNativeProcessId() {
     return getpid();
 }
 #endif
+
+#ifdef _WIN32
+inline NativeProcessId getCurrentNativeThreadId() {
+    return GetCurrentThreadId();
+}
+#elif __APPLE__
+inline NativeProcessId getCurrentNativeThreadId() {
+    // macOS deprecated syscall in 10.12.
+    uint64_t tid;
+    invariant(::pthread_threadid_np(NULL, &tid) == 0);
+    return tid;
+}
+#else
+inline NativeProcessId getCurrentNativeThreadId() {
+    return ::syscall(SYS_gettid);
+}
+#endif
 }  // namespace
 
 ProcessId ProcessId::getCurrent() {
     return fromNative(getCurrentNativeProcessId());
 }
 
+ProcessId ProcessId::getCurrentThreadId() {
+    return fromNative(getCurrentNativeThreadId());
+}
+
 int64_t ProcessId::asInt64() const {
     typedef std::numeric_limits<NativeProcessId> limits;
     if (limits::is_signed)
diff --git a/src/mongo/platform/process_id.h b/src/mongo/platform/process_id.h
index 2e9d1f721f7..2e2e6396de9 100644
--- a/src/mongo/platform/process_id.h
+++ b/src/mongo/platform/process_id.h
@@ -56,6 +56,11 @@ public:
     static ProcessId getCurrent();
 
     /**
+     * Gets the thread id for the currently executing process.
+     */
+    static ProcessId getCurrentThreadId();
+
+    /**
      * Constructs a ProcessId from a NativeProcessId.
      */
     static inline ProcessId fromNative(NativeProcessId npid) {