主页 / dba / 问题 / 252124
Accepted
suvrat
Asked: 2019-10-30 03:02:31 +0800 CST

在 unix 中将变量与 0 进行比较时获取错误的结果

  • 772

我在unix中有以下代码

 #!/bin/bash
 user=$1
 username=$user'%'
 DB_name=$2
 userpwd=$4
 export PGPASSWORD=$3

 psql -h localhost -U postgres -d $DB_name -At -c "SELECT COUNT(*) AS USER_COUNT FROM pg_roles WHERE rolname = '${user}' " \
| while read -a Record ; 
do
USER_COUNT=${Record[0]} 
echo 'user USER_COUNT ' $USER_COUNT
done

echo 'count' $count
if   [[ $(($USER_COUNT)) -eq 0 ]]
then
echo 'New User ' $user
else
echo 'User Exists ' $user

fi

它基本上接受给定的用户名并检查它是否存在于 Postgresql 中。在我们传递参数之后，它会检查它的计数，如果用户名存在则为 1，否则为 0，并将其存储在变量中USER_COUNT。然后它将它与 0 进行比较。但比较结果是错误的。不管用户名是否存在，它只运行 if 的“真”部分。

Edit-1正如 Phill 所建议的那样，我使用if [[ $(($USER_COUNT)) -eq 0 ]]而不是 的 原因if [$USER_COUNT -eq 0 ]是因为我收到了一条错误消息，否则会说一元运算符是预期的。我在某处读到这是因为它$USER_COUNT是一个未设置的变量。

Edit-2感谢您的建议 Lennart。按照他的建议，我添加-x到我的第一行，发现即使$USER_COUNT用户存在时的值为 1，否则为 0，但是在比较时，即在 step if [[ $(($USER_COUNT)) -eq 0 ]]中，$USER_COUNT取值为 0，即使之前它是 1 . 因此，它总是评估为真。但我不确定如何解决上述情况。

Edit-3非常感谢 Lennart。你的代码解决了我的问题。

postgresql scripting

2 个回答

  1. Best Answer

    你可以考虑这样的事情：

    #!/bin/bash
...
cnt=$(psql -h localhost -U postgres -d $DB_name -At -c "SELECT COUNT(*) AS USER_COUNT FROM pg_roles WHERE rolname = '${user}' ")

if [[ ${cnt} -eq 0 ]]
then
    echo 'New User ' $user
else
    echo 'User Exists ' $user
fi

    您说一元运算符预期的错误确实是因为变量未设置。有多种处理方法，但在这种情况下，我认为应该没问题

    • 1

  2. 如果 [[ $(($USER_COUNT)) -eq 0 ]]

    鉴于这是 bash 并且您只是在执行一个简单的数字比较，那么“if”条件不应该看起来更像这样吗？

    if [ $USER_COUNT -eq 0 ]

    双方括号做一些“更多”的事情，允许使用正则表达式和其他你不需要的怪异。

    就个人而言，我不会依靠计数来检测任何东西。从记录中检索一些确定的东西（比如名字）并确保你得到了一些东西——你可能会因为任何原因得到一个零计数。

    另外，请阅读使用 Postgres 密码文件将密码输入 psql。 https://www.postgresql.org/docs/9.1/libpq-pgpass.html

    • 0

相关问题