rapiddb-web 0.1.32

A reasonably fast configurable embedded key-value sensor database
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143

use crate::api::helpers::with_db;
use rapiddb::traits::IAsyncDatabase;

use warp::{Filter, Rejection, Reply};

/// GET /api/v0/:String
pub fn get(
  db: std::sync::Arc<tokio::sync::RwLock<impl IAsyncDatabase + ?Sized>>,
) -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
  warp::path!("api" / "v0" / String)
    .and(warp::get())
    .and(with_db(db))
    .and_then(_get)
}

pub async fn _get(
  id: String,
  db: std::sync::Arc<tokio::sync::RwLock<impl IAsyncDatabase + ?Sized>>,
) -> Result<impl warp::Reply, std::convert::Infallible> {
  if db.write().await.contains(&id).await {
    return Ok(warp::hyper::Response::builder()
        .status(warp::http::StatusCode::OK)
        .body(format!("{}", &serde_json::json!({
          "resources": [
            {"endpoint": format!("/api/v0/{id}/latest"), "description": format!("GET latest measurment from {id}")},
            {"endpoint": format!("/api/v0/{id}/latest/:count"), "description": format!("GET latest :count measurments from {id}")},
            {"endpoint": format!("/api/v0/{id}/:id"), "description": format!("GET measurment by id from {id}")},
            {"endpoint": format!("/api/v0/{id}/:start/:end"), "description": format!("GET measurment by id in range :start to :end from {id}")},
            {"endpoint": format!("/api/v0/{id}/meta"), "description": format!("GET metadata from {id}")},
            {"endpoint": format!("/api/v0/{id}/aggregates"), "description": format!("GET aggregates from {id}")},
            {"endpoint": format!("/api/v0/{id}"), "description": format!("POST data to {id}")},
            {"endpoint": format!("/api/v0/{id}/meta"), "description": format!("POST metadata to {id}")},
          ],
          "description": format!("Discover resources available under {id}"),
        }))));
  }

  Ok(
    warp::hyper::Response::builder()
      .status(warp::http::StatusCode::NOT_FOUND)
      .body(Default::default()),
  )
}

/// POST /api/v0/:String
pub fn post(
  db: std::sync::Arc<tokio::sync::RwLock<impl IAsyncDatabase + ?Sized>>,
) -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
  warp::path!("api" / "v0" / String)
    .and(warp::post())
    .and(warp::body::content_length_limit(1024 * 16))
    .and(warp::body::bytes())
    .and(with_db(db))
    .and_then(_post)
}

/// POST /api/v0/:String
pub async fn _post(
  id: String,
  data: warp::hyper::body::Bytes,
  db: std::sync::Arc<tokio::sync::RwLock<impl IAsyncDatabase + ?Sized>>,
) -> Result<impl warp::Reply, std::convert::Infallible> {
  db.write().await.post(&id, &data).await;

  Ok(
    warp::hyper::Response::builder()
      .status(warp::http::StatusCode::ACCEPTED)
      .body(""),
  )
}

#[tokio::test]
async fn test_get() {
  let database_test_factory =
    rapiddb::db::DatabaseTestFactory::new(".temp/test/sensor/test_get");

  for db in database_test_factory.get_instance().values() {
    let api = super::endpoints((*db).clone());

    let id = "test-0";

    let resp = warp::test::request()
      .method("GET")
      .path(&format!("/api/v0/{id}"))
      .reply(&api)
      .await;
    assert_eq!(resp.status(), 404);

    db.write()
      .await
      .post(id, serde_json::json!({ "id": &id }).to_string().as_bytes())
      .await;

    let resp = warp::test::request()
      .method("GET")
      .path(&format!("/api/v0/{id}"))
      .reply(&api)
      .await;
    assert_eq!(resp.status(), 200);
  }
}

#[tokio::test]
async fn test_post() {
  let database_test_factory =
    rapiddb::db::DatabaseTestFactory::new(".temp/test/sensor/test_post");

  for db in database_test_factory.get_instance().values() {
    let api = super::endpoints((*db).clone());

    let id = "test-0";

    let resp = warp::test::request()
      .method("POST")
      .path(&format!("/api/v0/{id}"))
      .reply(&api)
      .await;
    assert_eq!(resp.status(), 411);

    let resp = warp::test::request()
      .method("POST")
      .body(id)
      .path(&format!("/api/v0/{id}"))
      .reply(&api)
      .await;
    assert_eq!(resp.status(), 202);

    let resp = warp::test::request()
      .method("POST")
      .json(&serde_json::json!({ "id": &id }))
      .path(&format!("/api/v0/{id}"))
      .reply(&api)
      .await;
    assert_eq!(resp.status(), 202);
    assert_eq!(resp.body().len(), 0);

    let id_db = db.write().await.get_latest(id).await;
    assert_eq!(
      id_db,
      serde_json::json!({ "id": &id }).to_string().as_bytes().to_vec()
    );
  }
}