Skip to content

Commit 349db38

Browse files
committed
fixes
1 parent b23dd03 commit 349db38

File tree

1 file changed

+3
-11
lines changed

1 file changed

+3
-11
lines changed

source/includes/fundamentals/code-snippets/enterprise-auth.rs

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -69,18 +69,14 @@ async fn main() -> mongodb::error::Result<()> {
6969
async move {
7070
let token_file_path = std::env::var("AWS_WEB_IDENTITY_TOKEN_FILE").map_err(mongodb::error::Error::custom)?;
7171
let access_token = tokio::fs::read_to_string(token_file_path).await?;
72-
Ok(IdpServerResponse {
73-
access_token,
74-
expires: None,
75-
refresh_token: None,
76-
})
72+
Ok(IdpServerResponse::builder().access_token(access_token).build())
7773
}
7874
.boxed()
7975
}))
8076
.build()
8177
.into();
8278

83-
credential_options.credentials = Some(credential);
79+
client_options.credential = Some(credential);
8480
let client = Client::with_options(client_options)?;
8581

8682
let res = client
@@ -94,11 +90,7 @@ async fn main() -> mongodb::error::Result<()> {
9490
async fn cb(params: CallbackContext) -> mongodb::error::Result<IdpServerResponse> {
9591
let idp_info = params.idp_info.ok_or(Error::NoIDPInfo)?;
9692
let (access_token, expires, refresh_token) = negotiate_with_idp(ctx, idpInfo.Issuer).await?;
97-
Ok(oidc::IdpServerResponse {
98-
access_token,
99-
expires: Some(expires),
100-
refresh_token: Some(refresh_token),
101-
})
93+
Ok(IdpServerResponse::builder().access_token(access_token).expires(expires).refresh_token(refresh_token).build())
10294
}
10395
client_options.credential = Credential::builder()
10496
.mechanism(AuthMechanism::MongoDbOidc)

0 commit comments

Comments
 (0)