Java tutorial
/* * Copyright 2008-2010 Oleg Sukhoroslov * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package jlite.cli; import java.io.File; import java.util.ArrayList; import java.util.List; import jlite.GridAPIException; import jlite.GridSession; import jlite.GridSessionConfig; import jlite.GridSessionFactory; import jlite.util.Util; import org.apache.commons.cli.CommandLine; import org.apache.commons.cli.CommandLineParser; import org.apache.commons.cli.GnuParser; import org.apache.commons.cli.HelpFormatter; import org.apache.commons.cli.OptionBuilder; import org.apache.commons.cli.Options; import org.apache.commons.cli.ParseException; public class JobOutput { private static final String COMMAND = "job-output [options] <jobId> ..."; public static void main(String[] args) { System.out.println(); // extra line CommandLineParser parser = new GnuParser(); Options options = setupOptions(); HelpFormatter helpFormatter = new HelpFormatter(); helpFormatter.setSyntaxPrefix("Usage: "); CommandLine line = null; try { line = parser.parse(options, args); if (line.hasOption("help")) { helpFormatter.printHelp(100, COMMAND, "\noptions:", options, "\n" + CLI.FOOTER, false); System.out.println(); // extra line System.exit(0); } else { if (line.hasOption("xml")) { System.out.println("<output>"); } run(line.getArgs(), line); } } catch (ParseException e) { System.err.println(e.getMessage() + "\n"); helpFormatter.printHelp(100, COMMAND, "\noptions:", options, "\n" + CLI.FOOTER, false); System.out.println(); // extra line System.exit(-1); } catch (Exception e) { if (line.hasOption("xml")) { System.out.println("<error>" + e.getMessage() + "</error>"); } else { System.err.println(e.getMessage()); } } finally { if (line.hasOption("xml")) { System.out.println("</output>"); } } System.out.println(); // extra line } private static Options setupOptions() { Options options = new Options(); options.addOption(OptionBuilder.withDescription("displays usage").create("help")); options.addOption(OptionBuilder.withArgName("file_path") .withDescription("select JobId(s) from the specified file").hasArg().create("i")); options.addOption(OptionBuilder.withArgName("service_URL").withDescription("WMProxy service endpoint") .hasArg().create("e")); options.addOption(OptionBuilder.withArgName("dir_path") .withDescription("store output files in the specified directory (default is CURRENT_DIR/JOB_ID)") .hasArg().create("dir")); options.addOption( OptionBuilder.withDescription("do not purge job output after retrieval").create("nopurge")); options.addOption(OptionBuilder.withDescription("do not retrieve output files, just print their URIs") .create("list")); options.addOption(OptionBuilder.withArgName("proxyfile") .withDescription("non-standard location of proxy cert").hasArg().create("proxypath")); options.addOption(OptionBuilder.withArgName("xml").withDescription("output as xml").create("xml")); // options.addOption(OptionBuilder // .withArgName("protocol") // .withDescription("protocol to be used for file tranfer {gsiftp,https} (default is gsiftp)") // .hasArg() // .create("proto")); return options; } private static void run(String[] jobIdArgs, CommandLine line) throws Exception { GridSessionConfig conf = new GridSessionConfig(); if (line.hasOption("proxypath")) { conf.setProxyPath(line.getOptionValue("proxypath")); } GridSession grid = GridSessionFactory.create(conf); List<String> jobIds = new ArrayList<String>(); ; if (jobIdArgs.length > 0) { for (String jobId : jobIdArgs) { jobIds.add(jobId); } } if (line.hasOption("i")) { for (String jobId : JobsSelector.select(line.getOptionValue("i"))) { jobIds.add(jobId); } } if (jobIds.size() == 0) { throw new Exception("JobId(s) not found"); } String vo = Util.readVOFromVOMSProxy(conf.getProxyPath()); if (line.hasOption("xml")) { System.out.println("<vo>" + vo + "</vo>"); } else { System.out.println("Working VO: " + vo); } String wmProxyURL = conf.getWMProxies().get(vo); if (line.hasOption("e")) { wmProxyURL = line.getOptionValue("e"); } if (wmProxyURL == null) { throw new GridAPIException("Could not find WMProxy server for VO: " + vo); } if (line.hasOption("xml")) { System.out.println("<wmProxy>" + wmProxyURL + "</wmProxy>"); } else { System.out.println("Connecting to WMProxy service: " + wmProxyURL + "\n"); } if (line.hasOption("xml")) { System.out.println("<jobOutputs>"); } for (String jobId : jobIds) { if (line.hasOption("xml")) { System.out.println("<jobOutput>"); } if (line.hasOption("xml")) { System.out.println("<jobId>" + jobId + "</jobId>"); } else { System.out.println("Requesting output for job: " + jobId); } try { if (line.hasOption("list")) { List<String> files = grid.listJobOutput(wmProxyURL, jobId); if (line.hasOption("xml")) { System.out.println("<fileList>"); } for (String file : files) { if (line.hasOption("xml")) { System.out.println("<file>" + file + "</file>"); } else { System.out.println("- " + file); } } if (line.hasOption("xml")) { System.out.println("</fileList>"); } } else { String outboxDir = line.getOptionValue("dir", System.getProperty("user.dir") + "/" + Util.getShortJobId(jobId)); grid.getJobOutput(wmProxyURL, jobId, outboxDir, !line.hasOption("nopurge")); if (line.hasOption("xml")) { System.out.println("<dir>" + new File(outboxDir).getAbsolutePath() + "</dir>"); } else { System.out.println("Job output is uploaded to: " + new File(outboxDir).getAbsolutePath()); } } } catch (GridAPIException e) { if (line.hasOption("xml")) { System.out.println("<error>" + e.getMessage() + "</error>"); } else { System.err.println(e.getMessage()); } } if (jobIds.size() > 1) { System.out.println(); } if (line.hasOption("xml")) { System.out.println("</jobOutput>"); } } if (line.hasOption("xml")) { System.out.println("</jobOutputs>"); } } }