如何配置Gevent监控仅打印阻塞Celery Worker的堆栈信息？

要实现只在gevent检测到Celery Worker阻塞时打印堆栈，需要结合gevent的阻塞检测机制和Celery的Worker初始化流程，具体方案如下：

1. 提前配置GEVENT_MAX_BLOCKING_TIME #

GEVENT_MAX_BLOCKING_TIME是gevent检测阻塞的核心配置，必须在gevent初始化前设置（即Celery Worker启动前），可通过两种方式配置：

• 环境变量设置（推荐，无需修改代码）：

  export GEVENT_MAX_BLOCKING_TIME=2.0
  

  这里设置为2秒，意味着当某个greenlet阻塞时长超过2秒时，gevent会触发阻塞检测回调。
• 代码中设置（需确保在Worker启动前执行）：

  import os
  os.environ['GEVENT_MAX_BLOCKING_TIME'] = '2.0'
  

2. 注册阻塞检测回调函数 #

gevent在检测到阻塞时，会调用greenlet.settrace注册的trace函数，我们可以在该函数中判断阻塞事件并打印对应堆栈：

import sys
import traceback
import greenlet
import os
from celery.signals import worker_ready

# 公司内部日志器
import log

def blocking_trace(event, args):
    # 仅处理阻塞事件
    if event == 'block':
        greenlet_obj, block_time = args
        # 过滤掉未达到阈值的短时间阻塞
        threshold = float(os.environ.get('GEVENT_MAX_BLOCKING_TIME', 2.0))
        if block_time >= threshold:
            # 获取当前阻塞greenlet的堆栈信息
            frame = sys._current_frames().get(greenlet_obj.thread_ident)
            if frame:
                log.info(
                    extra={
                        "block_duration": f"{block_time:.2f}s",
                        "stack": traceback.extract_stack(frame)
                    }
                )

@worker_ready.connect
def on_worker_ready(**kwargs):
    # 在Worker启动完成后注册阻塞检测回调
    greenlet.settrace(blocking_trace)

3. 移除原有周期性监控逻辑 #

原有的setup_monitoring函数会定时轮询所有线程堆栈，导致持续输出，需完全移除该部分代码，避免重复检测。

关键说明 #

• GEVENT_MAX_BLOCKING_TIME依赖gevent默认的libev hub实现，无需额外配置即可生效。
• block事件会在greenlet阻塞结束后触发，此时能获取到实际阻塞时长，可精准过滤无效的短阻塞。
• 若需仅监控Celery任务执行的greenlet，可在trace函数中额外判断greenlet的上下文（比如检查greenlet的名称或执行函数是否为Celery任务处理逻辑），进一步缩小监控范围。

内容的提问来源于stack exchange，提问作者Amizan J.