[PATCH] D42742: [clangd] Use pthread instead of thread_local to support more runtimes.

[Sam McCall via Phabricator via cfe-commits]

This revision was automatically updated to reflect the committed changes.
Closed by commit rL323949: [clangd] Use pthread instead of thread_local to support more runtimes. (authored by sammccall, committed by ).
Herald added a subscriber: llvm-commits.

Repository:
  rL LLVM

https://reviews.llvm.org/D42742

Files:
  clang-tools-extra/trunk/clangd/Context.cpp


Index: clang-tools-extra/trunk/clangd/Context.cpp
===================================================================
--- clang-tools-extra/trunk/clangd/Context.cpp
+++ clang-tools-extra/trunk/clangd/Context.cpp
@@ -8,8 +8,48 @@
 //===---------------------------------------------------------------------===//
 
 #include "Context.h"
+#include "llvm/Config/config.h"
 #include <cassert>
 
+// The thread-local Context is scoped in a function to avoid init-order issues.
+// It's created by currentContext() when first needed.
+
+#ifdef HAVE_PTHREAD_GETSPECIFIC
+// We'd love to use thread_local everywhere.
+// It requires support from the runtime: __cxa_thread_atexit.
+// Rather than detect this, we use the pthread API where available.
+#include <pthread.h>
+static clang::clangd::Context &currentContext() {
+  using clang::clangd::Context;
+  static pthread_key_t CtxKey;
+
+  // Once (across threads), set up pthread TLS for Context, and its destructor.
+  static int Dummy = [] { // Create key only once, for all threads.
+    if (auto Err = pthread_key_create(&CtxKey, /*destructor=*/+[](void *Ctx) {
+          delete reinterpret_cast<Context *>(Ctx);
+        }))
+      llvm_unreachable(strerror(Err));
+    return 0;
+  }();
+  (void)Dummy;
+
+  // Now grab the current context from TLS, and create it if it doesn't exist.
+  void *Ctx = pthread_getspecific(CtxKey);
+  if (!Ctx) {
+    Ctx = new Context();
+    if (auto Err = pthread_setspecific(CtxKey, Ctx))
+      llvm_unreachable(strerror(Err));
+  }
+  return *reinterpret_cast<Context *>(Ctx);
+}
+#else
+// Only supported platform without pthread is windows, and thread_local works.
+static clang::clangd::Context &currentContext() {
+  static thread_local auto C = clang::clangd::Context::empty();
+  return C;
+}
+#endif
+
 namespace clang {
 namespace clangd {
 
@@ -20,13 +60,6 @@
 
 Context Context::clone() const { return Context(DataPtr); }
 
-// The thread-local Context is scoped in a function to avoid
-// initialization-order issues. It's created when first needed.
-static Context &currentContext() {
-  static thread_local Context C = Context::empty();
-  return C;
-}
-
 const Context &Context::current() { return currentContext(); }
 
 Context Context::swapCurrent(Context Replacement) {


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D42742.132339.patch
Type: text/x-patch
Size: 2274 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/cfe-commits/attachments/20180201/a07762da/attachment-0001.bin>