Java 和 MySQL 控制台 REPL

Question

我需要编写一个 Java 代码来创建一个运行 MySQL 控制台命令的 shell，并且可以保留交互式会话，直到它被故意破坏。

为了模拟它，这是我的代码。

    ProcessBuilder pb;
    pb = new ProcessBuilder(
        "cmd.exe","/c","c://xampp//mysql//bin//mysql","-uroot");

    Process p = pb.start();

    OutputStream stdout = p.getOutputStream();
    InputStream stdin = p.getInputStream();

    BufferedReader reader = new BufferedReader(new InputStreamReader(stdin));
    BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(stdout));

    boolean written = false;
    while(true) {
        System.out.println("loop... ");

        writer.write("show databases;\r\n\r\n");
        writer.flush();
        writer.newLine();
        writer.newLine();
        writer.flush();
        writer.close(); //If not close, it cannot produce result

        String line;
        line = reader.readLine();
        while (line != null) {
            System.out.println(line);
            line = reader.readLine();
        }
    }

问题是，代码只运行一次，因为在循环中调用了writer.close()。但是如果我不打电话给writer.close()，它永远不会响应。

我怎样才能实现我的目标？

Answer 1

您的问题来自line = reader.readLine()。在找到完整的行之前它不会返回，这由输出流中的换行符指示。但是当你在处理来自mysql的输出时，你不会得到一个完整的行，而是一个“提示”，比如

mysql> _

此外，它永远不会返回null，直到它正在读取的进程终止或以其他方式关闭其输出流。因此，您不能将其用作循环终止条件。相反，您应该逐个字符地阅读流，寻找您提交的命令已完成的指示。

我没有安装 mysql，因为我安装了 sqlite3，所以我创建了一个脚本，它使用测试数据库“sql”启动它，查询表 t1（单文本列，两行），并打印输出。重复 3 次。

ProcessBuilder pb = new ProcessBuilder("/usr/bin/sqlite3", "sql");
Process p = pb.start();
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(p.getOutputStream()));

// Don't use a BufferedReader; it waits for '\n' characters.
InputStreamReader isr = new InputStreamReader(p.getInputStream());

// Our own "Buffered Reader" workspace 
StringBuilder sb = new StringBuilder();

// Execute 3 times only ...
for(int i=0; i<3; i++) {

   System.out.println("Command: select * from t1;");
   bw.write("select * from t1;\n");
   bw.write(".print !-DONE-!\n");
   bw.flush();

   System.out.println("Result:");
   // Repeat until magic phrase in read buffer
   while (true) {

      // Read characters one at a time...
      char ch = (char) isr.read();

      // Do our own newline processing.
      if (ch == '\n') {
         String line = sb.toString();   // Buffer to string
         sb.setLength(0);               // ... and clear buffer

         if (line.equals("!-DONE-!"))
            break;

         System.out.println(line);
      } else {
         sb.append(ch);
      }
   }
   System.out.println();
}

jshell pb.java 的输出：

Command: select * from t1;
Result:
Hello world
Goodbye world!

Command: select * from t1;
Result:
Hello world
Goodbye world!

Command: select * from t1;
Result:
Hello world
Goodbye world!

|  Welcome to JShell -- Version 9
|  For an introduction type: /help intro

jshell> _

似乎stdin 来自另一个进程，stdout 去往另一个进程，我的sqlite3 拒绝在stdout 上发出sqlite> 提示。这让我很伤心，在与各种设置和命令行选项斗争了一个小时后，我放弃了，只是添加了一个打印命令，可以打印出神奇的字词!-DONE-!。输出处理会寻找这个魔法词，并跳出结果读取循环。

如果您能找出使mysql 在stdout 上吐出mysql> _ 提示符所需的魔法咒语，那么您的中断条件将位于非换行路径中，例如：

      } else {
         sb.append(ch);
         if (sb.length() == 7 && sb.toString().equals("mysql> ")) {
            sb.setLength(0);
            break;
         }
      }