Поделиться через

Создание консольного приложения Rust с помощью Azure DocumentDB

В этом руководстве вы создадите консольное приложение Rust для подключения к кластеру Azure DocumentDB. В этом руководстве описывается настройка среды разработки, использование крейта azure_identity из Azure SDK для Rust для аутентификации и управления документами в базе данных.

Предпосылки

  • Существующий кластер Azure DocumentDB

Настройка консольного приложения

Затем создайте проект консольного приложения и импортируйте необходимые библиотеки для проверки подлинности в кластере.

  1. Создание проекта Rust с помощью cargo new.

    cargo new mongodb-app
cd mongodb-app

  2. Добавьте крейт azure_core к зависимостям.

    cargo add azure_core

  3. Добавьте azure_identity пакет для аутентификации.

    cargo add azure_identity

  4. Добавьте класс драйвера для взаимодействия с кластером mongodb .

    cargo add mongodb

  5. Для асинхронных операций также добавьте вспомогательные tokio, futures и serde библиотеки.

    cargo add tokio --features full
cargo add futures
cargo add serde --features derive

Подключение к кластеру

Теперь используйте библиотеку Azure.Identity, чтобы получить TokenCredential для подключения к вашему кластеру. Официальный драйвер MongoDB имеет специальный интерфейс, который должен быть реализован для получения маркеров из Microsoft Entra для использования при подключении к кластеру.

  1. Откройте main.rs файл и импортируйте необходимые крейты и модули.

    use azure_core::credentials::TokenCredential;
use azure_identity::DefaultAzureCredential;
use futures::{FutureExt, TryStreamExt};
use mongodb::{
    Client,
    bson::doc,
    options::{
        AuthMechanism, ClientOptions, Credential,
        oidc::{self, IdpServerResponse},
    },
};
use serde::{Deserialize, Serialize};

  2. Создайте основную асинхронную функцию с необходимой обработкой ошибок.

    #[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {

    Ok(())
}

  3. Создайте новый экземпляр структуры azure_identity::DefaultAzureCredential.

    let credential = DefaultAzureCredential::new()?;

  4. Создайте функцию обратного вызова для учетных данных, чтобы обрабатывать запросы токенов от клиента MongoDB.

    let azure_identity_token_credential = Credential::builder()
    .mechanism(AuthMechanism::MongoDbOidc)
    .oidc_callback(oidc::Callback::machine(move |_| {
        let azure_credential = credential.clone();
        async move {
            let access_token = azure_credential
                .get_token(&["https://ossrdbms-aad.database.windows.net/.default"])
                .await
                .map_err(|e| {
                    mongodb::error::Error::custom(format!("Azure token error: {}", e))
                })?;
            Ok(IdpServerResponse::builder()
                .access_token(access_token.token.secret().to_owned())
                .build())
        }
        .boxed()
    }))
    .build()
    .into();

  5. Определите универсальный индикатор ресурса (URI) из кластера, используя его имя, схему и глобальную конечную точку.

    let cluster_name = "<azure-documentdb-cluster-name>";

let uri = format!(
    "mongodb+srv://{}.global.mongocluster.cosmos.azure.com/",
    cluster_name
);

  6. Создайте экземпляр mongodb::ClientOptions с помощью конфигурации в соответствии с лучшими практиками, вашего URI и функции обратного вызова для предоставления учетных данных.

    let mut client_options = ClientOptions::parse(uri).await?;

client_options.connect_timeout = Some(std::time::Duration::from_secs(120));
client_options.tls = Some(mongodb::options::Tls::Enabled(Default::default()));
client_options.retry_writes = Some(true);

client_options.credential = Some(azure_identity_token_credential);

  7. Создайте новый экземпляр mongodb::Client с помощью созданных параметров.

    let client = Client::with_options(client_options)?;

println!("Client created");

Выполнение распространенных операций

Наконец, используйте официальную библиотеку для выполнения общих задач с базами данных, коллекциями и документами. Здесь вы используете те же классы и методы, которые будут использоваться для взаимодействия с MongoDB или DocumentDB для управления коллекциями и элементами.

  1. Создайте структуру Rust для представления Product документов с serde поддержкой сериализации.

    #[derive(Serialize, Deserialize, Debug)]
struct Product {
    _id: String,
    category: String,
    name: String,
    quantity: i32,
    price: f64,
    clearance: bool,
}

  2. Получите ссылку на базу данных по имени.

    let database = client.database("<database-name>");

println!("Database pointer created");

  3. Получите ссылку на коллекцию.

    let collection = database.collection::<Product>("<collection-name>");

println!("Collection pointer created");

  4. Создайте документ с помощью collection.update_one, и затем обновите или вставьте его в коллекцию.

    let document = Product {
    _id: "aaaaaaaa-0000-1111-2222-bbbbbbbbbbbb".to_string(),
    category: "gear-surf-surfboards".to_string(),
    name: "Yamba Surfboard".to_string(),
    quantity: 12,
    price: 850.00,
    clearance: false,
};

let response = collection
    .update_one(
        doc! { "_id": "aaaaaaaa-0000-1111-2222-bbbbbbbbbbbb" },
        doc! { "$set": mongodb::bson::to_document(&document)? },
    )
    .upsert(true)
    .await?;

println!("Documents upserted count:\t{}", response.modified_count);

  5. Чтение определенного документа из коллекции с помощью collection.find_one фильтра.

    let document = collection
    .find_one(doc! { "_id": "aaaaaaaa-0000-1111-2222-bbbbbbbbbbbb" })
    .await?;

println!("Read document _id:\t{:#?}", document.unwrap()._id);

  6. Запрос нескольких документов, соответствующих фильтру, с помощью collection.find.

    let filter = doc! { "category": "gear-surf-surfboards" };

let mut cursor = collection.find(filter).await?;

while let Some(document) = cursor.try_next().await? {
    println!("Found document:\t{:#?}", document);
}

Связанный контент

  • Last updated on