26195612357

Committed 20 May 2026 11:19PM UTC coverage: 49.859% (-2.3%) from 52.2%

Build # 26195612357

Build Type

push

github

Committed by

hariharan-devarajan

Commit Message

feat(aggregator): improve system metrics scanning and persistence error handling

Coverage Stats

16041 of 43831 branches covered (36.6%)

Branch coverage included in aggregate %.

6 of 17 new or added lines in 2 files covered. (35.29%)

1072 existing lines in 104 files now uncovered.

21423 of 31309 relevant lines covered (68.42%)

13054.31 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

51.81

/src/dftracer/utils/core/tasks/task_result.cpp

#include <dftracer/utils/core/tasks/task_result.h>

#include <cassert>

namespace dftracer::utils {

void TaskResult::set_value(std::any value) {
    value_ = std::move(value);
    publish(State::value);
}

void TaskResult::set_exception(std::exception_ptr ex) {
    exception_ = ex;
    publish(State::exception);
}

void TaskResult::set_cancelled() { publish(State::cancelled); }

void TaskResult::mark_running() {
    auto expected = static_cast<std::uint8_t>(State::pending);
    state_.compare_exchange_strong(
        expected, static_cast<std::uint8_t>(State::running),
        std::memory_order_release, std::memory_order_relaxed);
}

void TaskResult::add_reader() {
    pending_readers_.fetch_add(1, std::memory_order_relaxed);
}

void TaskResult::release_reader() {
    int prev = pending_readers_.fetch_sub(1, std::memory_order_acq_rel);
    if (prev == 1) {
        // Last reader released -- free value memory
        std::lock_guard<std::mutex> lock(mutex_);
        value_ = std::any{};
    }
}

void TaskResult::publish(State s) {
    std::vector<std::coroutine_handle<>> to_resume;
    {
        std::lock_guard<std::mutex> lock(mutex_);
        state_.store(static_cast<std::uint8_t>(s), std::memory_order_release);
        to_resume.swap(continuations_);
    }
    cv_.notify_all();
    // Resume coroutine continuations OUTSIDE the lock.
    for (auto h : to_resume) {
        if (h && !h.done()) {
            h.resume();
        }
    }
}

bool TaskResult::wait(std::chrono::milliseconds timeout) const {
    auto ready = [this] {
        auto s = static_cast<State>(state_.load(std::memory_order_acquire));
        return s == State::value || s == State::exception ||
               s == State::cancelled;
    };
    if (ready()) return true;

    std::unique_lock<std::mutex> lock(mutex_);
    if (timeout.count() == 0) {
        cv_.wait(lock, ready);
        return true;
    }
    return cv_.wait_for(lock, timeout, ready);
}

std::any TaskResult::get() const {
    wait();
    std::lock_guard<std::mutex> lock(mutex_);
    auto s = static_cast<State>(state_.load(std::memory_order_acquire));
    if (s == State::exception) {
        std::rethrow_exception(exception_);
    }
    if (s == State::cancelled) {
        throw std::runtime_error("Task was cancelled");
    }
    return value_;  // returns COPY
}

std::any TaskResult::get_ready() const {
    std::lock_guard<std::mutex> lock(mutex_);
    auto s = static_cast<State>(state_.load(std::memory_order_acquire));
    assert(s == State::value || s == State::exception || s == State::cancelled);
    if (s == State::exception) {
        std::rethrow_exception(exception_);
    }
    if (s == State::cancelled) {
        throw std::runtime_error("Task was cancelled");
    }
    return value_;  // returns COPY
}

std::exception_ptr TaskResult::get_exception() const { return exception_; }

bool TaskResult::is_ready() const {
    auto s = static_cast<State>(state_.load(std::memory_order_acquire));
    return s == State::value || s == State::exception || s == State::cancelled;
}

bool TaskResult::has_exception() const {
    return static_cast<State>(state_.load(std::memory_order_acquire)) ==
           State::exception;
}

bool TaskResult::is_cancelled() const {
    return static_cast<State>(state_.load(std::memory_order_acquire)) ==
           State::cancelled;
}

TaskResult::State TaskResult::state() const {
    return static_cast<State>(state_.load(std::memory_order_acquire));
}

// WhenReadyAwaitable
bool TaskResult::WhenReadyAwaitable::await_ready() const noexcept {
    return result.is_ready();
}

bool TaskResult::WhenReadyAwaitable::await_suspend(std::coroutine_handle<> h) {
    std::lock_guard<std::mutex> lock(result.mutex_);
    // Double-check under lock -- result may have been published
    // between await_ready() and await_suspend().
    if (result.is_ready()) {
        return false;  // Don't suspend, resume immediately
    }
    result.continuations_.push_back(h);
    return true;       // Suspend
}

void TaskResult::WhenReadyAwaitable::await_resume() {
    if (result.has_exception()) {
        std::rethrow_exception(result.exception_);
    }
}

}  // namespace dftracer::utils

1	#include <dftracer/utils/core/tasks/task_result.h>
2
3	#include <cassert>
4
5	namespace dftracer::utils {
6
7	void TaskResult::set_value(std::any value) {	745✔
8	value_ = std::move(value);	745✔
9	publish(State::value);	749✔
10	}	745✔
11
12	void TaskResult::set_exception(std::exception_ptr ex) {	17✔
13	exception_ = ex;	17✔
14	publish(State::exception);	17✔
15	}	17✔
16
17	void TaskResult::set_cancelled() { publish(State::cancelled); }	×
18
19	void TaskResult::mark_running() {	771✔
20	auto expected = static_cast<std::uint8_t>(State::pending);	771✔
21	state_.compare_exchange_strong(	771✔
22	expected, static_cast<std::uint8_t>(State::running),
23	std::memory_order_release, std::memory_order_relaxed);
24	}	771✔
25
26	void TaskResult::add_reader() {	561✔
27	pending_readers_.fetch_add(1, std::memory_order_relaxed);	561✔
28	}	561✔
29
30	void TaskResult::release_reader() {	526✔
31	int prev = pending_readers_.fetch_sub(1, std::memory_order_acq_rel);	526✔
32	if (prev == 1) {	526✔
33	// Last reader released -- free value memory
34	std::lock_guard<std::mutex> lock(mutex_);	326!
35	value_ = std::any{};	326✔
36	}	326✔
37	}	526✔
38
39	void TaskResult::publish(State s) {	771✔
40	std::vector<std::coroutine_handle<>> to_resume;	771✔
41	{
42	std::lock_guard<std::mutex> lock(mutex_);	762!
43	state_.store(static_cast<std::uint8_t>(s), std::memory_order_release);	764✔
44	to_resume.swap(continuations_);	764✔
45	}	766✔
46	cv_.notify_all();	765✔
47	// Resume coroutine continuations OUTSIDE the lock.
48	for (auto h : to_resume) {	771!
UNCOV 49	if (h && !h.done()) {	×
50	h.resume();	×
51	}
52	}
53	}	768✔
54
55	bool TaskResult::wait(std::chrono::milliseconds timeout) const {	426✔
56	auto ready = [this] {	426✔
57	auto s = static_cast<State>(state_.load(std::memory_order_acquire));	426✔
58	return s == State::value \|\| s == State::exception \|\|	426!
59	s == State::cancelled;	426✔
60	};	426✔
61	if (ready()) return true;	426!
62
63	std::unique_lock<std::mutex> lock(mutex_);	×
64	if (timeout.count() == 0) {	×
65	cv_.wait(lock, ready);	×
66	return true;	×
67	}
68	return cv_.wait_for(lock, timeout, ready);	×
UNCOV 69	}	×
70
71	std::any TaskResult::get() const {	337✔
72	wait();	337!
73	std::lock_guard<std::mutex> lock(mutex_);	337!
74	auto s = static_cast<State>(state_.load(std::memory_order_acquire));	337✔
75	if (s == State::exception) {	337✔
76	std::rethrow_exception(exception_);	6✔
77	}
78	if (s == State::cancelled) {	334!
79	throw std::runtime_error("Task was cancelled");	×
80	}
81	return value_; // returns COPY	668!
82	}	337✔
83
84	std::any TaskResult::get_ready() const {	527✔
85	std::lock_guard<std::mutex> lock(mutex_);	527!
86	auto s = static_cast<State>(state_.load(std::memory_order_acquire));	527✔
87	assert(s == State::value \|\| s == State::exception \|\| s == State::cancelled);	527!
88	if (s == State::exception) {	527!
89	std::rethrow_exception(exception_);	×
90	}
91	if (s == State::cancelled) {	527!
92	throw std::runtime_error("Task was cancelled");	×
93	}
94	return value_; // returns COPY	1,054!
95	}	526✔
96
97	std::exception_ptr TaskResult::get_exception() const { return exception_; }	5✔
98
99	bool TaskResult::is_ready() const {	7✔
100	auto s = static_cast<State>(state_.load(std::memory_order_acquire));	7✔
101	return s == State::value \|\| s == State::exception \|\| s == State::cancelled;	7!
102	}
103
104	bool TaskResult::has_exception() const {	781✔
105	return static_cast<State>(state_.load(std::memory_order_acquire)) ==	781✔
106	State::exception;	781✔
107	}
108
109	bool TaskResult::is_cancelled() const {	×
110	return static_cast<State>(state_.load(std::memory_order_acquire)) ==	×
UNCOV 111	State::cancelled;	×
112	}
113
114	TaskResult::State TaskResult::state() const {	×
115	return static_cast<State>(state_.load(std::memory_order_acquire));	×
116	}
117
118	// WhenReadyAwaitable
119	bool TaskResult::WhenReadyAwaitable::await_ready() const noexcept {	×
120	return result.is_ready();	×
121	}
122
123	bool TaskResult::WhenReadyAwaitable::await_suspend(std::coroutine_handle<> h) {	×
124	std::lock_guard<std::mutex> lock(result.mutex_);	×
125	// Double-check under lock -- result may have been published
126	// between await_ready() and await_suspend().
127	if (result.is_ready()) {	×
128	return false; // Don't suspend, resume immediately	×
129	}
130	result.continuations_.push_back(h);	×
131	return true; // Suspend	×
132	}	×
133
134	void TaskResult::WhenReadyAwaitable::await_resume() {	×
135	if (result.has_exception()) {	×
136	std::rethrow_exception(result.exception_);	×
137	}
138	}	×
139
140	} // namespace dftracer::utils

llnl / dftracer-utils / 26195612357

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous