sql - 用于运行 SQL 查询并将结果返回给变量的 Bash 脚本

Question

我运行这个 bash 脚本是为了连接到 PostgreSQL 数据库，运行一些查询，并简单地检查它是否返回空结果。然后，我想打印出是否返回了任何行。

#!/bin/sh
DATABASE=dbname
USERNAME=user
HOSTNAME=somehost.com
export PGPASSWORD=password

queryResult () {
psql -h $HOSTNAME -U $USERNAME -d $DATABASE <<SQL
SELECT * FROM myTable WHERE valueA > 5.0 OR valueB > 5.0;
IF @@ROWCOUNT > 0 THEN 1 ELSE 0;
SQL
}

valuesFound=$(queryResult)

echo $valuesFound

我遇到了两个问题:

1) 它将第一个查询 (SELECT * FROM myTable...) 的结果存储到 valuesFound 中并打印出来，我不希望这样。我只关心 IF 语句返回 1 还是 0。

2) 第二个查询 (IF @@ROWCOUNT...) 抛出一个语法错误:Syntax error at or near IF

Answer 1

我认为您的脚本可以通过将逻辑移动到一个独特的语句来简化:

queryResult () {
    psql -h $HOSTNAME -U $USERNAME -d $DATABASE <<SQL
        SELECT CASE WHEN COUNT(*) = 0 THEN 0 ELSE 1 END
        FROM myTable 
        WHERE valueA > 5.0 OR valueB > 5.0;
    SQL
}

或者更好的是，使用 EXISTS 来避免潜在的昂贵聚合:

queryResult () {
    psql -h $HOSTNAME -U $USERNAME -d $DATABASE <<SQL
        SELECT CASE WHEN EXISTS (
            SELECT 1 FROM myTable WHERE valueA > 5.0 OR valueB > 5.0
        ) THEN 1 ELSE 0 END;
    SQL
}