land-runtime 0.0.4

Database migration for Runtime Land
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

use hyper::body::{Body, Sender};
use std::collections::HashMap;
use std::sync::atomic::AtomicU32;

mod guest;
pub use guest::exports::land::http::http_incoming::{Request, Response};
pub use guest::HttpHandler;

mod host;
pub use host::HttpService;

pub mod http_body;
pub mod http_outgoing;
pub mod http_types;

pub struct HttpContext {
    /// req_id set related request id from main request
    pub req_id: String,
    /// counter is used to count fetch times, limit 10. avoid creating too many requests
    pub counter: u16,
    /// body hash map
    body_map: HashMap<u32, Body>,
    /// body sender
    body_sender_map: HashMap<u32, Sender>,
    /// atomic increment id for body
    body_id: AtomicU32,
}

impl HttpContext {
    pub fn new(req_id: String) -> Self {
        HttpContext {
            req_id,
            counter: 10,
            body_map: HashMap::new(),
            body_sender_map: HashMap::new(),
            body_id: AtomicU32::new(1),
        }
    }
    pub fn set_body(&mut self, body: Body) -> u32 {
        let id = self
            .body_id
            .fetch_add(1, std::sync::atomic::Ordering::SeqCst);
        self.body_map.insert(id, body);
        id
    }

    pub fn replace_body(&mut self, id: u32, body: Body) -> Option<Body> {
        self.body_map.insert(id, body)
    }

    pub fn take_body(&mut self, id: u32) -> Option<Body> {
        self.body_map.remove(&id)
    }

    fn set_body_sender(&mut self, id: u32, sender: Sender) {
        self.body_sender_map.insert(id, sender);
    }
}