Shell脚本函数与数组操作实战指南

1. Shell脚本中的函数基础

在Shell编程中，函数是将一组命令封装为可重复调用单元的核心机制。与C语言等编译型语言不同，Shell函数的定义更加灵活，不需要声明返回类型和参数类型。典型的函数定义语法如下：

bash复制function_name() {
    command1
    command2
    ...
    return $?
}

注意：虽然可以使用function关键字定义（如function name {...}），但POSIX标准推荐使用更简洁的name() {...}形式以保证兼容性。

函数调用时直接写函数名即可，参数传递通过位置参数实现。例如调用print_msg "Hello"时，函数内通过$1获取第一个参数。这种设计使得Shell函数特别适合封装常用操作逻辑。

我在实际脚本开发中发现几个关键点：

1. 函数内定义的变量默认是全局的，除非使用local关键字声明
2. 返回值只能是0-255的整数，通过return返回
3. 要返回字符串数据，通常采用输出到stdout再捕获的方式

2. 数组操作的核心技巧

2.1 数组定义与初始化

Shell数组分为索引数组和关联数组两种。索引数组是最常用的形式，定义方式如下：

bash复制# 直接赋值
files=(/etc/*.conf)

# 逐个元素赋值
colors[0]="red"
colors[1]="green"

# 从命令输出初始化
processes=($(ps -ef | awk '{print $2}'))

关联数组需要先声明类型，这在处理键值对数据时非常有用：

bash复制declare -A user_info
user_info["name"]="Alice"
user_info["age"]=25

2.2 数组遍历的三种方式

1. 标准for循环遍历：

bash复制for i in "${files[@]}"; do
    echo "Processing file: $i"
done

2. C风格for循环（适用于索引数组）：

bash复制for ((i=0; i<${#colors[@]}; i++)); do
    echo "Color $i: ${colors[i]}"
done

3. 键值遍历（关联数组专用）：

bash复制for key in "${!user_info[@]}"; do
    echo "$key => ${user_info[$key]}"
done

经验：使用"${array[@]}"而非${array[*]}可以正确处理包含空格的元素

3. 函数与数组的实战组合

3.1 数组作为函数参数

虽然Shell函数不能直接接收数组作为参数，但可以通过以下技巧实现：

bash复制process_array() {
    local -n arr=$1  # 使用nameref特性
    for item in "${arr[@]}"; do
        # 处理每个元素
    done
}

files=(/etc/*.conf)
process_array files

3.2 函数返回数组

通过echo输出数组元素并用括号捕获：

bash复制get_users() {
    local users=($(cut -d: -f1 /etc/passwd))
    echo "${users[@]}"
}

user_array=($(get_users))

4. 高级应用场景解析

4.1 矩阵运算模拟

利用二维数组模拟矩阵操作：

bash复制declare -A matrix
rows=3
cols=3

# 初始化矩阵
for ((i=0;i<rows;i++)) {
    for ((j=0;j<cols;j++)) {
        matrix[$i,$j]=$((i+j))
    }
}

# 矩阵打印函数
print_matrix() {
    for ((i=0;i<rows;i++)) {
        for ((j=0;j<cols;j++)) {
            printf "%2d" ${matrix[$i,$j]}
        }
        echo
    }
}

4.2 配置文件解析器

结合函数和数组实现配置解析：

bash复制parse_config() {
    declare -gA config
    while IFS='=' read -r key value; do
        config["$key"]="$value"
    done < "$1"
}

parse_config "/etc/myapp.conf"
echo "Server: ${config[server]}"

5. 性能优化与陷阱规避

5.1 数组操作的性能考量

1. 大数组操作时，避免频繁的数组复制
2. 关联数组的查找比索引数组慢，数据量大时应考虑其他方案
3. 使用unset删除数组元素比置空更高效

5.2 常见错误排查

1. 未声明关联数组导致错误：

bash复制declare -A dict  # 必须先声明
dict["key"]="value"

2. 数组元素包含特殊字符时的处理：

bash复制# 错误方式
items=($(find . -type f))

# 正确方式
mapfile -t items < <(find . -type f -print0)

3. 函数内修改全局数组的注意事项：

bash复制modify_array() {
    local -n arr=$1
    arr+=(new_item)  # 实际修改了原数组
}

6. 实用代码片段库

6.1 数组去重函数

bash复制array_unique() {
    local -n arr=$1
    local -A seen
    local -a result
    for i in "${arr[@]}"; do
        [ -z "${seen[$i]}" ] && {
            result+=("$i")
            seen[$i]=1
        }
    done
    echo "${result[@]}"
}

6.2 数组排序实现

bash复制array_sort() {
    local -n arr=$1
    IFS=$'\n' sorted=($(sort <<<"${arr[*]}"))
    echo "${sorted[@]}"
}

6.3 数组交集/差集计算

bash复制array_intersect() {
    local -a arr1=("${!1}") arr2=("${!2}")
    declare -A tmp
    local -a res
    
    for i in "${arr1[@]}"; do
        tmp["$i"]=1
    done
    
    for i in "${arr2[@]}"; do
        [[ ${tmp["$i"]} ]] && res+=("$i")
    done
    
    echo "${res[@]}"
}

在实际脚本开发中，我发现将常用数组操作封装成函数库可以显著提高开发效率。特别是在处理日志分析、批量文件操作等场景时，合理运用数组和函数能让代码更加清晰可维护。