Lubuntu/ci-tools
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use futures for queuing pull and tarball tasks

Browse Source
This commit is contained in:
Simon Quigley 2025-01-27 00:10:50 -06:00
parent 406fddb27b
commit 8af3fe19b4
1 changed files with 51 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
cpp/ci_logic.cpp
View File

@ -639,60 +639,66 @@ std::string CiLogic::queue_pull_tarball(std::vector<std::shared_ptr<PackageConf>
std::mutex task_assignment_mutex; std::mutex task_assignment_mutex;
try { try {
std::vector<std::future<void>> futures;
for (auto &r : repos) { for (auto &r : repos) {
{ futures.push_back(std::async(std::launch::async, [this, &task_queue, &job_statuses,
std::lock_guard<std::mutex> lock(task_assignment_mutex); &encountered_items, &task_assignment_mutex,
auto found_it = encountered_items.find(r->package->name); &r]() mutable {
if (found_it != encountered_items.end()) { {
// GHOST pull std::lock_guard<std::mutex> lock(task_assignment_mutex);
auto existing_item = found_it->second; auto found_it = encountered_items.find(r->package->name);
if (found_it != encountered_items.end()) {
// GHOST pull
auto existing_item = found_it->second;
// Assign tasks (reuse the same Task objects for "pull"/"tarball") // Assign tasks (reuse the same Task objects for "pull"/"tarball")
r->assign_task(job_statuses->at("pull"), existing_item->first_pull_task, r); r->assign_task(job_statuses->at("pull"), existing_item->first_pull_task, r);
r->assign_task(job_statuses->at("tarball"), existing_item->first_tarball_task, r); r->assign_task(job_statuses->at("tarball"), existing_item->first_tarball_task, r);
// Point packaging_commit/upstream_commit to the real pkgconf's pointers // Point packaging_commit/upstream_commit to the real pkgconf's pointers
r->packaging_commit = existing_item->first_pkgconf->packaging_commit; r->packaging_commit = existing_item->first_pkgconf->packaging_commit;
r->upstream_commit = existing_item->first_pkgconf->upstream_commit; r->upstream_commit = existing_item->first_pkgconf->upstream_commit;
r->sync(); r->sync();
continue; return;
}
} }
} // REAL pull
// REAL pull auto new_item = std::make_shared<package_conf_item>();
auto new_item = std::make_shared<package_conf_item>(); new_item->first_pkgconf = r;
new_item->first_pkgconf = r;
// Enqueue "pull" // Enqueue "pull"
task_queue->enqueue( task_queue->enqueue(
job_statuses->at("pull"), job_statuses->at("pull"),
[this, r](std::shared_ptr<Log> log) mutable { [this, r](std::shared_ptr<Log> log) mutable {
pull_project(r, log); pull_project(r, log);
r->sync(); r->sync();
}, },
r r
); );
{ {
std::lock_guard<std::mutex> lock(task_assignment_mutex); std::lock_guard<std::mutex> lock(task_assignment_mutex);
new_item->first_pull_task = r->get_task_by_jobstatus(job_statuses->at("pull")); new_item->first_pull_task = r->get_task_by_jobstatus(job_statuses->at("pull"));
} }
// Enqueue "tarball" // Enqueue "tarball"
task_queue->enqueue( task_queue->enqueue(
job_statuses->at("tarball"), job_statuses->at("tarball"),
[this, r](std::shared_ptr<Log> log) mutable { [this, r](std::shared_ptr<Log> log) mutable {
create_project_tarball(r, log); create_project_tarball(r, log);
r->sync(); r->sync();
}, },
r r
); );
{ {
std::lock_guard<std::mutex> lock(task_assignment_mutex); std::lock_guard<std::mutex> lock(task_assignment_mutex);
new_item->first_tarball_task = r->get_task_by_jobstatus(job_statuses->at("tarball")); new_item->first_tarball_task = r->get_task_by_jobstatus(job_statuses->at("tarball"));
encountered_items[r->package->name] = new_item; encountered_items[r->package->name] = new_item;
} }
}));
} }
for (auto& future : futures) future.get();
msg = "Succeeded"; msg = "Succeeded";
} catch (...) { } catch (...) {
msg = "Failed"; msg = "Failed";