JavaSE ネイティブコマンド実行(旧) の Java 5 近代化改修版


概要

ソースコード

コマンド実行クラス

package com.snail.grappaexam;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.Charset;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.SynchronousQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public final class CommandExecutor {

    /** コマンド処理が継続中かのフラグ. */
    private boolean isRunning = false;
    /** HOME Directory. */
    private String pHomeDir = null;
    /** スレッドプール. */
    private static ExecutorService thPool;

    static {
        // 3秒間使われなかったらプールしているスレッドを解放する
        // (すべてのスレッドが解放されるまでアプリが終了しない)
        thPool = new ThreadPoolExecutor(
                0, Integer.MAX_VALUE,
                3L, TimeUnit.SECONDS,
                new SynchronousQueue<Runnable>());
    }

    /**
     * コンストラクタ.
     * ホームディレクトリはカレントディレクトリになります
     */
    public CommandExecutor() {
        super();
    }

    /**
     * コンストラクタ.
     * @param homeDir ホームディレクトリ
     */
    public CommandExecutor(final String homeDir) {
        this();
        pHomeDir = homeDir;
    }

    /**
     * シェルコマンドを実行します.
     * @param cmdArray コマンド
     * @return 実行結果
     * @throws InterruptedException プロセス実行例外
     * @throws IOException プロセスの出力ストリームが異常
     */
    public CommandResult exec(final String stdin, final String... cmdArray)
            throws IOException, InterruptedException, ExecutionException {

        long startTime = System.currentTimeMillis();

        CommandResult res = new CommandResult();

        ProcessBuilder pb = new ProcessBuilder(cmdArray);
        if (pHomeDir != null) {
            pb.directory(new File(pHomeDir));
        }

        Process process = pb.start();

        isRunning = true;
        Future<String> stdout = thPool.submit(new InputHandler(process.getInputStream()));
        Future<String> stderr = thPool.submit(new InputHandler(process.getErrorStream()));
        Thread watchdog = new Thread(new WatchDog(process));
        watchdog.start();

        OutputStream out = process.getOutputStream();
        out.write(stdin.getBytes(Charset.forName("UTF-8")));
        out.close();

        res.setStatus(process.waitFor());

        long endTime = System.currentTimeMillis();

        isRunning = false;

        if (watchdog.isAlive()) {
            watchdog.interrupt();
        }
        res.setStdout(stdout.get());
        res.setStderr(stderr.get());
        process.destroy();

        res.setProcessingTime(endTime - startTime);
        return res;
    }

    /**
     * ProcessのStdout、Stderrの受け取り処理を行う.
     * <pre>
     * JavaVMの子プロセスからの出力を受け取るバッファは非常に小さいので
     * 出力を別スレッドで逐次取り出す必要がある
     * </pre>
     */
    private class InputHandler implements Callable<String> {

        /**
         * シェルコマンドからの出力を受け取るInputStream
         */
        private InputStream pIn;

        /**
         * デフォルトコンストラクタの無効化
         */
        private InputHandler() {
            super();
        }

        /**
         * コンストラクタ
         * @param argIn シェルコマンドからの出力を受け取るInputStream
         */
        private InputHandler(final InputStream in) {
            this();
            pIn = in;
        }

        @Override
        public String call() throws Exception {
            ByteArrayOutputStream bArray = new ByteArrayOutputStream();
            try {
                byte[] tmp = new byte[128]; // 大きくしすぎるとバッファが詰まる
                int size;
                while((size = pIn.read(tmp)) > 0) {
                    bArray.write(tmp, 0, size);
                }
            } catch (IOException ex) {
                ex = null; // なにもしないし、できない
            } finally {
                try {
                    pIn.close();
                } catch (IOException ex) {
                    ex = null; // なにもしないし、できない
                }
            }
            return bArray.toString();
        }
    }

    /**
     * 起動したシェルコマンドが長い間終了しない場合に強制終了します.
     * <pre>
     * Apache Ant を参考に作成した。
     * </pre>
     */
    private class WatchDog implements Runnable {

        /**
         * 実行中のプロセス
         */
        private Process process;

        /**
         * Creates a new WatchDog object.
         *
         * @param param 実行予定のProcess
         */
        public WatchDog(final Process param) {
            this();
            process = param;
        }

        /**
         * デフォルトコンストラクタの無効化
         */
        private WatchDog() {
            super();
        }

        /**
         * 30000ミリ秒待ってまだプロセスが動いていれば、プロセスを強制終了します
         */
        @Override
        public synchronized void run() {
            long until = System.currentTimeMillis() + 30000L;
            long now;
            while (isRunning && (until > (now = System.currentTimeMillis()))) {

                try {
                    wait(until - now);
                } catch (InterruptedException ignoreEx) {
                    // Ignore exception
                    ignoreEx = null;
                }
            }


            if (isRunning) {
                process.destroy();
            }
        }
    }
}

コマンド実行結果格納Bean

package com.snail.grappaexam;

import java.io.Serializable;

public final class CommandResult implements Serializable {
    private String stderr;
    private String stdout;
    private int status;
    private long processingTime;

    public void setProcessingTime(long processingTime) {
        this.processingTime = processingTime;
    }

    public void setStatus(int status) {
        this.status = status;
    }

    public void setStderr(String stderr) {
        this.stderr = stderr;
    }

    public void setStdout(String stdout) {
        this.stdout = stdout;
    }

    public long getProcessingTime() {
        return processingTime;
    }

    public int getStatus() {
        return status;
    }

    public String getStderr() {
        return stderr;
    }

    public String getStdout() {
        return stdout;
    }
}

使い方

public class App 
{
    public static void main( String[] args ) throws ExecutionException
    {
        try {
            CommandExecutor cmd = new CommandExecutor("/Users/Atsushi/");
            CommandResult res = cmd.exec("","ls","-la"); // 第一引数は標準入力に与える文字列
            System.out.println("STATUS:" + res.getStatus());
            System.out.println("STDOUT:\n" + res.getStdout());
        } catch (IOException ex) {
            Logger.getLogger(App.class.getName()).log(Level.SEVERE, null, ex);
        } catch (InterruptedException ex) {
            Logger.getLogger(App.class.getName()).log(Level.SEVERE, null, ex);
        }
    }
}

Java#JavaSE


トップ   編集 凍結 差分 バックアップ 添付 複製 名前変更 リロード   新規 一覧 単語検索 最終更新   ヘルプ   最終更新のRSS   sitemap
Last-modified: 2013-01-27 (日) 01:02:16 (1407d)
ISBN10
ISBN13
9784061426061