如何让Gunicorn应用仅单个Worker执行指定初始化函数？

方法一：启动Gunicorn前先跑初始化脚本 #

把拉取制品的逻辑从Flask应用里拆出来，做成独立脚本，让容器先完成初始化再启动Gunicorn，这样不管多少个Worker，拉取只做一次：

• 单独写一个初始化脚本init_artifacts.py，把load_artifacts的逻辑放进去，确保脚本执行完才退出。
• 修改Docker启动命令，先执行初始化脚本再启动Gunicorn，比如：

  CMD ["sh", "-c", "python init_artifacts.py && gunicorn --workers=4 --threads=2 app:app"]
  

  容器启动时会先拉取好制品，所有Worker启动后直接用现成的文件，不会重复执行。

方法二：用文件锁控制单Worker执行 #

利用容器内共享的Docker卷做文件锁，保证只有第一个拿到锁的Worker执行拉取，其他直接跳过：

• 在load_artifacts函数里加文件锁逻辑，比如用Linux的fcntl模块（跨平台可以用portalocker库）：

  import fcntl
  import os
  
  def load_artifacts():
      lock_path = "/your/volume/path/artifact.lock"
      with open(lock_path, 'w') as lock_file:
          try:
              # 非阻塞方式拿排他锁
              fcntl.flock(lock_file, fcntl.LOCK_EX | fcntl.LOCK_NB)
              # 拿到锁后再检查文件是否存在，避免并发问题
              if not os.path.exists("/your/volume/path/artifact.file"):
                  pull_from_s3()  # 你的S3拉取逻辑
          except BlockingIOError:
              # 拿不到锁，说明已有Worker在拉取，直接跳过
              pass
          finally:
              fcntl.flock(lock_file, fcntl.LOCK_UN)
  

• 注意锁文件要存在于Docker卷中，确保所有Worker都能访问到。

方法三：用Gunicorn的pre_fork钩子 #

Gunicorn的pre_fork钩子在创建所有Worker前只执行一次（主进程中运行），把拉取逻辑放这里：

• 写一个Gunicorn配置文件gunicorn.conf.py：

  def pre_fork(server, worker):
      if not os.path.exists("/your/volume/path/artifact.file"):
          load_artifacts()  # 你的拉取函数
  

• 启动Gunicorn时指定配置文件：

  gunicorn --config gunicorn.conf.py --workers=4 app:app
  

  主进程先拉取好制品，Worker是fork出来的，直接继承文件系统状态，不会重复拉取。

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