NiFi中DistributedMapCacheClientService的handShakeHandler为null问题求助

问题背景 #

基于NiFi 1.20.0版本搭建DistributedMapCacheClientService与DistributedMapCacheServer集群，客户端连接服务端时偶尔抛出如下异常：

java.io.IOException: Request invocation failed
    at org.apache.nifi.distributed.cache.client.CacheClientRequestHandler.invoke(CacheClientRequestHandler.java:103)
    at org.apache.nifi.distributed.cache.client.MyDistributedCacheClient.invoke(MyDistributedCacheClient.java:69)
    at org.apache.nifi.distributed.cache.client.NettyDistributedMapCacheClient.getAndPutIfAbsent(NettyDistributedMapCacheClient.java:140)
    ...
Caused by: java.io.IOException: Channel unregistered before processing completed: [id: 0x7288e0ed, L:0.0.0.0/0.0.0.0:38868]
    at org.apache.nifi.distributed.cache.client.CacheClientRequestHandler.channelUnregistered(CacheClientRequestHandler.java:70)
    ...

经远程调试NiFi JVM实例，确认异常触发时CacheClientRequestHandler中的handShakeHandler字段为null。

核心原因分析 #

1. 握手流程未完成前通道被注销 #

NiFi分布式缓存客户端基于Netty实现，CacheClientRequestHandler依赖HandshakeHandler完成与服务端的握手（版本协商、安全校验等）。handShakeHandler仅在通道初始化且握手流程完成后才会被赋值，若通道在握手阶段就被关闭/注销，会导致该字段始终为null：

• 服务端因资源耗尽、负载过高主动拒绝新连接或关闭未完成握手的连接
• 网络波动导致客户端与服务端的连接在握手阶段中断
• 客户端连接超时配置过短，握手未完成就触发超时关闭

2. 并发请求与通道初始化的竞态 #

高并发场景下，业务线程发起缓存请求的时机早于Netty IO线程完成通道初始化与握手的时机：

• 业务线程调用getAndPutIfAbsent等方法时，通道仍处于握手流程中，handShakeHandler尚未被初始化
• 若此时通道因异常触发注销逻辑，CacheClientRequestHandler.channelUnregistered方法被调用时，handShakeHandler仍为null，最终抛出异常

3. 连接池复用失效的连接 #

DistributedMapCacheClientService默认复用连接池中的连接，若连接池内的连接因服务端超时被回收，但客户端未及时感知：

• 复用该失效连接时，握手流程无法正常完成，handShakeHandler无法被初始化
• 后续请求触发通道注销，导致handShakeHandler为null的异常

触发场景总结 #

• 服务端负载突增，主动拒绝新连接或关闭空闲/未完成握手的连接
• 网络不稳定，握手阶段出现丢包、延迟过高导致连接中断
• 客户端连接超时、空闲超时等参数配置不合理，连接在握手完成前被关闭
• 高并发场景下，业务请求与通道初始化流程存在竞态

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