关于java调用autoit脚本的问题

由于selenium本身的局限性,我们在做web测试的时候,不得不使用autoit作为辅助工具来处理一些windows窗口的操作。

先来段autoit代码:

    Sleep($CmdLine[1])

编译成exe文件,文件名为sleep.exe,放在c盘根目录下,代码的功能很简单,就是根据传入的时间休眠。cmd下执行:c:\sleep.exe 2000,表示休眠2秒。

我们在Java中调用这个程序:

   long time1 = System.currentTimeMillis();
                    try {
                            Runtime.getRuntime().exec("c://sleep.exe 10000");
                            long time2 = System.currentTimeMillis();
                            System.out.println((time2 - time1) + "毫秒。");
                    } catch (Exception e) {
                            e.printStackTrace();
                    }

用上面这段代码去调用的话,java不会等待autoit的执行,sleep启动后,继续往下执行,这样几乎没有实际的应用价值。我们通常都是要等autoit去把该做的事情都做完再做后面的事情。

    long time1 = System.currentTimeMillis();
                    try {
                            Runtime.getRuntime().exec("c://sleep.exe 10000").waitFor();
                            long time2 = System.currentTimeMillis();
                            System.out.println((time2 - time1) + "毫秒。");
                    } catch (Exception e) {
                            e.printStackTrace();
                    }

我们在代码中加入waitFor(),可以达到上面的效果,但是又带来了另外一个问题,玩意autoit执行出了问题,一直不退出怎么办?程序就hang住了。

向高人讨教了一番,知道了如何控制autoit的超时问题,我们可以利用检查内存中是否存在au3进程的方式来做判断程序是否执行完毕,同时加一个超时控制,超时则继续往后执行,下面是demo 代码:

Java代码:

    
    import java.io.BufferedReader;
    import java.io.File;
    import java.io.InputStreamReader;
     
    public class Runau3 {
            public static void main(String[] args) {
                    // 执行au3脚本
                    // 休眠的那个脚本,脚本所在的目录(秒),等待时间,脚本的休眠时间(毫秒)
                    runAu3("sleep.exe", "c://", 5, "10000");
            }
     
            /**
             * 
             * @param filename
             *            au3的文件名,必须编译成exe
             * @param filepath
             *            au3的文件的路径
             * @param timeout
             *            超时时间(秒)
             * @param args
             *            au3的参数
             */
            public static void runAu3(String filename, String filepath, int timeout,
                            String... args) {
                    try {
                            // 如果发现进程,先杀了
                            if (findProcess(filename)) {
                                    killProcess(filename);
                            }
                            // 拼装命令
                            String cmd = filename;
                            for (String arg : args) {
                                    cmd += " " + arg;
                            }
     
                            // 执行au3脚本
                            Runtime.getRuntime()
                                            .exec("cmd /C " + cmd, null, new File(filepath));
                            int count = 0;
                            while (findProcess(filename) && count < timeout) {
                                    System.out.println(count);
                                    Thread.sleep(1000);
                                    count++;
                            }
                            // 如果进程未结束,杀了
                            if (findProcess(filename)) {
                                    killProcess(filename);
                            }
                    } catch (Exception e) {
                            e.printStackTrace();
                    }
     
            }
     
            /**
             * 查找进程
             * @param processname
             * @return
             */
            public static boolean findProcess(String processname) {
                    BufferedReader bufferedreader = null;
                    try {
                            Process proc = Runtime.getRuntime().exec(
                                            "tasklist /fi \" imagename eq " + processname + " \" ");
                            bufferedreader = new BufferedReader(new InputStreamReader(proc
                                            .getInputStream()));
                            String line = null;
                            while ((line = bufferedreader.readLine()) != null) {
                                    if (line.contains(processname)) {
                                            return true;
                                    }
                            }
                            return false;
                    } catch (Exception ex) {
                            ex.printStackTrace();
                            return false;
                    } finally {
                            if (bufferedreader != null) {
                                    try {
                                            bufferedreader.close();
                                    } catch (Exception ex) {
                                    }
                            }
                    }
            }
     
            /**
             * 杀进程
             * @param processname
             */
            public static void killProcess(String processname) {
                    BufferedReader bufferedreader = null;
                    try {
                            Process proc = Runtime.getRuntime().exec(
                                            "taskkill /F /IM " + processname);
                            bufferedreader = new BufferedReader(new InputStreamReader(proc
                                            .getInputStream()));
                            String line = null;
                            while ((line = bufferedreader.readLine()) != null) {
                                    System.out.println(line);
                            }
                    } catch (Exception ex) {
                            ex.printStackTrace();
                    } finally {
                            if (bufferedreader != null) {
                                    try {
                                            bufferedreader.close();
                                    } catch (Exception ex) {
                                    }
                            }
                    }
            }
    }

你可能感兴趣的:(java)