如何在Spring Integration中移除MessageHandlingException中的PII信息？

要避免Spring Integration抛出的MessageHandlingException日志中包含payload和headers里的PII数据，有几种可行的方案，以下是具体实现：

1. 自定义脱敏消息类 #

默认GenericMessage的toString()方法会完整打印payload和headers，我们可以继承它并重写该方法，只输出非敏感信息（如消息ID）：

public class SafeGenericMessage<T> extends GenericMessage<T> {

    public SafeGenericMessage(T payload) {
        super(payload);
    }

    public SafeGenericMessage(T payload, Map<String, Object> headers) {
        super(payload, headers);
    }

    @Override
    public String toString() {
        return "SafeGenericMessage [id=" + getId() + "]";
    }
}

在构建消息时使用这个子类：

return MessageBuilder.withPayload("Example")
        .build(SafeGenericMessage::new);

优点：从源头控制消息的字符串输出，所有涉及该消息的日志都会自动脱敏；缺点：需要统一替换所有消息构建的地方，适合新项目或重构场景。

2. 自定义异常处理器过滤敏感信息 #

通过Spring Integration的错误处理机制，捕获MessageHandlingException后重新包装异常，移除敏感的消息上下文：

@Bean
public ErrorHandler sanitizingErrorHandler() {
    return throwable -> {
        if (throwable instanceof MessageHandlingException) {
            MessageHandlingException mhe = (MessageHandlingException) throwable;
            String sanitizedMsg = String.format("处理消息出错，消息ID: %s", mhe.getFailedMessage().getId());
            // 打印脱敏后的异常日志
            LoggerFactory.getLogger(getClass())
                    .error(sanitizedMsg, mhe.getCause());
        } else {
            // 其他异常正常处理
            LoggerFactory.getLogger(getClass())
                    .error("未知错误", throwable);
        }
    };
}

将该处理器配置到你的Integration组件中，例如：

@Bean
public IntegrationFlow errorHandlingFlow() {
    return IntegrationFlows.from("errorChannel")
            .handle(sanitizingErrorHandler())
            .get();
}

优点：无需修改消息构建逻辑，集中处理异常日志；缺点：需要确保所有异常都流经该错误通道。

3. 日志框架层面过滤敏感内容 #

借助日志框架（如Logback）的自定义转换器，在日志输出阶段替换掉MessageHandlingException中的敏感部分：

步骤1：实现Logback转换器 #

public class SanitizedExceptionConverter extends ThrowableConverter {

    @Override
    protected String throwableToString(Throwable throwable) {
        if (throwable instanceof MessageHandlingException) {
            MessageHandlingException mhe = (MessageHandlingException) throwable;
            String originalLog = mhe.toString();
            // 用正则匹配并替换敏感的消息内容
            return originalLog.replaceAll(
                "failedMessage=GenericMessage\\[payload=.*?, headers=\\{.*?\\}]",
                "failedMessage=SanitizedMessage[id=" + mhe.getFailedMessage().getId() + "]"
            );
        }
        return super.throwableToString(throwable);
    }
}

步骤2：配置Logback #

在logback.xml中注册转换器并使用：

<conversionRule conversionWord="sanitizedException" 
                converterClass="com.example.SanitizedExceptionConverter" />

<appender name="CONSOLE" class="ch.qos.logback.core.ConsoleAppender">
    <encoder>
        <pattern>%d{HH:mm:ss.SSS} [%thread] %-5level %logger{36} - %sanitizedException%n</pattern>
    </encoder>
</appender>

优点：无需修改业务代码，对应用透明；缺点：依赖日志框架的特性，正则匹配可能存在边界情况。

4. 手动捕获异常并脱敏上下文 #

在业务方法中提前捕获异常，抛出不包含原始消息的自定义异常：

public Message<?> hasErrorAndPii(Message<?> message) throws Exception {
    try {
        if (1 == 1)
            throw new Exception("Example");
        return MessageBuilder.withPayload("Example").build();
    } catch (Exception e) {
        // 只携带消息ID等非敏感信息
        throw new RuntimeException("处理消息失败，ID: " + message.getId(), e);
    }
}

注意：Spring Integration仍会将该异常封装为MessageHandlingException，但此时可结合前面的脱敏消息类或日志过滤方案进一步处理。

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