use derive_builder::Builder;
use http::{HeaderMap, HeaderName, HeaderValue};
use crate::api::rest_endpoint_prelude::*;
use serde::Deserialize;
use serde::Serialize;
use std::borrow::Cow;
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum HttpMethod {
#[serde(rename = "CONNECT")]
Connect,
#[serde(rename = "DELETE")]
Delete,
#[serde(rename = "GET")]
Get,
#[serde(rename = "HEAD")]
Head,
#[serde(rename = "OPTIONS")]
Options,
#[serde(rename = "PATCH")]
Patch,
#[serde(rename = "POST")]
Post,
#[serde(rename = "PUT")]
Put,
#[serde(rename = "TRACE")]
Trace,
}
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum HealthmonitorType {
#[serde(rename = "HTTP")]
Http,
#[serde(rename = "HTTPS")]
Https,
#[serde(rename = "PING")]
Ping,
#[serde(rename = "SCTP")]
Sctp,
#[serde(rename = "TCP")]
Tcp,
#[serde(rename = "TLS-HELLO")]
TlsHello,
#[serde(rename = "UDP-CONNECT")]
UdpConnect,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Healthmonitor<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) admin_state_up: Option<bool>,
#[serde()]
#[builder(setter(into))]
pub(crate) delay: i32,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) domain_name: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) expected_codes: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default)]
pub(crate) http_method: Option<HttpMethod>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) http_version: Option<f32>,
#[serde()]
#[builder(setter(into))]
pub(crate) max_retries: i32,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) max_retries_down: Option<i32>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) tags: Option<Vec<Cow<'a, str>>>,
#[serde()]
#[builder(setter(into))]
pub(crate) timeout: i32,
#[serde(rename = "type")]
#[builder()]
pub(crate) _type: HealthmonitorType,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) url_path: Option<Cow<'a, str>>,
}
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum LbAlgorithm {
#[serde(rename = "LEAST_CONNECTIONS")]
LeastConnections,
#[serde(rename = "ROUND_ROBIN")]
RoundRobin,
#[serde(rename = "SOURCE_IP")]
SourceIp,
#[serde(rename = "SOURCE_IP_PORT")]
SourceIpPort,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Members<'a> {
#[serde()]
#[builder(setter(into))]
pub(crate) address: Cow<'a, str>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) admin_state_up: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) backup: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) monitor_address: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) monitor_port: Option<i32>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
#[serde()]
#[builder(setter(into))]
pub(crate) protocol_port: i32,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) request_sriov: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) subnet_id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) tags: Option<Vec<Cow<'a, str>>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) weight: Option<i32>,
}
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum Protocol {
#[serde(rename = "HTTP")]
Http,
#[serde(rename = "HTTPS")]
Https,
#[serde(rename = "PROXY")]
Proxy,
#[serde(rename = "PROXYV2")]
Proxyv2,
#[serde(rename = "SCTP")]
Sctp,
#[serde(rename = "TCP")]
Tcp,
#[serde(rename = "UDP")]
Udp,
}
#[derive(Debug, Deserialize, Clone, Serialize)]
pub enum SessionPersistenceType {
#[serde(rename = "APP_COOKIE")]
AppCookie,
#[serde(rename = "HTTP_COOKIE")]
HttpCookie,
#[serde(rename = "SOURCE_IP")]
SourceIp,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct SessionPersistence<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) cookie_name: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) persistence_granularity: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) persistence_timeout: Option<i32>,
#[serde(rename = "type")]
#[builder()]
pub(crate) _type: SessionPersistenceType,
}
#[derive(Builder, Debug, Deserialize, Clone, Serialize)]
#[builder(setter(strip_option))]
pub struct Pool<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) admin_state_up: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) alpn_protocols: Option<Vec<Cow<'a, str>>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) ca_tls_container_ref: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) crl_container_ref: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) description: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) healthmonitor: Option<Healthmonitor<'a>>,
#[serde()]
#[builder()]
pub(crate) lb_algorithm: LbAlgorithm,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) listener_id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) loadbalancer_id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) members: Option<Vec<Members<'a>>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) name: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) project_id: Option<Cow<'a, str>>,
#[serde()]
#[builder()]
pub(crate) protocol: Protocol,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) session_persistence: Option<SessionPersistence<'a>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) tags: Option<Vec<Cow<'a, str>>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) tenant_id: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) tls_ciphers: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) tls_container_ref: Option<Cow<'a, str>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) tls_enabled: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
#[builder(default, setter(into))]
pub(crate) tls_versions: Option<Vec<Cow<'a, str>>>,
}
#[derive(Builder, Debug, Clone)]
#[builder(setter(strip_option))]
pub struct Request<'a> {
#[builder(setter(into))]
pub(crate) pool: Pool<'a>,
#[builder(setter(name = "_headers"), default, private)]
_headers: Option<HeaderMap>,
}
impl<'a> Request<'a> {
pub fn builder() -> RequestBuilder<'a> {
RequestBuilder::default()
}
}
impl<'a> RequestBuilder<'a> {
pub fn header<K, V>(&mut self, header_name: K, header_value: V) -> &mut Self
where
K: Into<HeaderName>,
V: Into<HeaderValue>,
{
self._headers
.get_or_insert(None)
.get_or_insert_with(HeaderMap::new)
.insert(header_name.into(), header_value.into());
self
}
pub fn headers<I, T>(&mut self, iter: I) -> &mut Self
where
I: Iterator<Item = T>,
T: Into<(Option<HeaderName>, HeaderValue)>,
{
self._headers
.get_or_insert(None)
.get_or_insert_with(HeaderMap::new)
.extend(iter.map(Into::into));
self
}
}
impl RestEndpoint for Request<'_> {
fn method(&self) -> http::Method {
http::Method::POST
}
fn endpoint(&self) -> Cow<'static, str> {
"lbaas/pools".to_string().into()
}
fn parameters(&self) -> QueryParams<'_> {
QueryParams::default()
}
fn body(&self) -> Result<Option<(&'static str, Vec<u8>)>, BodyError> {
let mut params = JsonBodyParams::default();
params.push("pool", serde_json::to_value(&self.pool)?);
params.into_body()
}
fn service_type(&self) -> ServiceType {
ServiceType::LoadBalancer
}
fn response_key(&self) -> Option<Cow<'static, str>> {
Some("pool".into())
}
fn request_headers(&self) -> Option<&HeaderMap> {
self._headers.as_ref()
}
fn api_version(&self) -> Option<ApiVersion> {
Some(ApiVersion::new(2, 0))
}
}
#[cfg(test)]
mod tests {
use super::*;
#[cfg(feature = "sync")]
use crate::api::Query;
use crate::test::client::FakeOpenStackClient;
use crate::types::ServiceType;
use http::{HeaderName, HeaderValue};
use httpmock::MockServer;
use serde_json::json;
#[test]
fn test_service_type() {
assert_eq!(
Request::builder()
.pool(
PoolBuilder::default()
.lb_algorithm(LbAlgorithm::LeastConnections)
.protocol(Protocol::Http)
.build()
.unwrap()
)
.build()
.unwrap()
.service_type(),
ServiceType::LoadBalancer
);
}
#[test]
fn test_response_key() {
assert_eq!(
Request::builder()
.pool(
PoolBuilder::default()
.lb_algorithm(LbAlgorithm::LeastConnections)
.protocol(Protocol::Http)
.build()
.unwrap()
)
.build()
.unwrap()
.response_key()
.unwrap(),
"pool"
);
}
#[cfg(feature = "sync")]
#[test]
fn endpoint() {
let server = MockServer::start();
let client = FakeOpenStackClient::new(server.base_url());
let mock = server.mock(|when, then| {
when.method(httpmock::Method::POST)
.path("/lbaas/pools".to_string());
then.status(200)
.header("content-type", "application/json")
.json_body(json!({ "pool": {} }));
});
let endpoint = Request::builder()
.pool(
PoolBuilder::default()
.lb_algorithm(LbAlgorithm::LeastConnections)
.protocol(Protocol::Http)
.build()
.unwrap(),
)
.build()
.unwrap();
let _: serde_json::Value = endpoint.query(&client).unwrap();
mock.assert();
}
#[cfg(feature = "sync")]
#[test]
fn endpoint_headers() {
let server = MockServer::start();
let client = FakeOpenStackClient::new(server.base_url());
let mock = server.mock(|when, then| {
when.method(httpmock::Method::POST)
.path("/lbaas/pools".to_string())
.header("foo", "bar")
.header("not_foo", "not_bar");
then.status(200)
.header("content-type", "application/json")
.json_body(json!({ "pool": {} }));
});
let endpoint = Request::builder()
.pool(
PoolBuilder::default()
.lb_algorithm(LbAlgorithm::LeastConnections)
.protocol(Protocol::Http)
.build()
.unwrap(),
)
.headers(
[(
Some(HeaderName::from_static("foo")),
HeaderValue::from_static("bar"),
)]
.into_iter(),
)
.header(
HeaderName::from_static("not_foo"),
HeaderValue::from_static("not_bar"),
)
.build()
.unwrap();
let _: serde_json::Value = endpoint.query(&client).unwrap();
mock.assert();
}
}