Skip to content

Commit 904c66c

Browse files
authored
fix cosmos_query sample
1 parent 78026bd commit 904c66c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

sdk/cosmos/azure_data_cosmos/examples/cosmos_query.rs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,10 @@ use azure_data_cosmos::{
55
clients::{ContainerClientMethods, DatabaseClientMethods},
66
CosmosClient, CosmosClientMethods, PartitionKey,
77
};
8+
use azure_identity::DefaultAzureCredential;
89
use clap::Parser;
910
use futures::StreamExt;
11+
use std::sync::Arc;
1012

1113
/// An example to show querying a Cosmos DB container.
1214
#[derive(Parser)]
@@ -69,13 +71,13 @@ fn create_client(args: &Args) -> CosmosClient {
6971
if let Some(key) = args.key.as_ref() {
7072
CosmosClient::with_key(&args.endpoint, key.clone(), None).unwrap()
7173
} else {
72-
let cred = azure_identity::create_default_credential().unwrap();
74+
let cred = DefaultAzureCredential::new().map(Arc::new).unwrap();
7375
CosmosClient::new(&args.endpoint, cred, None).unwrap()
7476
}
7577
}
7678

7779
#[cfg(not(feature = "key_auth"))]
7880
fn create_client(args: &Args) -> CosmosClient {
79-
let cred = azure_identity::create_default_credential().unwrap();
81+
let cred = DefaultAzureCredential::new().map(Arc::new).unwrap();
8082
CosmosClient::new(&args.endpoint, cred, None).unwrap()
8183
}

0 commit comments

Comments
 (0)