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.drivers.resManagers;
18  
19  import java.util.HashMap;
20  import java.util.Map;
21  
22  import org.apache.commons.logging.Log;
23  import org.apache.commons.logging.LogFactory;
24  import org.gwe.drivers.netAccess.ShellCommand;
25  import org.gwe.persistence.model.AllocationInfo;
26  import org.gwe.utils.IOUtils;
27  import org.gwe.utils.VelocityUtils;
28  
29  /**
30   * @author Marco Ruiz
31   * @since Oct 26, 2008
32   */
33  public abstract class CommandLineResourceManagerDriver extends ResourceManagerDriver {
34  
35  	private static Log log = LogFactory.getLog(CommandLineResourceManagerDriver.class);
36  
37  	protected String discoveryCommand;
38  	protected String submissionCommand;
39  	protected String template;
40  	
41  	public CommandLineResourceManagerDriver(String discoveryCommand, String submissionCommand, String submissionTemplateFile) {
42  		this.discoveryCommand = discoveryCommand;
43  		this.submissionCommand = submissionCommand;
44  		this.template = IOUtils.readClassPathFile(submissionTemplateFile);
45  	}
46  
47  	public boolean isSupportedJobManagerAvailable() {
48  		try {
49  			log.info("Opening a local shell connection to discover if resource manager associated with driver " + this.getClass() + " is installed...");
50  			new ShellCommand(discoveryCommand).runLocally();
51  			return true;
52  		} catch (Exception e) {
53  			log.info("Driver " + this.getClass() + " could not detect an associated resource manager installed! - " + e.getMessage());
54  			return false;
55  		}
56  	}
57  	
58  	public String queueAllocationRequest(AllocationInfo alloc) throws ResourceAllocationException {
59  		writeAllocationFile(alloc.getAgentScriptFileName(), agentScriptPreffix + alloc.getId(), alloc);
60  		String[] results = runQueueCommand(submissionCommand + " " + writeSubmitFile(alloc), alloc);
61  		return extractIdFromSubmissionOutput(results);
62      }
63  	
64      protected abstract String extractIdFromSubmissionOutput(String[] results);
65  
66      protected String writeSubmitFile(AllocationInfo alloc) throws ResourceAllocationException {
67  	    writeAllocationFile(alloc.getAgentSubmitFileName(), getSubmitContent(alloc), alloc);
68  	    return IOUtils.concatenatePaths(getAllocWorkspace(alloc), alloc.getAgentSubmitFileName());
69      }
70  
71  	protected final String getSubmitContent(AllocationInfo alloc) {
72      	Map<String, Object> ctx = new HashMap<String, Object>();
73      	ctx.put("GWE_AGENT_ID", alloc.getId());
74      	ctx.put("GWE_AGENT_SCRIPT", alloc.getAgentScriptFileName());
75      	ctx.put("GWE_AGENT_WORKSPACE", getAllocWorkspace(alloc));
76  		return VelocityUtils.merge(ctx, template);
77      }
78  }