forked from spring-projects/spring-amqp
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
spring-projectsGH-1419: Add RestTemplateNodeLocator
- also remove hard dependency on `spring-webflux` from `spring-rabbit-junit`.
- Loading branch information
1 parent
c143c5b
commit d8171b4
Showing
8 changed files
with
204 additions
and
25 deletions.
There are no files selected for viewing
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
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
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
41 changes: 41 additions & 0 deletions
41
...g-rabbit/src/main/java/org/springframework/amqp/rabbit/connection/RestTemplateHolder.java
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,41 @@ | ||
| /* | ||
| * Copyright 2022 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 | ||
| * | ||
| * https://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.amqp.rabbit.connection; | ||
|
|
||
| import org.springframework.web.client.RestTemplate; | ||
|
|
||
| /** | ||
| * Holder for a {@link RestTemplate} and credentials. | ||
| * | ||
| * @author Gary Russell | ||
| * @since 2.4.8 | ||
| * | ||
| */ | ||
| class RestTemplateHolder { | ||
|
|
||
| final String userName; // NOSONAR | ||
|
|
||
| final String password; // NOSONAR | ||
|
|
||
| RestTemplate template; // NOSONAR | ||
|
|
||
| RestTemplateHolder(String userName, String password) { | ||
| this.userName = userName; | ||
| this.password = password; | ||
| } | ||
|
|
||
| } |
97 changes: 97 additions & 0 deletions
97
...bit/src/main/java/org/springframework/amqp/rabbit/connection/RestTemplateNodeLocator.java
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,97 @@ | ||
| /* | ||
| * Copyright 2022 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 | ||
| * | ||
| * https://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.amqp.rabbit.connection; | ||
|
|
||
| import java.io.IOException; | ||
| import java.net.URI; | ||
| import java.net.URISyntaxException; | ||
| import java.nio.charset.StandardCharsets; | ||
| import java.util.Map; | ||
|
|
||
| import org.apache.hc.client5.http.auth.AuthCache; | ||
| import org.apache.hc.client5.http.impl.auth.BasicAuthCache; | ||
| import org.apache.hc.client5.http.impl.auth.BasicScheme; | ||
| import org.apache.hc.client5.http.protocol.HttpClientContext; | ||
| import org.apache.hc.core5.http.HttpHost; | ||
| import org.apache.hc.core5.http.protocol.BasicHttpContext; | ||
| import org.apache.hc.core5.http.protocol.HttpContext; | ||
|
|
||
| import org.springframework.amqp.rabbit.connection.LocalizedQueueConnectionFactory.NodeLocator; | ||
| import org.springframework.http.HttpMethod; | ||
| import org.springframework.http.HttpStatus; | ||
| import org.springframework.http.ResponseEntity; | ||
| import org.springframework.http.client.HttpComponentsClientHttpRequestFactory; | ||
| import org.springframework.http.client.support.BasicAuthenticationInterceptor; | ||
| import org.springframework.lang.Nullable; | ||
| import org.springframework.web.client.RestTemplate; | ||
| import org.springframework.web.util.UriUtils; | ||
|
|
||
| /** | ||
| * A {@link NodeLocator} using the {@link RestTemplate}. | ||
| * | ||
| * @author Gary Russell | ||
| * @since 3.0 | ||
| * | ||
| */ | ||
| public class RestTemplateNodeLocator implements NodeLocator<RestTemplateHolder> { | ||
|
|
||
| @Override | ||
| public RestTemplateHolder createClient(String userName, String password) { | ||
| return new RestTemplateHolder(userName, password); | ||
| } | ||
|
|
||
| @SuppressWarnings({ "unchecked", "rawtypes" }) | ||
| @Override | ||
| @Nullable | ||
| public Map<String, Object> restCall(RestTemplateHolder client, String baseUri, String vhost, String queue) | ||
| throws URISyntaxException { | ||
|
|
||
| if (client.template == null) { | ||
| URI uri = new URI(baseUri); | ||
| HttpHost host = new HttpHost(uri.getHost(), uri.getPort()); | ||
| client.template = new RestTemplate(new HttpComponentsClientHttpRequestFactory() { | ||
|
|
||
| @Override | ||
| @Nullable | ||
| protected HttpContext createHttpContext(HttpMethod httpMethod, URI uri) { | ||
| AuthCache cache = new BasicAuthCache(); | ||
| BasicScheme scheme = new BasicScheme(); | ||
| cache.put(host, scheme); | ||
| BasicHttpContext context = new BasicHttpContext(); | ||
| context.setAttribute(HttpClientContext.AUTH_CACHE, cache); | ||
| return context; | ||
| } | ||
|
|
||
| }); | ||
| client.template.getInterceptors().add(new BasicAuthenticationInterceptor(client.userName, client.password)); | ||
| } | ||
| URI uri = new URI(baseUri) | ||
| .resolve("/api/queues/" + UriUtils.encodePathSegment(vhost, StandardCharsets.UTF_8) + "/" + queue); | ||
| ResponseEntity<Map> response = client.template.exchange(uri, HttpMethod.GET, null, Map.class); | ||
| return response.getStatusCode().equals(HttpStatus.OK) ? response.getBody() : null; | ||
| } | ||
|
|
||
| @Override | ||
| public void close(RestTemplateHolder client) { | ||
| try { | ||
| client.template.close(); | ||
| } | ||
| catch (IOException e) { | ||
| } | ||
| } | ||
|
|
||
| } |
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
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
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