mirror of
https://github.com/intel/llvm.git
synced 2026-01-13 19:08:21 +08:00
be1a504228
Moves all Session member variables dedicated to shutdown into a new ShutdownInfo struct, and uses the presence / absence of this struct as the flag to indicate that we've entered the "shutting down" state. This simplifies the implementation of the shutdown process.
76 lines
1.9 KiB
C++
76 lines
1.9 KiB
C++
//===- Session.cpp --------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Contains the implementation of the Session class and related APIs.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "orc-rt/Session.h"
|
|
|
|
namespace orc_rt {
|
|
|
|
Session::~Session() { waitForShutdown(); }
|
|
|
|
void Session::shutdown(OnShutdownCompleteFn OnShutdownComplete) {
|
|
{
|
|
std::scoped_lock<std::mutex> Lock(M);
|
|
if (SI) {
|
|
SI->OnCompletes.push_back(std::move(OnShutdownComplete));
|
|
return;
|
|
}
|
|
|
|
SI = std::make_unique<ShutdownInfo>();
|
|
SI->OnCompletes.push_back(std::move(OnShutdownComplete));
|
|
std::swap(SI->ResourceMgrs, ResourceMgrs);
|
|
}
|
|
|
|
shutdownNext(Error::success());
|
|
}
|
|
|
|
void Session::waitForShutdown() {
|
|
shutdown([]() {});
|
|
std::unique_lock<std::mutex> Lock(M);
|
|
SI->CompleteCV.wait(Lock, [&]() { return SI->Complete; });
|
|
}
|
|
|
|
void Session::shutdownNext(Error Err) {
|
|
if (Err)
|
|
reportError(std::move(Err));
|
|
|
|
if (SI->ResourceMgrs.empty())
|
|
return shutdownComplete();
|
|
|
|
// Get the next ResourceManager to shut down.
|
|
auto NextRM = std::move(SI->ResourceMgrs.back());
|
|
SI->ResourceMgrs.pop_back();
|
|
NextRM->shutdown([this](Error Err) { shutdownNext(std::move(Err)); });
|
|
}
|
|
|
|
void Session::shutdownComplete() {
|
|
|
|
std::unique_ptr<TaskDispatcher> TmpDispatcher;
|
|
{
|
|
std::lock_guard<std::mutex> Lock(M);
|
|
TmpDispatcher = std::move(Dispatcher);
|
|
}
|
|
|
|
TmpDispatcher->shutdown();
|
|
|
|
for (auto &OnShutdownComplete : SI->OnCompletes)
|
|
OnShutdownComplete();
|
|
|
|
{
|
|
std::lock_guard<std::mutex> Lock(M);
|
|
SI->Complete = true;
|
|
}
|
|
|
|
SI->CompleteCV.notify_all();
|
|
}
|
|
|
|
} // namespace orc_rt
|