Add as_ref and as_mut functions to Toggle (#1684)

libp2p · Aug 10, 2020 · b6ec95e · b6ec95e
1 parent 32df8ea
commit b6ec95e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/swarm/src/toggle.rs b/swarm/src/toggle.rs
@@ -51,6 +51,16 @@ impl<TBehaviour> Toggle<TBehaviour> {
     pub fn is_enabled(&self) -> bool {
         self.inner.is_some()
     }
+
+    /// Returns a reference to the inner `NetworkBehaviour`.
+    pub fn as_ref(&self) -> Option<&TBehaviour> {
+        self.inner.as_ref()
+    }
+
+    /// Returns a mutable reference to the inner `NetworkBehaviour`.
+    pub fn as_mut(&mut self) -> Option<&mut TBehaviour> {
+        self.inner.as_mut()
+    }
 }
 
 impl<TBehaviour> From<Option<TBehaviour>> for Toggle<TBehaviour> {