artisan 0.10.0

Api RequesT Framework U Like - 你喜欢的 Rust API 请求框架
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

//! 解析响应插件
//!
//! 执行 HTTP 请求并解析响应。
//!
//! # 行为
//!
//! - 检查 rocket.config.direction，决定是否发起请求
//! - 执行 HTTP 请求，存入 rocket.destination_origin
//! - 根据 DirectionKind 解析响应
//! - 结果存入 rocket.destination

use async_trait::async_trait;

use crate::Rocket;
use crate::direction::{Destination, Direction, DirectionKind};
use crate::directions::JsonDirection;
use crate::error::ArtfulError;
use crate::flow_ctrl::Next;
use crate::http::get_client;
use crate::plugin::Plugin;

/// 解析响应插件
#[derive(Clone, Copy, Debug, Default)]
pub struct ParserPlugin;

#[async_trait]
impl Plugin for ParserPlugin {
    async fn assembly(&self, rocket: &mut Rocket, next: Next<'_>) -> crate::Result<()> {
        // NoRequest - 不发起请求，直接调用下一层
        if let DirectionKind::NoRequest = rocket.config.direction {
            return next.call(rocket).await;
        }

        // 发送 HTTP 请求
        rocket.destination_origin = Some(
            get_client()
                .execute(rocket.radar.take().ok_or(ArtfulError::MissingRequest)?)
                .await
                .map_err(ArtfulError::RequestFailed)?,
        );

        // 解析响应
        let destination = match &rocket.config.direction {
            DirectionKind::Json => JsonDirection.parse(rocket).await?,
            DirectionKind::Response => rocket
                .destination_origin
                .take()
                .map(Destination::Response)
                .ok_or(ArtfulError::MissingResponse)?,
            DirectionKind::Custom(direction) => direction.clone().parse(rocket).await?,
            DirectionKind::NoRequest => Destination::None,
        };

        rocket.destination = Some(destination);

        next.call(rocket).await
    }
}