commit 9aeaa4d65074c3f4b25551f7b20adf5c87734ad7
author Douglas Gregor <dgregor@apple.com> Tue Dec 07 00:05:48 2010 +0000
committer Douglas Gregor <dgregor@apple.com> Tue Dec 07 00:05:48 2010 +0000
tree 79a83aa18df886e832f62f122c1c6db16e6f5958
parent 9e1994698d3da8cecec5a6cc5e95b18db913ef4b

Use atomic operations for libclang's object count tracking.

llvm-svn: 121073

clang/lib/Frontend/ASTUnit.cpp

diff --git a/clang/lib/Frontend/ASTUnit.cpp b/clang/lib/Frontend/ASTUnit.cpp
index 586002d..97e0326 100644
--- a/clang/lib/Frontend/ASTUnit.cpp
+++ b/clang/lib/Frontend/ASTUnit.cpp
@@ -35,6 +35,7 @@
 #include "clang/Basic/TargetInfo.h"
 #include "clang/Basic/Diagnostic.h"
 #include "llvm/ADT/StringSet.h"
+#include "llvm/Support/Atomic.h"
 #include "llvm/Support/MemoryBuffer.h"
 #include "llvm/Support/Host.h"
 #include "llvm/Support/Path.h"
@@ -85,7 +86,7 @@
 /// \brief Tracks the number of ASTUnit objects that are currently active.
 ///
 /// Used for debugging purposes only.
-static unsigned ActiveASTUnitObjects;
+static llvm::sys::cas_flag ActiveASTUnitObjects;
 
 ASTUnit::ASTUnit(bool _MainFileIsAST)
   : CaptureDiagnostics(false), MainFileIsAST(_MainFileIsAST), 
@@ -98,7 +99,7 @@
     CacheCodeCompletionCoolDown(0),
     UnsafeToFree(false) { 
   if (getenv("LIBCLANG_OBJTRACKING")) {
-    ++ActiveASTUnitObjects;
+    llvm::sys::AtomicIncrement(&ActiveASTUnitObjects);
     fprintf(stderr, "+++ %d translation units\n", ActiveASTUnitObjects);
   }    
 }
@@ -129,7 +130,7 @@
   ClearCachedCompletionResults();  
   
   if (getenv("LIBCLANG_OBJTRACKING")) {
-    --ActiveASTUnitObjects;
+    llvm::sys::AtomicDecrement(&ActiveASTUnitObjects);
     fprintf(stderr, "--- %d translation units\n", ActiveASTUnitObjects);
   }    
 }