Remove code duplicates

systream · Jun 5, 2016 · 9d64efc · 9d64efc
1 parent 445fbab
commit 9d64efc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/FeatureSwitch.php b/src/FeatureSwitch.php
@@ -38,7 +38,8 @@ public function addFeature(Feature\SwitchableFeatureInterface $feature)
 		if (isset($this->features[$feature->getKey()])) {
 			throw new \Exception(sprintf('There are already a feature with this key: %s', $feature->getKey()));
 		}
-		$this->features[$feature->getKey()] = $feature;
+
+		$this->storeFeature($feature);
 	}
 
 	/**
@@ -53,4 +54,12 @@ public function isEnabled($key)
 		}
 		return $this->features[$key]->isEnabled();
 	}
+
+	/**
+	 * @param Feature\SwitchableFeatureInterface $feature
+	 */
+	protected function storeFeature(Feature\SwitchableFeatureInterface $feature)
+	{
+		$this->features[$feature->getKey()] = $feature;
+	}
 }
diff --git a/src/FeatureSwitchArray.php b/src/FeatureSwitchArray.php
@@ -128,8 +128,7 @@ public function offsetSet($offset, $value)
 			return;
 		}
 
-		$this->features[$value->getKey()] = $value;
-
+		$this->storeFeature($value);
 	}
 
 	/**