use datadog_api_client::datadog;
use datadog_api_client::datadogV2::api_feature_flags::FeatureFlagsAPI;
use datadog_api_client::datadogV2::model::AllocationDataRequest;
use datadog_api_client::datadogV2::model::AllocationDataType;
use datadog_api_client::datadogV2::model::AllocationType;
use datadog_api_client::datadogV2::model::CreateAllocationsRequest;
use datadog_api_client::datadogV2::model::UpsertAllocationRequest;
use datadog_api_client::datadogV2::model::VariantWeightRequest;
#[tokio::main]
async fn main() {
let feature_flag_data_attributes_variants_0_id = uuid::Uuid::parse_str(
&std::env::var("FEATURE_FLAG_DATA_ATTRIBUTES_VARIANTS_0_ID").unwrap(),
)
.expect("Invalid UUID");
let feature_flag_data_id =
uuid::Uuid::parse_str(&std::env::var("FEATURE_FLAG_DATA_ID").unwrap())
.expect("Invalid UUID");
let environment_data_id = uuid::Uuid::parse_str(&std::env::var("ENVIRONMENT_DATA_ID").unwrap())
.expect("Invalid UUID");
let body = CreateAllocationsRequest::new(AllocationDataRequest::new(
UpsertAllocationRequest::new(
"new-targeting-rule-example-feature-flag".to_string(),
"New targeting rule Example-Feature-Flag".to_string(),
AllocationType::CANARY,
)
.guardrail_metrics(vec![])
.targeting_rules(vec![])
.variant_weights(vec![VariantWeightRequest::new(100.0)
.variant_id(feature_flag_data_attributes_variants_0_id.clone())]),
AllocationDataType::ALLOCATIONS,
));
let configuration = datadog::Configuration::new();
let api = FeatureFlagsAPI::with_config(configuration);
let resp = api
.create_allocations_for_feature_flag_in_environment(
feature_flag_data_id.clone(),
environment_data_id.clone(),
body,
)
.await;
if let Ok(value) = resp {
println!("{:#?}", value);
} else {
println!("{:#?}", resp.unwrap_err());
}
}