Use get_sticker_set_names in process_new_business_callback_query_queue.

main
levlam 2 months ago
parent b29a9952af
commit cef85b3a1f

@ -7968,7 +7968,7 @@ void Client::on_get_sticker_set(int64 set_id, int64 new_callback_query_user_id,
process_new_business_message_queue(new_message_business_connection_id); process_new_business_message_queue(new_message_business_connection_id);
} }
if (new_business_callback_query_user_id != 0) { if (new_business_callback_query_user_id != 0) {
process_new_business_callback_query_queue(new_business_callback_query_user_id); process_new_business_callback_query_queue(new_business_callback_query_user_id, 1);
} }
} }
@ -16708,17 +16708,24 @@ void Client::add_new_business_callback_query(object_ptr<td_api::updateNewBusines
CHECK(query != nullptr); CHECK(query != nullptr);
auto user_id = query->sender_user_id_; auto user_id = query->sender_user_id_;
CHECK(user_id != 0); CHECK(user_id != 0);
new_business_callback_query_queues_[user_id].queue_.push(std::move(query)); NewBusinessCallbackQuery new_query;
process_new_business_callback_query_queue(user_id); new_query.message_info_ = create_business_message(query->connection_id_, std::move(query->message_));
new_query.update_ = std::move(query);
new_business_callback_query_queues_[user_id].queue_.push(std::move(new_query));
process_new_business_callback_query_queue(user_id, 0);
} }
void Client::process_new_business_callback_query_queue(int64 user_id) { void Client::process_new_business_callback_query_queue(int64 user_id, int state) {
auto &queue = new_business_callback_query_queues_[user_id]; auto &queue = new_business_callback_query_queues_[user_id];
if (queue.has_active_request_) { if (queue.has_active_request_) {
CHECK(!queue.queue_.empty()); CHECK(!queue.queue_.empty());
LOG(INFO) << "Have an active request in business callback query queue of size " << queue.queue_.size() if (state != 0) {
<< " for user " << user_id; queue.has_active_request_ = false;
return; } else {
LOG(INFO) << "Have an active request in business callback query queue of size " << queue.queue_.size()
<< " for user " << user_id;
return;
}
} }
if (logging_out_ || closing_) { if (logging_out_ || closing_) {
LOG(INFO) << "Ignore business callback query while closing for user " << user_id; LOG(INFO) << "Ignore business callback query while closing for user " << user_id;
@ -16727,36 +16734,30 @@ void Client::process_new_business_callback_query_queue(int64 user_id) {
} }
while (!queue.queue_.empty()) { while (!queue.queue_.empty()) {
auto &query = queue.queue_.front(); auto &query = queue.queue_.front();
auto &message_ref = query->message_;
LOG(INFO) << "Process business callback query from user " << user_id; LOG(INFO) << "Process business callback query from user " << user_id;
drop_internal_reply_to_message_in_another_chat(message_ref->message_); if (state == 0) {
auto sticker_set_ids = get_message_sticker_set_ids(query.message_info_.get());
auto message_sticker_set_id = get_sticker_set_id(message_ref->message_->content_); if (!sticker_set_ids.empty()) {
if (!have_sticker_set_name(message_sticker_set_id)) {
queue.has_active_request_ = true;
return send_request(
make_object<td_api::getStickerSetName>(message_sticker_set_id),
td::make_unique<TdOnGetStickerSetCallback>(this, message_sticker_set_id, 0, 0, td::string(), user_id));
}
if (message_ref->reply_to_message_ != nullptr) {
drop_internal_reply_to_message_in_another_chat(message_ref->reply_to_message_);
auto reply_sticker_set_id = get_sticker_set_id(message_ref->reply_to_message_->content_);
if (!have_sticker_set_name(reply_sticker_set_id)) {
queue.has_active_request_ = true; queue.has_active_request_ = true;
return send_request( return get_sticker_set_names(std::move(sticker_set_ids),
make_object<td_api::getStickerSetName>(reply_sticker_set_id), td::PromiseCreator::lambda([actor_id = actor_id(this), user_id](td::Unit) mutable {
td::make_unique<TdOnGetStickerSetCallback>(this, reply_sticker_set_id, 0, 0, td::string(), user_id)); send_closure(actor_id, &Client::process_new_business_callback_query_queue,
user_id, 1);
}));
} }
state = 1;
} }
CHECK(state == 1);
CHECK(user_id == query->sender_user_id_); auto update = std::move(query.update_);
auto message_info = create_business_message(query->connection_id_, std::move(message_ref)); CHECK(user_id == update->sender_user_id_);
add_update(UpdateType::CallbackQuery, add_update(UpdateType::CallbackQuery,
JsonCallbackQuery(query->id_, user_id, 0, 0, message_info.get(), query->chat_instance_, JsonCallbackQuery(update->id_, user_id, 0, 0, query.message_info_.get(), update->chat_instance_,
query->payload_.get(), this), update->payload_.get(), this),
150, user_id + (static_cast<int64>(3) << 33)); 150, user_id + (static_cast<int64>(3) << 33));
queue.queue_.pop(); queue.queue_.pop();
state = 0;
} }
new_business_callback_query_queues_.erase(user_id); new_business_callback_query_queues_.erase(user_id);
} }
@ -17399,6 +17400,9 @@ td::vector<td::int64> Client::get_message_sticker_set_ids(const MessageInfo *mes
sticker_set_ids.push_back(reply_sticker_set_id); sticker_set_ids.push_back(reply_sticker_set_id);
} }
} }
if (message_info->business_reply_to_message != nullptr) {
td::combine(sticker_set_ids, get_message_sticker_set_ids(message_info->business_reply_to_message.get()));
}
return sticker_set_ids; return sticker_set_ids;
} }

@ -1311,7 +1311,7 @@ class Client final : public WebhookActor::Callback {
void add_new_business_callback_query(object_ptr<td_api::updateNewBusinessCallbackQuery> &&query); void add_new_business_callback_query(object_ptr<td_api::updateNewBusinessCallbackQuery> &&query);
void process_new_business_callback_query_queue(int64 user_id); void process_new_business_callback_query_queue(int64 user_id, int state);
void add_new_inline_callback_query(object_ptr<td_api::updateNewInlineCallbackQuery> &&query); void add_new_inline_callback_query(object_ptr<td_api::updateNewInlineCallbackQuery> &&query);
@ -1501,8 +1501,12 @@ class Client final : public WebhookActor::Callback {
}; };
td::FlatHashMap<int64, NewCallbackQueryQueue> new_callback_query_queues_; // sender_user_id -> queue td::FlatHashMap<int64, NewCallbackQueryQueue> new_callback_query_queues_; // sender_user_id -> queue
struct NewBusinessCallbackQuery {
object_ptr<td_api::updateNewBusinessCallbackQuery> update_;
td::unique_ptr<MessageInfo> message_info_;
};
struct NewBusinessCallbackQueryQueue { struct NewBusinessCallbackQueryQueue {
std::queue<object_ptr<td_api::updateNewBusinessCallbackQuery>> queue_; std::queue<NewBusinessCallbackQuery> queue_;
bool has_active_request_ = false; bool has_active_request_ = false;
}; };
td::FlatHashMap<int64, NewBusinessCallbackQueryQueue> new_business_callback_query_queues_; // sender_user_id -> queue td::FlatHashMap<int64, NewBusinessCallbackQueryQueue> new_business_callback_query_queues_; // sender_user_id -> queue

Loading…
Cancel
Save