Linux Shell 脚本

Published on 2022-08-01

Shell for 循环列表输出

# 定义数组
export a=("我" "美丽")
export b=("爱" "的")
export c=("你" "中国")

#循环 当 i = 0 时 ，输出列表 a 中下标为 0 的值，当 i = 1 时，输出列表 a 中下标为 1 的值
for i in 0 1
do
	echo ${a[i]} ${b[i]} ${c[i]}
done

Shell 原格式输出内容到一个文件，如 `docker-compose.yml` 与 `.env`

# 1. 原内容输出至 docker-compose.yml ，变量不解析，'EOF'
function ECHO_DOCKER-COMPOSE.YML() {
  cat > ./docker-compose.yml <<'EOF'
version: '3.5'
services:
  postgres:
    image: postgres:13-alpine
    restart: always
    env_file: .env
    volumes:
      - ./data/pg_data:/var/lib/postgresql/data
  redis:
    image: redis:6-alpine
    restart: always
    command: [ "redis-server", "--save", '""', "--appendonly", "no" ]
  clickhouse:
    image: yandex/clickhouse-server:21.8
    restart: always
    ulimits:
      nofile:
        soft: 262144
        hard: 262144
    volumes:
      - ./data/ch-data:/var/lib/clickhouse
  minio:
    image: quay.io/minio/minio:RELEASE.2022-08-02T23-59-16Z.fips
    restart: always
    env_file: .env
    ports:
      - ${MINIO_PORT}:9000
    volumes:
      - ./data/minio-data:/data
    command: server /data 
EOF
}

# 2. 原内容变量解析输出至 .env ，变量解析，EOF
function ECHO_ENV() {
HOSTNAME_IP=$(hostname -I | awk '{print $1}')
MINIO_PORT=9000
  cat > ./.env <<EOF
#变量
MINIO_SERVER_URL=http://${HOSTNAME_IP}:${MINIO_PORT}/
MINIO_ROOT_USER=datarc
MINIO_ROOT_PASSWORD=
MINIO_BUCKET=datarc

# PostgreSQL 相关配置
POSTGRES_USER=postgres
POSTGRES_PASSWORD=example
POSTGRES_DB=postgres
PGDATA=/var/lib/postgresql/data

# 端口冲突时请修改
MINIO_PORT=9000
EOF
}

Shell 设置 echo 输出字体颜色

一般输出为三种颜色，正确，警告，错误，错误会退出 exit，不会向下执行。

cat > ./echo.sh <<'EOF'
#!/bin/bash
###############  设置 echo 输出字体颜色   ############################################
function echo_info() {
  echo '========================================================================================='
  echo -en "\033[1;32m""\033[3m$1 \033[0m\033[90G \033[1;32m\033[3m [ SUCCES ] \033[0m\n"
}

function echo_warning() {
  echo '========================================================================================='
  echo -en "\033[1;33m""\033[3m$1 \033[0m\033[90G \033[1;33m\033[3m [ warning ] \033[0m\n"
}

function echo_error() {
  echo '========================================================================================='
  echo -en "\033[1;31m""\033[3m$1 \033[0m\033[90G \033[1;31m\033[3m [ ERROR ] \033[0m\n"
  exit 2
}
EOF

# 测试
cat > ./echo.sh <<'EOF'
#!/bin/bash
###############  设置 echo 输出字体颜色   ############################################
function echo_info() {
  echo '========================================================================================='
  echo -en "\033[1;32m""\033[3m$1 \033[0m\033[90G \033[1;32m\033[3m [ SUCCES ] \033[0m\n"
}

function echo_warning() {
  echo '========================================================================================='
  echo -en "\033[1;33m""\033[3m$1 \033[0m\033[90G \033[1;33m\033[3m [ warning ] \033[0m\n"
}

function echo_error() {
  echo '========================================================================================='
  echo -en "\033[1;31m""\033[3m$1 \033[0m\033[90G \033[1;31m\033[3m [ ERROR ] \033[0m\n"
  exit 2
}
echo 1
echo_info "开始计数"
echo 2
echo_warning "上次输出为 2"
echo 3
echo_error "上次输出为 3 错误，不会执行 4"
echo 4
EOF

输出结果为下，错误不会向下执行，exit 退出

Shell 接受外部传参，例如–version=1.1.1，如没有传参，设置默认值为1.1.0

# 1. 设置 -- 外部传参，并设置默认值

cat > ./install.sh <<'EOF'
#!/bin/bash
# 执行命令 bash install.sh --path=<路径> --username=<用户名> --image_version=<X.X.X>  
# 说明 --username 为必填值
###############  设置 echo 输出字体颜色   ############################################
function echo_info() {
  echo '==================================================================================================='
  echo -en "\033[1;32m""\033[3m$1 \033[0m\033[90G \033[1;32m\033[3m [ SUCCES ] \033[0m\n"
}

function echo_warning() {
  echo '==================================================================================================='
  echo -en "\033[1;33m""\033[3m$1 \033[0m\033[90G \033[1;33m\033[3m [ warning ] \033[0m\n"
}

function echo_error() {
  echo '==================================================================================================='
  echo -en "\033[1;31m""\033[3m$1 \033[0m\033[90G \033[1;31m\033[3m [ ERROR ] \033[0m\n"
  exit 2
}
###############  --path  可设置参数 ############################################
function main() {
  ARGS=$(getArgs "$@")
  path=$(echo "$ARGS" | getNamedArg path) ; [ ! $path ] && path='/opt/datarc'
  username=$(echo "$ARGS" | getNamedArg username) ; [ ! $username ] && echo_error "--username 为必填参数，请重新执行"
  image_version=$(echo "$ARGS" | getNamedArg image_version) ; [ ! $image_version ] && image_version='2.8.0'
}
function getArgs() {
  for arg in "$@"; do
    echo "$arg"
  done
}
function getNamedArg() {
  ARG_NAME=$1

  sed --regexp-extended --quiet --expression="
        s/^--$ARG_NAME=(.*)\$/\1/p  # Get arguments in format '--arg=value': [s]ubstitute '--arg=value' by 'value', and [p]rint
        /^--$ARG_NAME\$/ {          # Get arguments in format '--arg value' ou '--arg'
            n                       # - [n]ext, because in this format, if value exists, it will be the next argument
            /^--/! p                # - If next doesn't starts with '--', it is the value of the actual argument
            /^--/ {                 # - If next do starts with '--', it is the next argument and the actual argument is a boolean one
                # Then just repla[c]ed by TRUE
                c TRUE
            }
        }
    "
}
main "$@"
#####################################################################
echo_info "您的 安装路径为 ${path} , 您的 安装版本为 ${image_version}"
EOF