Created
October 4, 2018 14:08
-
-
Save mp911de/e2480afd4d5d5c27038e2b1b49b30cfa to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
/* | |
* Copyright 2018 the original author or authors. | |
* | |
* Licensed under the Apache License, Version 2.0 (the "License"); | |
* you may not use this file except in compliance with the License. | |
* You may obtain a copy of the License at | |
* | |
* http://www.apache.org/licenses/LICENSE-2.0 | |
* | |
* Unless required by applicable law or agreed to in writing, software | |
* distributed under the License is distributed on an "AS IS" BASIS, | |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
* See the License for the specific language governing permissions and | |
* limitations under the License. | |
*/ | |
package org.springframework.vault.core; | |
import org.springframework.http.client.ClientHttpRequestFactory; | |
import org.springframework.vault.authentication.ClientAuthentication; | |
import org.springframework.vault.authentication.SessionManager; | |
import org.springframework.vault.client.VaultEndpoint; | |
import org.springframework.vault.client.VaultEndpointProvider; | |
import org.springframework.web.client.RestTemplate; | |
/** | |
* @author Mark Paluch | |
*/ | |
public class NamespaceScopedVaultTemplate extends VaultTemplate { | |
private final String namespace; | |
public NamespaceScopedVaultTemplate(VaultEndpoint vaultEndpoint, | |
ClientAuthentication clientAuthentication, String namespace) { | |
super(vaultEndpoint, clientAuthentication); | |
this.namespace = namespace; | |
} | |
public NamespaceScopedVaultTemplate(VaultEndpoint vaultEndpoint, | |
ClientHttpRequestFactory clientHttpRequestFactory, | |
SessionManager sessionManager, String namespace) { | |
super(vaultEndpoint, clientHttpRequestFactory, sessionManager); | |
this.namespace = namespace; | |
} | |
public NamespaceScopedVaultTemplate(VaultEndpointProvider endpointProvider, | |
ClientHttpRequestFactory requestFactory, SessionManager sessionManager, | |
String namespace) { | |
super(endpointProvider, requestFactory, sessionManager); | |
this.namespace = namespace; | |
} | |
/** | |
* Associate namespace interceptor that adds the namespace header to each Vault | |
* request. | |
* | |
* @param endpointProvider must not be {@literal null}. | |
* @param requestFactory must not be {@literal null}. | |
* @return | |
*/ | |
@Override | |
protected RestTemplate doCreateRestTemplate(VaultEndpointProvider endpointProvider, | |
ClientHttpRequestFactory requestFactory) { | |
RestTemplate restTemplate = super.doCreateRestTemplate(endpointProvider, | |
requestFactory); | |
restTemplate.getInterceptors().add((request, body, execution) -> { | |
request.getHeaders().add("X-Vault-Namespace", namespace); | |
return execution.execute(request, body); | |
}); | |
return restTemplate; | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment