org/objectweb/proactive/core/process/rsh/maprsh/MapRshProcess.java

/* 
 * ################################################################
 * 
 * ProActive: The Java(TM) library for Parallel, Distributed, 
 *            Concurrent computing with Security and Mobility
 * 
 * Copyright (C) 1997-2007 INRIA/University of Nice-Sophia Antipolis
 * Contact: proactive@objectweb.org
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or any later version.
 *  
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
 * USA
 *  
 *  Initial developer(s):               The ProActive Team
 *                        http://www.inria.fr/oasis/ProActive/contacts.html
 *  Contributor(s): 
 * 
 * ################################################################
 */ 
package org.objectweb.proactive.core.process.rsh.maprsh;

import org.objectweb.proactive.core.process.AbstractExternalProcessDecorator;
import org.objectweb.proactive.core.process.ExternalProcess;
import org.objectweb.proactive.core.process.JVMProcess;
import org.objectweb.proactive.core.process.JVMProcessImpl;
import org.objectweb.proactive.core.process.UniversalProcess;


public class MapRshProcess extends AbstractExternalProcessDecorator {
    private static final String FILE_SEPARATOR = System.getProperty(
            "file.separator");
    private static final String DEFAULT_SCRIPT_LOCATION = System.getProperty(
            "user.home") + FILE_SEPARATOR + "ProActive" + FILE_SEPARATOR +
        "scripts" + FILE_SEPARATOR + "unix" + FILE_SEPARATOR +
        "gridexperiment" + FILE_SEPARATOR + "oasis-exp";
    protected String scriptLocation = DEFAULT_SCRIPT_LOCATION;
    private String parallelize = null;

    //private String logFile = System.getProperty("user.home")+System.getProperty("file.separator")+"oasisgridlog.txt";
    public MapRshProcess() {
        super();
        setCompositionType(GIVE_COMMAND_AS_PARAMETER);
    }

    public MapRshProcess(ExternalProcess targetProcess) {
        super(targetProcess);
        setCompositionType(GIVE_COMMAND_AS_PARAMETER);
    }

    public void setParallelization(String parallelize) {
        this.parallelize = parallelize;
    }

    public String getParallelization() {
        return this.parallelize;
    }

    public void setScriptLocation(String scriptLocation) {
        this.scriptLocation = scriptLocation;
    }

    public String getScriptLocation() {
        return scriptLocation;
    }

    public String getProcessId() {
        return "maprsh_" + targetProcess.getProcessId();
    }

    public int getNodeNumber() {
        return targetProcess.getNodeNumber();
    }

    public UniversalProcess getFinalProcess() {
        checkStarted();
        return targetProcess.getFinalProcess();
    }

    //
    // -- PROTECTED METHODS -----------------------------------------------
    //
    protected String internalBuildCommand() {
        return buildMapRshCommand() + buildEnvironmentCommand();
    }

    protected String buildMapRshCommand() {
        StringBuilder command = new StringBuilder();
        try {
            java.io.File script = new java.io.File(scriptLocation);
            byte[] b = getBytesFromInputStream(new java.io.FileInputStream(
                        script));
            String scriptText = new String(b);
            scriptText = removeJavaCommand(scriptText);
            //System.out.println(scriptText);
            scriptText = appendJavaCommand(scriptText);
            if (logger.isDebugEnabled()) {
                logger.debug(scriptText);
            }
            b = scriptText.getBytes();
            // script.delete();
            java.io.OutputStream out = new java.io.BufferedOutputStream(new java.io.FileOutputStream(
                        script));
            out.write(b, 0, b.length);
            out.flush();
            out.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        command.append("/usr/local/bin/maprsh ");
        if (parallelize != null) {
            command.append(parallelize + " ");
        }
        command.append(scriptLocation + " " + hostname);
        if (logger.isDebugEnabled()) {
            logger.debug(command.toString());
        }
        return command.toString();
    }

    private String appendJavaCommand(String scriptText) {
        StringBuilder newScriptText = new StringBuilder(scriptText.length());
        String targetCommand = targetProcess.getCommand();

        newScriptText.append(scriptText);
        newScriptText.append("\ntime " + targetCommand + " ) & \n");
        return newScriptText.toString();
    }

    private String removeJavaCommand(String scriptText) {
        int marker = scriptText.lastIndexOf("}");
        String newScriptText = scriptText.substring(0, marker + 1);

        //System.out.println(newScriptText);
        return newScriptText;
    }

    public static void main(String[] args) {
        try {
            JVMProcess process = new JVMProcessImpl(new StandardOutputMessageLogger());
            process.setParameters("///toto");

            //ExternalProcess process = new SimpleExternalProcess("ls -la");
            MapRshProcess maprsh = new MapRshProcess(process);
            maprsh.setHostname("waha owenii");
            maprsh.startProcess();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private static byte[] getBytesFromInputStream(java.io.InputStream in)
        throws java.io.IOException {
        java.io.DataInputStream din = new java.io.DataInputStream(in);
        byte[] bytecodes = new byte[in.available()];
        try {
            din.readFully(bytecodes);
        } finally {
            if (din != null) {
                din.close();
            }
        }
        return bytecodes;
    }
}

---