fix super nit in ServerBuilder (#512)

paritytech · Oct 12, 2021 · 07df2f8 · 07df2f8
1 parent f949d9b
commit 07df2f8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/http-server/src/server.rs b/http-server/src/server.rs
@@ -98,8 +98,9 @@ impl Builder {
 	/// Configure a custom [`tokio::runtime::Handle`] to run the server on.
 	///
 	/// Default: [`tokio::spawn`]
-	pub fn custom_tokio_runtime(mut self, rt: tokio::runtime::Handle) {
+	pub fn custom_tokio_runtime(mut self, rt: tokio::runtime::Handle) -> Self {
 		self.tokio_runtime = Some(rt);
+		self
 	}
 
 	/// Finalizes the configuration of the server.

diff --git a/ws-server/src/server.rs b/ws-server/src/server.rs
@@ -487,8 +487,9 @@ impl Builder {
 	/// Configure a custom [`tokio::runtime::Handle`] to run the server on.
 	///
 	/// Default: [`tokio::spawn`]
-	pub fn custom_tokio_runtime(mut self, rt: tokio::runtime::Handle) {
+	pub fn custom_tokio_runtime(mut self, rt: tokio::runtime::Handle) -> Self {
 		self.settings.tokio_runtime = Some(rt);
+		self
 	}
 
 	/// Finalize the configuration of the server. Consumes the [`Builder`].