☰

如何捕获 bash 数组中的 MySQL 结果集？

tag-icon • tag-icon bash shell-script mysql variable array
如何捕获 bash 数组中的 MySQL 结果集？

我在 CentOS 7 上使用 bash shell。我想从 shell 脚本运行 MySQL 查询并迭代每行结果。如果返回 4 行，我想我可以捕获数组中的四行，如下所示：

query="select p.id, p.ebook_id, es.id FROM ...";
echo "$query" > /tmp/query.sql
mysql -u user --password=pass db_id < /tmp/query.sql > /tmp/query.csv

linesIN=`cat /tmp/query.csv | sed 's/\t/,/g'`
arraylength=${#linesIN[@]}
echo $arraylength

但是，即使我可以看到返回的多个结果行，也$arraylength始终输出。1如何修改上述内容以正确创建结果数组，其中数组中的每个元素代表结果集中的一行？

答案1

随着--batch选项，mysql应将结果输出为一行中的一条记录，并且各列之间用制表符分隔。您可以使用以下命令将行读取到数组中巴什的mapfile和进程替换，或命令替换和数组赋值：

mapfile results  < <( mysql --batch ... < query.sql )

或者

set -f        # disable globbing
IFS=$'\n'     # set field separator to NL (only)
results=( $(mysql --batch ... ) )

（请注意，IFS在此之后保持修改状态并禁用通配符。）

然后，如果您想将一行的列拆分为一些变量：

IFS=$'\t' read -r col1 col2 col2 ... <<< "${results[0]}"

你的任务

linesIN=`cat /tmp/query.csv | sed 's/\t/,/g'`

不是一个数组赋值（它缺少括号）。它只是将命令替换的输出分配给常规字符串变量。 （任何换行符都将嵌入其中，但它仍然是单个字符串。）${#linesIN[@]}仍然有效，因为在 Bash/ksh 中单元素数组和标量变量的行为相同。

答案2

另一种方法是将命令的输出通过管道传输到 while 循环。请注意，您要包含 -N 或结果包含列名称。

#!/bin/bash
#Script to read output from a mysql command line by line 

mysql -uroot -p example -N -e "select column from table" | while IFS= read -r loop
do
    echo "$loop"
done

如果您只想计数，您可以执行 aselect count(columnName)并打印出结果。

答案3

如果您想要文件中的行数等于查询返回的行数，只需使用wc计算文件中的行数即可

  arraylength=$( wc -l < /tmp/query.csv)
  echo $arraylength

相关内容