Merge pull request #228 from UiPath/fix-writes-on-shutdown

lifupan · web-flow · commit 152ac12c30f6 · 2024-05-11T09:56:48.000+08:00
send all pending responses before shutting down
diff --git a/src/asynchronous/server.rs b/src/asynchronous/server.rs
@@ -327,13 +327,14 @@ impl Builder for ServerBuilder {
                 server_shutdown: self.shutdown_waiter.clone(),
                 handler_shutdown: disconnect_notifier,
             },
-            ServerWriter { rx },
+            ServerWriter { rx, _server_shutdown: self.shutdown_waiter.clone() },
         )
     }
 }
 
 struct ServerWriter {
     rx: MessageReceiver,
+    _server_shutdown: shutdown::Waiter
 }
 
 #[async_trait]

Original file line number	Diff line number	Diff line change
`@@ -327,13 +327,14 @@ impl Builder for ServerBuilder {`
`327`	`327`	`server_shutdown: self.shutdown_waiter.clone(),`
`328`	`328`	`handler_shutdown: disconnect_notifier,`
`329`	`329`	`},`
`330`		`- ServerWriter { rx },`
	`330`	`+ ServerWriter { rx, _server_shutdown: self.shutdown_waiter.clone() },`
`331`	`331`	`)`
`332`	`332`	`}`
`333`	`333`	`}`
`334`	`334`
`335`	`335`	`struct ServerWriter {`
`336`	`336`	`rx: MessageReceiver,`
	`337`	`+ _server_shutdown: shutdown::Waiter`
`337`	`338`	`}`
`338`	`339`
`339`	`340`	`#[async_trait]`