如何使用Shell命令自动化生成用于Postgres的Docker镜像

#!/bin/bash

# 创建文件并写入内容
 > docker-compose.yml

cat<>docker-compose.yml
services:
    postgres:
        image: postgres:18-alpine
        container_name: ${container_name:= $db}

        environment:
            POSTGRES_USER: $user
            POSTGRES_PASSWORD: $pwd
            POSTGRES_database: $db

        ports:
            - ${port:=5432}:5432
        volumes:
            - ${vol_name:=database-data}:/var/postgresql

 volumes:
    ${vol_name:=database-data}:

EOF

Bash参数与标志

通过使用Bash参数，你可以在执行脚本时在同一行中传递多个值。要使用参数，只需在执行脚本时将这些值放在同一行中即可，例如：

bash script val1 val2 

如果要读取某个参数的值，可以使用美元符号$后跟上该参数在输入时的位置编号（位置编号从1开始，而不是0）。

举个例子，如果你想输出第二个参数的内容，就可以使用echo $2命令。

为了更好地理解这一点，让我们通过使用参数来动态传递数据库凭据，从而优化数据输入流程：

#!/bin/bash

# 数据库凭据

if [[ -z $1 ]]; then
    read -p '数据库名称(*): ' db
 else
    db="$1"
fi

# 检查并验证数据库信息

if [[ -z $2 ]; then
    read -p '数据库用户名(*): ' user
 else
    user="$2"
fi

# 检查并验证用户名

if [[ -z $1 ]]; then
    read -sp '数据库密码(*): ' pwd
 else
    pwd=$3
fi

# 验证密码

首先，脚本会检查是否有参数传入。如果没有参数，就会提示用户输入相关信息；如果有参数，这些参数的值会被赋给相应的变量，用于表示数据库的名称、密码等信息。现在你可以带着参数来运行这个脚本了：

bash image.sh customers ABC-Company secret123!

通过上述脚本创建的容器将使用customers作为数据库名称，ABC-Company作为用户名，密码为secret123!。

Bash标志/选项

脚本中有一部分会要求用户提供关于所创建容器的信息，但这些信息其实是可选的。通过使用-D标志，你可以跳过这一环节，从而让脚本直接使用默认的容器配置信息来运行。

下面是一个例子：

#!/bin/bash

while getopts "ny" options; do
	case "$options" in
		n) echo "你选择了“不”；";
		y) echo "你选择了“是”；";
		*) echo “不存在这样的选项/标志”； exit 1;;
    esac
done

shift $((OPTIND - 1))

在上述脚本中：

1. 只要`getopts`的结果不为空，while循环就会持续运行。
2. `getopts`用于获取脚本中传递的参数选项。
3. `"ny"`是我们期望用户在运行脚本时输入的选项。
4. `options`是一个变量，用于记录用户选择的“n”或“y”选项。
5. `do`命令用于启动循环。
6. `case`语句通过`in`关键字判断用户选择的选项是否在预定义的选项列表中。
7. 我们通过双分号`;)`来定义当用户使用`-n`或`-y`选项运行脚本时应该执行的操作。
8. 使用通配符`*`可以处理那些未被明确指定的输入选项。
9. `esac`命令用于结束`case`语句块。
10. `done`命令用于终止循环，防止循环无限期地继续执行。
11. `shift $((OPTIND - 1))`这一操作可以避免脚本在同时接收参数和选项时可能出现的冲突，它会将剩余的参数向左移动，从而将选项从参数列表中移除。

如果在运行脚本时传递了`-D`选项，就可以使用这种语法来跳过请求容器信息的步骤。

#!/bin/bash

ask_container_info=true

while getopts "D" options; do
	case "$options" in
		D) ask_container_info=false;;
		*) echo "没有这样的选项/标志"; exit 1;;
    esac
done

shift $((OPTIND - 1))

if [[ $ask_container_info == true ]]; then
	# 提示用户输入自定义容器信息
    read -p '容器名称: ' container_name
    read -p '容器端口（如果省略，容器将使用PostgreSQL的默认端口5432）: ' port
    read -p '卷名: ' vol_name
fi

这里只有一个选项，因此选项列表中只列出了`D`。变量`ask_container_info`的初始值为`true`，这意味着除非用户明确输入了`-D`选项，否则脚本会提示用户输入容器信息。当用户执行脚本并传递`-D`选项时，`ask_container_info`的值会被设置为`false`。

如何使你的脚本可在电脑路径中被执行

下一步是让这个脚本能够在你使用的所有终端中被执行。目前，这个脚本只能在你创建和编写它的当前目录中被运行。

首先，我们需要将脚本文件设置为可执行文件，并去掉文件的扩展名，这样就不必每次运行脚本时都加上`.sh`扩展名了。

如何使文件可执行

要使一个文件可执行，可以使用Linux系统的`chmod`命令。这个命令用于修改文件的权限设置，比如写入权限、读取权限和执行权限等。

chmod +x image

在这里，x命令会使文件具有可执行性，而你指定的文件名image就是想要被设置为可执行文件的文件。

如何将你的目录添加到system/machine路径中

为了让你在打开的任何终端中都能使用这些脚本，无论当前处于哪个目录下，你都需要将这些脚本添加到系统中存储所有全局脚本和命令的路径中。

具体操作方法是：复制你的目录的完整路径，然后在终端中输入以下命令：

export PATH="$PATH:<YOUR_DIRECTORY_LOCATION>"

将替换成你的目录路径。这条命令会将该脚本目录添加到当前工作的终端环境中。但是，如果你尝试切换到另一个终端来运行这个脚本，它将会失败。

现在让我们将这个目录设置为全局路径，这样所有终端就能够执行这些脚本了：

nano ~/.bash_profile

执行这条命令后，你应该会看到类似下面的内容：

从上面的图片中你可以看到，我已经在文件中配置了路径信息。

如果你的文件中还没有这个配置，你可以将export PATH="$PATH:<YOUR_DIRECTORY_LOCATION>"这段代码复制并粘贴到文件中。完成后，按Ctrl+O保存文件，然后按Ctrl+x退出Nano编辑器。

接下来，再次重新加载系统路径，以便让这些脚本能够被包含在系统中：

source ~/.bash_profile

这样，你就可以在机器上的任何终端中，仅通过输入脚本文件名来运行它了（我给这个文件取名为image）。

总结

通过本指南，你学会了如何创建自己的脚本来自动化工作流程。希望这些内容能帮助你减少在那些可以通过脚本命令自动完成的重复性任务上花费的时间和精力。

我们在这个指南中作为示例使用的完整CLI工具，你可以从这个git仓库中获取。在那里，你还可以找到安装指南和使用技巧。如果你觉得这个资源有用，请考虑给这个仓库添加星标，并与可能从中受益的同事们分享它。

谢谢！