From a3e9c0c3f09acf199529dd837bce252ece7906b7 Mon Sep 17 00:00:00 2001 From: Mark Paluch Date: Mon, 27 Sep 2021 15:44:20 +0200 Subject: [PATCH] Enable authentication steps for AzureMsiAuthentication for reactive usage. Closes gh-665. --- .../vault/authentication/AzureMsiAuthentication.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/spring-vault-core/src/main/java/org/springframework/vault/authentication/AzureMsiAuthentication.java b/spring-vault-core/src/main/java/org/springframework/vault/authentication/AzureMsiAuthentication.java index ee00d0c7..20b6b74e 100644 --- a/spring-vault-core/src/main/java/org/springframework/vault/authentication/AzureMsiAuthentication.java +++ b/spring-vault-core/src/main/java/org/springframework/vault/authentication/AzureMsiAuthentication.java @@ -52,7 +52,7 @@ import org.springframework.web.client.RestOperations; * "https://docs.microsoft.com/en-us/azure/virtual-machines/windows/instance-metadata-service" * >Azure Instance Metadata service */ -public class AzureMsiAuthentication implements ClientAuthentication { +public class AzureMsiAuthentication implements ClientAuthentication, AuthenticationStepsFactory { private static final Log logger = LogFactory.getLog(AzureMsiAuthentication.class); @@ -144,6 +144,11 @@ public class AzureMsiAuthentication implements ClientAuthentication { return createTokenUsingAzureMsiCompute(); } + @Override + public AuthenticationSteps getAuthenticationSteps() { + return createAuthenticationSteps(this.options); + } + private VaultToken createTokenUsingAzureMsiCompute() { Map login = getAzureLogin(this.options.getRole(), getVmEnvironment(), getAccessToken());