Add insert_message function.

GitOrigin-RevId: c78957f5489e5efcc906c93b9ad9b9b5e2644240
This commit is contained in:
levlam 2018-11-22 21:27:43 +03:00
parent 9118cbc71c
commit 9841b28693
2 changed files with 36 additions and 34 deletions

View File

@ -19743,6 +19743,39 @@ const unique_ptr<MessagesManager::Message> *MessagesManager::find_message(const
return v; return v;
} }
MessagesManager::Message *MessagesManager::insert_message(unique_ptr<Message> *v, unique_ptr<Message> message) {
auto message_id = message->message_id;
while (*v != nullptr && (*v)->random_y >= message->random_y) {
if ((*v)->message_id.get() < message_id.get()) {
v = &(*v)->right;
} else if ((*v)->message_id == message_id) {
UNREACHABLE();
} else {
v = &(*v)->left;
}
}
unique_ptr<Message> *left = &message->left;
unique_ptr<Message> *right = &message->right;
unique_ptr<Message> cur = std::move(*v);
while (cur != nullptr) {
if (cur->message_id.get() < message_id.get()) {
*left = std::move(cur);
left = &((*left)->right);
cur = std::move(*left);
} else {
*right = std::move(cur);
right = &((*right)->left);
cur = std::move(*right);
}
}
CHECK(*left == nullptr);
CHECK(*right == nullptr);
*v = std::move(message);
return v->get();
}
MessagesManager::Message *MessagesManager::get_message(Dialog *d, MessageId message_id) { MessagesManager::Message *MessagesManager::get_message(Dialog *d, MessageId message_id) {
return const_cast<Message *>(get_message(static_cast<const Dialog *>(d), message_id)); return const_cast<Message *>(get_message(static_cast<const Dialog *>(d), message_id));
} }
@ -20408,40 +20441,7 @@ MessagesManager::Message *MessagesManager::add_message_to_dialog(Dialog *d, uniq
} }
} }
Message *result_message = nullptr; Message *result_message = insert_message(&d->messages, std::move(message));
{
// TODO function
auto v = &d->messages;
while (*v != nullptr && (*v)->random_y >= m->random_y) {
if ((*v)->message_id.get() < message_id.get()) {
v = &(*v)->right;
} else if ((*v)->message_id == message_id) {
UNREACHABLE();
} else {
v = &(*v)->left;
}
}
unique_ptr<Message> *left = &message->left;
unique_ptr<Message> *right = &message->right;
unique_ptr<Message> cur = std::move(*v);
while (cur != nullptr) {
if (cur->message_id.get() < message_id.get()) {
*left = std::move(cur);
left = &((*left)->right);
cur = std::move(*left);
} else {
*right = std::move(cur);
right = &((*right)->left);
cur = std::move(*right);
}
}
CHECK(*left == nullptr);
CHECK(*right == nullptr);
*v = std::move(message);
result_message = v->get();
}
CHECK(result_message != nullptr); CHECK(result_message != nullptr);
CHECK(d->messages != nullptr); CHECK(d->messages != nullptr);

View File

@ -1625,6 +1625,8 @@ class MessagesManager : public Actor {
static unique_ptr<Message> *find_message(unique_ptr<Message> *v, MessageId message_id); static unique_ptr<Message> *find_message(unique_ptr<Message> *v, MessageId message_id);
static const unique_ptr<Message> *find_message(const unique_ptr<Message> *v, MessageId message_id); static const unique_ptr<Message> *find_message(const unique_ptr<Message> *v, MessageId message_id);
static Message *insert_message(unique_ptr<Message> *v, unique_ptr<Message> message);
static Message *get_message(Dialog *d, MessageId message_id); static Message *get_message(Dialog *d, MessageId message_id);
static const Message *get_message(const Dialog *d, MessageId message_id); static const Message *get_message(const Dialog *d, MessageId message_id);