Question

我有一个要求，在使用Java代码的地方我需要创建/删除/更改postgres表。

我编写了如下程序：

public static void main(String[] args) throws IOException {
        System.out.println("Hello World!");
        Process p = Runtime.getRuntime().exec("psql -U postgres -d testdb -h localhost -p 5433 -f D:\test.sql");
    }

test.sql文件如下所示，

Create TABLE MyTable1
(
    VersionNumber VARCHAR(32) NOT NUll
);

Create TABLE MyTable2
(
    VersionNumber VARCHAR(32) NOT NUll
);

Create TABLE MyTable3
(
    VersionNumber VARCHAR(32) NOT NUll
);

问题：

如果我运行相同的psql命令：

psql -U postgres -d testdb -h localhost -p 5433 -f D:\test.sql

在命令行中，要求输入密码并创建表。

但是在Java程序中，没有提供密码的规定。请让我知道如何实现它。

Answer 1

您可以改用connection URL：

psql -f d:\test.sql postgresql://postgres:password@localhost:5433/testdb

Answer 2

首先，最好实际使用JDBC连接到数据库并运行SQL语句。如果您设置使用命令行psql，则可以使用PGPASSWORD环境变量来设置密码：

String command = "psql -U postgres -d testdb -h localhost -p 5433 -f D:\test.sql";
String[] envVars = { "PGPASSWORD=yourpassword" };
Process p = Runtime.getRuntime().exec(command, envVars);

如有必要，您可以从stdin中读取密码。但是，再次，最好通过JDBC进行此操作。

Answer 3

关于a_horse_with_no_name answer 并在Windows OS上使用Java代码启动psql 并使用ProcessBuilder 对我稍加修改即可：

    ProcessBuilder builder = new ProcessBuilder();
    String connUrl = "postgresql://user:password@host:port/dbname";
    String sqlFileToProcess = "--file=Disk:\\path\\to\\file.sql";
    builder.command("psql.exe", sqlFileToProcess, connUrl);
    builder.directory(new File("Disk:\\Path\\to\\psql\\containing\\folder"));
    Process process = builder.start();
    int exitCode = 0;
    try {
         exitCode = process.waitFor();
         int len;
         if ((len = process.getErrorStream().available()) > 0) {
             byte[] buf = new byte[len];
             process.getErrorStream().read(buf);
             System.err.println("Command error:\t\""+new String(buf)+"\"");
         }
    } catch (InterruptedException e) {
         e.printStackTrace();
    }
   assert exitCode == 0;

不知道为什么，但是：

-f Disk:\\path\\to\\file.sql

作为Java代码中的参数抛出：

Command error:  "unrecognized win32 error code: 123 psql: error:  Disk:/path/to/file.sql: Invalid argument
"

（注意otput和input中斜杠和反斜杠的重定向）

在Java程序中运行psql命令

3 个答案: