andreacavalli/tdlib-fork
Archived
1
0
Fork 0
Code Pull Requests Releases Activity

Add support for scheduled message_id.

Browse Source 
GitOrigin-RevId: 1e56677062f8b59501fe64760ede980b9cb9bceb
This commit is contained in:
levlam 2019-11-26 18:52:59 +03:00
parent 727d0f1c6f
commit 4c5dbc4160
1 changed files with 54 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
td/telegram/MessageId.h
View File

@ -9,6 +9,7 @@
#include "td/telegram/DialogId.h" #include "td/telegram/DialogId.h"
#include "td/utils/common.h" #include "td/utils/common.h"
#include "td/utils/logging.h"
#include "td/utils/misc.h" #include "td/utils/misc.h"
#include "td/utils/StringBuilder.h" #include "td/utils/StringBuilder.h"
#include "td/utils/tl_helpers.h" #include "td/utils/tl_helpers.h"
@ -63,13 +64,23 @@ class MessageId {
int64 id = 0; int64 id = 0;
static constexpr int32 SERVER_ID_SHIFT = 20; static constexpr int32 SERVER_ID_SHIFT = 20;
static constexpr int32 SHORT_TYPE_MASK = (1 << 2) - 1;
static constexpr int32 TYPE_MASK = (1 << 3) - 1; static constexpr int32 TYPE_MASK = (1 << 3) - 1;
static constexpr int32 FULL_TYPE_MASK = (1 << SERVER_ID_SHIFT) - 1; static constexpr int32 FULL_TYPE_MASK = (1 << SERVER_ID_SHIFT) - 1;
static constexpr int32 SCHEDULED_MASK = 4;
static constexpr int32 TYPE_YET_UNSENT = 1; static constexpr int32 TYPE_YET_UNSENT = 1;
static constexpr int32 TYPE_LOCAL = 2; static constexpr int32 TYPE_LOCAL = 2;
friend StringBuilder &operator<<(StringBuilder &string_builder, MessageId message_id); friend StringBuilder &operator<<(StringBuilder &string_builder, MessageId message_id);
// ordinary message ID layout
// |-------31--------|---17---|1|--2-|
// |server_message_id|local_id|0|type|
// scheduled message ID layout
// |-------30-------|----18---|1|--2-|
// |send_date-2**30 |server_id|1|type|
public: public:
MessageId() = default; MessageId() = default;
@ -77,6 +88,22 @@ class MessageId {
: id(static_cast<int64>(server_message_id.get()) << SERVER_ID_SHIFT) { : id(static_cast<int64>(server_message_id.get()) << SERVER_ID_SHIFT) {
} }
static MessageId get_scheduled_message_id(int32 server_message_id, int32 send_date) {
if (send_date <= (1 << 30)) {
LOG(ERROR) << "Scheduled message send date " << send_date << " is in the past";
return MessageId();
}
if (server_message_id <= 0) {
LOG(ERROR) << "Scheduled message ID " << server_message_id << " is non-positive";
return MessageId();
}
if (server_message_id >= (1 << 18)) {
LOG(ERROR) << "Scheduled message ID " << server_message_id << " is too big";
return MessageId();
}
return MessageId((static_cast<int64>(send_date - (1 << 30)) << 21) | (server_message_id << 3) | SCHEDULED_MASK);
}
explicit constexpr MessageId(int64 message_id) : id(message_id) { explicit constexpr MessageId(int64 message_id) : id(message_id) {
} }
template <class T, typename = std::enable_if_t<std::is_convertible<T, int64>::value>> template <class T, typename = std::enable_if_t<std::is_convertible<T, int64>::value>>
@ -100,6 +127,15 @@ class MessageId {
return type == TYPE_YET_UNSENT || type == TYPE_LOCAL; return type == TYPE_YET_UNSENT || type == TYPE_LOCAL;
} }
bool is_valid_scheduled() const {
if (id <= 0 || id > max().get()) {
return false;
}
int32 type = (id & TYPE_MASK);
return type == SCHEDULED_MASK || type == (SCHEDULED_MASK | TYPE_YET_UNSENT) ||
type == (SCHEDULED_MASK | TYPE_LOCAL);
}
int64 get() const { int64 get() const {
return id; return id;
} }
@ -121,14 +157,18 @@ class MessageId {
} }
} }
bool is_scheduled() const {
return (id & SCHEDULED_MASK) != 0;
}
bool is_yet_unsent() const { bool is_yet_unsent() const {
CHECK(is_valid()); CHECK(is_valid() || is_scheduled());
return (id & TYPE_MASK) == TYPE_YET_UNSENT; return (id & SHORT_TYPE_MASK) == TYPE_YET_UNSENT;
} }
bool is_local() const { bool is_local() const {
CHECK(is_valid()); CHECK(is_valid() || is_scheduled());
return (id & TYPE_MASK) == TYPE_LOCAL; return (id & SHORT_TYPE_MASK) == TYPE_LOCAL;
} }
bool is_server() const { bool is_server() const {
@ -136,6 +176,11 @@ class MessageId {
return (id & FULL_TYPE_MASK) == 0; return (id & FULL_TYPE_MASK) == 0;
} }
bool is_scheduled_server() const {
CHECK(is_valid_scheduled());
return (id & SHORT_TYPE_MASK) == 0;
}
ServerMessageId get_server_message_id() const { ServerMessageId get_server_message_id() const {
CHECK(id == 0 || is_server()); CHECK(id == 0 || is_server());
return ServerMessageId(narrow_cast<int32>(id >> SERVER_ID_SHIFT)); return ServerMessageId(narrow_cast<int32>(id >> SERVER_ID_SHIFT));
@ -166,6 +211,11 @@ class MessageId {
} }
} }
int32 get_scheduled_server_message_id() const {
CHECK(is_scheduled_server());
return static_cast<int32>((id >> 3) & ((1 << 18) - 1));
}
bool operator==(const MessageId &other) const { bool operator==(const MessageId &other) const {
return id == other.id; return id == other.id;
} }