#include "llvm/ADT/ScopeExit.h"
#include "llvm/Support/FormatVariadic.h"
#include "llvm/Support/Threading.h"
+#include <atomic>
#include <thread>
#ifdef __USE_POSIX
#include <pthread.h>
CV.wait_until(Lock, D.time());
}
+static std::atomic<bool> AvoidThreadStarvation = {false};
+
void setThreadPriority(std::thread &T, ThreadPriority Priority) {
// Some *really* old glibcs are missing SCHED_IDLE.
#if defined(__linux__) && defined(SCHED_IDLE)
priority.sched_priority = 0;
pthread_setschedparam(
T.native_handle(),
- Priority == ThreadPriority::Low ? SCHED_IDLE : SCHED_OTHER, &priority);
+ Priority == ThreadPriority::Low && !AvoidThreadStarvation ? SCHED_IDLE
+ : SCHED_OTHER,
+ &priority);
#endif
}
+void preventThreadStarvationInTests() { AvoidThreadStarvation = true; }
+
} // namespace clangd
} // namespace clang
Normal = 1,
};
void setThreadPriority(std::thread &T, ThreadPriority Priority);
+// Avoid the use of scheduler policies that may starve low-priority threads.
+// This prevents tests from timing out on loaded systems.
+// Affects subsequent setThreadPriority() calls.
+void preventThreadStarvationInTests();
} // namespace clangd
} // namespace clang
RunSynchronously = true;
InputStyle = JSONStreamStyle::Delimited;
PrettyPrint = true;
+ preventThreadStarvationInTests(); // Ensure background index makes progress.
}
if (Test || EnableTestScheme) {
static URISchemeRegistry::Add<TestScheme> X(
}
};
-TEST(BackgroundIndexTest, IndexTwoFiles) {
+class BackgroundIndexTest : public ::testing::Test {
+protected:
+ BackgroundIndexTest() { preventThreadStarvationInTests(); }
+};
+
+TEST_F(BackgroundIndexTest, IndexTwoFiles) {
MockFSProvider FS;
// a.h yields different symbols when included by A.cc vs B.cc.
FS.Files[testPath("root/A.h")] = R"cpp(
FileURI("unittest:///root/B.cc")}));
}
-TEST(BackgroundIndexTest, ShardStorageWriteTest) {
+TEST_F(BackgroundIndexTest, ShardStorageWriteTest) {
MockFSProvider FS;
FS.Files[testPath("root/A.h")] = R"cpp(
void common();