dex-sdk 1.38.0

Strongly typed Rust SDK for Dex
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

// Portions of this file are derived from indeedeng/iwf-java-sdk.
// Those portions are licensed under the Apache License, Version 2.0.
// See LICENSES/Apache-2.0.txt and LEGACY_NOTICES.md.
//
// Modifications Copyright (c) 2026 Super Durable, Inc.
//
// Modifications are licensed under the Sustainable Use License 1.0.
// Third-Party Materials remain under the Apache License, Version 2.0.
// See LICENSE and LEGACY_NOTICES.md.

use std::sync::LazyLock;

use dex_sdk::{
    Attribute, Channel, Context, Flow, HandlerError, HandlerResult, PersistenceSchema, Step,
    StepDecision, StepList, StepMovement, StepOptions, Wait,
};

pub(crate) static WAIT_FOR_COUNT: LazyLock<Attribute<i32>> =
    LazyLock::new(|| Attribute::new("step-lock-wait-for-count"));
pub(crate) static EXECUTE_COUNT: LazyLock<Attribute<i32>> =
    LazyLock::new(|| Attribute::new("step-lock-execute-count"));
static COMPLETED: LazyLock<Channel<()>> = LazyLock::new(|| Channel::new("step-lock-completed"));

pub(crate) struct StateOptionsLockingWorkflow {
    start: StartStep,
    locked: LockedStep,
    complete: CompleteStep,
}

impl StateOptionsLockingWorkflow {
    pub(crate) fn new() -> Self {
        Self {
            start: StartStep,
            locked: LockedStep,
            complete: CompleteStep,
        }
    }
}

impl Flow for StateOptionsLockingWorkflow {
    type StartInput = i32;

    fn steps(&self) -> StepList<'_, Self::StartInput> {
        StepList::start(&self.start)
            .and(&self.locked)
            .and(&self.complete)
    }

    fn persistence(&self) -> PersistenceSchema {
        PersistenceSchema::new()
            .attribute(&WAIT_FOR_COUNT)
            .attribute(&EXECUTE_COUNT)
            .channel(&COMPLETED)
    }
}

struct StartStep;

impl Step for StartStep {
    type Input = i32;

    fn execute(&self, _context: &mut Context, parallelism: i32) -> HandlerResult<StepDecision> {
        let mut movements = (0..parallelism)
            .map(|index| StepMovement::to(&LockedStep, index))
            .collect::<Vec<_>>();
        movements.push(StepMovement::to(&CompleteStep, parallelism));
        Ok(StepDecision::go_to_many(movements))
    }
}

struct LockedStep;

impl Step for LockedStep {
    type Input = i32;

    fn wait_for(&self, context: &mut Context, _input: i32) -> HandlerResult<Wait> {
        let next = WAIT_FOR_COUNT.get(context)?.unwrap_or_default() + 1;
        WAIT_FOR_COUNT.set(context, next)?;
        Ok(Wait::skip_immediately())
    }

    fn execute(&self, context: &mut Context, _input: i32) -> HandlerResult<StepDecision> {
        let next = EXECUTE_COUNT.get(context)?.unwrap_or_default() + 1;
        EXECUTE_COUNT.set(context, next)?;
        COMPLETED.publish(context, ())?;
        Ok(StepDecision::dead_end())
    }

    fn options(&self) -> StepOptions<Self::Input> {
        StepOptions::new()
            .wait_for_lock(WAIT_FOR_COUNT.lock())
            .execute_lock(EXECUTE_COUNT.lock())
    }
}

struct CompleteStep;

impl Step for CompleteStep {
    type Input = i32;

    fn wait_for(&self, _context: &mut Context, parallelism: i32) -> HandlerResult<Wait> {
        Ok(Wait::until(COMPLETED.for_n(parallelism as usize)))
    }

    fn execute(&self, context: &mut Context, parallelism: i32) -> HandlerResult<StepDecision> {
        if COMPLETED.condition_results(context)?.len() != parallelism as usize {
            return Err(HandlerError::new(
                "StateOptionsLockingFailure",
                "not all locked Steps completed",
            ));
        }
        Ok(StepDecision::graceful_complete(format!(
            "{}:{}",
            WAIT_FOR_COUNT.get_required(context)?,
            EXECUTE_COUNT.get_required(context)?
        )))
    }
}