Java tutorial
/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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 org.apache.axis2.wsdl.codegen.jaxws; import java.lang.reflect.Field; import java.util.ArrayList; import java.util.Arrays; import java.util.Iterator; import java.util.List; import java.util.Map; import org.apache.axiom.om.util.LogOutputStream; import org.apache.axis2.util.CommandLineOption; import org.apache.axis2.util.CommandLineOptionConstants; import org.apache.axis2.util.CommandLineOptionParser; import org.apache.axis2.wsdl.codegen.CodeGenConfiguration; import org.apache.axis2.wsdl.codegen.CodeGenerationException; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import com.sun.tools.ws.wscompile.WsimportTool; /** * The Class JAXWSCodeGenerationEngine. */ public class JAXWSCodeGenerationEngine { private static final Log log = LogFactory.getLog(JAXWSCodeGenerationEngine.class); private CodeGenConfiguration configuration; private CommandLineOptionParser commandLineOptionParser; private String[] originalArgs; /** * Instantiates a new jAXWS code generation engine. * * @param configuration * the configuration * @param originalArgs * the original args * @throws CodeGenerationException * the code generation exception */ public JAXWSCodeGenerationEngine(CodeGenConfiguration configuration, String[] originalArgs) throws CodeGenerationException { this.configuration = configuration; this.originalArgs = originalArgs; // loadExtensions(); } /** * Instantiates a new jAXWS code generation engine. * * @param commandLineOptionParser * the command line option parser * @param originalArgs * the original args */ public JAXWSCodeGenerationEngine(CommandLineOptionParser commandLineOptionParser, String[] originalArgs) { this.commandLineOptionParser = commandLineOptionParser; this.originalArgs = originalArgs; // loadExtensions(); } /** * Generate. * * @throws CodeGenerationException * the code generation exception */ public void generate() throws CodeGenerationException { LogOutputStream logOutputStream = new LogOutputStream(log, Integer.MAX_VALUE); WsimportTool importTool = new WsimportTool(logOutputStream); ArrayList<String> args = new ArrayList<String>(); configurImportToolOptions(args); mergeOriginalArgs(args); boolean success = importTool.run(args.toArray(new String[args.size()])); if (success) { log.info("Code generation completed"); } } /** * Merge original args. * * @param args * the args */ private void mergeOriginalArgs(ArrayList<String> args) { Map<String, CommandLineOption> allOptions = commandLineOptionParser.getAllOptions(); List<String> axisOptionList = new ArrayList<String>(); List<String> originalArgsOps = new ArrayList<String>(Arrays.asList(originalArgs)); originalArgsOps.remove("-".concat(CommandLineOptionConstants.WSDL2JavaConstants.JAX_WS_SERVICE_OPTION)); originalArgsOps .remove("-".concat(CommandLineOptionConstants.WSDL2JavaConstants.JAX_WS_SERVICE_OPTION_LONG)); Field[] allFields = CommandLineOptionConstants.WSDL2JavaConstants.class.getFields(); Iterator<String> mapItr = allOptions.keySet().iterator(); for (Field field : allFields) { if (String.class.equals(field.getType())) { try { axisOptionList.add((String) field.get(CommandLineOptionConstants.WSDL2JavaConstants.class)); } catch (IllegalArgumentException e) { e.printStackTrace(); } catch (IllegalAccessException e) { e.printStackTrace(); } } } while (mapItr.hasNext()) { CommandLineOption op = allOptions.get(mapItr.next()); if (axisOptionList.contains(op.getOptionType())) { if (op.getOptionType() != null) { originalArgsOps.remove("-".concat(op.getOptionType())); } if (op.getOptionValue() != null) { originalArgsOps.remove(op.getOptionValue()); } } } args.addAll(originalArgsOps); log.info("Original WSImport options" + Arrays.asList(originalArgsOps)); log.info("All WSImport options" + Arrays.asList(args)); } /** * Configur import tool options. * * @param args * the args */ private void configurImportToolOptions(ArrayList<String> args) { Map allOptions = commandLineOptionParser.getAllOptions(); // Set some default options args.add(WS_IMPORT_EXTENSION); args.add(WS_IMPORT_NO_COMPILE); args.add(WS_IMPORT_KEEP_FILE); // Set some properties based on AXIS2 WSDL2JAVA options String uri = getOptionValue(allOptions, CommandLineOptionConstants.WSDL2JavaConstants.WSDL_LOCATION_URI_OPTION); if (uri != null) { args.add(uri); } String location = getOptionValue(allOptions, CommandLineOptionConstants.WSDL2JavaConstants.OUTPUT_LOCATION_OPTION); if (location == null) { location = getOptionValue(allOptions, CommandLineOptionConstants.WSDL2JavaConstants.OUTPUT_LOCATION_OPTION_LONG); } if (location != null) { args.add(WS_IMPORT_FILE_OUTPUT_DIR); args.add(location); } String pkg = getOptionValue(allOptions, CommandLineOptionConstants.WSDL2JavaConstants.PACKAGE_OPTION); if (pkg == null) { pkg = getOptionValue(allOptions, CommandLineOptionConstants.WSDL2JavaConstants.PACKAGE_OPTION_LONG); } if (pkg != null) { args.add(WS_IMPORT_PKG); args.add(pkg); } String proxcyHost = getOptionValue(allOptions, CommandLineOptionConstants.WSDL2JavaConstants.HTTP_PROXY_HOST_OPTION_LONG); String proxcyPort = getOptionValue(allOptions, CommandLineOptionConstants.WSDL2JavaConstants.HTTP_PROXY_PORT_OPTION_LONG); if (pkg != null) { args.add(WS_IMPORT_PROXY); args.add(proxcyHost + ":" + proxcyPort); } } /** * Gets the option value. * * @param allOptions * the all options * @param optionName * the option name * @return the option value */ private static String getOptionValue(Map allOptions, String optionName) { CommandLineOption option = (CommandLineOption) allOptions.get(optionName); if (option != null) { return option.getOptionValue().toString(); } return null; } public static final String WS_IMPORT_EXTENSION = "-extension"; public static final String WS_IMPORT_NO_COMPILE = "-Xnocompile"; public static final String WS_IMPORT_FILE_OUTPUT_DIR = "-d"; public static final String WS_IMPORT_SOURCE_OUTPUT_DIR = "-s"; public static final String WS_IMPORT_KEEP_FILE = "-keep"; public static final String WS_IMPORT_PKG = "-p"; public static final String WS_IMPORT_PROXY = "-httpproxy"; }