linux下编写一个函数，计算目录下总文件大小

**Linux 下编写一个函数，计算目录下总文件大小**

在 Linux 系统中，计算目录下总文件大小是一个常见的需求。我们可以使用 `find` 命令来实现这个功能，但是如果要在脚本中使用，这个命令可能会比较麻烦。因此，我们可以编写一个自定义函数来完成这个任务。

**函数名称和参数**

我们将这个函数命名为 `get_dir_size()`，它接受两个参数：目录路径和选项（可选）。目录路径是必填的，而选项则是可选的，可以用于控制函数的行为。

bash#!/bin/bashget_dir_size() {
 local dir_path=$1 local options=${2:-}

 # ...
}

**函数内部逻辑**

在函数内部，我们首先需要检查传入的目录路径是否存在。如果不存在，则返回错误信息。然后，我们使用 `find` 命令来查找指定目录下的所有文件和子目录。

bashif [ ! -d "$dir_path" ]; then echo "Error: Directory '$dir_path' not found." >&2 return1filocal files=$(find "$dir_path" -type f)

接下来，我们使用 `du` 命令来计算这些文件的大小。由于我们可能需要考虑子目录中的文件，因此我们使用 `-r`选项来递归地计算所有子目录下的文件大小。

bashlocal size=$(du -r "$dir_path" | awk '{print $1}')

**函数返回值**

最后，我们将计算出的总文件大小作为函数的返回值。由于 `du` 命令输出的是字节数，因此我们需要进行转换以得到更易于理解的单位（例如 KB、MB 等）。

bash# Convert size to human-readable format (KB, MB, etc.)
local unit="B"
while ((size >=1024)); do size=$(bc -l <<< "scale=2; $size /1024")
 unit=${unit,,}
 if [ "$unit" = "kb" ]; then unit="MB"
 elif [ "$unit" = "mb" ]; then unit="GB"
 fidoneecho "${size} ${unit}"

**示例使用**

下面是一个示例，演示如何使用 `get_dir_size()` 函数：

bashget_dir_size /path/to/directory

这将输出目录 `/path/to/directory` 下的总文件大小。

**注意事项**

请注意，这个函数假设传入的目录路径是有效的 Linux 文件系统路径。如果传入的路径不正确，函数可能会返回错误信息。