pdk-core 1.8.0

PDK Core
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

// Copyright (c) 2026, Salesforce, Inc.,
// All rights reserved.
// For full license text, see the LICENSE.txt file

//! Middleware to fix missing metadata propagation in the filter context.

use crate::policy_context::metadata::PolicyMetadata;
use crate::policy_context::static_policy_context_cache::StaticPolicyContextCache;
use classy::proxy_wasm::traits::{Context, HttpContext};
use classy::proxy_wasm::types::Action;
use std::rc::Rc;

pub struct HttpContextAdapter {
    http_context: Box<dyn HttpContext>,
    policy_metadata: Rc<PolicyMetadata>,
    plugin_name_api_id: Rc<String>,
}

impl HttpContextAdapter {
    pub fn new(
        http_context: Box<dyn HttpContext>,
        policy_metadata: Rc<PolicyMetadata>,
        plugin_name_api_id: Rc<String>,
    ) -> Self {
        Self {
            http_context,
            policy_metadata,
            plugin_name_api_id,
        }
    }

    pub fn boxed(self) -> Box<dyn HttpContext> {
        Box::new(self)
    }

    fn fix_current_context(&self) {
        StaticPolicyContextCache::fix_metadata(&self.policy_metadata);
        StaticPolicyContextCache::fix_plugin_name_api_id(&self.plugin_name_api_id);
    }
}

impl Context for HttpContextAdapter {
    fn on_http_call_response(
        &mut self,
        token_id: u32,
        num_headers: usize,
        body_size: usize,
        num_trailers: usize,
    ) {
        self.fix_current_context();
        self.http_context
            .on_http_call_response(token_id, num_headers, body_size, num_trailers)
    }

    fn on_grpc_call_response(&mut self, token_id: u32, status_code: u32, response_size: usize) {
        self.fix_current_context();
        self.http_context
            .on_grpc_call_response(token_id, status_code, response_size)
    }

    fn on_done(&mut self) -> bool {
        self.fix_current_context();
        self.http_context.on_done()
    }
}

impl HttpContext for HttpContextAdapter {
    fn on_http_request_headers(&mut self, num_headers: usize, end_of_stream: bool) -> Action {
        self.fix_current_context();
        self.http_context
            .on_http_request_headers(num_headers, end_of_stream)
    }

    fn on_http_request_body(&mut self, body_size: usize, end_of_stream: bool) -> Action {
        self.fix_current_context();
        self.http_context
            .on_http_request_body(body_size, end_of_stream)
    }

    fn on_http_response_body(&mut self, body_size: usize, end_of_stream: bool) -> Action {
        self.fix_current_context();
        self.http_context
            .on_http_response_body(body_size, end_of_stream)
    }

    fn on_http_response_headers(&mut self, num_headers: usize, end_of_stream: bool) -> Action {
        self.fix_current_context();
        self.http_context
            .on_http_response_headers(num_headers, end_of_stream)
    }
}