3

首先,这是我的代码:

import java.io.*;
import java.util.Date;

import com.banctecmtl.ca.vlp.shared.exceptions.*;

public class PowershellTest implements Runnable {

public static final String PATH_TO_SCRIPT = "C:\\Scripts\\ScriptTest.ps1";
public static final String SERVER_IP = "XX.XX.XX.XXX";
public static final String MACHINE_TO_MOD = "MachineTest";

/**
 * @param args
 * @throws OperationException 
 */
public static void main(String[] args) throws OperationException {

    new PowershellTest().run();

}

public PowershellTest(){}

@Override
public synchronized void run() {
    String input = "";
    String error = "";

    boolean isHanging = false;

    try {

        Runtime runtime = Runtime.getRuntime();
        Process proc = runtime.exec("powershell -file " + PATH_TO_SCRIPT +" "+ SERVER_IP +" "+ MACHINE_TO_MOD);
        proc.getOutputStream().close();
        InputStream inputstream = proc.getInputStream();
        InputStreamReader inputstreamreader = new InputStreamReader(inputstream);
        BufferedReader bufferedreader = new BufferedReader(inputstreamreader);

        proc.waitFor();

        String line;
        while (!isHanging && (line = bufferedreader.readLine()) != null) {
            input += (line + "\n");
            Date date = new Date();
            while(!bufferedreader.ready()){

                this.wait(1000);
                //if its been more then 1 minute since a line has been read, its hanging. 
                if(new Date().getTime() - date.getTime() >= 60000){

                    isHanging = true;
                    break;

                }
            }
        }

        inputstream.close();
        inputstream = proc.getErrorStream();
        inputstreamreader = new InputStreamReader(inputstream);
        bufferedreader = new BufferedReader(inputstreamreader);
        isHanging = false;

        while (!isHanging && (line = bufferedreader.readLine()) != null) {
            error += (line + "\n");
            Date date = new Date();
            while(!bufferedreader.ready()){

                this.wait(1000);
                //if its been more then 1 minute since a line has been read, its hanging. 
                if(new Date().getTime() - date.getTime() >= 60000){

                    isHanging = true;
                    break;

                }
            }
        }
        inputstream.close();

        proc.destroy();

    } catch (IOException e) {
        //throw new OperationException("File IO problem.", e);
    } catch (InterruptedException e) {
        //throw new OperationException("Script thread problem.",e);
    }
    System.out.println("Error : " + error + "\nInput : " + input);
}

}

我目前正在尝试运行将在远程服务器上启动/停止 vm (VMWARE) 的 powershell 脚本。该脚本从命令行工作,此代码也是如此。问题是,我讨厌我必须为这样的工作使用线程(并让它等待脚本响应,如进一步解释的那样)。我不得不这样做,因为 BufferedReader.readline() 和 proc.waitFor() 都永远挂起。

该脚本从 cmd 运行时执行时间很长。从验证服务器身份验证和执行实际脚本开始,它会停止 30 秒到 1 分钟。根据我从调试中看到的情况,当它开始从脚本接收这些延迟时,readline 挂起。

我也很确定这不是内存问题,因为我在任何调试会话中都没有任何 OOM 错误。

现在我明白 Process.waitFor() 需要我从错误流和常规流中刷新缓冲区才能工作,这就是我不使用它的主要原因(我需要输出来管理 VM 特定错误、证书问题, ETC。)。

我想知道是否有人可以向我解释它为什么会挂起,以及是否有办法只使用典型的 readline() 而不会让它挂得那么难。即使脚本应该已经结束了一段时间,它仍然挂起(我尝试使用我在 java 应用程序中使用的完全相同的东西同时运行 java 应用程序和 cmd 命令,让它运行 1 小时,没有任何效果)。它不只是停留while 循环中,readline() 是挂起的地方。

这也是一个测试版本,离最终代码还很远,所以请给我留点:这应该是一个常数,这是无用的,等等。我稍后会清理代码。显然,IP 在我的代码中也不是 XX.XX.XX.XXX。

关于如何修复的解释或建议将不胜感激。

顺便说一句,这里是我目前使用的脚本:

Add-PSSnapin vmware.vimautomation.core

Connect-VIServer -server $args[0]

Start-VM -VM "MachineTest"

如果您需要更多详细信息,我会尽力提供。

在此先感谢您的帮助!

编辑:我之前还使用要求不高的脚本测试了代码,该工作是获取文件的内容并打印它。由于无需等待即可获取信息,因此 readline() 运行良好。因此,我相当确定问题出在脚本执行的等待时间上。

另外,请原谅我的错误,英语不是我的主要语言。

在此先感谢您的帮助!

EDIT2:因为我无法回答我自己的问题:

这是我使用线程后的“最终”代码:

import java.io.*;

public class PowershellTest implements Runnable {

public InputStream is;

public PowershellTest(InputStream newIs){

    this.is = newIs;
}

@Override
public synchronized void run() {
    String input = "";
    String error = "";

    try {


        InputStreamReader inputstreamreader = new InputStreamReader(is);
        BufferedReader bufferedreader = new BufferedReader(inputstreamreader);

        String line;
        while ((line = bufferedreader.readLine()) != null) {
            input += (line + "\n");

        }

        is.close();

    } catch (IOException e) {
        //throw new OperationException("File IO problem.", e);
    }
    System.out.println("Error : " + error + "\nInput : " + input);
}

}

主要简单地创建并启动 2 个线程(PowerShellTest 实例),1 个使用 errorStream,1 个使用 inputStream。

我相信当我第一次编写应用程序并以某种方式修复它时,我犯了一个愚蠢的错误,因为我一遍又一遍地重新编写代码。运行仍然需要 5-6 分钟,如果不长于我之前的代码,这在某种程度上是相似的(这是合乎逻辑的,因为在我的情况下,errorStream 和 inputStream 按顺序获取它们的信息)。

无论如何,感谢您的所有回答,尤其是 Miserable Variable 对线程的提示。

4

2 回答 2

5

waitFor()首先,在你读完信息流之前不要打电话。我强烈建议您查看ProcessBuilder而不是简单地使用Runtime.exec,并自己拆分命令,而不是依靠 Java 为您完成:

ProcessBuilder pb = new ProcessBuilder("powershell", "-file", PATH_TO_SCRIPT,
        SERVER_IP, MACHINE_TO_MOD);
pb.redirectErrorStream(true); // merge stdout and stderr
Process proc = pb.start();

redirectErrorStream将错误输出合并到正常输出中,因此您只需阅读proc.getInputStream(). 然后,您应该能够在 EOF 之前读取该流,然后调用proc.waitFor().

于 2012-10-09T20:19:42.063 回答
4

您目前正在等待从inputStream开始阅读之前完成阅读errorStreamstderr如果该进程之前写入它,stdout那么您可能会陷入僵局。

尝试从同时运行的线程中读取两个流。当你在它的时候,也删除proc.getOutputStream().close();. 它不应该影响行为,但也不是必需的。

于 2012-10-09T20:42:28.473 回答