bash

Question

我的一些实用程序函数通常共享完全相同的初始代码，这允许使用参数或管道输入调用函数：

RED=$(tput setaf 1)
YELLOW=$(tput setaf 3)
RESET=$(tput sgr0)

# -----------

function yellow {
  local arg
  if (( "$#" == 0 )); then
      IFS= read -r arg
      set -- "$arg"
  fi
  echo "${YELLOW}${RESET}"
}
function red {
  local arg
  if (( "$#" == 0 )); then
      IFS= read -r arg
      set -- "$arg"
  fi
  echo "${RED}${RESET}"
}

是否有一种技术可以让我“重用”或以某种方式将代码的相同部分引入函数定义中？在我的示例中，这将是：

local arg
if (( "$#" == 0 )); then
  IFS= read -r arg     
  set -- "$arg"
fi

Answer 1

不，但您可以通过以下方式缩短此时间：

set -- "${1:-$(cat)}"

Answer 2

要从某些脚本中获取部分源变量 and/or 函数，请使用此方法：

$ cat test.sh
#!/bin/bash

test=123
mess() { echo ok; }
# Vars and functions ends here ^^
[[  =~ source ]] && return
mess

然后像这样制作

. ./test source

bash - 在不同的函数定义中重用代码片段

bash - reuse code snippet in different function definitions

code-reuse

function