# 设置默认环境变量
export web_worker=${web_worker:-4}
export sandbox_worker=${sandbox_worker:-4}
export worker_concurrency=${worker_concurrency:-4}
export worker_queue=${worker_queue:-default}
export worker_process=${worker_process:-prefork}

# 删除旧的 socket 文件（如果存在）
if [ -e /var/run/supervisor.sock ]; then
  unlink /var/run/supervisor.sock
fi

supervisord -c supervisord.ini
if [ "${run_upgrade:-0}" == 1 ]; then
  echo "检查升级版本"
  # 执行命令，检查升级版本
  # 按逗号切分字符串，遍历每个包名执行 pip install --upgrade 命令
  IFS=',' read -ra package_list <<< "$upgrade_packages"
  for package in "${package_list[@]}"; do
      # 移除首尾空格
      package=$(echo "$package" | awk '{$1=$1};1')
      # 执行 pip install --upgrade 命令
      pip install --upgrade "$package" -i https://pypi.doubanio.com/simple
  done
fi
python init_system.py
if [ "${run_web:-0}" == 1 ]; then
  echo "开启web服务"
  # 执行命令，开启web服务
  supervisorctl start web_api
fi

if [ "${run_scheduler:-0}" == 1 ]; then
  echo "开启任务调度"
    supervisorctl start scheduler
fi

if [ "${run_sandbox:-0}" == 1 ]; then
  echo "开启sandbox"
    supervisorctl start sandbox
fi

if [ "${run_flower:-0}" == 1 ]; then
  echo "开启flower"
  # 以守护进程启动
  supervisorctl start celery_flower
fi

if [ "${run_worker:-0}" == 1 ]; then
  echo "开启celery worker"
  supervisorctl start celery_worker
fi

tail -f /dev/null