obzenflow_core 0.2.4

Core domain layer for ObzenFlow - pure abstractions with minimal dependencies
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

// SPDX-License-Identifier: MIT OR Apache-2.0
// SPDX-FileCopyrightText: 2025-2026 ObzenFlow Contributors
// https://obzenflow.dev

//! Existing durable composite protocol records. The event descriptor selects the
//! closed protocol shape; application values nested inside it remain untouched.

use crate::ai::{
    AiMapReduceChunkFailed, AiMapReduceFinaliseFailed, AiMapReduceJobFailed, AiMapReduceMapInput,
    AiMapReducePlanningFailed, AiMapReducePlanningManifest, AiMapReduceReduceInput,
    AiMapReduceTaggedPartial, ChunkEnvelope,
};
use serde::{Serialize, Serializer};
use serde_json::Value;

#[derive(Debug, Clone)]
pub enum CompositeDataPayload {
    MapInput(AiMapReduceMapInput<ChunkEnvelope<Value>>),
    PlanningManifest(AiMapReducePlanningManifest),
    TaggedPartial(AiMapReduceTaggedPartial<Value>),
    ReduceInput(AiMapReduceReduceInput<Value, Value>),
    PlanningFailed(AiMapReducePlanningFailed),
    ChunkFailed(AiMapReduceChunkFailed),
    JobFailed(AiMapReduceJobFailed),
    FinaliseFailed(AiMapReduceFinaliseFailed),
}

impl CompositeDataPayload {
    pub const fn event_type(&self) -> &'static str {
        match self {
            Self::MapInput(_) => "ai.map_reduce.map_input.v1",
            Self::PlanningManifest(_) => "ai.map_reduce.planning_manifest.v1",
            Self::TaggedPartial(_) => "ai.map_reduce.tagged_partial.v1",
            Self::ReduceInput(_) => "ai.map_reduce.reduce_input.v2",
            Self::PlanningFailed(_) => "ai.map_reduce.planning_failed.v1",
            Self::ChunkFailed(_) => "ai.map_reduce.chunk_failed.v2",
            Self::JobFailed(_) => "ai.map_reduce.job_failed.v1",
            Self::FinaliseFailed(_) => "ai.map_reduce.finalise_failed.v1",
        }
    }

    pub fn decode(event_type: &str, payload: Value) -> Result<Self, serde_json::Error> {
        match event_type {
            "ai.map_reduce.map_input.v1" => serde_json::from_value(payload).map(Self::MapInput),
            "ai.map_reduce.planning_manifest.v1" => {
                serde_json::from_value(payload).map(Self::PlanningManifest)
            }
            "ai.map_reduce.tagged_partial.v1" => {
                serde_json::from_value(payload).map(Self::TaggedPartial)
            }
            "ai.map_reduce.reduce_input.v2" => {
                serde_json::from_value(payload).map(Self::ReduceInput)
            }
            "ai.map_reduce.planning_failed.v1" => {
                serde_json::from_value(payload).map(Self::PlanningFailed)
            }
            "ai.map_reduce.chunk_failed.v2" => {
                serde_json::from_value(payload).map(Self::ChunkFailed)
            }
            "ai.map_reduce.job_failed.v1" => serde_json::from_value(payload).map(Self::JobFailed),
            "ai.map_reduce.finalise_failed.v1" => {
                serde_json::from_value(payload).map(Self::FinaliseFailed)
            }
            _ => Err(<serde_json::Error as serde::de::Error>::custom(
                "event descriptor does not match payload",
            )),
        }
    }
}

impl Serialize for CompositeDataPayload {
    fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
        match self {
            Self::MapInput(payload) => payload.serialize(serializer),
            Self::PlanningManifest(payload) => payload.serialize(serializer),
            Self::TaggedPartial(payload) => payload.serialize(serializer),
            Self::ReduceInput(payload) => payload.serialize(serializer),
            Self::PlanningFailed(payload) => payload.serialize(serializer),
            Self::ChunkFailed(payload) => payload.serialize(serializer),
            Self::JobFailed(payload) => payload.serialize(serializer),
            Self::FinaliseFailed(payload) => payload.serialize(serializer),
        }
    }
}