cratestack_sqlx/query/batch/
create.rs

1//! `batch_create` driver — opens the outer tx, fans the inputs out to
2//! [`super::create_item::run_create_item`] (one savepoint per input),
3//! commits, drains the outbox.
4
5use cratestack_core::{BatchResponse, CoolContext, CoolError, ModelEventKind};
6
7use crate::audit::{dispatch_audit_sink, ensure_audit_table};
8use crate::descriptor::ensure_event_outbox_table;
9use crate::{CreateModelInput, ModelDescriptor, SqlxRuntime, cool_error_from_sqlx, sqlx};
10
11use super::create_item::run_create_item;
12use super::validate::validate_batch_size;
13
14#[derive(Debug, Clone)]
15pub struct BatchCreate<'a, M: 'static, PK: 'static, I> {
16    pub(crate) runtime: &'a SqlxRuntime,
17    pub(crate) descriptor: &'static ModelDescriptor<M, PK>,
18    pub(crate) inputs: Vec<I>,
19}
20
21impl<'a, M: 'static, PK: 'static, I> BatchCreate<'a, M, PK, I>
22where
23    I: CreateModelInput<M> + Send,
24{
25    pub async fn run(self, ctx: &CoolContext) -> Result<BatchResponse<M>, CoolError>
26    where
27        for<'r> M: Send + Unpin + sqlx::FromRow<'r, sqlx::postgres::PgRow> + serde::Serialize,
28    {
29        validate_batch_size(self.inputs.len())?;
30        // No PK dedup — `CreateModelInput` doesn't expose the PK
31        // generically (and server-generated PKs make duplicates
32        // impossible). Client-supplied PK collisions trip the DB
33        // uniqueness constraint and surface as `CoolError::Database`.
34        // The right primitive for idempotent client-PK ingestion is
35        // `.batch_upsert(...)`.
36        if self.inputs.is_empty() {
37            return Ok(BatchResponse::from_results(vec![]));
38        }
39
40        let emits_event = self.descriptor.emits(ModelEventKind::Created);
41        let audit_enabled = self.descriptor.audit_enabled;
42
43        let mut tx = self
44            .runtime
45            .pool()
46            .begin()
47            .await
48            .map_err(cool_error_from_sqlx)?;
49        if emits_event {
50            ensure_event_outbox_table(&mut *tx).await?;
51        }
52        if audit_enabled {
53            ensure_audit_table(self.runtime).await?;
54        }
55
56        let mut per_item: Vec<Result<M, CoolError>> = Vec::with_capacity(self.inputs.len());
57        let mut audit_events = Vec::new();
58        for input in self.inputs {
59            let (outcome, audit_event) = run_create_item(
60                &mut tx,
61                self.runtime.pool(),
62                self.descriptor,
63                input,
64                ctx,
65                emits_event,
66                audit_enabled,
67            )
68            .await?;
69            per_item.push(outcome);
70            audit_events.extend(audit_event);
71        }
72
73        tx.commit().await.map_err(cool_error_from_sqlx)?;
74
75        if emits_event {
76            let _ = self.runtime.drain_event_outbox().await;
77        }
78        dispatch_audit_sink(self.runtime, &audit_events).await;
79
80        Ok(BatchResponse::from_results(per_item))
81    }
82}
cratestack_sqlx/query/batch/create.rs

cratestack_sqlx/query/batch/
create.rs
