tqcq/sled
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix crash for static ThreadPool
All checks were successful
linux-mips64-gcc / linux-gcc-mips64el (Debug) (push) Successful in 1m57s
Details
linux-x64-gcc / linux-gcc (Debug) (push) Successful in 2m20s
Details
linux-mips64-gcc / linux-gcc-mips64el (Release) (push) Successful in 2m35s
Details
linux-x64-gcc / linux-gcc (Release) (push) Successful in 2m48s
Details
linux-aarch64-cpu-gcc / linux-gcc-aarch64 (push) Successful in 3m41s
Details
linux-arm-gcc / linux-gcc-armhf (push) Successful in 6m57s
Details

Browse Source
This commit is contained in:
tqcq 2024-04-24 15:36:09 +08:00
parent 85852ee2e9
commit 0a2c5e3ca7
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/sled/futures/future.cc
View File

@ -14,23 +14,30 @@ DecrementFuturesUsage()
}// namespace future_detail
static ThreadPool g_default_thread_pool;
TaskQueueBase *g_default_scheduler = &g_default_thread_pool;
static std::atomic<TaskQueueBase *> g_default_scheduler{nullptr};
// static ThreadPool default_thread_pool;
static std::unique_ptr<Thread> g_default_thread;
void
SetDefaultScheduler(TaskQueueBase *scheduler) noexcept
{
if (scheduler == nullptr) {
g_default_scheduler = &g_default_thread_pool;
} else {
g_default_scheduler = scheduler;
}
SLED_ASSERT(scheduler, "scheduler is nullptr");
g_default_scheduler.store(scheduler, std::memory_order_release);
}
TaskQueueBase *
GetDefaultScheduler() noexcept
{
return g_default_scheduler;
static std::once_flag flag;
std::call_once(flag, [&] {
g_default_thread = sled::Thread::Create();
g_default_thread->Start();
TaskQueueBase *null_scheduler = nullptr;
while (g_default_scheduler.load() == nullptr) {
g_default_scheduler.compare_exchange_weak(null_scheduler, g_default_thread.get());
}
});
return g_default_scheduler.load(std::memory_order_acquire);
}
}// namespace sled