fabian/act-cluster-lib
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

implement get_task_status

Browse source
This commit is contained in:
Fabian Posch 2024-01-17 13:22:44 -05:00
parent e5f54e4a29
commit 11714d09db
1 changed files with 22 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/db_client.cpp
View file

@ -211,9 +211,28 @@ JobStatusType Connection::get_job_status(std::string job) {
}
JobStatusType Connection::get_task_status(db::uuid_t) {
std::cout << "JOB STATUS called, implement me pls" << std::endl;
return JobStatusType::IN_PROGRESS;
JobStatusType Connection::get_task_status(db::uuid_t task) {
auto get_task_status_lambda = [](pqxx::work *txn, db::uuid_t *task, db::JobStatusType *status){
auto res = txn->exec_params("SELECT part_status FROM artifacts WHERE id = $1 LIMIT 1;", *task);
if (res.size() < 1) {
*status = db::JobStatusType::UNKNOWN;
} else {
*status = res[0]["part_status"].as<db::JobStatusType>();
}
};
std::function<void(pqxx::work*, db::uuid_t*, db::JobStatusType*)> get_task_status_func = get_task_status_lambda;
db::JobStatusType status;
if (!this->send_request(&get_task_status_func, &task, &status)) {
std::cerr << "Error: Status for task " << db::to_string(task) << " could not be fetched, returning unknown." << std::endl;
return db::JobStatusType::UNKNOWN;
}
return status;
}
}