mongor 0.1.10

Ergonomic MongoDB ODM
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

// Authors: Robert Lopez
// License: MIT (See `LICENSE.md`)

use crate::{error::Error, util::convert_bson_to_oid};
use mongodb::{bson::oid::ObjectId, options::*, ClientSession, Collection};
use serde::{Deserialize, Serialize};

/// Executes a `insertOne` query on a `Collection`.
///
/// https://www.mongodb.com/docs/manual/reference/method/db.collection.insertOne/
///
/// ---
/// Example Usage:
/// ```
///
/// let collection: Collection<D> = ...;
///
/// let document: D = ...;
///
/// let inserted_oid: ObjectId = insert_one(
///     &collection,
///     document,
///     None::<InsertOneOptions>,
///     None::<&mut ClientSession>,
/// ).await?;
/// ```
pub async fn insert_one<D>(
    collection: &Collection<D>,
    document: D,
    options: Option<InsertOneOptions>,
    session: Option<&mut ClientSession>,
) -> Result<ObjectId, Error>
where
    D: Serialize + for<'a> Deserialize<'a> + Send + Sync,
{
    let insert_one_result = if let Some(session) = session {
        collection
            .insert_one(document)
            .with_options(options)
            .session(session)
            .await
    } else {
        collection.insert_one(document).with_options(options).await
    }
    .map_err(Error::Mongo)?;

    convert_bson_to_oid(insert_one_result.inserted_id)
}

/// Executes a `insertMany` query on a `Collection`.
///
/// https://www.mongodb.com/docs/manual/reference/method/db.collection.insertMany/
///
/// ---
/// Example Usage:
/// ```
///
/// let collection: Collection<D> = ...;
///
/// let documents: Vec<D> = ...;
///
/// let inserted_oids: Vec<ObjectId> = insert_many(
///     &collection,
///     documents,
///     None::<InsertManyOptions>,
///     None::<&mut ClientSession>,
/// ).await?;
/// ```
pub async fn insert_many<D>(
    collection: &Collection<D>,
    documents: Vec<D>,
    options: Option<InsertManyOptions>,
    session: Option<&mut ClientSession>,
) -> Result<Vec<ObjectId>, Error>
where
    D: Serialize + for<'a> Deserialize<'a> + Send + Sync,
{
    let insert_many_result = if let Some(session) = session {
        collection
            .insert_many(documents)
            .with_options(options)
            .session(session)
            .await
    } else {
        collection
            .insert_many(documents)
            .with_options(options)
            .await
    }
    .map_err(Error::Mongo)?;

    let mut inserted_ids = vec![];

    for (_, bson) in insert_many_result.inserted_ids {
        inserted_ids.push(convert_bson_to_oid(bson)?);
    }

    Ok(inserted_ids)
}