IoHandlerExtension for Option (#535)

paritytech · Jan 28, 2020 · 2048538 · 2048538
1 parent 596ed3a
commit 2048538
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/core/src/io.rs b/core/src/io.rs
@@ -378,6 +378,14 @@ impl<M: Metadata, S2: Middleware<M>> IoHandlerExtension<M> for MetaIoHandler<M,
 	}
 }
 
+impl<M: Metadata, T: IoHandlerExtension<M>> IoHandlerExtension<M> for Option<T> {
+	fn augment<S: Middleware<M>>(self, handler: &mut MetaIoHandler<M, S>) {
+		if let Some(x) = self {
+			x.augment(handler)
+		}
+	}
+}
+
 /// Simplified `IoHandler` with no `Metadata` associated with each request.
 #[derive(Clone, Debug, Default)]
 pub struct IoHandler<M: Metadata = ()>(MetaIoHandler<M>);