From b73f7cdd7ece090c0c66225fd9bada1d4b7aca45 Mon Sep 17 00:00:00 2001 From: Wenjie Guo Date: Wed, 18 Dec 2024 15:58:15 +0800 Subject: [PATCH] Add timeout for fetching token --- .../metadata/client/util/CustomClientRequestFilter.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/main/java/org/commonjava/service/metadata/client/util/CustomClientRequestFilter.java b/src/main/java/org/commonjava/service/metadata/client/util/CustomClientRequestFilter.java index b4bf3b7..3e7148b 100644 --- a/src/main/java/org/commonjava/service/metadata/client/util/CustomClientRequestFilter.java +++ b/src/main/java/org/commonjava/service/metadata/client/util/CustomClientRequestFilter.java @@ -29,6 +29,8 @@ import jakarta.ws.rs.core.HttpHeaders; import jakarta.ws.rs.ext.Provider; +import java.time.Duration; + @Provider @Priority(Priorities.AUTHENTICATION) public class CustomClientRequestFilter implements ClientRequestFilter @@ -51,7 +53,7 @@ public void filter( ClientRequestContext requestContext ) if ( tokens == null || tokens.isAccessTokenExpired() ) { logger.debug("Security enabled, get oidc Tokens"); - tokens = client.getTokens().await().indefinitely(); + tokens = client.getTokens().await().atMost( Duration.ofSeconds( 30 ) ); } requestContext.getHeaders().add(HttpHeaders.AUTHORIZATION, "Bearer " + tokens.getAccessToken()); }