#![deny(missing_docs)]
use std::marker::PhantomData;
use std::sync::Arc;
use type_bridge_contract::id::is_canonical_thing_iid;
use type_bridge_orm::_manager::DynamicEntityManager;
use type_bridge_orm::session::backend::TxType;
use type_bridge_orm::{
ProjectedBatchOperation, ProjectedBatchRow, ProjectedCrudExecutor, ProjectedManagerComparison,
};
use crate::__codegen::{
CompleteModel, EncodedCreate, EntityModel, FieldToken, HydrationCapability, IntoEncodedCreate,
Model, QueryValued, SubtypeRootModel,
};
use crate::entity_codec::{
hydrate_entity, lower_entity_create, map_validation_error, resolve_discovered_entity,
resolve_entity_authority,
};
use crate::error::{Error, ModelValidationPhase};
use crate::hooks::{CrudOperation, HookRunner, LifecycleHook, ModelKind};
use crate::projected_batch::{
checked_batch_ordinal, create_rows, delete_rows, encode_batch_create, execute_owned_delete,
execute_owned_things, prepare_batch, project_encoded_batch_create, reserved_binding_vec,
uses_successor_batch_runtime, validate_binding_row_count,
};
use crate::projected_codec::{materialize_projected, project_create};
use crate::projected_filter::ProjectedEntityFilter;
use crate::schema::Schema;
use crate::{Database, Result};
#[cfg(test)]
mod tests;
fn invalid_iid() -> Error {
Error::model_validation(
ModelValidationPhase::Input,
"invalid_iid",
vec!["iid".into()],
"IID is not canonical",
None,
)
}
fn schema_not_bound() -> Error {
Error::model_validation(
ModelValidationPhase::Input,
"schema_not_bound",
vec![],
"database is not schema-bound",
None,
)
}
pub(crate) async fn rehydrate_written_entity<M>(
manager: &DynamicEntityManager<'_>,
iid: &str,
id: &type_bridge_contract::id::TypeId,
installed: &type_bridge_orm::InstalledRuntimeProjection,
) -> Result<M>
where
M: crate::__codegen::CompleteModel,
{
let row = manager
.get_by_iid_exact(iid)
.await
.map_err(Error::from_orm)?
.ok_or_else(|| {
Error::model_validation(
ModelValidationPhase::Hydration,
"missing_post_write_row",
vec!["iid".into()],
"written entity was not returned",
None,
)
})?;
let hydrated = hydrate_entity(row, id, installed)?;
M::materialize(&hydrated, &HydrationCapability::new())
.map_err(|error| map_validation_error(error, ModelValidationPhase::Hydration))
}
impl<S, M> EntitySubtypeManager<'_, S, M>
where
S: Schema,
M: SubtypeRootModel<Schema = S> + EntityModel<Schema = S>,
{
pub async fn get_by_iid(&self, _iid: &str) -> Result<Option<M::Subtypes>> {
if !is_canonical_thing_iid(_iid) {
return Err(invalid_iid());
}
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (_id, descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
false,
)?;
let tx = self
.db
.inner_orm()
.transaction_context(TxType::Read)
.await
.map_err(Error::from_orm)?;
let manager =
DynamicEntityManager::with_canonical_transaction(tx.clone(), Arc::new(descriptor));
let identity = match manager.discover_by_iid(_iid).await {
Ok(v) => v,
Err(e) => {
let _ = tx.close().await;
return Err(Error::from_orm(e));
}
};
let out = match identity {
None => None,
Some(identity) => {
let (child_id, child_descriptor) =
match resolve_discovered_entity(&identity.type_name, installed) {
Ok(v) => v,
Err(e) => {
let _ = tx.close().await;
return Err(e);
}
};
let child = DynamicEntityManager::with_canonical_transaction(
tx.clone(),
Arc::new(child_descriptor),
);
let row = match child.get_by_iid_exact(&identity.iid).await {
Ok(Some(v)) => v,
Ok(None) => {
let _ = tx.close().await;
return Err(Error::model_validation(
ModelValidationPhase::Hydration,
"missing_concrete_row",
vec!["iid".into()],
"discovered entity row is missing",
None,
));
}
Err(e) => {
let _ = tx.close().await;
return Err(Error::from_orm(e));
}
};
let h = match hydrate_entity(row, &child_id, installed) {
Ok(v) => v,
Err(e) => {
let _ = tx.close().await;
return Err(e);
}
};
Some(
match M::__tb_dispatch_subtype(&h, &HydrationCapability::new()) {
Ok(v) => v,
Err(e) => {
let _ = tx.close().await;
return Err(map_validation_error(e, ModelValidationPhase::Hydration));
}
},
)
}
};
tx.close().await.map_err(Error::from_orm)?;
Ok(out)
}
pub async fn all(&self) -> Result<Vec<M::Subtypes>> {
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (_id, descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
false,
)?;
let tx = self
.db
.inner_orm()
.transaction_context(TxType::Read)
.await
.map_err(Error::from_orm)?;
let manager =
DynamicEntityManager::with_canonical_transaction(tx.clone(), Arc::new(descriptor));
let identities = match manager.discover_all().await {
Ok(v) => v,
Err(e) => {
let _ = tx.close().await;
return Err(Error::from_orm(e));
}
};
let mut out = Vec::with_capacity(identities.len());
for identity in identities {
let type_json = identity.type_name;
let (child_id, child_descriptor) =
match resolve_discovered_entity(&type_json, installed) {
Ok(v) => v,
Err(e) => {
let _ = tx.close().await;
return Err(e);
}
};
let child = DynamicEntityManager::with_canonical_transaction(
tx.clone(),
Arc::new(child_descriptor),
);
let row = match child.get_by_iid_exact(&identity.iid).await {
Ok(Some(v)) => v,
Ok(None) => {
let _ = tx.close().await;
return Err(Error::model_validation(
ModelValidationPhase::Hydration,
"missing_concrete_row",
vec!["iid".into()],
"discovered entity row is missing",
None,
));
}
Err(e) => {
let _ = tx.close().await;
return Err(Error::from_orm(e));
}
};
let h = match hydrate_entity(row, &child_id, installed) {
Ok(v) => v,
Err(e) => {
let _ = tx.close().await;
return Err(e);
}
};
out.push(
match M::__tb_dispatch_subtype(&h, &HydrationCapability::new()) {
Ok(v) => v,
Err(e) => {
let _ = tx.close().await;
return Err(map_validation_error(e, ModelValidationPhase::Hydration));
}
},
);
}
tx.close().await.map_err(Error::from_orm)?;
Ok(out)
}
pub async fn count(&self) -> Result<u64> {
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (_id, descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
false,
)?;
DynamicEntityManager::new_canonical(self.db.inner_orm(), Arc::new(descriptor))
.count()
.await
.map_err(Error::from_orm)
}
}
pub struct EntityManager<'db, S: Schema, M: EntityModel<Schema = S>> {
db: &'db Database<S>,
hooks: HookRunner,
marker: PhantomData<M>,
}
pub struct EntitySubtypeManager<
'db,
S: Schema,
M: SubtypeRootModel<Schema = S> + EntityModel<Schema = S>,
> {
db: &'db Database<S>,
marker: PhantomData<M>,
}
impl<'db, S: Schema, M: SubtypeRootModel<Schema = S> + EntityModel<Schema = S>>
EntitySubtypeManager<'db, S, M>
{
pub(crate) fn new(db: &'db Database<S>) -> Self {
Self {
db,
marker: PhantomData,
}
}
}
impl<'db, S, M> EntityManager<'db, S, M>
where
S: Schema,
M: SubtypeRootModel<Schema = S> + EntityModel<Schema = S>,
{
pub fn subtypes(&self) -> EntitySubtypeManager<'db, S, M> {
EntitySubtypeManager::new(self.db)
}
}
impl<'db, S: Schema, M: EntityModel<Schema = S>> Clone for EntityManager<'db, S, M> {
fn clone(&self) -> Self {
Self {
db: self.db,
hooks: self.hooks.clone(),
marker: PhantomData,
}
}
}
impl<S: Schema, M: EntityModel<Schema = S>> EntityManager<'_, S, M> {
pub(crate) fn new(db: &Database<S>) -> EntityManager<'_, S, M> {
EntityManager {
db,
hooks: HookRunner::default(),
marker: PhantomData,
}
}
}
impl<'db, S, M> EntityManager<'db, S, M>
where
S: Schema,
M: EntityModel<Schema = S> + CompleteModel,
{
pub fn filter(&self) -> Result<ProjectedEntityFilter<'db, S, M>> {
ProjectedEntityFilter::for_database(self.db)
}
pub fn where_<V>(
&self,
field: FieldToken<M, V>,
operator: ProjectedManagerComparison,
value: &V,
) -> Result<ProjectedEntityFilter<'db, S, M>>
where
V: Model<Schema = S> + QueryValued,
{
self.filter()?.where_(field, operator, value)
}
fn successor_batches_enabled(&self) -> bool {
self.db
.installed_schema()
.is_some_and(|installed| uses_successor_batch_runtime(installed))
}
pub fn add_hook(&mut self, hook: Arc<dyn LifecycleHook>) -> &mut Self {
self.hooks.add(hook);
self
}
fn encode_hook_input(input: &M::Create) -> Result<EncodedCreate> {
input
.clone()
.into_encoded_create()
.map_err(|error| map_validation_error(error, ModelValidationPhase::Input))
}
pub async fn insert(&self, input: M::Create) -> Result<M> {
if !self.hooks.has_hooks() {
return self
.projected_write(input, CrudOperation::Insert, None)
.await;
}
let encoded = Self::encode_hook_input(&input)?;
let metadata = self
.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Insert,
None,
Some(&encoded),
)
.await?;
let output = self
.projected_write(input, CrudOperation::Insert, None)
.await?;
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Insert,
Some(output.iid()),
Some(&encoded),
metadata,
)
.await;
Ok(output)
}
pub async fn put(&self, input: M::Create) -> Result<M> {
if !self.hooks.has_hooks() {
return self.projected_write(input, CrudOperation::Put, None).await;
}
let encoded = Self::encode_hook_input(&input)?;
let metadata = self
.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Put,
None,
Some(&encoded),
)
.await?;
let output = self
.projected_write(input, CrudOperation::Put, None)
.await?;
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Put,
Some(output.iid()),
Some(&encoded),
metadata,
)
.await;
Ok(output)
}
pub async fn insert_many(&self, inputs: Vec<M::Create>) -> Result<Vec<M>> {
let successor = self.successor_batches_enabled();
if successor {
validate_binding_row_count(inputs.len())?;
}
if inputs.is_empty() && !successor {
return Ok(Vec::new());
}
if !self.hooks.has_hooks() {
return self.write_many(inputs, false).await;
}
if successor {
return self
.successor_write_many_with_hooks(
inputs,
CrudOperation::Insert,
ProjectedBatchOperation::Insert,
)
.await;
}
let encoded = inputs
.iter()
.map(Self::encode_hook_input)
.collect::<Result<Vec<_>>>()?;
let mut metadata = Vec::with_capacity(inputs.len());
for input in &encoded {
metadata.push(
self.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Insert,
None,
Some(input),
)
.await?,
);
}
let outputs = self.write_many(inputs.clone(), false).await?;
for ((input, output), metadata) in encoded.iter().zip(&outputs).zip(metadata) {
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Insert,
Some(output.iid()),
Some(input),
metadata,
)
.await;
}
Ok(outputs)
}
pub async fn put_many(&self, inputs: Vec<M::Create>) -> Result<Vec<M>> {
let successor = self.successor_batches_enabled();
if successor {
validate_binding_row_count(inputs.len())?;
}
if inputs.is_empty() && !successor {
return Ok(Vec::new());
}
if !self.hooks.has_hooks() {
return self.write_many(inputs, true).await;
}
if successor {
return self
.successor_write_many_with_hooks(
inputs,
CrudOperation::Put,
ProjectedBatchOperation::Put,
)
.await;
}
let encoded = inputs
.iter()
.map(Self::encode_hook_input)
.collect::<Result<Vec<_>>>()?;
let mut metadata = Vec::with_capacity(inputs.len());
for input in &encoded {
metadata.push(
self.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Put,
None,
Some(input),
)
.await?,
);
}
let outputs = self.write_many(inputs.clone(), true).await?;
for ((input, output), metadata) in encoded.iter().zip(&outputs).zip(metadata) {
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Put,
Some(output.iid()),
Some(input),
metadata,
)
.await;
}
Ok(outputs)
}
pub async fn update(&self, iid: &str, input: M::Create) -> Result<M> {
if !is_canonical_thing_iid(iid) {
return Err(invalid_iid());
}
if !self.hooks.has_hooks() {
return self
.projected_write(input, CrudOperation::Update, Some(iid))
.await;
}
let encoded = Self::encode_hook_input(&input)?;
let state = self
.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Update,
Some(iid),
Some(&encoded),
)
.await?;
let output = self
.projected_write(input, CrudOperation::Update, Some(iid))
.await?;
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Update,
Some(output.iid()),
Some(&encoded),
state,
)
.await;
Ok(output)
}
pub async fn delete(&self, iid: &str) -> Result<()> {
if !is_canonical_thing_iid(iid) {
return Err(invalid_iid());
}
if !self.hooks.has_hooks() {
return self.projected_delete(iid).await;
}
let state = self
.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Delete,
Some(iid),
None,
)
.await?;
self.projected_delete(iid).await?;
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Delete,
Some(iid),
None,
state,
)
.await;
Ok(())
}
pub async fn update_many(&self, inputs: Vec<(String, M::Create)>) -> Result<Vec<M>> {
let successor = self.successor_batches_enabled();
if successor {
validate_binding_row_count(inputs.len())?;
}
if inputs.is_empty() && !successor {
return Ok(Vec::new());
}
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
if uses_successor_batch_runtime(installed) {
if !self.hooks.has_hooks() {
let rows = crate::projected_batch::update_rows(installed, &id, inputs)?;
let batch = prepare_batch(installed, id, ProjectedBatchOperation::Update, rows)?;
return execute_owned_things(self.db.inner_orm(), installed, &batch).await;
}
let mut hook_inputs = reserved_binding_vec(inputs.len())?;
let mut rows = reserved_binding_vec(inputs.len())?;
for (ordinal, (iid, input)) in inputs.into_iter().enumerate() {
let encoded = encode_batch_create(input, ordinal)?;
rows.push(ProjectedBatchRow::Update {
iid: iid.clone(),
replacement: project_encoded_batch_create(&encoded, &id, installed, ordinal)?,
});
hook_inputs.push((iid, encoded));
}
let batch = prepare_batch(installed, id, ProjectedBatchOperation::Update, rows)?;
let mut states = reserved_binding_vec(hook_inputs.len())?;
for (ordinal, (iid, encoded)) in hook_inputs.iter().enumerate() {
states.push(
self.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Update,
Some(iid),
Some(encoded),
)
.await
.map_err(|error| {
error.with_projected_batch_row(checked_batch_ordinal(ordinal))
})?,
);
}
let outputs = execute_owned_things::<M>(self.db.inner_orm(), installed, &batch).await?;
for (((_, encoded), output), state) in hook_inputs.iter().zip(&outputs).zip(states) {
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Update,
Some(output.iid()),
Some(encoded),
state,
)
.await;
}
return Ok(outputs);
}
let mut prepared = Vec::with_capacity(inputs.len());
for (iid, input) in inputs {
if !is_canonical_thing_iid(&iid) {
return Err(invalid_iid());
}
let encoded = Self::encode_hook_input(&input)?;
let attributes = lower_entity_create(input, &id, installed)?;
prepared.push((iid, encoded, attributes));
}
let mut states = Vec::new();
if self.hooks.has_hooks() {
states.reserve(prepared.len());
for (iid, encoded, _) in &prepared {
states.push(
self.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Update,
Some(iid),
Some(encoded),
)
.await?,
);
}
}
let tx = self
.db
.inner_orm()
.transaction_context(TxType::Write)
.await
.map_err(Error::from_orm)?;
let manager =
DynamicEntityManager::with_canonical_transaction(tx.clone(), Arc::new(descriptor));
let mut outputs = Vec::with_capacity(prepared.len());
for (iid, _, attributes) in &prepared {
if let Err(error) = manager.update_exact(iid, attributes).await {
let _ = tx.rollback().await;
return Err(Error::from_orm(error));
}
match rehydrate_written_entity::<M>(&manager, iid, &id, installed).await {
Ok(output) => outputs.push(output),
Err(error) => {
let _ = tx.rollback().await;
return Err(error);
}
}
}
tx.commit().await.map_err(Error::from_orm)?;
if self.hooks.has_hooks() {
for (((_, encoded, _), output), state) in prepared.iter().zip(&outputs).zip(states) {
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Update,
Some(output.iid()),
Some(encoded),
state,
)
.await;
}
}
Ok(outputs)
}
pub async fn delete_many(&self, iids: &[String]) -> Result<()> {
let successor = self.successor_batches_enabled();
if successor {
validate_binding_row_count(iids.len())?;
}
if iids.is_empty() && !successor {
return Ok(());
}
if !successor && iids.iter().any(|iid| !is_canonical_thing_iid(iid)) {
return Err(invalid_iid());
}
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
let batch = if successor {
Some(prepare_batch(
installed,
id.clone(),
ProjectedBatchOperation::Delete,
delete_rows(iids)?,
)?)
} else {
None
};
let mut states = if successor && self.hooks.has_hooks() {
reserved_binding_vec(iids.len())?
} else {
Vec::new()
};
if self.hooks.has_hooks() {
if !successor {
states.reserve(iids.len());
}
for (ordinal, iid) in iids.iter().enumerate() {
states.push(
self.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Delete,
Some(iid),
None,
)
.await
.map_err(|error| {
if successor {
error.with_projected_batch_row(checked_batch_ordinal(ordinal))
} else {
error
}
})?,
);
}
}
if uses_successor_batch_runtime(installed) {
execute_owned_delete(
self.db.inner_orm(),
installed,
batch.as_ref().expect("successor batch was prepared"),
)
.await?;
} else {
let tx = self
.db
.inner_orm()
.transaction_context(TxType::Write)
.await
.map_err(Error::from_orm)?;
let manager =
DynamicEntityManager::with_canonical_transaction(tx.clone(), Arc::new(descriptor));
for iid in iids {
if let Err(error) = manager.delete_by_iid_exact(iid).await {
let _ = tx.rollback().await;
return Err(Error::from_orm(error));
}
}
tx.commit().await.map_err(Error::from_orm)?;
}
if self.hooks.has_hooks() {
for (iid, state) in iids.iter().zip(states) {
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
CrudOperation::Delete,
Some(iid),
None,
state,
)
.await;
}
}
Ok(())
}
async fn projected_write(
&self,
input: M::Create,
operation: CrudOperation,
iid: Option<&str>,
) -> Result<M> {
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, _descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
let create = project_create(input, &id, installed)?;
let tx = self
.db
.inner_orm()
.transaction_context(TxType::Write)
.await
.map_err(Error::from_orm)?;
let executor = ProjectedCrudExecutor::new(installed);
let projected = match operation {
CrudOperation::Insert => {
executor
.insert_entity_in_transaction_with_compatibility(&tx, &create)
.await
}
CrudOperation::Put => {
executor
.put_entity_in_transaction_with_compatibility(&tx, &create)
.await
}
CrudOperation::Update => {
executor
.update_entity_in_transaction_with_compatibility(
&tx,
iid.expect("projected entity update requires a checked IID"),
&create,
)
.await
}
CrudOperation::Delete => unreachable!("delete has no generated create payload"),
};
let projected = match projected {
Ok(value) => value,
Err(error) => {
let mapped = Error::from_projected_crud(error, ModelKind::Entity, Some(operation));
let _ = tx.rollback().await;
return Err(mapped);
}
};
let value = match materialize_projected(projected, installed) {
Ok(value) => value,
Err(error) => {
let _ = tx.rollback().await;
return Err(error);
}
};
tx.commit_classified()
.await
.map_err(|error| Error::from_orm(error.into_orm_error()))?;
Ok(value)
}
async fn projected_delete(&self, iid: &str) -> Result<()> {
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, _descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
let tx = self
.db
.inner_orm()
.transaction_context(TxType::Write)
.await
.map_err(Error::from_orm)?;
let result = ProjectedCrudExecutor::new(installed)
.delete_entity_by_iid_in_transaction_with_compatibility(&tx, &id, iid)
.await;
if let Err(error) = result {
let mapped =
Error::from_projected_crud(error, ModelKind::Entity, Some(CrudOperation::Delete));
let _ = tx.rollback().await;
return Err(mapped);
}
tx.commit_classified()
.await
.map_err(|error| Error::from_orm(error.into_orm_error()))
}
async fn successor_write_many_with_hooks(
&self,
inputs: Vec<M::Create>,
crud_operation: CrudOperation,
batch_operation: ProjectedBatchOperation,
) -> Result<Vec<M>> {
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, _descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
let mut encoded = reserved_binding_vec(inputs.len())?;
let mut rows = reserved_binding_vec(inputs.len())?;
for (ordinal, input) in inputs.into_iter().enumerate() {
let input = encode_batch_create(input, ordinal)?;
rows.push(ProjectedBatchRow::Create(project_encoded_batch_create(
&input, &id, installed, ordinal,
)?));
encoded.push(input);
}
let batch = prepare_batch(installed, id, batch_operation, rows)?;
let mut metadata = reserved_binding_vec(encoded.len())?;
for (ordinal, input) in encoded.iter().enumerate() {
metadata.push(
self.hooks
.run_pre(
M::TYPE_ID_JSON,
ModelKind::Entity,
crud_operation,
None,
Some(input),
)
.await
.map_err(|error| {
error.with_projected_batch_row(checked_batch_ordinal(ordinal))
})?,
);
}
let outputs = execute_owned_things::<M>(self.db.inner_orm(), installed, &batch).await?;
for ((input, output), metadata) in encoded.iter().zip(&outputs).zip(metadata) {
self.hooks
.run_post(
M::TYPE_ID_JSON,
ModelKind::Entity,
crud_operation,
Some(output.iid()),
Some(input),
metadata,
)
.await;
}
Ok(outputs)
}
async fn write_many(&self, inputs: Vec<M::Create>, put: bool) -> Result<Vec<M>> {
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
if uses_successor_batch_runtime(installed) {
let rows = create_rows(installed, &id, inputs)?;
let operation = if put {
ProjectedBatchOperation::Put
} else {
ProjectedBatchOperation::Insert
};
let batch = prepare_batch(installed, id, operation, rows)?;
return execute_owned_things(self.db.inner_orm(), installed, &batch).await;
}
let mut lowered = Vec::with_capacity(inputs.len());
for input in inputs {
lowered.push(lower_entity_create(input, &id, installed)?);
}
let tx = self
.db
.inner_orm()
.transaction_context(TxType::Write)
.await
.map_err(Error::from_orm)?;
let manager =
DynamicEntityManager::with_canonical_transaction(tx.clone(), Arc::new(descriptor));
let iids = match if put {
manager.put_many_exact(&lowered).await
} else {
manager.insert_many(&lowered).await
} {
Ok(v) if v.len() == lowered.len() => v,
Ok(_) => {
let _ = tx.rollback().await;
return Err(Error::model_validation(
ModelValidationPhase::Hydration,
"iid_count_mismatch",
vec!["iid".into()],
"provider returned an unexpected IID count",
None,
));
}
Err(e) => {
let _ = tx.rollback().await;
return Err(Error::from_orm(e));
}
};
let mut out = Vec::with_capacity(iids.len());
for iid in iids {
let row = match manager.get_by_iid_exact(&iid).await {
Ok(Some(r)) => r,
Ok(None) => {
let _ = tx.rollback().await;
return Err(Error::model_validation(
ModelValidationPhase::Hydration,
"missing_post_write_row",
vec!["iid".into()],
"written entity was not returned",
None,
));
}
Err(e) => {
let _ = tx.rollback().await;
return Err(Error::from_orm(e));
}
};
let h = match hydrate_entity(row, &id, installed) {
Ok(v) => v,
Err(e) => {
let _ = tx.rollback().await;
return Err(e);
}
};
let value = match M::materialize(&h, &HydrationCapability::new()) {
Ok(v) => v,
Err(e) => {
let _ = tx.rollback().await;
return Err(map_validation_error(e, ModelValidationPhase::Hydration));
}
};
out.push(value);
}
tx.commit().await.map_err(Error::from_orm)?;
Ok(out)
}
pub async fn count(&self) -> Result<u64> {
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, _descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
ProjectedCrudExecutor::new(installed)
.count_entities_with_compatibility(self.db.inner_orm(), &id)
.await
.map_err(|error| Error::from_projected_crud(error, ModelKind::Entity, None))
}
pub async fn get_by_iid(&self, iid: &str) -> Result<Option<M>> {
if !is_canonical_thing_iid(iid) {
return Err(invalid_iid());
}
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, _descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
ProjectedCrudExecutor::new(installed)
.get_entity_by_iid_with_compatibility(self.db.inner_orm(), &id, iid)
.await
.map_err(|error| Error::from_projected_crud(error, ModelKind::Entity, None))?
.map(|projected| materialize_projected(projected, installed))
.transpose()
}
pub async fn all(&self) -> Result<Vec<M>> {
let installed = self.db.installed_schema().ok_or_else(schema_not_bound)?;
let (id, descriptor) = resolve_entity_authority(
M::TYPE_ID_JSON,
installed,
ModelValidationPhase::Input,
true,
)?;
let rows =
DynamicEntityManager::new_canonical(self.db.inner_orm(), Arc::new(descriptor.clone()))
.all_exact()
.await
.map_err(Error::from_orm)?;
rows.into_iter()
.map(|r| {
let h = hydrate_entity(r, &id, installed)?;
M::materialize(&h, &HydrationCapability::new())
.map_err(|e| map_validation_error(e, ModelValidationPhase::Hydration))
})
.collect()
}
}