HttpConnection: safe migrate between schedulers

GitOrigin-RevId: 2bcc79fc100613a86a8048a2c35e4791903ccb11

1 files changed, 8 insertions, 0 deletions

diff --git a/tdnet/td/net/HttpConnectionBase.cpp b/tdnet/td/net/HttpConnectionBase.cpp
index 3a7e5f261..652def782 100644
--- a/tdnet/td/net/HttpConnectionBase.cpp
+++ b/tdnet/td/net/HttpConnectionBase.cpp
@@ -190,5 +190,13 @@ void HttpConnectionBase::loop() {
   }
 }
 
+void HttpConnectionBase::on_start_migrate(int32 sched_id) {
+  Scheduler::unsubscribe(fd_.get_poll_info().get_pollable_fd_ref());
+}
+
+void HttpConnectionBase::on_finish_migrate() {
+  Scheduler::subscribe(fd_.get_poll_info().extract_pollable_fd(this));
+}
+
 }  // namespace detail
 }  // namespace td