GCC Code Coverage Report

Directory: cvmfs/
File: cvmfs/ingestion/task.h
Date: 2025-11-16 02:35:16
Exec Total Coverage
Lines: 43 44 97.7%
Branches: 18 24 75.0%
Line Branch Exec Source
1 /**
2 * This file is part of the CernVM File System.
3 */
4
5 #ifndef CVMFS_INGESTION_TASK_H_
6 #define CVMFS_INGESTION_TASK_H_
7
8 #include <errno.h>
9 #include <pthread.h>
10 #include <unistd.h>
11
12 #include <cassert>
13 #include <vector>
14
15 #include "util/exception.h"
16 #include "util/single_copy.h"
17 #include "util/tube.h"
18
19 /**
20 * Forward declaration of TubeConsumerGroup so that it can be used as a friend
21 * class to TubeConsumer.
22 */
23 template<typename ItemT>
24 class TubeConsumerGroup;
25
26
27 /**
28 * Base class for threads that processes items from a tube one by one. Concrete
29 * implementations overwrite the Process() method.
30 */
31 template<class ItemT>
32 class TubeConsumer : SingleCopy {
33 friend class TubeConsumerGroup<ItemT>;
34
35 public:
36 502024 virtual ~TubeConsumer() { }
37
38 protected:
39 499003 explicit TubeConsumer(Tube<ItemT> *tube) : tube_(tube) { }
40 virtual void Process(ItemT *item) = 0;
41 452380 virtual void OnTerminate() { }
42
43 Tube<ItemT> *tube_;
44
45 private:
46 462907 static void *MainConsumer(void *data) {
47 462907 TubeConsumer<ItemT> *consumer = reinterpret_cast<TubeConsumer<ItemT> *>(
48 data);
49
50 256144103 while (true) {
51 256607010 ItemT *item = consumer->tube_->PopFront();
52
2/2
✓ Branch 1 taken 230965 times.
✓ Branch 2 taken 139635095 times.
 254006705 if (item->IsQuitBeacon()) {
53
2/2
✓ Branch 0 taken 230476 times.
✓ Branch 1 taken 489 times.
 458636 delete item;
54 458054 break;
55 }
56 252895339 consumer->Process(item);
57 }
58 458054 consumer->OnTerminate();
59 457220 return NULL;
60 }
61 };
62
63
64 template<class ItemT>
65 class TubeConsumerGroup : SingleCopy {
66 public:
67 25415 TubeConsumerGroup() : is_active_(false) { }
68
69 25402 ~TubeConsumerGroup() {
70
2/2
✓ Branch 1 taken 251012 times.
✓ Branch 2 taken 14306 times.
 524132 for (unsigned i = 0; i < consumers_.size(); ++i)
71
1/2
✓ Branch 1 taken 251012 times.
✗ Branch 2 not taken.
 498730 delete consumers_[i];
72 25402 }
73
74 499003 void TakeConsumer(TubeConsumer<ItemT> *consumer) {
75
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 251149 times.
 499003 assert(!is_active_);
76 499003 consumers_.push_back(consumer);
77 499003 }
78
79 24287 void Spawn() {
80
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13749 times.
 24287 assert(!is_active_);
81 24287 unsigned N = consumers_.size();
82 24287 threads_.resize(N);
83
2/2
✓ Branch 0 taken 233101 times.
✓ Branch 1 taken 13749 times.
 487194 for (unsigned i = 0; i < N; ++i) {
84 462907 int retval = pthread_create(
85 462907 &threads_[i], NULL, TubeConsumer<ItemT>::MainConsumer, consumers_[i]);
86
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 233101 times.
 462907 if (retval != 0) {
87 ✗ PANIC(kLogStderr, "failed to create new thread (error: %d, pid: %d)",
88 errno, getpid());
89 }
90 }
91 24287 is_active_ = true;
92 24287 }
93
94 24274 void Terminate() {
95
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13742 times.
 24274 assert(is_active_);
96 24274 unsigned N = consumers_.size();
97
2/2
✓ Branch 0 taken 232964 times.
✓ Branch 1 taken 13742 times.
 486908 for (unsigned i = 0; i < N; ++i) {
98 462634 consumers_[i]->tube_->EnqueueBack(ItemT::CreateQuitBeacon());
99 }
100
2/2
✓ Branch 0 taken 232964 times.
✓ Branch 1 taken 13742 times.
 486908 for (unsigned i = 0; i < N; ++i) {
101 462634 int retval = pthread_join(threads_[i], NULL);
102
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 232964 times.
 462634 assert(retval == 0);
103 }
104 24274 is_active_ = false;
105 24274 }
106
107 3210 bool is_active() { return is_active_; }
108
109 private:
110 bool is_active_;
111 std::vector<TubeConsumer<ItemT> *> consumers_;
112 std::vector<pthread_t> threads_;
113 };
114
115 #endif // CVMFS_INGESTION_TASK_H_
116