import java.util.HashMap;

import java.util.Map;

public class HeaderCache {

private static final Map<String, Map<String, String>> cache = new HashMap<>(); // Cache for each staging environment

public static Map<String, String> getHeaders(String stagingEnvironment) {

// Check if headers are already cached for this staging environment

Map<String, String> cachedHeaders = cache.get(stagingEnvironment);

if (cachedHeaders != null) {

return cachedHeaders; // Return cached headers

}

// If not cached, simulate a header request (replace with actual request logic)

Map<String, String> headers = simulateHeaderRequest(stagingEnvironment);

// Cache the headers for this staging environment

cache.put(stagingEnvironment, headers);

return headers;

}

private static Map<String, String> simulateHeaderRequest(String stagingEnvironment) {

// Simulate fetching headers from a staging environment (replace with actual logic)

Map<String, String> headers = new HashMap<>();

headers.put("X-Staging-Id", stagingEnvironment);

headers.put("Content-Type", "application/json");

return headers;

}

public static void main(String[] args) {

// Example usage:

Map<String, String> staging1Headers = getHeaders("staging-env-1");

System.out.println("Staging 1 Headers: " + staging1Headers);

Map<String, String> staging1HeadersAgain = getHeaders("staging-env-1"); // Should be from cache

System.out.println("Staging 1 Headers (from cache): " + staging1HeadersAgain);

Map<String, String> staging2Headers = getHeaders("staging-env-2");

System.out.println("Staging 2 Headers: " + staging2Headers);

}

}