use super::{configuration, Error};
use crate::apis::ResponseContent;
use ::qcs_api_client_common::backoff::{
duration_from_io_error, duration_from_reqwest_error, duration_from_response, ExponentialBackoff,
};
#[cfg(feature = "tracing")]
use qcs_api_client_common::configuration::TokenRefresher;
use reqwest::StatusCode;
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(untagged)]
pub enum CreateReservationError {
Status401(crate::models::Error),
Status402(crate::models::Error),
Status403(crate::models::Error),
Status409(crate::models::Error),
Status422(crate::models::Error),
UnknownValue(serde_json::Value),
}
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(untagged)]
pub enum DeleteReservationError {
Status401(crate::models::Error),
Status403(crate::models::Error),
Status404(crate::models::Error),
UnknownValue(serde_json::Value),
}
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(untagged)]
pub enum FindAvailableReservationsError {
Status401(crate::models::Error),
Status422(crate::models::Error),
UnknownValue(serde_json::Value),
}
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(untagged)]
pub enum ListGroupReservationsError {
Status401(crate::models::Error),
Status422(crate::models::Error),
UnknownValue(serde_json::Value),
}
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(untagged)]
pub enum ListReservationsError {
Status401(crate::models::Error),
Status422(crate::models::Error),
UnknownValue(serde_json::Value),
}
async fn create_reservation_inner(
configuration: &configuration::Configuration,
backoff: &mut ExponentialBackoff,
create_reservation_request: crate::models::CreateReservationRequest,
x_qcs_account_id: Option<&str>,
x_qcs_account_type: Option<crate::models::AccountType>,
) -> Result<crate::models::Reservation, Error<CreateReservationError>> {
let local_var_configuration = configuration;
let local_var_client = &local_var_configuration.client;
let local_var_uri_str = format!(
"{}/v1/reservations",
local_var_configuration.qcs_config.api_url()
);
let mut local_var_req_builder =
local_var_client.request(reqwest::Method::POST, local_var_uri_str.as_str());
#[cfg(feature = "tracing")]
{
let local_var_do_tracing =
local_var_uri_str
.parse::<::url::Url>()
.ok()
.map_or(true, |url| {
configuration
.qcs_config
.should_trace(&::urlpattern::UrlPatternMatchInput::Url(url))
});
if local_var_do_tracing {
::tracing::debug!(
url=%local_var_uri_str,
method="POST",
"making create_reservation request",
);
}
}
if let Some(local_var_param_value) = x_qcs_account_id {
local_var_req_builder =
local_var_req_builder.header("x-qcs-account-id", local_var_param_value.to_string());
}
if let Some(local_var_param_value) = x_qcs_account_type {
local_var_req_builder =
local_var_req_builder.header("x-qcs-account-type", local_var_param_value.to_string());
}
let token = configuration.qcs_config.get_bearer_access_token().await?;
local_var_req_builder = local_var_req_builder.bearer_auth(token);
local_var_req_builder = local_var_req_builder.json(&create_reservation_request);
let local_var_req = local_var_req_builder.build()?;
let local_var_resp = local_var_client.execute(local_var_req).await?;
let local_var_status = local_var_resp.status();
if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
let local_var_content = local_var_resp.text().await?;
serde_json::from_str(&local_var_content).map_err(Error::from)
} else {
let local_var_retry_delay =
duration_from_response(local_var_resp.status(), local_var_resp.headers(), backoff);
let local_var_content = local_var_resp.text().await?;
let local_var_entity: Option<CreateReservationError> =
serde_json::from_str(&local_var_content).ok();
let local_var_error = ResponseContent {
status: local_var_status,
content: local_var_content,
entity: local_var_entity,
retry_delay: local_var_retry_delay,
};
Err(Error::ResponseError(local_var_error))
}
}
pub async fn create_reservation(
configuration: &configuration::Configuration,
create_reservation_request: crate::models::CreateReservationRequest,
x_qcs_account_id: Option<&str>,
x_qcs_account_type: Option<crate::models::AccountType>,
) -> Result<crate::models::Reservation, Error<CreateReservationError>> {
let mut backoff = configuration.backoff.clone();
let mut refreshed_credentials = false;
let method = reqwest::Method::POST;
loop {
let result = create_reservation_inner(
configuration,
&mut backoff,
create_reservation_request.clone(),
x_qcs_account_id.clone(),
x_qcs_account_type.clone(),
)
.await;
match result {
Ok(result) => return Ok(result),
Err(Error::ResponseError(response)) => {
if !refreshed_credentials
&& matches!(
response.status,
StatusCode::FORBIDDEN | StatusCode::UNAUTHORIZED
)
{
configuration.qcs_config.refresh().await?;
refreshed_credentials = true;
continue;
} else if let Some(duration) = response.retry_delay {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::ResponseError(response));
}
Err(Error::Reqwest(error)) => {
if let Some(duration) = duration_from_reqwest_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Reqwest(error));
}
Err(Error::Io(error)) => {
if let Some(duration) = duration_from_io_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Io(error));
}
Err(error) => return Err(error),
}
}
}
async fn delete_reservation_inner(
configuration: &configuration::Configuration,
backoff: &mut ExponentialBackoff,
reservation_id: i64,
) -> Result<crate::models::Reservation, Error<DeleteReservationError>> {
let local_var_configuration = configuration;
let local_var_client = &local_var_configuration.client;
let local_var_uri_str = format!(
"{}/v1/reservations/{reservationId}",
local_var_configuration.qcs_config.api_url(),
reservationId = reservation_id
);
let mut local_var_req_builder =
local_var_client.request(reqwest::Method::DELETE, local_var_uri_str.as_str());
#[cfg(feature = "tracing")]
{
let local_var_do_tracing =
local_var_uri_str
.parse::<::url::Url>()
.ok()
.map_or(true, |url| {
configuration
.qcs_config
.should_trace(&::urlpattern::UrlPatternMatchInput::Url(url))
});
if local_var_do_tracing {
::tracing::debug!(
url=%local_var_uri_str,
method="DELETE",
"making delete_reservation request",
);
}
}
let token = configuration.qcs_config.get_bearer_access_token().await?;
local_var_req_builder = local_var_req_builder.bearer_auth(token);
let local_var_req = local_var_req_builder.build()?;
let local_var_resp = local_var_client.execute(local_var_req).await?;
let local_var_status = local_var_resp.status();
if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
let local_var_content = local_var_resp.text().await?;
serde_json::from_str(&local_var_content).map_err(Error::from)
} else {
let local_var_retry_delay =
duration_from_response(local_var_resp.status(), local_var_resp.headers(), backoff);
let local_var_content = local_var_resp.text().await?;
let local_var_entity: Option<DeleteReservationError> =
serde_json::from_str(&local_var_content).ok();
let local_var_error = ResponseContent {
status: local_var_status,
content: local_var_content,
entity: local_var_entity,
retry_delay: local_var_retry_delay,
};
Err(Error::ResponseError(local_var_error))
}
}
pub async fn delete_reservation(
configuration: &configuration::Configuration,
reservation_id: i64,
) -> Result<crate::models::Reservation, Error<DeleteReservationError>> {
let mut backoff = configuration.backoff.clone();
let mut refreshed_credentials = false;
let method = reqwest::Method::DELETE;
loop {
let result =
delete_reservation_inner(configuration, &mut backoff, reservation_id.clone()).await;
match result {
Ok(result) => return Ok(result),
Err(Error::ResponseError(response)) => {
if !refreshed_credentials
&& matches!(
response.status,
StatusCode::FORBIDDEN | StatusCode::UNAUTHORIZED
)
{
configuration.qcs_config.refresh().await?;
refreshed_credentials = true;
continue;
} else if let Some(duration) = response.retry_delay {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::ResponseError(response));
}
Err(Error::Reqwest(error)) => {
if let Some(duration) = duration_from_reqwest_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Reqwest(error));
}
Err(Error::Io(error)) => {
if let Some(duration) = duration_from_io_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Io(error));
}
Err(error) => return Err(error),
}
}
}
async fn find_available_reservations_inner(
configuration: &configuration::Configuration,
backoff: &mut ExponentialBackoff,
quantum_processor_id: &str,
start_time_from: String,
duration: &str,
page_size: Option<i64>,
page_token: Option<&str>,
) -> Result<crate::models::FindAvailableReservationsResponse, Error<FindAvailableReservationsError>>
{
let local_var_configuration = configuration;
let local_var_client = &local_var_configuration.client;
let local_var_uri_str = format!(
"{}/v1/reservations:findAvailable",
local_var_configuration.qcs_config.api_url()
);
let mut local_var_req_builder =
local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
#[cfg(feature = "tracing")]
{
let local_var_do_tracing =
local_var_uri_str
.parse::<::url::Url>()
.ok()
.map_or(true, |url| {
configuration
.qcs_config
.should_trace(&::urlpattern::UrlPatternMatchInput::Url(url))
});
if local_var_do_tracing {
::tracing::debug!(
url=%local_var_uri_str,
method="GET",
"making find_available_reservations request",
);
}
}
if let Some(ref local_var_str) = page_size {
local_var_req_builder =
local_var_req_builder.query(&[("pageSize", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = page_token {
local_var_req_builder =
local_var_req_builder.query(&[("pageToken", &local_var_str.to_string())]);
}
local_var_req_builder =
local_var_req_builder.query(&[("quantumProcessorId", &quantum_processor_id.to_string())]);
local_var_req_builder =
local_var_req_builder.query(&[("startTimeFrom", &start_time_from.to_string())]);
local_var_req_builder = local_var_req_builder.query(&[("duration", &duration.to_string())]);
let token = configuration.qcs_config.get_bearer_access_token().await?;
local_var_req_builder = local_var_req_builder.bearer_auth(token);
let local_var_req = local_var_req_builder.build()?;
let local_var_resp = local_var_client.execute(local_var_req).await?;
let local_var_status = local_var_resp.status();
if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
let local_var_content = local_var_resp.text().await?;
serde_json::from_str(&local_var_content).map_err(Error::from)
} else {
let local_var_retry_delay =
duration_from_response(local_var_resp.status(), local_var_resp.headers(), backoff);
let local_var_content = local_var_resp.text().await?;
let local_var_entity: Option<FindAvailableReservationsError> =
serde_json::from_str(&local_var_content).ok();
let local_var_error = ResponseContent {
status: local_var_status,
content: local_var_content,
entity: local_var_entity,
retry_delay: local_var_retry_delay,
};
Err(Error::ResponseError(local_var_error))
}
}
pub async fn find_available_reservations(
configuration: &configuration::Configuration,
quantum_processor_id: &str,
start_time_from: String,
duration: &str,
page_size: Option<i64>,
page_token: Option<&str>,
) -> Result<crate::models::FindAvailableReservationsResponse, Error<FindAvailableReservationsError>>
{
let mut backoff = configuration.backoff.clone();
let mut refreshed_credentials = false;
let method = reqwest::Method::GET;
loop {
let result = find_available_reservations_inner(
configuration,
&mut backoff,
quantum_processor_id.clone(),
start_time_from.clone(),
duration.clone(),
page_size.clone(),
page_token.clone(),
)
.await;
match result {
Ok(result) => return Ok(result),
Err(Error::ResponseError(response)) => {
if !refreshed_credentials
&& matches!(
response.status,
StatusCode::FORBIDDEN | StatusCode::UNAUTHORIZED
)
{
configuration.qcs_config.refresh().await?;
refreshed_credentials = true;
continue;
} else if let Some(duration) = response.retry_delay {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::ResponseError(response));
}
Err(Error::Reqwest(error)) => {
if let Some(duration) = duration_from_reqwest_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Reqwest(error));
}
Err(Error::Io(error)) => {
if let Some(duration) = duration_from_io_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Io(error));
}
Err(error) => return Err(error),
}
}
}
async fn list_group_reservations_inner(
configuration: &configuration::Configuration,
backoff: &mut ExponentialBackoff,
group_name: &str,
filter: Option<&str>,
order: Option<&str>,
page_size: Option<i64>,
page_token: Option<&str>,
show_deleted: Option<&str>,
) -> Result<crate::models::ListReservationsResponse, Error<ListGroupReservationsError>> {
let local_var_configuration = configuration;
let local_var_client = &local_var_configuration.client;
let local_var_uri_str = format!(
"{}/v1/groups/{groupName}/reservations",
local_var_configuration.qcs_config.api_url(),
groupName = crate::apis::urlencode(group_name)
);
let mut local_var_req_builder =
local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
#[cfg(feature = "tracing")]
{
let local_var_do_tracing =
local_var_uri_str
.parse::<::url::Url>()
.ok()
.map_or(true, |url| {
configuration
.qcs_config
.should_trace(&::urlpattern::UrlPatternMatchInput::Url(url))
});
if local_var_do_tracing {
::tracing::debug!(
url=%local_var_uri_str,
method="GET",
"making list_group_reservations request",
);
}
}
if let Some(ref local_var_str) = filter {
local_var_req_builder =
local_var_req_builder.query(&[("filter", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = order {
local_var_req_builder =
local_var_req_builder.query(&[("order", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = page_size {
local_var_req_builder =
local_var_req_builder.query(&[("pageSize", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = page_token {
local_var_req_builder =
local_var_req_builder.query(&[("pageToken", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = show_deleted {
local_var_req_builder =
local_var_req_builder.query(&[("showDeleted", &local_var_str.to_string())]);
}
let token = configuration.qcs_config.get_bearer_access_token().await?;
local_var_req_builder = local_var_req_builder.bearer_auth(token);
let local_var_req = local_var_req_builder.build()?;
let local_var_resp = local_var_client.execute(local_var_req).await?;
let local_var_status = local_var_resp.status();
if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
let local_var_content = local_var_resp.text().await?;
serde_json::from_str(&local_var_content).map_err(Error::from)
} else {
let local_var_retry_delay =
duration_from_response(local_var_resp.status(), local_var_resp.headers(), backoff);
let local_var_content = local_var_resp.text().await?;
let local_var_entity: Option<ListGroupReservationsError> =
serde_json::from_str(&local_var_content).ok();
let local_var_error = ResponseContent {
status: local_var_status,
content: local_var_content,
entity: local_var_entity,
retry_delay: local_var_retry_delay,
};
Err(Error::ResponseError(local_var_error))
}
}
pub async fn list_group_reservations(
configuration: &configuration::Configuration,
group_name: &str,
filter: Option<&str>,
order: Option<&str>,
page_size: Option<i64>,
page_token: Option<&str>,
show_deleted: Option<&str>,
) -> Result<crate::models::ListReservationsResponse, Error<ListGroupReservationsError>> {
let mut backoff = configuration.backoff.clone();
let mut refreshed_credentials = false;
let method = reqwest::Method::GET;
loop {
let result = list_group_reservations_inner(
configuration,
&mut backoff,
group_name.clone(),
filter.clone(),
order.clone(),
page_size.clone(),
page_token.clone(),
show_deleted.clone(),
)
.await;
match result {
Ok(result) => return Ok(result),
Err(Error::ResponseError(response)) => {
if !refreshed_credentials
&& matches!(
response.status,
StatusCode::FORBIDDEN | StatusCode::UNAUTHORIZED
)
{
configuration.qcs_config.refresh().await?;
refreshed_credentials = true;
continue;
} else if let Some(duration) = response.retry_delay {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::ResponseError(response));
}
Err(Error::Reqwest(error)) => {
if let Some(duration) = duration_from_reqwest_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Reqwest(error));
}
Err(Error::Io(error)) => {
if let Some(duration) = duration_from_io_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Io(error));
}
Err(error) => return Err(error),
}
}
}
async fn list_reservations_inner(
configuration: &configuration::Configuration,
backoff: &mut ExponentialBackoff,
filter: Option<&str>,
order: Option<&str>,
page_size: Option<i64>,
page_token: Option<&str>,
show_deleted: Option<&str>,
x_qcs_account_id: Option<&str>,
x_qcs_account_type: Option<crate::models::AccountType>,
) -> Result<crate::models::ListReservationsResponse, Error<ListReservationsError>> {
let local_var_configuration = configuration;
let local_var_client = &local_var_configuration.client;
let local_var_uri_str = format!(
"{}/v1/reservations",
local_var_configuration.qcs_config.api_url()
);
let mut local_var_req_builder =
local_var_client.request(reqwest::Method::GET, local_var_uri_str.as_str());
#[cfg(feature = "tracing")]
{
let local_var_do_tracing =
local_var_uri_str
.parse::<::url::Url>()
.ok()
.map_or(true, |url| {
configuration
.qcs_config
.should_trace(&::urlpattern::UrlPatternMatchInput::Url(url))
});
if local_var_do_tracing {
::tracing::debug!(
url=%local_var_uri_str,
method="GET",
"making list_reservations request",
);
}
}
if let Some(ref local_var_str) = filter {
local_var_req_builder =
local_var_req_builder.query(&[("filter", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = order {
local_var_req_builder =
local_var_req_builder.query(&[("order", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = page_size {
local_var_req_builder =
local_var_req_builder.query(&[("pageSize", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = page_token {
local_var_req_builder =
local_var_req_builder.query(&[("pageToken", &local_var_str.to_string())]);
}
if let Some(ref local_var_str) = show_deleted {
local_var_req_builder =
local_var_req_builder.query(&[("showDeleted", &local_var_str.to_string())]);
}
if let Some(local_var_param_value) = x_qcs_account_id {
local_var_req_builder =
local_var_req_builder.header("x-qcs-account-id", local_var_param_value.to_string());
}
if let Some(local_var_param_value) = x_qcs_account_type {
local_var_req_builder =
local_var_req_builder.header("x-qcs-account-type", local_var_param_value.to_string());
}
let token = configuration.qcs_config.get_bearer_access_token().await?;
local_var_req_builder = local_var_req_builder.bearer_auth(token);
let local_var_req = local_var_req_builder.build()?;
let local_var_resp = local_var_client.execute(local_var_req).await?;
let local_var_status = local_var_resp.status();
if !local_var_status.is_client_error() && !local_var_status.is_server_error() {
let local_var_content = local_var_resp.text().await?;
serde_json::from_str(&local_var_content).map_err(Error::from)
} else {
let local_var_retry_delay =
duration_from_response(local_var_resp.status(), local_var_resp.headers(), backoff);
let local_var_content = local_var_resp.text().await?;
let local_var_entity: Option<ListReservationsError> =
serde_json::from_str(&local_var_content).ok();
let local_var_error = ResponseContent {
status: local_var_status,
content: local_var_content,
entity: local_var_entity,
retry_delay: local_var_retry_delay,
};
Err(Error::ResponseError(local_var_error))
}
}
pub async fn list_reservations(
configuration: &configuration::Configuration,
filter: Option<&str>,
order: Option<&str>,
page_size: Option<i64>,
page_token: Option<&str>,
show_deleted: Option<&str>,
x_qcs_account_id: Option<&str>,
x_qcs_account_type: Option<crate::models::AccountType>,
) -> Result<crate::models::ListReservationsResponse, Error<ListReservationsError>> {
let mut backoff = configuration.backoff.clone();
let mut refreshed_credentials = false;
let method = reqwest::Method::GET;
loop {
let result = list_reservations_inner(
configuration,
&mut backoff,
filter.clone(),
order.clone(),
page_size.clone(),
page_token.clone(),
show_deleted.clone(),
x_qcs_account_id.clone(),
x_qcs_account_type.clone(),
)
.await;
match result {
Ok(result) => return Ok(result),
Err(Error::ResponseError(response)) => {
if !refreshed_credentials
&& matches!(
response.status,
StatusCode::FORBIDDEN | StatusCode::UNAUTHORIZED
)
{
configuration.qcs_config.refresh().await?;
refreshed_credentials = true;
continue;
} else if let Some(duration) = response.retry_delay {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::ResponseError(response));
}
Err(Error::Reqwest(error)) => {
if let Some(duration) = duration_from_reqwest_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Reqwest(error));
}
Err(Error::Io(error)) => {
if let Some(duration) = duration_from_io_error(&method, &error, &mut backoff) {
tokio::time::sleep(duration).await;
continue;
}
return Err(Error::Io(error));
}
Err(error) => return Err(error),
}
}
}