• file
• revisions
• annotate
• diff
• comparison
• raw

src/share/jaxws_classes/com/sun/tools/internal/ws/processor/generator/CustomExceptionGenerator.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/tools/internal/ws/processor/generator/CustomExceptionGenerator.java

Thu, 31 Aug 2017 15:18:52 +0800

author

aoqi

date

Thu, 31 Aug 2017 15:18:52 +0800

changeset 637

9c07ef4934dd

parent 368

0989ad8c0860

parent 0

373ffda63c9a

permissions

-rw-r--r--

merge

aoqi@0	1	/*
aoqi@0	2	* Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
aoqi@0	3	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0	4	*
aoqi@0	5	* This code is free software; you can redistribute it and/or modify it
aoqi@0	6	* under the terms of the GNU General Public License version 2 only, as
aoqi@0	7	* published by the Free Software Foundation. Oracle designates this
aoqi@0	8	* particular file as subject to the "Classpath" exception as provided
aoqi@0	9	* by Oracle in the LICENSE file that accompanied this code.
aoqi@0	10	*
aoqi@0	11	* This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0	12	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0	13	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
aoqi@0	14	* version 2 for more details (a copy is included in the LICENSE file that
aoqi@0	15	* accompanied this code).
aoqi@0	16	*
aoqi@0	17	* You should have received a copy of the GNU General Public License version
aoqi@0	18	* 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0	19	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0	20	*
aoqi@0	21	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0	22	* or visit www.oracle.com if you need additional information or have any
aoqi@0	23	* questions.
aoqi@0	24	*/
aoqi@0	25
aoqi@0	26	package com.sun.tools.internal.ws.processor.generator;
aoqi@0	27
aoqi@0	28	import com.sun.codemodel.internal.ClassType;
aoqi@0	29	import com.sun.codemodel.internal.JAnnotationUse;
aoqi@0	30	import com.sun.codemodel.internal.JBlock;
aoqi@0	31	import com.sun.codemodel.internal.JClass;
aoqi@0	32	import com.sun.codemodel.internal.JClassAlreadyExistsException;
aoqi@0	33	import com.sun.codemodel.internal.JDefinedClass;
aoqi@0	34	import com.sun.codemodel.internal.JDocComment;
aoqi@0	35	import com.sun.codemodel.internal.JExpr;
aoqi@0	36	import com.sun.codemodel.internal.JFieldRef;
aoqi@0	37	import com.sun.codemodel.internal.JFieldVar;
aoqi@0	38	import com.sun.codemodel.internal.JMethod;
aoqi@0	39	import com.sun.codemodel.internal.JMod;
aoqi@0	40	import com.sun.codemodel.internal.JType;
aoqi@0	41	import com.sun.codemodel.internal.JVar;
aoqi@0	42	import com.sun.tools.internal.ws.processor.model.Fault;
aoqi@0	43	import com.sun.tools.internal.ws.processor.model.Model;
aoqi@0	44	import com.sun.tools.internal.ws.wscompile.ErrorReceiver;
aoqi@0	45	import com.sun.tools.internal.ws.wscompile.WsimportOptions;
aoqi@0	46
aoqi@0	47	import javax.xml.ws.WebFault;
aoqi@0	48	import java.util.HashMap;
aoqi@0	49	import java.util.Map;
aoqi@0	50
aoqi@0	51	/**
aoqi@0	52	*
aoqi@0	53	* @author WS Development Team
aoqi@0	54	*/
aoqi@0	55	public class CustomExceptionGenerator extends GeneratorBase {
aoqi@0	56	private Map<String, JClass> faults = new HashMap<String, JClass>();
aoqi@0	57
aoqi@0	58	public static void generate(Model model,
aoqi@0	59	WsimportOptions options,
aoqi@0	60	ErrorReceiver receiver){
aoqi@0	61	CustomExceptionGenerator exceptionGen = new CustomExceptionGenerator();
aoqi@0	62	exceptionGen.init(model, options, receiver);
aoqi@0	63	exceptionGen.doGeneration();
aoqi@0	64	}
aoqi@0	65
aoqi@0	66	public GeneratorBase getGenerator(Model model, WsimportOptions options, ErrorReceiver receiver) {
aoqi@0	67	GeneratorBase g = new CustomExceptionGenerator();
aoqi@0	68	g.init(model, options, receiver);
aoqi@0	69	return g;
aoqi@0	70	}
aoqi@0	71
aoqi@0	72	@Override
aoqi@0	73	public void visit(Fault fault) throws Exception {
aoqi@0	74	if (isRegistered(fault))
aoqi@0	75	return;
aoqi@0	76	registerFault(fault);
aoqi@0	77	}
aoqi@0	78
aoqi@0	79	private boolean isRegistered(Fault fault) {
aoqi@0	80	if(faults.keySet().contains(fault.getJavaException().getName())){
aoqi@0	81	fault.setExceptionClass(faults.get(fault.getJavaException().getName()));
aoqi@0	82	return true;
aoqi@0	83	}
aoqi@0	84	return false;
aoqi@0	85	}
aoqi@0	86
aoqi@0	87	private void registerFault(Fault fault) {
aoqi@0	88	try {
aoqi@0	89	write(fault);
aoqi@0	90	faults.put(fault.getJavaException().getName(), fault.getExceptionClass());
aoqi@0	91	} catch (JClassAlreadyExistsException e) {
aoqi@0	92	throw new GeneratorException("generator.nestedGeneratorError",e);
aoqi@0	93	}
aoqi@0	94	}
aoqi@0	95
aoqi@0	96	private void write(Fault fault) throws JClassAlreadyExistsException {
aoqi@0	97	String className = Names.customExceptionClassName(fault);
aoqi@0	98
aoqi@0	99	JDefinedClass cls = cm._class(className, ClassType.CLASS);
aoqi@0	100	JDocComment comment = cls.javadoc();
aoqi@0	101	if(fault.getJavaDoc() != null){
aoqi@0	102	comment.add(fault.getJavaDoc());
aoqi@0	103	comment.add("\n\n");
aoqi@0	104	}
aoqi@0	105
aoqi@0	106	for (String doc : getJAXWSClassComment()) {
aoqi@0	107	comment.add(doc);
aoqi@0	108	}
aoqi@0	109
aoqi@0	110	cls._extends(java.lang.Exception.class);
aoqi@0	111
aoqi@0	112	//@WebFault
aoqi@0	113	JAnnotationUse faultAnn = cls.annotate(WebFault.class);
aoqi@0	114	faultAnn.param("name", fault.getBlock().getName().getLocalPart());
aoqi@0	115	faultAnn.param("targetNamespace", fault.getBlock().getName().getNamespaceURI());
aoqi@0	116
aoqi@0	117	JType faultBean = fault.getBlock().getType().getJavaType().getType().getType();
aoqi@0	118
aoqi@0	119	//faultInfo filed
aoqi@0	120	JFieldVar fi = cls.field(JMod.PRIVATE, faultBean, "faultInfo");
aoqi@0	121
aoqi@0	122	//add jaxb annotations
aoqi@0	123	fault.getBlock().getType().getJavaType().getType().annotate(fi);
aoqi@0	124
aoqi@0	125	fi.javadoc().add("Java type that goes as soapenv:Fault detail element.");
aoqi@0	126	JFieldRef fr = JExpr.ref(JExpr._this(), fi);
aoqi@0	127
aoqi@0	128	//Constructor
aoqi@0	129	JMethod constrc1 = cls.constructor(JMod.PUBLIC);
aoqi@0	130	JVar var1 = constrc1.param(String.class, "message");
aoqi@0	131	JVar var2 = constrc1.param(faultBean, "faultInfo");
aoqi@0	132	constrc1.javadoc().addParam(var1);
aoqi@0	133	constrc1.javadoc().addParam(var2);
aoqi@0	134	JBlock cb1 = constrc1.body();
aoqi@0	135	cb1.invoke("super").arg(var1);
aoqi@0	136
aoqi@0	137	cb1.assign(fr, var2);
aoqi@0	138
aoqi@0	139	//constructor with Throwable
aoqi@0	140	JMethod constrc2 = cls.constructor(JMod.PUBLIC);
aoqi@0	141	var1 = constrc2.param(String.class, "message");
aoqi@0	142	var2 = constrc2.param(faultBean, "faultInfo");
aoqi@0	143	JVar var3 = constrc2.param(Throwable.class, "cause");
aoqi@0	144	constrc2.javadoc().addParam(var1);
aoqi@0	145	constrc2.javadoc().addParam(var2);
aoqi@0	146	constrc2.javadoc().addParam(var3);
aoqi@0	147	JBlock cb2 = constrc2.body();
aoqi@0	148	cb2.invoke("super").arg(var1).arg(var3);
aoqi@0	149	cb2.assign(fr, var2);
aoqi@0	150
aoqi@0	151
aoqi@0	152	//getFaultInfo() method
aoqi@0	153	JMethod fim = cls.method(JMod.PUBLIC, faultBean, "getFaultInfo");
aoqi@0	154	fim.javadoc().addReturn().add("returns fault bean: "+faultBean.fullName());
aoqi@0	155	JBlock fib = fim.body();
aoqi@0	156	fib._return(fi);
aoqi@0	157	fault.setExceptionClass(cls);
aoqi@0	158
aoqi@0	159	}
aoqi@0	160	}