teo_runtime/handler/default/
upsert.rs

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
use key_path::path;
use crate::value::Value;
use crate::response::Response;
use crate::action::action::*;
use crate::connection::transaction;
use crate::handler::default::internal::create::create_internal;
use crate::handler::default::internal::update::update_internal;
use crate::request::Request;

pub async fn upsert(request: &Request) -> teo_result::Result<Response> {
    let model = request.transaction_ctx().namespace().model_at_path(&request.handler_match().unwrap().path()).unwrap();
    let action = UPSERT | SINGLE | ENTRY;
    let value: Value = request.transaction_ctx().run_transaction(|ctx: transaction::Ctx| async move {
        let find_result = ctx.find_unique_internal(model, request.body_value()?, true, action, Some(request.clone()), path![]).await?;
        let include = request.body_value()?.get("include");
        let select = request.body_value()?.get("select");
        match find_result {
            Some(object) => {
                let update = request.body_value()?.get("update");
                Ok(update_internal(object, update, include, select, &path![]).await?)
            }
            None => {
                let create = request.body_value()?.get("update");
                Ok(create_internal(ctx.clone(), request.clone(), create, include, select, model, &path![], action).await?)
            }
        }
    }).await?;
    Ok(Response::data(value))
}