View Javadoc

1   /*
2    * Copyright 2007-2008 the original author or authors.
3    * 
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    * 
8    *      http://www.apache.org/licenses/LICENSE-2.0
9    * 
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  
17  package org.gwe.persistence.model.order;
18  
19  import java.io.File;
20  import java.io.FileNotFoundException;
21  import java.io.FileOutputStream;
22  import java.io.IOException;
23  import java.io.Serializable;
24  import java.util.HashMap;
25  import java.util.Map;
26  
27  import org.apache.commons.logging.Log;
28  import org.apache.commons.logging.LogFactory;
29  import org.gwe.drivers.netAccess.ConnectorException;
30  import org.gwe.drivers.netAccess.ShellCommand;
31  import org.gwe.utils.IOUtils;
32  
33  /**
34   * @author Marco Ruiz
35   * @since Sep 27, 2007
36   */
37  public class OSCommandDaemonRequest<DRP_TYPE extends Serializable> extends DaemonRequest<DRP_TYPE> {
38  
39  	private static Log log = LogFactory.getLog(OSCommandDaemonRequest.class);
40  
41  	private static String EXIT_CODE_TOKEN = "EXIT_CODE=";
42  	private static String EXIT_CODE_COMMAND = "\necho '" + EXIT_CODE_TOKEN + "'$?";
43  	
44  	protected String localizedCmd;
45  	protected String workspacePath;
46  
47  	public void setOSCommand(String localizedCmd) {
48  		this.localizedCmd = localizedCmd;
49  	}
50  
51  	public void setWorkspacePath(String workspacePath) {
52  		this.workspacePath = workspacePath;
53  	}
54  
55  	public final Serializable process(String allocWorkspace) throws Exception {
56  		setup();
57  		String result = IOUtils.concatenatePaths(allocWorkspace, "result-" + execId + ".out");
58  		FileOutputStream fos = null;
59  		try {
60  			log.info("Preparing daemon request execution.\n\tLocalized command: '" + localizedCmd + "' \n\tWorkspace: '" + workspacePath + "'");
61  			if (localizedCmd != null && !localizedCmd.equals("")) {
62  				// Create wrapper script
63  				String wrapperScript = createFile(allocWorkspace, "job-" + execId + ".sh", "#!/bin/sh\n" + localizedCmd + "\n");
64  				pause(1000);
65  	
66  				// Execute wrapper script 
67  				ShellCommand cmd = new ShellCommand(wrapperScript, workspacePath, null);
68  				String outResult = cmd.runLocally();
69  				
70  				// Parse results and exit code
71  /*
72  				int lastIndex = outResult.lastIndexOf(EXIT_CODE_TOKEN);
73  				String exitCode = outResult.substring(lastIndex + EXIT_CODE_TOKEN.length());
74  				if (!"0".equals(exitCode)) {
75  					
76  				}
77  */
78  				// Save results
79  				fos = writeResultOutputFile(result, outResult.getBytes());
80  			}
81  			return result;
82  		} catch(ConnectorException e) {
83  			fos = writeResultOutputFile(result, e.getMessage().getBytes());
84  			e.setCommand(localizedCmd);
85  			throw e;
86  		} catch(Exception e) {
87  			fos = writeResultOutputFile(result, e.getMessage().getBytes());
88  			throw e;
89  		} finally {
90  			if (fos != null)
91  				try {
92  					fos.close();
93  				} catch (IOException e) {
94  					log.warn("Result file could not be closed", e);
95  				}
96  			tearDown();
97  		}
98  	}
99  
100 	private FileOutputStream writeResultOutputFile(String result, byte[] bytes) throws FileNotFoundException, IOException {
101 	    FileOutputStream fos = new FileOutputStream(result, false);
102 	    fos.write(bytes);
103 	    return fos;
104     }
105 	
106 	private void pause(long millis) {
107 		try {
108 	        Thread.sleep(millis);
109         } catch (InterruptedException e) {
110         }
111 		Thread.yield();
112 	}
113 	
114 	public String toString() {
115 		return localizedCmd;
116 	}
117 
118 	protected Map<String, String> getEnvironment() { return new HashMap<String, String>(); }
119 
120 	protected void setup() {}
121 
122 	protected void tearDown() {}
123 	
124 	protected String createFileUnderWorkspace(String fileName, String contents) {
125 		return createFile(workspacePath, fileName, contents);
126 	}
127 
128 	protected String createFile(String root, String fileName, String contents) {
129 		return createFile(IOUtils.concatenatePaths(root, fileName), contents);
130 	}
131 
132 	protected String createFile(String fullFileName, String contents) {
133         try {
134         	return IOUtils.createLocalExecutableFile(fullFileName, contents);
135         } catch (IOException e) {
136 			// TODO: throw a real exception
137         	log.warn("Couldn't create file '" + fullFileName + "'", e);
138 			throw new RuntimeException("Exception thrown when agent tried to create a file for a job. Please contact developers", e);
139         }
140 	}
141 	
142 	public static void main(String[] args) throws IOException {
143 		OSCommandDaemonRequest<Serializable> request = new OSCommandDaemonRequest<Serializable>();
144 		String path = "/Users/admin/temp/";
145 		request.setWorkspacePath(path);
146 		String cmd = "ps aux | grep admin";
147 		String wrapperScript = request.createFileUnderWorkspace("process4job-1.sh", "#!/bin/sh\n" + cmd + "\n");
148 
149 		ProcessBuilder pb = new ProcessBuilder(wrapperScript);
150         pb.directory(new File(path));
151         Process proc = pb.start();
152 		FileOutputStream fos = new FileOutputStream(IOUtils.concatenatePaths(path, "result.txt"), false);
153         String results = IOUtils.readStream(proc.getInputStream(), fos).toString();
154         System.out.print(results);
155 	}
156 }