记一次通过脚本来实现自定义容器的自动重启

1. 场景还原

现在我有一个自定义的Docker镜像，是基于基础镜像来构建的带有多个服务的镜像，镜像里面包含了两个服务，其中需要A服务先启动，然后B服务依赖A服务，需要等A服务启动成功后，才能正常对外提供服务。其中A服务启动成功后，可以通过 5005 端口来判断，A服务的端口就是 5005；

在刚开始使用的时候，每次都是先启动一个这个镜像的容器，然后进入容器内部，手动执行A服务的启动命令，然后等待A服务启动完成，再执行B服务的启动命令。

这种场景，如果在服务器不关机的情况下，服务一直是正常运行的。但是如果服务器关机重启了，或者需要更新镜像版本了，这就得人工来重复操作了。这就显得有点不太智能了，于是想着折腾一番，能不能整一个在容器启动的时候，执行某个脚本，这个启动脚本里面包含A服务、B服务的启动命令，并且还能维护下启动顺序，如果能把启动服务的日志输出到Docker容器的日志里，使用 docker logs 命令能实时的查看服务的日志，这样也是极好的。于是带着这个目的，开始折腾了。

2. 自定义启动脚本

针对上面的场景，先写一个shell脚本，来启动需要启动的服务；
其中 A服务的启动命令是：

rasa run --enable-api

A服务启动成功后，可以对外提供 5005 端口的 web服务；但是 A服务的启动过程有点慢，比如要几十秒或者一分钟，这个时间不确定；

B服务的启动命令是：

python3 /home/rasa_dev/bot_api.py

B服务启动成功后，可以对外提供 5006 端口的 web服务；

自定义脚本 wait_for_port.sh 内容如下：

#!/bin/bash

# 第一步：运行 rasa run --enable-api 命令
nohup rasa run --enable-api &

# 第二步：等待端口5005可用
while ! nc -z localhost 5005; do
  echo "Waiting for port 5005 to become available..."
  sleep 5
done

# 第三步：一旦端口可用，执行 bot_api.py
python3 /home/rasa_dev/bot_api.py &

# 防止容器退出
sleep infinity
[root@VM-0-5-centos aikg-bot_build]# 

脚本包含了需要启动的命令，并且有个判断A服务 5005端口的过程，在端口可用之后再启动 B 服务；

在这个脚本中，nc（netcat）命令用于检查端口是否可访问。while循环将持续检测端口，直到它变为可访问状态。一旦端口可用，脚本将执行bot_api.py。

刚开始我将上面两个启动命令的日志重定向到 /logs 文件夹下、比如 启动 B服务的命令为：

python3 /home/rasa_dev/bot_api.py >> /logs/bot_api.log 2>&1 &

这样在 B服务启动时，启动日志就会追加到 /logs/bot_api.log 文件中了。

然后启动镜像的时候将日志目录挂载出来，这样就可以在宿主机上查看服务的启动日志了，但是我还是感觉麻烦，所以又改成了上面的命令，移除了/logs/bot_api.log 2>&1 这个日志重定向的内容，这样就不会将日志重定向到文件，而是直接输出到stdout和stderr。当容器运行时，所有输出都会被捕获，这样就可以使用docker logs命令来查看它们：

docker logs <container-name-or-id>

如果日志文件很重要，需要留痕或者方便后面追溯的话，可以将日志重定向到文件中，这样方便追溯或者排查原因，不过不考虑日志内容丢失的话，直接输出到stdout和stderr是最简便的方式。

刚开始我的镜像里面是不包含 nc 这个命令的，我先使用当前版本的镜像，启动一个容器，然后进入容器内部安装这个nc命令：

比如我之前的镜像是：test-bot:1.0.0 ，使用下面命令直接启动一个容器：