From b6ec95eb772b371dbde3563fe2a24bd13b3d11a7 Mon Sep 17 00:00:00 2001
From: Ashley <ashley.ruglys@gmail.com>
Date: Mon, 10 Aug 2020 10:11:58 +0200
Subject: [PATCH] Add as_ref and as_mut functions to Toggle (#1684)

---
 swarm/src/toggle.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/swarm/src/toggle.rs b/swarm/src/toggle.rs
index efd61776..2dbd5268 100644
--- 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> {