Передать stdio в функцию

Проблема с xargs заключается в том, что он использует подоболочку, поэтому ваша функция не наследуется в новой оболочке.

Если вы создадите новую команду, доступную в PATH, вы можете использовать xargs и добавить функцию, позволяющую динамически использовать STDIN или аргументы::

~/bin/sqr:

#!/bin/bash

if [[ $1 ]]; then
    arg=$1
else
    arg=$(cat)
fi
echo $(( $arg * $arg ))

chmod +x ~/bin/sqr
PATH=~/bin:$PATH

Сейчас:

$ echo 6 | xargs -n1 sqr
36

$ echo 6 | sqr 
36

$ sqr 6
36

Часть arg=$(cat) хлебает STDIN в переменной arg.

Одним из способов может быть использование расширения параметра на $1, чтобы cat сбросить STDIN в качестве резерва:

$ sqr() { n=${1:-$(cat)}; echo $((n*n)); }

$ sqr 6
36

$ echo 5 | sqr
25

$ echo 5 | sqr 6  # $1 takes precedence over STDIN
36

Параметры и стандартный ввод — это два совершенно разных способа взаимодействия вызывающего объекта с функцией:

sqr_param () {
    local n
    n=$1
    echo $(( $n * $n ))
}

sqr_stdin () {
    local n
    read -r n
    echo $(( $n * $n ))
}

Записав их обе в довольно подробной форме, вы увидите, что одна («нечистая») версия может быть тривиально реализована в терминах другой («чистой») версии.

# This one focuses on math
sqr () {
    echo $(( $1 * $1 ))
}

# This one provides an interface with the outside world
sqr_stdin () {
    local n
    read -r n
    sqr "$n"
}

Я бы сопротивлялся искушению реализовать одну функцию, которая могла бы обрабатывать либо аргумент, либо стандартный ввод. «Столько сложностей в программном обеспечении возникает из-за попыток заставить одну вещь выполнять две вещи».

$ sqr() {
    local n tmpfd
    exec {tmpfd}<&0             # Save value of FD 0 (stdin) in "tmpfd"
    (( $# )) && exec <<<"$1"    # If an argument passed in create a
                                # temp file, store that value in it,
                                # then set FD 0 to point to that file.
    IFS= read -r n              # Read a value from FD 0
    exec 0<&"$tmpfd"            # Restore FD 0 to the value in "tmpfd"
    echo $(( $n * $n ))         # Print the result of the math
}

$ sqr 4
16

$ echo 4 | sqr
16

В приведенном выше примере используется именованный файловый дескриптор tmpfd, доступный в bash 4.1 или более поздней версии. См. https://mywiki.wooledge.org/FileDescriptor для получения дополнительной информации о файловых дескрипторах.

Альтернативно, если вы определяете тело функции в подоболочке, вам не нужно сохранять и восстанавливать FD 0, поскольку директива exec будет применяться только внутри подоболочки:

$ sqr() {
  (                             # Start a subshell
    local n
    (( $# )) && exec <<<"$1"    # If an argument passed in create a
                                # temp file, store that value in it,
                                # then set FD 0 to point to that file.
    IFS= read -r n              # Read a value from FD 0
    echo $(( $n * $n ))         # Print the result of the math
  )                             # End the subshell
}

$ sqr 4
16

$ echo 4 | sqr
16