如何在SpringBoot中提取GCP PubSub消息data中的event_type并处理？

问题分析 #

你收到的PubSub消息是嵌套JSON结构，外层data键对应一个子对象，event_type就在这个子对象里。原代码用HashMap<String, String>映射会因为类型不匹配（子对象不是字符串）导致无法直接获取event_type，需要调整解析方式。

解决方案 #

1. 嵌套HashMap取值（快速适配现有代码）

修改HashMap泛型，先获取内层data对象，再提取event_type：

public void process(final PubsubMessage message) throws ConsumerException {
    String data = message.getData().toStringUtf8();

    try {
        ObjectMapper mapper = new ObjectMapper();
        // 外层Map的value为Object类型，对应内层JSON对象
        HashMap<String, Object> outerMap = mapper.readValue(data, HashMap.class);
        // 强转内层Map
        HashMap<String, String> innerData = (HashMap<String, String>) outerMap.get("data");
        String eventType = innerData.get("event_type");
        
        // 业务逻辑处理
        if ("REGISTRATION".equals(eventType)) {
            registrationHandler.handle(message);
        } else {
            // 其他类型处理逻辑
        }
    } catch (JsonProcessingException e) {
        throw new ConsumerException("解析PubSub消息失败", e);
    }
}

2. 实体类映射（SpringBoot推荐方案）

创建对应JSON结构的实体类，通过对象映射更规范，避免类型转换问题：

首先定义内层数据实体类：

import com.fasterxml.jackson.annotation.JsonProperty;

public class EventData {
    private String systemId;
    private String userId;
    private String departmentId;
    private String enrolmentDateTime;
    
    @JsonProperty("event_type")
    private String eventType;

    // 生成getter、setter方法
    public String getEventType() {
        return eventType;
    }

    public void setEventType(String eventType) {
        this.eventType = eventType;
    }

    // 其他字段的getter/setter省略
}

再定义外层消息实体类：

public class PubSubEventMessage {
    private EventData data;

    // 生成getter、setter方法
    public EventData getData() {
        return data;
    }

    public void setData(EventData data) {
        this.data = data;
    }
}

修改解析代码：

public void process(final PubsubMessage message) throws ConsumerException {
    String data = message.getData().toStringUtf8();

    try {
        ObjectMapper mapper = new ObjectMapper();
        PubSubEventMessage eventMessage = mapper.readValue(data, PubSubEventMessage.class);
        String eventType = eventMessage.getData().getEventType();
        
        if ("REGISTRATION".equals(eventType)) {
            registrationHandler.handle(message);
        } else {
            // 其他处理逻辑
        }
    } catch (JsonProcessingException e) {
        throw new ConsumerException("解析PubSub消息失败", e);
    }
}

3. JsonNode解析（灵活处理不确定结构）

如果消息结构可能变化，用JsonNode可避免实体类限制：

public void process(final PubsubMessage message) throws ConsumerException {
    String data = message.getData().toStringUtf8();

    try {
        ObjectMapper mapper = new ObjectMapper();
        JsonNode rootNode = mapper.readTree(data);
        JsonNode dataNode = rootNode.get("data");
        // 提取event_type，字段不存在时设置默认值
        String eventType = dataNode.get("event_type").asText("UNKNOWN");
        
        if ("REGISTRATION".equals(eventType)) {
            registrationHandler.handle(message);
        } else {
            // 其他处理逻辑
        }
    } catch (JsonProcessingException e) {
        throw new ConsumerException("解析PubSub消息失败", e);
    }
}

注意事项 #

• 原代码HashMap<String, String>泛型错误，外层data对应的值是JSON对象而非字符串，直接映射会抛出类型转换异常。
• 建议使用"REGISTRATION".equals(eventType)避免eventType为null时触发空指针异常。

内容的提问来源于stack exchange，提问作者Namrata