kong_rs 0.2.0

Plugin Development Kit for developing Kong plugins in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use kong_rs_protos::Target;
use request::ServiceRequestPDK;
use response::ServiceResponsePDK;
use strum::{EnumString, IntoStaticStr};

use crate::{stream::Stream, KongResult};

pub mod request;
pub mod response;

#[derive(Debug, PartialEq, IntoStaticStr, EnumString)]
pub(crate) enum Methods {
  #[strum(serialize = "kong.service.set_upstream")]
  SetUpstream,
  #[strum(serialize = "kong.service.set_target")]
  SetTarget,
}

#[derive(Clone)]
pub struct ServicePDK {
  stream: Stream,

  request: ServiceRequestPDK,
  response: ServiceResponsePDK
}

impl ServicePDK {
  pub fn new(stream: Stream) -> Self {
    Self {
      stream: stream.clone(),
      request: ServiceRequestPDK::new(stream.clone()),
      response: ServiceResponsePDK::new(stream.clone()),
    }
  }

  pub async fn set_upstream<A: Into<String>>(&self, addr: A) -> KongResult<bool> {
    let r: kong_rs_protos::Bool = self.stream.ask_message_with_args(Methods::SetUpstream.into(), &kong_rs_protos::String { v: addr.into() }).await?;
    Ok(r.v)
  }

  pub async fn set_target<H: Into<String>>(&self, host: H, port: usize) -> KongResult<()> {
    self.stream.ask_message_with_args(Methods::SetTarget.into(), &Target { host: host.into(), port: port as i32 }).await
  }

  pub fn request(&self) -> &ServiceRequestPDK {
    &self.request
  }

  pub fn response(&self) -> &ServiceResponsePDK {
    &self.response
  }
}