Closes #35

Author: michaelklishin

src/api.rs

@@ -24,7 +24,10 @@ use std::fmt;
 
 use crate::error::Error;
 use crate::error::Error::{ClientErrorResponse, NotFound, ServerErrorResponse};
-use crate::responses::{DeprecatedFeatureList, FeatureFlagList, MessageList};
+use crate::responses::{
+    DeprecatedFeatureList, FeatureFlag, FeatureFlagList, FeatureFlagStability, FeatureFlagState,
+    MessageList,
+};
 use crate::{
     commons::{BindingDestinationType, UserLimitTarget, VirtualHostLimitTarget},
     path,
@@ -1176,12 +1179,52 @@ where
     // Feature flags
     //
 
+    /// Enables a feature flag.
+    /// This function is idempotent: enabling an already enabled feature flag
+    /// will succeed.
     pub async fn list_feature_flags(&self) -> Result<FeatureFlagList> {
         let response = self.http_get("feature-flags", None, None).await?;
         let response = response.json().await?;
         Ok(response)
     }
 
+    /// Enables all stable feature flags.
+    /// This function is idempotent: enabling an already enabled feature flag
+    /// will succeed.
+    pub async fn enable_feature_flag(&self, name: &str) -> Result<()> {
+        let body = serde_json::json!({
+            "name": name
+        });
+        let _response = self
+            .http_put(path!("feature-flags", name, "enable"), &body, None, None)
+            .await?;
+        Ok(())
+    }
+
+    /// Enables all stable feature flags.
+    /// This function is idempotent: enabling an already enabled feature flag
+    /// will succeed.
+    pub async fn enable_all_stable_feature_flags(&self) -> Result<()> {
+        // PUT /api/feature-flags/{name}/enable does not support the special 'all' value like 'rabbitmqctl enable_feature_flag' does.
+        // Thus we do what management UI does: discover the stable disabled flags and enable
+        // them one by one.
+        let discovered_flags = self.list_feature_flags().await?;
+        let flags_to_enable: Vec<&FeatureFlag> = discovered_flags
+            .0
+            .iter()
+            .filter(|&ff| {
+                ff.state == FeatureFlagState::Disabled
+                    && ff.stability == FeatureFlagStability::Stable
+            })
+            .collect();
+
+        for ff in flags_to_enable {
+            self.enable_feature_flag(&ff.name).await?;
+        }
+
+        Ok(())
+    }
+
     //
     // Deprecated Features
     //

src/blocking_api.rs

@@ -15,7 +15,10 @@
 
 use crate::error::Error;
 use crate::error::Error::{ClientErrorResponse, NotFound, ServerErrorResponse};
-use crate::responses::{DeprecatedFeatureList, FeatureFlagList, OAuthConfiguration};
+use crate::responses::{
+    DeprecatedFeatureList, FeatureFlag, FeatureFlagList, FeatureFlagStability, FeatureFlagState,
+    OAuthConfiguration,
+};
 use crate::{
     commons::{BindingDestinationType, UserLimitTarget, VirtualHostLimitTarget},
     path,
@@ -1039,6 +1042,41 @@ where
         Ok(response)
     }
 
+    /// Enables a feature flag.
+    /// This function is idempotent: enabling an already enabled feature flag
+    /// will succeed.
+    pub fn enable_feature_flag(&self, name: &str) -> Result<()> {
+        let body = serde_json::json!({
+            "name": name
+        });
+        let _response = self.http_put(path!("feature-flags", name, "enable"), &body, None, None)?;
+        Ok(())
+    }
+
+    /// Enables all stable feature flags.
+    /// This function is idempotent: enabling an already enabled feature flag
+    /// will succeed.
+    pub fn enable_all_stable_feature_flags(&self) -> Result<()> {
+        // PUT /api/feature-flags/{name}/enable does not support the special 'all' value like 'rabbitmqctl enable_feature_flag' does.
+        // Thus we do what management UI does: discover the stable disabled flags and enable
+        // them one by one.
+        let discovered_flags = self.list_feature_flags()?;
+        let flags_to_enable: Vec<&FeatureFlag> = discovered_flags
+            .0
+            .iter()
+            .filter(|&ff| {
+                ff.state == FeatureFlagState::Disabled
+                    && ff.stability == FeatureFlagStability::Stable
+            })
+            .collect();
+
+        for ff in flags_to_enable {
+            self.enable_feature_flag(&ff.name)?;
+        }
+
+        Ok(())
+    }
+
     //
     // Deprecated Features
     //

tests/feature_flag_tests.rs

@@ -11,7 +11,10 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
-use rabbitmq_http_client::{blocking_api::Client, responses::FeatureFlagStability};
+use rabbitmq_http_client::{
+    blocking_api::Client,
+    responses::{FeatureFlagStability, FeatureFlagState},
+};
 
 mod test_helpers;
 use crate::test_helpers::{endpoint, PASSWORD, USERNAME};
@@ -29,3 +32,41 @@ fn test_list_feature_flags() {
         .into_iter()
         .any(|ff| ff.name == "rabbitmq_4.0.0" && ff.stability == FeatureFlagStability::Stable));
 }
+
+#[test]
+fn test_enable_a_feature_flag() {
+    let endpoint = endpoint();
+    let rc = Client::new(&endpoint, USERNAME, PASSWORD);
+    let ff_name = "detailed_queues_endpoint";
+
+    let result1 = rc.enable_feature_flag(ff_name);
+    assert!(result1.is_ok());
+
+    let result2 = rc.list_feature_flags();
+
+    assert!(result2.is_ok());
+    let vec = result2.unwrap();
+    assert!(vec
+        .0
+        .into_iter()
+        .any(|ff| ff.name == ff_name && ff.state == FeatureFlagState::Enabled));
+}
+
+#[test]
+fn test_enable_all_stable_feature_flags() {
+    let endpoint = endpoint();
+    let rc = Client::new(&endpoint, USERNAME, PASSWORD);
+    let ff_name = "rabbitmq_4.0.0";
+
+    let result1 = rc.enable_all_stable_feature_flags();
+    assert!(result1.is_ok());
+
+    let result2 = rc.list_feature_flags();
+
+    assert!(result2.is_ok());
+    let vec = result2.unwrap();
+    assert!(vec
+        .0
+        .into_iter()
+        .any(|ff| ff.name == ff_name && ff.state == FeatureFlagState::Enabled));
+}