如何在Bash函数中用Awk动态列号优化IP分段提取？

问题背景 #

需要编写一个Bash函数提取给定IP字符串的指定分段（第1-4段），原实现用case结构虽可行但代码冗余，尝试简化时遇到awk变量传递问题，无法正确输出对应分段。

原冗余代码 #

#!/bin/bash

function get_ip_segment() {
    case "$2" in
        1)
            printf "%s" "$(echo $1 | awk -F '.' '{print $1}')"
            ;;
        2)
            printf "%s" "$(echo $1 | awk -F '.' '{print $2}')"
            ;;
        3)
            printf "%s" "$(echo $1 | awk -F '.' '{print $3}')"
            ;;
        4)
            printf "%s" "$(echo $1 | awk -F '.' '{print $4}')"
            ;;
        *)
            printf "%s" "$1"
            ;;
    esac
}

IP="12.34.56.78"
echo $IP

echo "$(get_ip_segment $IP 1)"
echo "$(get_ip_segment $IP 2)"
echo "$(get_ip_segment $IP 3)"
echo "$(get_ip_segment $IP 4)"

原代码输出：

12.34.56.78
12
34
56
78

失败的简化尝试 #

尝试移除case结构，但输出不符合预期：

printf "%s" "$(echo $1 | awk -F '.' '{print column_id}' column_id=$2)"

错误输出：

1
2
3
4

问题原因 #

你的awk写法存在两个核心问题：

1. 直接在print语句中写column_id，awk会将其视为字符串字面量输出，而非引用对应的字段。
2. 通过命令行参数传递变量（column_id=$2）的方式，需要通过ARGV数组访问，而非直接使用变量名，这种方式不适合字段引用场景。

优化后的单行函数 #

用-v选项将外部变量传递给awk，同时用$col引用对应字段，实现简洁的单行函数：

get_ip_segment() { awk -v col="$2" -F '.' '{print (col >= 1 && col <= 4) ? $col : $0}' <<< "$1"; }

代码解释 #

• -v col="$2"：将Bash变量$2（指定的分段序号）传递给awk内部的col变量。
• (col >=1 && col <=4) ? $col : $0：三元表达式，判断分段序号是否在1-4范围内，是则输出对应字段，否则输出整个原字符串。
• <<< "$1"：用Here字符串替代echo $1 |，避免创建子进程，执行更高效。

测试验证 #

IP="12.34.56.78"
echo "$IP"
echo "$(get_ip_segment "$IP" 1)"
echo "$(get_ip_segment "$IP" 2)"
echo "$(get_ip_segment "$IP" 3)"
echo "$(get_ip_segment "$IP" 4)"
echo "$(get_ip_segment "$IP" 5)" # 测试超出范围的情况

输出结果：

12.34.56.78
12
34
56
78
12.34.56.78

内容的提问来源于stack exchange，提问作者zappee