popey/claude-code-skills/azure-cosmos-rust

cargo add azure_data_cosmos azure_identity

COSMOS_ENDPOINT=https://<account>.documents.azure.com:443/
COSMOS_DATABASE=mydb
COSMOS_CONTAINER=mycontainer

use azure_identity::DeveloperToolsCredential;
use azure_data_cosmos::CosmosClient;
 
let credential = DeveloperToolsCredential::new(None)?;
let client = CosmosClient::new(
    "https://<account>.documents.azure.com:443/",
    credential.clone(),
    None,
)?;

let database = client.database_client("myDatabase");
let container = database.container_client("myContainer");

use serde::{Serialize, Deserialize};
 
#[derive(Serialize, Deserialize)]
struct Item {
    pub id: String,
    pub partition_key: String,
    pub value: String,
}
 
let item = Item {
    id: "1".into(),
    partition_key: "partition1".into(),
    value: "hello".into(),
};
 
container.create_item("partition1", item, None).await?;

let response = container.read_item("partition1", "1", None).await?;
let item: Item = response.into_model()?;

let mut item: Item = container.read_item("partition1", "1", None).await?.into_model()?;
item.value = "updated".into();
 
container.replace_item("partition1", "1", item, None).await?;

use azure_data_cosmos::models::PatchDocument;
 
let patch = PatchDocument::default()
    .with_add("/newField", "newValue")?
    .with_remove("/oldField")?;
 
container.patch_item("partition1", "1", patch, None).await?;

container.delete_item("partition1", "1", None).await?;

cargo add azure_data_cosmos --features key_auth