fix timeout (#1621)

Signed-off-by: turuslan <turuslan.devbox@gmail.com>
qdrvm · May 18, 2023 · c87ef1c · c87ef1c
1 parent aea722a
commit c87ef1c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/core/api/transport/impl/ws/ws_session.cpp b/core/api/transport/impl/ws/ws_session.cpp
@@ -285,6 +285,7 @@ namespace kagome::api {
         stream_.next_layer().buffer(),
         http_request_->get(),
         [self{shared_from_this()}](boost::system::error_code ec, size_t) {
+          self->stream_.next_layer().next_layer().expires_never();
           if (ec) {
             self->httpClose();
             return;
@@ -313,10 +314,12 @@ namespace kagome::api {
 
   void WsSession::httpWrite() {
     http_response_->prepare_payload();
+    stream_.next_layer().next_layer().expires_after(config_.operation_timeout);
     boost::beast::http::async_write(
         stream_.next_layer(),
         *http_response_,
         [self = shared_from_this()](boost::system::error_code ec, size_t) {
+          self->stream_.next_layer().next_layer().expires_never();
           if (ec) {
             self->httpClose();
             return;