This repository has been archived on 2020-05-25. You can view files and clone it, but cannot push or open issues or pull requests.
tdlib-fork/td/telegram/net/DcId.h
levlam c2a370262c Better DcId logging.
GitOrigin-RevId: b1fe45b3ecd06ca6869a60124078930b2129b972
2019-05-09 05:01:45 +03:00

110 lines
2.3 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2019
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once
#include "td/utils/common.h"
#include "td/utils/StringBuilder.h"
#include <tuple>
namespace td {
class DcId {
public:
DcId() = default;
static bool is_valid(int32 dc_id) {
return 1 <= dc_id && dc_id <= 1000;
}
static DcId main() {
return DcId{MainDc, false};
}
static DcId invalid() {
return DcId{Invalid, false};
}
static DcId internal(int32 id) {
CHECK(is_valid(id));
return DcId{id, false};
}
static DcId external(int32 id) {
CHECK(is_valid(id));
return DcId{id, true};
}
static DcId empty() {
return {};
}
static DcId from_value(int32 value) {
return DcId{value, false};
}
bool is_empty() const {
return !is_valid();
}
bool is_main() const {
return dc_id_ == MainDc;
}
int32 get_raw_id() const {
CHECK(is_exact());
return dc_id_;
}
int32 get_value() const {
return dc_id_;
}
bool is_internal() const {
return !is_external();
}
bool is_external() const {
return is_external_;
}
bool is_exact() const {
return dc_id_ > 0;
}
bool operator==(DcId other) const {
return dc_id_ == other.dc_id_ && is_external_ == other.is_external_;
}
bool operator<(DcId other) const {
return std::tie(dc_id_, is_external_) < std::tie(other.dc_id_, other.is_external_);
}
bool operator!=(DcId other) const {
return !(*this == other);
}
private:
enum : int32 { Empty = 0, MainDc = -1, Invalid = -2 };
int32 dc_id_{Empty};
bool is_external_{false};
DcId(int32 dc_id, bool is_external) : dc_id_(dc_id), is_external_(is_external) {
}
bool is_valid() const {
return is_exact() || is_main();
}
};
inline StringBuilder &operator<<(StringBuilder &sb, const DcId &dc_id) {
sb << "DcId{";
if (dc_id == DcId::invalid()) {
sb << "invalid";
} else if (dc_id == DcId()) {
sb << "empty";
} else if (dc_id.is_empty()) {
sb << "is_empty";
} else if (dc_id.is_main()) {
sb << "main";
} else {
sb << dc_id.get_raw_id();
if (dc_id.is_external()) {
sb << " external";
}
}
sb << "}";
return sb;
}
} // namespace td