1. import org.json.JSONObject;
  3. public class JsonResponseExtender {
  5. /**
  6. * Extends the existing JSON response logic for isolated environments.
  7. * Adds environment-specific information to the JSON response.
  8. *
  9. * @param originalResponse The original JSON response as a JSONObject.
  10. * @param environmentName The name of the isolated environment (e.g., "dev", "staging").
  11. * @return The extended JSON response as a JSONObject. Returns null if the input is invalid.
  12. */
  13. public static JSONObject extendResponseForIsolatedEnvironment(JSONObject originalResponse, String environmentName) {
  15. // Check if the input is a valid JSONObject
  16. if (originalResponse == null || !(originalResponse instanceof JSONObject)) {
  17. System.err.println("Invalid input: originalResponse must be a JSONObject.");
  18. return null;
  19. }
  21. // Create a copy of the original response to avoid modifying the original
  22. JSONObject extendedResponse = originalResponse.copy();
  24. // Add environment-specific information to the extended response
  25. try {
  26. extendedResponse.put("environment", environmentName); // Add the environment name
  27. extendedResponse.put("environment_details", "Isolated environment"); // Add a general description
  28. extendedResponse.put("timestamp", System.currentTimeMillis()); // Add a timestamp
  29. } catch (Exception e) {
  30. System.err.println("Error adding environment details: " + e.getMessage());
  31. return null;
  32. }
  35. return extendedResponse;
  36. }
  38. public static void main(String[] args) {
  39. // Example Usage
  40. JSONObject originalResponse = new JSONObject();
  41. originalResponse.put("status", "success");
  42. originalResponse.put("data", "some data");
  44. String environment = "dev";
  46. JSONObject extendedResponse = extendResponseForIsolatedEnvironment(originalResponse, environment);
  48. if (extendedResponse != null) {
  49. System.out.println(extendedResponse.toString(2)); // Print the extended JSON with indentation
  50. }
  51. }
  52. }

Add your comment