moduforge-core 0.7.0

moduforge 核心模块
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

use std::{sync::Arc, time::Duration};

use crate::{
    runtime::sync_processor::{
        ProcessorError, SyncProcessor, TaskProcessor, TaskResult,
    },
    types::{ProcessorResult, TaskParams, TransactionStatus},
    ForgeResult,
};
use async_trait::async_trait;

/// 事务处理器
pub struct TransactionProcessor;

#[async_trait]
impl TaskProcessor<TaskParams, ProcessorResult> for TransactionProcessor {
    async fn process(
        &self,
        (state, tr): TaskParams,
    ) -> std::result::Result<ProcessorResult, ProcessorError> {
        match state.apply(tr).await {
            Ok(result) => Ok(ProcessorResult {
                status: TransactionStatus::Completed,
                error: None,
                result: Some(result),
            }),
            Err(e) => Ok(ProcessorResult {
                status: TransactionStatus::Failed(e.to_string()),
                error: None,
                result: None,
            }),
        }
    }
}

#[derive(Clone)]
pub struct FlowEngine {
    processor:
        Arc<SyncProcessor<TaskParams, ProcessorResult, TransactionProcessor>>,
}

impl FlowEngine {
    pub fn new() -> ForgeResult<Self> {
        let processor =
            SyncProcessor::new(TransactionProcessor, 3, Duration::from_secs(1));
        Ok(Self { processor: Arc::new(processor) })
    }

    pub async fn submit(
        &self,
        params: TaskParams,
    ) -> TaskResult<TaskParams, ProcessorResult> {
        self.processor.process_task(params).await
    }
}