Directory: | cvmfs/ |
---|---|
File: | cvmfs/bigqueue.h |
Date: | 2025-06-22 02:36:02 |
Exec | Total | Coverage | |
---|---|---|---|
Lines: | 76 | 77 | 98.7% |
Branches: | 23 | 32 | 71.9% |
Line | Branch | Exec | Source |
---|---|---|---|
1 | /** | ||
2 | * This file is part of the CernVM File System. | ||
3 | * | ||
4 | * Similar to bigvector, but queue semantics. Used by the negative entry | ||
5 | * tracker. Allocates with mmap in order to avoid memory fragmentation. | ||
6 | */ | ||
7 | |||
8 | #ifndef CVMFS_BIGQUEUE_H_ | ||
9 | #define CVMFS_BIGQUEUE_H_ | ||
10 | |||
11 | #include <algorithm> | ||
12 | #include <cassert> | ||
13 | #include <cstdlib> | ||
14 | #include <new> | ||
15 | |||
16 | #include "util/smalloc.h" | ||
17 | |||
18 | template<class Item> | ||
19 | class BigQueue { | ||
20 | public: | ||
21 | 977 | BigQueue() { | |
22 | 977 | Alloc(kNumInit); | |
23 | 977 | size_ = 0; | |
24 | 977 | } | |
25 | |||
26 | explicit BigQueue(const size_t num_items) { | ||
27 | const size_t min_items = kNumInit; | ||
28 | Alloc(std::max(num_items, min_items)); | ||
29 | size_ = 0; | ||
30 | } | ||
31 | |||
32 | 14 | BigQueue(const BigQueue<Item> &other) { CopyFrom(other); } | |
33 | |||
34 | 56 | BigQueue<Item> &operator=(const BigQueue<Item> &other) { | |
35 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
|
56 | if (&other == this) |
36 | ✗ | return *this; | |
37 | |||
38 | 56 | Dealloc(); | |
39 | 56 | CopyFrom(other); | |
40 | 56 | return *this; | |
41 | } | ||
42 | |||
43 | 977 | ~BigQueue() { Dealloc(); } | |
44 | |||
45 | 280056728 | void PushBack(const Item &item) { | |
46 |
2/2✓ Branch 1 taken 812 times.
✓ Branch 2 taken 280055916 times.
|
280056728 | if (GetAvailableSpace() == 0) { |
47 | 812 | Migrate(1.9 * static_cast<float>(capacity_)); | |
48 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 812 times.
|
812 | assert(GetAvailableSpace() > 0); |
49 | } | ||
50 |
1/2✓ Branch 2 taken 28728 times.
✗ Branch 3 not taken.
|
280056728 | new (head_ + size_) Item(item); |
51 | 280056728 | size_++; | |
52 | 280056728 | } | |
53 | |||
54 | 280021042 | void PopFront() { | |
55 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 280021042 times.
|
280021042 | assert(!IsEmpty()); |
56 | 280021042 | head_++; | |
57 | 280021042 | size_--; | |
58 |
4/4✓ Branch 0 taken 280018270 times.
✓ Branch 1 taken 2772 times.
✓ Branch 2 taken 714 times.
✓ Branch 3 taken 280017556 times.
|
280021042 | if ((size_ > kCompactThreshold) && (size_ < (capacity_ / 2))) |
59 | 714 | Migrate(static_cast<int>(static_cast<float>(capacity_ * 0.6))); | |
60 | 280021042 | } | |
61 | |||
62 | 280050064 | bool Peek(Item **item) { | |
63 |
2/2✓ Branch 1 taken 196 times.
✓ Branch 2 taken 280049868 times.
|
280050064 | if (IsEmpty()) |
64 | 196 | return false; | |
65 | 280049868 | *item = head_; | |
66 | 280049868 | return true; | |
67 | } | ||
68 | |||
69 | 560071106 | bool IsEmpty() const { return size_ == 0; } | |
70 | |||
71 | 56 | void Clear() { | |
72 | 56 | Dealloc(); | |
73 | 56 | Alloc(kNumInit); | |
74 | 56 | } | |
75 | |||
76 | 18382 | size_t size() const { return size_; } | |
77 | 140 | size_t capacity() const { return capacity_; } | |
78 | |||
79 | private: | ||
80 | static const size_t kNumInit = 64; | ||
81 | static const size_t kCompactThreshold = 64; | ||
82 | |||
83 | 350088841 | size_t GetHeadOffset() const { return head_ - buffer_; } | |
84 | 280057540 | size_t GetAvailableSpace() const { | |
85 | 280057540 | return capacity_ - (size_ + GetHeadOffset()); | |
86 | } | ||
87 | |||
88 | 2629 | void Alloc(const size_t num_elements) { | |
89 | 2629 | size_t num_bytes = sizeof(Item) * num_elements; | |
90 | 2629 | buffer_ = static_cast<Item *>(smmap(num_bytes)); | |
91 | 2629 | capacity_ = num_elements; | |
92 | 2629 | head_ = buffer_; | |
93 | 2629 | } | |
94 | |||
95 | 1089 | void Dealloc() { | |
96 | 1089 | FreeBuffer(buffer_, GetHeadOffset() + size_); | |
97 | 1089 | buffer_ = NULL; | |
98 | 1089 | head_ = NULL; | |
99 | 1089 | capacity_ = 0; | |
100 | 1089 | size_ = 0; | |
101 | 1089 | } | |
102 | |||
103 | 1526 | void Migrate(size_t new_capacity) { | |
104 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1526 times.
|
1526 | assert(new_capacity > 0); |
105 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1526 times.
|
1526 | assert(new_capacity >= size_); |
106 | |||
107 | 1526 | size_t head_offset = GetHeadOffset(); | |
108 | 1526 | Item *old_buffer = buffer_; | |
109 | |||
110 | 1526 | Alloc(new_capacity); | |
111 |
2/2✓ Branch 0 taken 784572642 times.
✓ Branch 1 taken 1526 times.
|
784574168 | for (size_t i = 0; i < size_; ++i) |
112 |
1/2✓ Branch 2 taken 46900 times.
✗ Branch 3 not taken.
|
784572642 | new (buffer_ + i) Item(old_buffer[head_offset + i]); |
113 | |||
114 | 1526 | FreeBuffer(old_buffer, head_offset + size_); | |
115 | 1526 | } | |
116 | |||
117 | 2615 | void FreeBuffer(Item *buf, const size_t nitems) { | |
118 |
2/2✓ Branch 0 taken 1134658042 times.
✓ Branch 1 taken 2615 times.
|
1134660657 | for (size_t i = 0; i < nitems; ++i) |
119 | 1134658042 | buf[i].~Item(); | |
120 | |||
121 |
1/2✓ Branch 0 taken 2615 times.
✗ Branch 1 not taken.
|
2615 | if (buf) |
122 | 2615 | smunmap(buf); | |
123 | 2615 | } | |
124 | |||
125 | 70 | void CopyFrom(const BigQueue<Item> &other) { | |
126 | 70 | size_t min_items = kNumInit; | |
127 | 70 | Alloc(std::max(other.size_, min_items)); | |
128 |
2/2✓ Branch 0 taken 70028686 times.
✓ Branch 1 taken 70 times.
|
70028756 | for (size_t i = 0; i < other.size_; ++i) { |
129 |
1/2✓ Branch 3 taken 28686 times.
✗ Branch 4 not taken.
|
70028686 | new (buffer_ + i) Item(*(other.buffer_ + other.GetHeadOffset() + i)); |
130 | } | ||
131 | 70 | size_ = other.size_; | |
132 | 70 | } | |
133 | |||
134 | Item *buffer_; | ||
135 | Item *head_; | ||
136 | size_t size_; | ||
137 | size_t capacity_; | ||
138 | }; | ||
139 | |||
140 | #endif // CVMFS_BIGQUEUE_H_ | ||
141 |