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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use kong_rs_protos::Kv;
use strum::{EnumString, IntoStaticStr};

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

use super::Value;

#[derive(Debug, PartialEq, IntoStaticStr, EnumString)]
pub(crate) enum Methods {
  #[strum(serialize = "kong.ctx.shared.set")]
  SharedSet,
  #[strum(serialize = "kong.ctx.shared.get")]
  SharedGet,
  #[strum(serialize = "kong.nginx.set_ctx")]
  Set,
  #[strum(serialize = "kong.nginx.get_ctx")]
  Get,
}

#[derive(Clone)]
pub struct CtxPDK {
  stream: Stream
}

impl CtxPDK {
  pub fn new(stream: Stream) -> Self {
    Self { stream }
  }

  pub async fn shared_set<K: Into<String>>(&self, key: K, value: Value) -> KongResult<()> {
    let kind = match value {
        Value::Null => None,
        x => Some(x.into())
    };

    let kv = Kv { k: key.into(), v: Some(prost_types::Value { kind: kind }) };
    self.stream.ask_message_with_args(Methods::SharedSet.into(), &kv).await
  }

  pub async fn shared_get<K: Into<String>>(&self, key: K) -> KongResult<Value> {
    let v: prost_types::Value = self.stream.ask_message_with_args(
      Methods::SharedGet.into(),
      &kong_rs_protos::String { v: key.into() }
    ).await?;

    Ok(v.kind.map(Into::into).unwrap_or(Value::Null))
  }

  pub async fn set<K: Into<String>>(&self, key: K, value: Value) -> KongResult<()> {
    let kind = match value {
        Value::Null => None,
        x => Some(x.into())
    };

    let kv = Kv { k: key.into(), v: Some(prost_types::Value { kind: kind }) };
    self.stream.ask_message_with_args(Methods::Set.into(), &kv).await
  }

  pub async fn get<K: Into<String>>(&self, key: K) -> KongResult<Value> {
    let v: prost_types::Value = self.stream.ask_message_with_args(
      Methods::Get.into(),
      &kong_rs_protos::String { v: key.into() }
    ).await?;

    Ok(v.kind.map(Into::into).unwrap_or(Value::Null))
  }
}