1. import java.util.HashMap;
  2. import java.util.Map;
  4. class HeaderTruncator {
  6. /**
  7. * Truncates header metadata for dry-run scenarios.
  8. * Handles edge cases like null headers and empty header values.
  9. *
  10. * @param headers A map of header names to values.
  11. * @param maxHeaderLength The maximum allowed length for each header value.
  12. * @return A new map with truncated header values. Returns an empty map if input is null.
  13. */
  14. public static Map<String, String> truncateHeaders(Map<String, String> headers, int maxHeaderLength) {
  15. if (headers == null) {
  16. return new HashMap<>(); // Return empty map if input is null
  17. }
  19. Map<String, String> truncatedHeaders = new HashMap<>();
  21. for (Map.Entry<String, String> entry : headers.entrySet()) {
  22. String headerName = entry.getKey();
  23. String headerValue = entry.getValue();
  25. if (headerValue == null) {
  26. truncatedHeaders.put(headerName, ""); // Handle null header values
  27. } else {
  28. if (headerValue.length() > maxHeaderLength) {
  29. truncatedHeaders.put(headerName, headerValue.substring(0, maxHeaderLength)); // Truncate
  30. } else {
  31. truncatedHeaders.put(headerName, headerValue); // No truncation needed
  32. }
  33. }
  34. }
  36. return truncatedHeaders;
  37. }
  39. public static void main(String[] args) {
  40. // Example Usage
  41. Map<String, String> headers = new HashMap<>();
  42. headers.put("Content-Type", "application/json");
  43. headers.put("X-Custom-Header", "This is a very long header value");
  44. headers.put("AnotherHeader", null);
  45. headers.put("ShortHeader", "short");
  47. int maxLen = 20;
  49. Map<String, String> truncated = truncateHeaders(headers, maxLen);
  51. for (Map.Entry<String, String> entry : truncated.entrySet()) {
  52. System.out.println(entry.getKey() + ": " + entry.getValue());
  53. }
  55. //Test case with null input
  56. Map<String, String> nullInput = null;
  57. Map<String, String> truncatedNull = truncateHeaders(nullInput, 20);
  58. System.out.println("Truncated Null Input: " + truncatedNull);
  61. //Test case with empty map
  62. Map<String, String> emptyMap = new HashMap<>();
  63. Map<String, String> truncatedEmpty = truncateHeaders(emptyMap, 20);
  64. System.out.println("Truncated Empty Input: " + truncatedEmpty);
  65. }
  66. }

Add your comment