Add support for caching based on Locale

Author: ianwallen

springdoc-openapi-common/src/main/java/org/springdoc/api/AbstractOpenApiResource.java

@@ -294,7 +294,7 @@ public static void addHiddenRestControllers(String... classes) {
 	 */
 	protected synchronized OpenAPI getOpenApi(Locale locale) {
 		OpenAPI openApi;
-		if (openAPIService.getCachedOpenAPI() == null || springDocConfigProperties.isCacheDisabled()) {
+		if (openAPIService.getCachedOpenAPI(locale) == null || springDocConfigProperties.isCacheDisabled()) {
 			Instant start = Instant.now();
 			openAPIService.build(locale);
 			Map<String, Object> mappingsMap = openAPIService.getMappingsMap().entrySet().stream()
@@ -325,15 +325,15 @@ protected synchronized OpenAPI getOpenApi(Locale locale) {
 			if ((!CollectionUtils.isEmpty(openApi.getServers()) && !Objects.equals(servers, openApi.getServers())))
 				openAPIService.setServersPresent(true);
 
-			openAPIService.setCachedOpenAPI(openApi);
+			openAPIService.setCachedOpenAPI(openApi, locale);
 			openAPIService.resetCalculatedOpenAPI();
 
 			LOGGER.info("Init duration for springdoc-openapi is: {} ms",
 					Duration.between(start, Instant.now()).toMillis());
 		}
 		else {
 			LOGGER.debug("Fetching openApi document from cache");
-			openApi = openAPIService.updateServers(openAPIService.getCachedOpenAPI());
+			openApi = openAPIService.updateServers(openAPIService.getCachedOpenAPI(locale));
 		}
 		return openApi;
 	}
@@ -1099,8 +1099,8 @@ else if (existingOperation != null) {
 	/**
 	 * Init open api builder.
 	 */
-	protected void initOpenAPIBuilder() {
-		if (openAPIService.getCachedOpenAPI() != null && springDocConfigProperties.isCacheDisabled()) {
+	protected void initOpenAPIBuilder(Locale locale) {
+		if (openAPIService.getCachedOpenAPI(locale) != null && springDocConfigProperties.isCacheDisabled()) {
 			openAPIService = openAPIBuilderObjectFactory.getObject();
 		}
 	}

springdoc-openapi-common/src/main/java/org/springdoc/core/OpenAPIService.java

@@ -129,7 +129,10 @@ public class OpenAPIService {
 	/**
 	 * The Cached open api.
 	 */
-	private OpenAPI cachedOpenAPI;
+	/**
+	 * The Mappings map.
+	 */
+	private final Map<String, OpenAPI> cachedOpenAPI = new HashMap<>();
 
 	/**
 	 * The Calculated open api.
@@ -724,17 +727,17 @@ public Map<String, Object> getControllerAdviceMap() {
 	 *
 	 * @return the cached open api
 	 */
-	public OpenAPI getCachedOpenAPI() {
-		return cachedOpenAPI;
+	public OpenAPI getCachedOpenAPI(Locale locale) {
+		return cachedOpenAPI.get(locale.getLanguage());
 	}
 
 	/**
 	 * Sets cached open api.
 	 *
 	 * @param cachedOpenAPI the cached open api
 	 */
-	public void setCachedOpenAPI(OpenAPI cachedOpenAPI) {
-		this.cachedOpenAPI = cachedOpenAPI;
+	public void setCachedOpenAPI(OpenAPI cachedOpenAPI, Locale locale) {
+		this.cachedOpenAPI.put(locale.getLanguage(), cachedOpenAPI);
 	}
 
 	/**

springdoc-openapi-common/src/test/java/org/springdoc/api/AbstractOpenApiResourceTest.java

@@ -167,11 +167,12 @@ void calculatePathFromRouterOperation() {
 
 	@Test
 	void preLoadingModeShouldNotOverwriteServers() throws InterruptedException {
+		Locale locale = new Locale("en_US");
 		when(openAPIService.updateServers(any())).thenCallRealMethod();
-		when(openAPIService.getCachedOpenAPI()).thenCallRealMethod();
+		when(openAPIService.getCachedOpenAPI(locale)).thenCallRealMethod();
 		doAnswer(new CallsRealMethods()).when(openAPIService).setServersPresent(true);
 		doAnswer(new CallsRealMethods()).when(openAPIService).setServerBaseUrl(any());
-		doAnswer(new CallsRealMethods()).when(openAPIService).setCachedOpenAPI(any());
+		doAnswer(new CallsRealMethods()).when(openAPIService).setCachedOpenAPI(any(), locale);
 
 		String customUrl = "https://custom.com";
 		String generatedUrl = "https://generated.com";

springdoc-openapi-webflux-core/src/main/java/org/springdoc/webflux/api/OpenApiActuatorResource.java

@@ -122,8 +122,8 @@ public Mono<String> openapiYaml(ServerHttpRequest serverHttpRequest, Locale loca
 	}
 
 	@Override
-	protected void calculateServerUrl(ServerHttpRequest serverHttpRequest, String apiDocsUrl) {
-		super.initOpenAPIBuilder();
+	protected void calculateServerUrl(ServerHttpRequest serverHttpRequest, String apiDocsUrl, Locale locale) {
+		super.initOpenAPIBuilder(locale);
 		URI uri = getActuatorURI(serverHttpRequest.getURI().getScheme(), serverHttpRequest.getURI().getHost());
 		openAPIService.setServerBaseUrl(uri.toString());
 	}

springdoc-openapi-webflux-core/src/main/java/org/springdoc/webflux/api/OpenApiResource.java

@@ -122,7 +122,7 @@ public OpenApiResource(ObjectFactory<OpenAPIService> openAPIBuilderObjectFactory
 	 */
 	protected Mono<String> openapiJson(ServerHttpRequest serverHttpRequest, String apiDocsUrl, Locale locale)
 			throws JsonProcessingException {
-		calculateServerUrl(serverHttpRequest, apiDocsUrl);
+		calculateServerUrl(serverHttpRequest, apiDocsUrl, locale);
 		OpenAPI openAPI = this.getOpenApi(locale);
 		return Mono.just(writeJsonValue(openAPI));
 	}
@@ -138,7 +138,7 @@ protected Mono<String> openapiJson(ServerHttpRequest serverHttpRequest, String a
 	 */
 	protected Mono<String> openapiYaml(ServerHttpRequest serverHttpRequest, String apiDocsUrl, Locale locale)
 			throws JsonProcessingException {
-		calculateServerUrl(serverHttpRequest, apiDocsUrl);
+		calculateServerUrl(serverHttpRequest, apiDocsUrl, locale);
 		OpenAPI openAPI = this.getOpenApi(locale);
 		return Mono.just(writeYamlValue(openAPI));
 	}
@@ -232,8 +232,8 @@ protected void getWebFluxRouterFunctionPaths(Locale locale) {
 	 * @param serverHttpRequest the server http request
 	 * @param apiDocsUrl the api docs url
 	 */
-	protected void calculateServerUrl(ServerHttpRequest serverHttpRequest, String apiDocsUrl) {
-		initOpenAPIBuilder();
+	protected void calculateServerUrl(ServerHttpRequest serverHttpRequest, String apiDocsUrl, Locale locale) {
+		initOpenAPIBuilder(locale);
 		String serverUrl = getServerUrl(serverHttpRequest,apiDocsUrl);
 		openAPIService.setServerBaseUrl(serverUrl);
 	}

springdoc-openapi-webmvc-core/src/main/java/org/springdoc/webmvc/api/OpenApiResource.java

@@ -157,7 +157,7 @@ public OpenApiResource(ObjectFactory<OpenAPIService> openAPIBuilderObjectFactory
 	public String openapiJson(HttpServletRequest request,
 			String apiDocsUrl, Locale locale)
 			throws JsonProcessingException {
-		calculateServerUrl(request, apiDocsUrl);
+		calculateServerUrl(request, apiDocsUrl, locale);
 		OpenAPI openAPI = this.getOpenApi(locale);
 		return writeJsonValue(openAPI);
 	}
@@ -174,7 +174,7 @@ public String openapiJson(HttpServletRequest request,
 	public String openapiYaml(HttpServletRequest request,
 			String apiDocsUrl, Locale locale)
 			throws JsonProcessingException {
-		calculateServerUrl(request, apiDocsUrl);
+		calculateServerUrl(request, apiDocsUrl, locale);
 		OpenAPI openAPI = this.getOpenApi(locale);
 		return writeYamlValue(openAPI);
 	}
@@ -290,8 +290,8 @@ private Comparator<Map.Entry<RequestMappingInfo, HandlerMethod>> byReversedReque
 	 * @param request the request
 	 * @param apiDocsUrl the api docs url
 	 */
-	protected void calculateServerUrl(HttpServletRequest request, String apiDocsUrl) {
-		super.initOpenAPIBuilder();
+	protected void calculateServerUrl(HttpServletRequest request, String apiDocsUrl, Locale locale) {
+		super.initOpenAPIBuilder(locale);
 		String calculatedUrl = getServerUrl(request, apiDocsUrl);
 		openAPIService.setServerBaseUrl(calculatedUrl);
 	}