recovery_worker.cpp

Go to the documentation of this file.

#include "recovery_worker.hpp"
#include "recovery_exporter.hpp"
 
RecoveryWorker::RecoveryWorker() = default;
 
RecoveryWorker::~RecoveryWorker()
{
    request_cancel();
    join();
}
 
void RecoveryWorker::start(const ScanReport& report,
                           const std::string& output_dir,
                           const std::vector<size_t>& selected_result_indices,
                           CompletionCallback cb)
{
    if (running_)
        return;
 
    cancel_requested_ = false;
    running_ = true;
 
    thread_ = std::thread([this, report, output_dir, selected_result_indices, cb]()
    {
        RecoveryResult rr;
 
        try
        {
            rr.ok = RecoveryExporter::export_results_and_carve(
                report,
                output_dir,
                selected_result_indices,
                rr.summary,
                rr.logs,
                rr.error,
                [this]() { return cancel_requested_.load(); }
            );
        }
        catch (const std::exception& e)
        {
            rr.ok = false;
            rr.error = std::string("Recovery exception: ") + e.what();
        }
        catch (...)
        {
            rr.ok = false;
            rr.error = "Unknown recovery exception";
        }
 
        running_ = false;
 
        if (!cancel_requested_ && cb)
            cb(rr);
    });
}
 
bool RecoveryWorker::running() const
{
    return running_;
}
 
void RecoveryWorker::join()
{
    if (thread_.joinable())
        thread_.join();
}
 
void RecoveryWorker::request_cancel()
{
    cancel_requested_ = true;
}