aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/ipmath/range_set.cpp
diff options
context:
space:
mode:
authormsherbakov <msherbakov@yandex-team.ru>2022-02-10 16:49:17 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:49:17 +0300
commita0ffafe83b7d6229709a32fa942c71d672ac989c (patch)
tree5d5cb817648f650d76cf1076100726fd9b8448e8 /library/cpp/ipmath/range_set.cpp
parentc224a621661ddd69699f9476922eb316607ef57e (diff)
downloadydb-a0ffafe83b7d6229709a32fa942c71d672ac989c.tar.gz
Restoring authorship annotation for <msherbakov@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/ipmath/range_set.cpp')
-rw-r--r--library/cpp/ipmath/range_set.cpp194
1 files changed, 97 insertions, 97 deletions
diff --git a/library/cpp/ipmath/range_set.cpp b/library/cpp/ipmath/range_set.cpp
index 503ccaf03f..55f42e451d 100644
--- a/library/cpp/ipmath/range_set.cpp
+++ b/library/cpp/ipmath/range_set.cpp
@@ -1,99 +1,99 @@
-#include "range_set.h"
-
-#include <util/generic/algorithm.h>
-
-namespace {
- bool ShouldJoin(const TIpAddressRange& lhs, const TIpAddressRange& rhs) {
- return lhs.Overlaps(rhs) || lhs.IsConsecutive(rhs);
- }
-}
-
-bool TIpRangeSet::TRangeLess::operator()(const TIpAddressRange& lhs, const TIpAddressRange& rhs) const {
- return *lhs.Begin() < *rhs.Begin();
-}
-
-TIpRangeSet::TIpRangeSet() = default;
-TIpRangeSet::~TIpRangeSet() = default;
-
-void TIpRangeSet::Add(TIpAddressRange r) {
- Y_ENSURE(IsEmpty() || r.Type() == Type(), "Mixing IPv4 and IPv6 ranges is disallowed");
-
+#include "range_set.h"
+
+#include <util/generic/algorithm.h>
+
+namespace {
+ bool ShouldJoin(const TIpAddressRange& lhs, const TIpAddressRange& rhs) {
+ return lhs.Overlaps(rhs) || lhs.IsConsecutive(rhs);
+ }
+}
+
+bool TIpRangeSet::TRangeLess::operator()(const TIpAddressRange& lhs, const TIpAddressRange& rhs) const {
+ return *lhs.Begin() < *rhs.Begin();
+}
+
+TIpRangeSet::TIpRangeSet() = default;
+TIpRangeSet::~TIpRangeSet() = default;
+
+void TIpRangeSet::Add(TIpAddressRange r) {
+ Y_ENSURE(IsEmpty() || r.Type() == Type(), "Mixing IPv4 and IPv6 ranges is disallowed");
+
auto lowerIt = Ranges_.lower_bound(r);
-
- // still may overlap the last interval in our tree
- if (IsEmpty()) {
- Ranges_.insert(r);
- return;
- } else if (lowerIt == Ranges_.end()) {
- if (auto it = Ranges_.rbegin(); ShouldJoin(*it, r)) {
- auto unitedRange = it->Union(r);
- Ranges_.erase(--it.base());
- Ranges_.insert(unitedRange);
- } else {
- Ranges_.insert(r);
- }
-
- return;
- }
-
-
- TIpAddressRange unitedRange{r};
-
- auto joined = lowerIt;
- if (lowerIt != Ranges_.begin()) {
- if (ShouldJoin(unitedRange, *(--joined))) {
- unitedRange = unitedRange.Union(*joined);
- } else {
- ++joined;
- }
- }
-
- auto it = lowerIt;
- for (; it != Ranges_.end() && ShouldJoin(*it, unitedRange); ++it) {
- unitedRange = unitedRange.Union(*it);
- }
-
- Ranges_.erase(joined, it);
- Ranges_.insert(unitedRange);
-}
-
-TIpAddressRange::TIpType TIpRangeSet::Type() const {
- return IsEmpty()
- ? TIpAddressRange::TIpType::LAST
- : Ranges_.begin()->Type();
-}
-
-bool TIpRangeSet::IsEmpty() const {
- return Ranges_.empty();
-}
-
-TIpRangeSet::TIterator TIpRangeSet::Find(TIpv6Address addr) const {
- if (IsEmpty() || addr.Type() != Type()) {
- return End();
- }
-
+
+ // still may overlap the last interval in our tree
+ if (IsEmpty()) {
+ Ranges_.insert(r);
+ return;
+ } else if (lowerIt == Ranges_.end()) {
+ if (auto it = Ranges_.rbegin(); ShouldJoin(*it, r)) {
+ auto unitedRange = it->Union(r);
+ Ranges_.erase(--it.base());
+ Ranges_.insert(unitedRange);
+ } else {
+ Ranges_.insert(r);
+ }
+
+ return;
+ }
+
+
+ TIpAddressRange unitedRange{r};
+
+ auto joined = lowerIt;
+ if (lowerIt != Ranges_.begin()) {
+ if (ShouldJoin(unitedRange, *(--joined))) {
+ unitedRange = unitedRange.Union(*joined);
+ } else {
+ ++joined;
+ }
+ }
+
+ auto it = lowerIt;
+ for (; it != Ranges_.end() && ShouldJoin(*it, unitedRange); ++it) {
+ unitedRange = unitedRange.Union(*it);
+ }
+
+ Ranges_.erase(joined, it);
+ Ranges_.insert(unitedRange);
+}
+
+TIpAddressRange::TIpType TIpRangeSet::Type() const {
+ return IsEmpty()
+ ? TIpAddressRange::TIpType::LAST
+ : Ranges_.begin()->Type();
+}
+
+bool TIpRangeSet::IsEmpty() const {
+ return Ranges_.empty();
+}
+
+TIpRangeSet::TIterator TIpRangeSet::Find(TIpv6Address addr) const {
+ if (IsEmpty() || addr.Type() != Type()) {
+ return End();
+ }
+
auto lowerIt = Ranges_.lower_bound(TIpAddressRange(addr, addr));
-
- if (lowerIt == Ranges_.begin()) {
- return lowerIt->Contains(addr)
- ? lowerIt
- : End();
- } else if (lowerIt == Ranges_.end()) {
- auto rbegin = Ranges_.crbegin();
- return rbegin->Contains(addr)
- ? (++rbegin).base()
- : End();
- } else if (lowerIt->Contains(addr)) {
- return lowerIt;
- }
-
- --lowerIt;
-
- return lowerIt->Contains(addr)
- ? lowerIt
- : End();
-}
-
-bool TIpRangeSet::Contains(TIpv6Address addr) const {
- return Find(addr) != End();
-}
+
+ if (lowerIt == Ranges_.begin()) {
+ return lowerIt->Contains(addr)
+ ? lowerIt
+ : End();
+ } else if (lowerIt == Ranges_.end()) {
+ auto rbegin = Ranges_.crbegin();
+ return rbegin->Contains(addr)
+ ? (++rbegin).base()
+ : End();
+ } else if (lowerIt->Contains(addr)) {
+ return lowerIt;
+ }
+
+ --lowerIt;
+
+ return lowerIt->Contains(addr)
+ ? lowerIt
+ : End();
+}
+
+bool TIpRangeSet::Contains(TIpv6Address addr) const {
+ return Find(addr) != End();
+}