Java RESTful 客户端中从 HTTP JSON 响应提取特定字段

import javax.ws.rs.core.Response;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.io.IOException;
import java.util.HashMap;

public class JsonResponseExtractor {

    /**
     * 模拟一个 JAX-RS 客户端调用，并从 Response 中提取 JSON 字段。
     * 在实际应用中，response 对象会通过 Resteasy Client 等方式获取。
     *
     * @param response 接收到的 javax.ws.rs.core.Response 对象
     * @param fieldName 待提取的字段名称
     * @return 提取到的字段值，如果不存在则返回 null
     * @throws IOException 如果 JSON 解析失败
     */
    public Object extractFieldFromResponse(Response response, String fieldName) throws IOException {
        if (response == null || response.getStatus() != Response.Status.OK.getStatusCode()) {
            System.err.println("HTTP 请求失败或响应为空。状态码: " + (response != null ? response.getStatus() : "N/A"));
            if (response != null) {
                response.close(); // 确保关闭资源
            }
            return null;
        }

        String jsonResponseString = null;
        try {
            // 关键步骤：从 Response 中读取实体为 String
            // 注意：readEntity() 只能调用一次
            jsonResponseString = response.readEntity(String.class);
            System.out.println("原始 JSON 响应字符串: " + jsonResponseString);

            // 使用 Jackson ObjectMapper 解析 JSON 字符串
            ObjectMapper mapper = new ObjectMapper();
            // 将 JSON 字符串解析为 HashMap
            // 也可以解析为自定义的 POJO 类，提供更好的类型安全性
            HashMap<String, Object> map = mapper.readValue(jsonResponseString, HashMap.class);

            // 提取特定字段
            return map.get(fieldName);

        } catch (IOException e) {
            System.err.println("解析 JSON 响应时发生错误: " + e.getMessage());
            throw e;
        } finally {
            // 务必关闭 Response 对象以释放底层连接资源
            if (response != null) {
                response.close();
            }
        }
    }

    // ----------------------------------------------------------------------------------
    // 模拟 Resteasy Client 调用和 Response 对象
    // 在实际应用中，这部分会由你的 Resteasy Client 接口和实现来完成
    // ----------------------------------------------------------------------------------

    // 假设这是你的 Resteasy Client 接口
    // @POST
    // @Path("/customers")
    // @Produces({ MediaType.APPLICATION_JSON })
    // Response createCustomer(Customer customer);

    // 模拟一个创建 Customer 后的响应
    public Response mockCreateCustomerResponse(String customerId) {
        // 构建一个模拟的 JSON 响应字符串
        String jsonPayload = String.format("{\"id\":\"%s\", \"name\":\"Test Customer\", \"email\":\"test@example.com\"}", customerId);

        // 创建一个模拟的 Response 对象
        // 在实际应用中，这是 Resteasy Client 返回的真实对象
        return Response.ok(jsonPayload).build();
    }

    public static void main(String[] args) {
        JsonResponseExtractor extractor = new JsonResponseExtractor();

        // 模拟一个成功的 HTTP 响应，包含一个 ID 字段
        Response successResponse = extractor.mockCreateCustomerResponse("CUST12345");
        try {
            Object id = extractor.extractFieldFromResponse(successResponse, "id");
            System.out.println("提取到的 'id' 字段值: " + id); // 预期输出: CUST12345
        } catch (IOException e) {
            System.err.println("处理成功响应时发生错误: " + e.getMessage());
        }

        System.out.println("\n----------------------------------\n");

        // 模拟一个不同的响应，尝试提取不存在的字段
        Response anotherResponse = extractor.mockCreateCustomerResponse("CUST67890");
        try {
            Object name = extractor.extractFieldFromResponse(anotherResponse, "name");
            System.out.println("提取到的 'name' 字段值: " + name); // 预期输出: Test Customer
            Object nonExistentField = extractor.extractFieldFromResponse(anotherResponse, "address");
            System.out.println("提取到的 'address' 字段值: " + nonExistentField); // 预期输出: null
        } catch (IOException e) {
            System.err.println("处理另一个响应时发生错误: " + e.getMessage());
        }

        System.out.println("\n----------------------------------\n");

        // 模拟一个空的或无效的响应（例如，状态码非200）
        Response badResponse = Response.status(Response.Status.BAD_REQUEST).entity("{\"error\":\"Invalid input\"}").build();
        try {
            Object error = extractor.extractFieldFromResponse(badResponse, "error");
            System.out.println("提取到的 'error' 字段值 (来自错误响应): " + error); // 预期输出: Invalid input
        } catch (IOException e) {
            System.err.println("处理错误响应时发生错误: " + e.getMessage());
        }
    }
}