src/share/jaxws_classes/com/sun/tools/internal/ws/processor/modeler/annotation/WebServiceWrapperGenerator.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, 2013, 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.modeler.annotation;
aoqi@0 27
aoqi@0 28 import com.sun.codemodel.internal.CodeWriter;
aoqi@0 29 import com.sun.codemodel.internal.JAnnotationArrayMember;
aoqi@0 30 import com.sun.codemodel.internal.JAnnotationUse;
aoqi@0 31 import com.sun.codemodel.internal.JBlock;
aoqi@0 32 import com.sun.codemodel.internal.JCodeModel;
aoqi@0 33 import com.sun.codemodel.internal.JCommentPart;
aoqi@0 34 import com.sun.codemodel.internal.JDefinedClass;
aoqi@0 35 import com.sun.codemodel.internal.JDocComment;
aoqi@0 36 import com.sun.codemodel.internal.JExpr;
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.codemodel.internal.writer.ProgressCodeWriter;
aoqi@0 43 import com.sun.tools.internal.jxc.ap.InlineAnnotationReaderImpl;
aoqi@0 44 import com.sun.tools.internal.jxc.model.nav.ApNavigator;
aoqi@0 45 import com.sun.tools.internal.ws.ToolVersion;
aoqi@0 46 import com.sun.tools.internal.ws.processor.generator.GeneratorBase;
aoqi@0 47 import com.sun.tools.internal.ws.processor.generator.GeneratorConstants;
aoqi@0 48 import com.sun.tools.internal.ws.processor.generator.Names;
aoqi@0 49 import com.sun.tools.internal.ws.processor.modeler.ModelerException;
aoqi@0 50 import com.sun.tools.internal.ws.processor.util.DirectoryUtil;
aoqi@0 51 import com.sun.tools.internal.ws.resources.WebserviceapMessages;
aoqi@0 52 import com.sun.tools.internal.ws.util.ClassNameInfo;
aoqi@0 53 import com.sun.tools.internal.ws.wscompile.FilerCodeWriter;
aoqi@0 54 import com.sun.tools.internal.ws.wscompile.WsgenOptions;
aoqi@0 55 import com.sun.tools.internal.ws.wsdl.document.soap.SOAPStyle;
aoqi@0 56 import com.sun.xml.internal.bind.v2.model.annotation.AnnotationReader;
aoqi@0 57 import com.sun.xml.internal.bind.v2.model.nav.Navigator;
aoqi@0 58 import com.sun.xml.internal.ws.model.AbstractWrapperBeanGenerator;
aoqi@0 59 import com.sun.xml.internal.ws.spi.db.BindingHelper;
aoqi@0 60 import com.sun.xml.internal.ws.util.StringUtils;
aoqi@0 61
aoqi@0 62 import javax.jws.Oneway;
aoqi@0 63 import javax.jws.WebMethod;
aoqi@0 64 import javax.jws.WebService;
aoqi@0 65 import javax.lang.model.element.ExecutableElement;
aoqi@0 66 import javax.lang.model.element.Name;
aoqi@0 67 import javax.lang.model.element.TypeElement;
aoqi@0 68 import javax.lang.model.type.DeclaredType;
aoqi@0 69 import javax.lang.model.type.MirroredTypeException;
aoqi@0 70 import javax.lang.model.type.TypeKind;
aoqi@0 71 import javax.lang.model.type.TypeMirror;
aoqi@0 72 import javax.xml.bind.annotation.XmlAccessType;
aoqi@0 73 import javax.xml.bind.annotation.XmlAccessorType;
aoqi@0 74 import javax.xml.bind.annotation.XmlAttachmentRef;
aoqi@0 75 import javax.xml.bind.annotation.XmlElement;
aoqi@0 76 import javax.xml.bind.annotation.XmlList;
aoqi@0 77 import javax.xml.bind.annotation.XmlMimeType;
aoqi@0 78 import javax.xml.bind.annotation.XmlRootElement;
aoqi@0 79 import javax.xml.bind.annotation.XmlType;
aoqi@0 80 import javax.xml.bind.annotation.adapters.XmlJavaTypeAdapter;
aoqi@0 81 import javax.xml.namespace.QName;
aoqi@0 82 import javax.xml.ws.RequestWrapper;
aoqi@0 83 import javax.xml.ws.ResponseWrapper;
aoqi@0 84 import javax.xml.ws.WebFault;
aoqi@0 85 import javax.xml.ws.WebServiceException;
aoqi@0 86 import java.io.File;
aoqi@0 87 import java.io.IOException;
aoqi@0 88 import java.lang.annotation.Annotation;
aoqi@0 89 import java.util.Collection;
aoqi@0 90 import java.util.HashSet;
aoqi@0 91 import java.util.List;
aoqi@0 92 import java.util.Set;
aoqi@0 93
aoqi@0 94 import static com.sun.codemodel.internal.ClassType.CLASS;
aoqi@0 95 import static com.sun.tools.internal.ws.processor.modeler.annotation.WebServiceConstants.BEAN;
aoqi@0 96 import static com.sun.tools.internal.ws.processor.modeler.annotation.WebServiceConstants.FAULT_INFO;
aoqi@0 97 import static com.sun.tools.internal.ws.processor.modeler.annotation.WebServiceConstants.JAXWS_PACKAGE_PD;
aoqi@0 98 import static com.sun.tools.internal.ws.processor.modeler.annotation.WebServiceConstants.PD_JAXWS_PACKAGE_PD;
aoqi@0 99 import static com.sun.tools.internal.ws.processor.modeler.annotation.WebServiceConstants.RESPONSE;
aoqi@0 100
aoqi@0 101 /**
aoqi@0 102 * This class generates the request/response and Exception Beans
aoqi@0 103 * used by the JAX-WS runtime.
aoqi@0 104 *
aoqi@0 105 * @author WS Development Team
aoqi@0 106 */
aoqi@0 107 public class WebServiceWrapperGenerator extends WebServiceVisitor {
aoqi@0 108 private Set<String> wrapperNames;
aoqi@0 109 private Set<String> processedExceptions;
aoqi@0 110 private JCodeModel cm;
aoqi@0 111 private final MakeSafeTypeVisitor makeSafeVisitor;
aoqi@0 112
aoqi@0 113 private static final FieldFactory FIELD_FACTORY = new FieldFactory();
aoqi@0 114
aoqi@0 115 private final AbstractWrapperBeanGenerator ap_generator =
aoqi@0 116 new ApWrapperBeanGenerator(InlineAnnotationReaderImpl.theInstance,
aoqi@0 117 new ApNavigator(builder.getProcessingEnvironment()), FIELD_FACTORY);
aoqi@0 118
aoqi@0 119 private final class ApWrapperBeanGenerator extends AbstractWrapperBeanGenerator<TypeMirror, TypeElement, ExecutableElement, MemberInfo> {
aoqi@0 120
aoqi@0 121 protected ApWrapperBeanGenerator(
aoqi@0 122 AnnotationReader<TypeMirror, TypeElement, ?, ExecutableElement> annReader,
aoqi@0 123 Navigator<TypeMirror, TypeElement, ?, ExecutableElement> nav, BeanMemberFactory<TypeMirror, MemberInfo> beanMemberFactory) {
aoqi@0 124 super(annReader, nav, beanMemberFactory);
aoqi@0 125 }
aoqi@0 126
aoqi@0 127 @Override
aoqi@0 128 protected TypeMirror getSafeType(TypeMirror type) {
aoqi@0 129 return WebServiceWrapperGenerator.this.getSafeType(type);
aoqi@0 130 }
aoqi@0 131
aoqi@0 132 @Override
aoqi@0 133 protected TypeMirror getHolderValueType(TypeMirror paramType) {
aoqi@0 134 return builder.getHolderValueType(paramType);
aoqi@0 135 }
aoqi@0 136
aoqi@0 137 @Override
aoqi@0 138 protected boolean isVoidType(TypeMirror type) {
aoqi@0 139 return type != null && type.getKind().equals(TypeKind.VOID);
aoqi@0 140 }
aoqi@0 141
aoqi@0 142 }
aoqi@0 143
aoqi@0 144 private static final class FieldFactory implements AbstractWrapperBeanGenerator.BeanMemberFactory<TypeMirror, MemberInfo> {
aoqi@0 145
aoqi@0 146 @Override
aoqi@0 147 public MemberInfo createWrapperBeanMember(TypeMirror paramType,
aoqi@0 148 String paramName, List<Annotation> jaxb) {
aoqi@0 149 return new MemberInfo(paramType, paramName, jaxb);
aoqi@0 150 }
aoqi@0 151 }
aoqi@0 152
aoqi@0 153 public WebServiceWrapperGenerator(ModelBuilder builder, AnnotationProcessorContext context) {
aoqi@0 154 super(builder, context);
aoqi@0 155 makeSafeVisitor = new MakeSafeTypeVisitor(builder.getProcessingEnvironment());
aoqi@0 156 }
aoqi@0 157
aoqi@0 158 @Override
aoqi@0 159 protected void processWebService(WebService webService, TypeElement d) {
aoqi@0 160 cm = new JCodeModel();
aoqi@0 161 wrapperNames = new HashSet<String>();
aoqi@0 162 processedExceptions = new HashSet<String>();
aoqi@0 163 }
aoqi@0 164
aoqi@0 165 @Override
aoqi@0 166 protected void postProcessWebService(WebService webService, TypeElement d) {
aoqi@0 167 super.postProcessWebService(webService, d);
aoqi@0 168 doPostProcessWebService(webService, d);
aoqi@0 169 }
aoqi@0 170
aoqi@0 171 @SuppressWarnings("CallToThreadDumpStack")
aoqi@0 172 protected void doPostProcessWebService(WebService webService, TypeElement d) {
aoqi@0 173 if (cm != null) {
aoqi@0 174 File sourceDir = builder.getSourceDir();
aoqi@0 175 assert(sourceDir != null);
aoqi@0 176 WsgenOptions options = builder.getOptions();
aoqi@0 177 try {
aoqi@0 178 CodeWriter cw = new FilerCodeWriter(sourceDir, options);
aoqi@0 179 if(options.verbose)
aoqi@0 180 cw = new ProgressCodeWriter(cw, System.out);
aoqi@0 181 cm.build(cw);
aoqi@0 182 } catch (IOException e) {
aoqi@0 183 e.printStackTrace();
aoqi@0 184 }
aoqi@0 185 }
aoqi@0 186 }
aoqi@0 187
aoqi@0 188 @Override
aoqi@0 189 protected void processMethod(ExecutableElement method, WebMethod webMethod) {
aoqi@0 190 builder.log("WrapperGen - method: "+method);
aoqi@0 191 builder.log("method.getDeclaringType(): " + method.asType());
aoqi@0 192 if (wrapped && soapStyle.equals(SOAPStyle.DOCUMENT)) {
aoqi@0 193 generateWrappers(method, webMethod);
aoqi@0 194 }
aoqi@0 195 generateExceptionBeans(method);
aoqi@0 196 }
aoqi@0 197
aoqi@0 198 private boolean generateExceptionBeans(ExecutableElement method) {
aoqi@0 199 String beanPackage = packageName + PD_JAXWS_PACKAGE_PD.getValue();
aoqi@0 200 if (packageName.length() == 0)
aoqi@0 201 beanPackage = JAXWS_PACKAGE_PD.getValue();
aoqi@0 202 boolean beanGenerated = false;
aoqi@0 203 for (TypeMirror thrownType : method.getThrownTypes()) {
aoqi@0 204 TypeElement typeDecl = (TypeElement) ((DeclaredType) thrownType).asElement();
aoqi@0 205 if (typeDecl == null) {
aoqi@0 206 builder.processError(WebserviceapMessages.WEBSERVICEAP_COULD_NOT_FIND_TYPEDECL(
aoqi@0 207 thrownType.toString(), context.getRound()));
aoqi@0 208 return false;
aoqi@0 209 }
aoqi@0 210 boolean tmp = generateExceptionBean(typeDecl, beanPackage);
aoqi@0 211 beanGenerated = beanGenerated || tmp;
aoqi@0 212 }
aoqi@0 213 return beanGenerated;
aoqi@0 214 }
aoqi@0 215
aoqi@0 216 private boolean duplicateName(String name) {
aoqi@0 217 for (String str : wrapperNames) {
aoqi@0 218 if (str.equalsIgnoreCase(name))
aoqi@0 219 return true;
aoqi@0 220 }
aoqi@0 221 wrapperNames.add(name);
aoqi@0 222 return false;
aoqi@0 223 }
aoqi@0 224
aoqi@0 225 private boolean generateWrappers(ExecutableElement method, WebMethod webMethod) {
aoqi@0 226 boolean isOneway = method.getAnnotation(Oneway.class) != null;
aoqi@0 227 String beanPackage = packageName + PD_JAXWS_PACKAGE_PD.getValue();
aoqi@0 228 if (packageName.length() == 0)
aoqi@0 229 beanPackage = JAXWS_PACKAGE_PD.getValue();
aoqi@0 230 Name methodName = method.getSimpleName();
aoqi@0 231 String operationName = builder.getOperationName(methodName);
aoqi@0 232 operationName = webMethod != null && webMethod.operationName().length() > 0 ?
aoqi@0 233 webMethod.operationName() : operationName;
aoqi@0 234 String reqName = operationName;
aoqi@0 235 String resName = operationName + WebServiceConstants.RESPONSE.getValue();
aoqi@0 236 String reqNamespace = typeNamespace;
aoqi@0 237 String resNamespace = typeNamespace;
aoqi@0 238
aoqi@0 239 String requestClassName = beanPackage + StringUtils.capitalize(method.getSimpleName().toString());
aoqi@0 240 RequestWrapper reqWrapper = method.getAnnotation(RequestWrapper.class);
aoqi@0 241 if (reqWrapper != null) {
aoqi@0 242 if (reqWrapper.className().length() > 0)
aoqi@0 243 requestClassName = reqWrapper.className();
aoqi@0 244 if (reqWrapper.localName().length() > 0)
aoqi@0 245 reqName = reqWrapper.localName();
aoqi@0 246 if (reqWrapper.targetNamespace().length() > 0)
aoqi@0 247 reqNamespace = reqWrapper.targetNamespace();
aoqi@0 248 }
aoqi@0 249 builder.log("requestWrapper: "+requestClassName);
aoqi@0 250 ///// fix for wsgen CR 6442344
aoqi@0 251 File file = new File(DirectoryUtil.getOutputDirectoryFor(requestClassName, builder.getSourceDir()),
aoqi@0 252 Names.stripQualifier(requestClassName) + GeneratorConstants.JAVA_SRC_SUFFIX.getValue());
aoqi@0 253 builder.getOptions().addGeneratedFile(file);
aoqi@0 254 //////////
aoqi@0 255 boolean canOverwriteRequest = builder.canOverWriteClass(requestClassName);
aoqi@0 256 if (!canOverwriteRequest) {
aoqi@0 257 builder.log("Class " + requestClassName + " exists. Not overwriting.");
aoqi@0 258 }
aoqi@0 259 if (duplicateName(requestClassName) && canOverwriteRequest) {
aoqi@0 260 builder.processError(WebserviceapMessages.WEBSERVICEAP_METHOD_REQUEST_WRAPPER_BEAN_NAME_NOT_UNIQUE(
aoqi@0 261 typeElement.getQualifiedName(), method.toString()));
aoqi@0 262 }
aoqi@0 263
aoqi@0 264 String responseClassName = null;
aoqi@0 265 boolean canOverwriteResponse = canOverwriteRequest;
aoqi@0 266 if (!isOneway) {
aoqi@0 267 responseClassName = beanPackage+StringUtils.capitalize(method.getSimpleName().toString())+RESPONSE.getValue();
aoqi@0 268 ResponseWrapper resWrapper = method.getAnnotation(ResponseWrapper.class);
aoqi@0 269 if(resWrapper != null) {
aoqi@0 270 if (resWrapper.className().length() > 0)
aoqi@0 271 responseClassName = resWrapper.className();
aoqi@0 272 if (resWrapper.localName().length() > 0)
aoqi@0 273 resName = resWrapper.localName();
aoqi@0 274 if (resWrapper.targetNamespace().length() > 0)
aoqi@0 275 resNamespace = resWrapper.targetNamespace();
aoqi@0 276 }
aoqi@0 277 canOverwriteResponse = builder.canOverWriteClass(responseClassName);
aoqi@0 278 if (!canOverwriteResponse) {
aoqi@0 279 builder.log("Class " + responseClassName + " exists. Not overwriting.");
aoqi@0 280 }
aoqi@0 281 if (duplicateName(responseClassName) && canOverwriteResponse) {
aoqi@0 282 builder.processError(WebserviceapMessages.WEBSERVICEAP_METHOD_RESPONSE_WRAPPER_BEAN_NAME_NOT_UNIQUE(
aoqi@0 283 typeElement.getQualifiedName(), method.toString()));
aoqi@0 284 }
aoqi@0 285 file = new File(DirectoryUtil.getOutputDirectoryFor(responseClassName, builder.getSourceDir()),
aoqi@0 286 Names.stripQualifier(responseClassName) + GeneratorConstants.JAVA_SRC_SUFFIX.getValue());
aoqi@0 287 builder.getOptions().addGeneratedFile(file);
aoqi@0 288 }
aoqi@0 289 //ArrayList<MemberInfo> reqMembers = new ArrayList<MemberInfo>();
aoqi@0 290 //ArrayList<MemberInfo> resMembers = new ArrayList<MemberInfo>();
aoqi@0 291 WrapperInfo reqWrapperInfo = new WrapperInfo(requestClassName);
aoqi@0 292 //reqWrapperInfo.setMembers(reqMembers);
aoqi@0 293 WrapperInfo resWrapperInfo = null;
aoqi@0 294 if (!isOneway) {
aoqi@0 295 resWrapperInfo = new WrapperInfo(responseClassName);
aoqi@0 296 //resWrapperInfo.setMembers(resMembers);
aoqi@0 297 }
aoqi@0 298 seiContext.setReqWrapperOperation(method, reqWrapperInfo);
aoqi@0 299 if (!isOneway)
aoqi@0 300 seiContext.setResWrapperOperation(method, resWrapperInfo);
aoqi@0 301 try {
aoqi@0 302 if (!canOverwriteRequest && !canOverwriteResponse) {
aoqi@0 303 return false;
aoqi@0 304 }
aoqi@0 305
aoqi@0 306 JDefinedClass reqCls = null;
aoqi@0 307 if (canOverwriteRequest) {
aoqi@0 308 reqCls = getCMClass(requestClassName, CLASS);
aoqi@0 309 }
aoqi@0 310
aoqi@0 311 JDefinedClass resCls = null;
aoqi@0 312 if (!isOneway && canOverwriteResponse) {
aoqi@0 313 resCls = getCMClass(responseClassName, CLASS);
aoqi@0 314 }
aoqi@0 315
aoqi@0 316 // XMLElement Declarations
aoqi@0 317 writeXmlElementDeclaration(reqCls, reqName,reqNamespace);
aoqi@0 318 writeXmlElementDeclaration(resCls, resName, resNamespace);
aoqi@0 319
aoqi@0 320 List<MemberInfo> reqMembers = ap_generator.collectRequestBeanMembers(method);
aoqi@0 321 List<MemberInfo> resMembers = ap_generator.collectResponseBeanMembers(method);
aoqi@0 322
aoqi@0 323 // XmlType
aoqi@0 324 writeXmlTypeDeclaration(reqCls, reqName, reqNamespace, reqMembers);
aoqi@0 325 writeXmlTypeDeclaration(resCls, resName, resNamespace, resMembers);
aoqi@0 326
aoqi@0 327 // class members
aoqi@0 328 writeMembers(reqCls, reqMembers);
aoqi@0 329 writeMembers(resCls, resMembers);
aoqi@0 330
aoqi@0 331 } catch (Exception e) {
aoqi@0 332 throw new ModelerException("modeler.nestedGeneratorError",e);
aoqi@0 333 }
aoqi@0 334 return true;
aoqi@0 335 }
aoqi@0 336
aoqi@0 337 // private List<Annotation> collectJAXBAnnotations(Declaration decl) {
aoqi@0 338 // List<Annotation> jaxbAnnotation = new ArrayList<Annotation>();
aoqi@0 339 // for(Class jaxbClass : jaxbAnns) {
aoqi@0 340 // Annotation ann = decl.getAnnotation(jaxbClass);
aoqi@0 341 // if (ann != null) {
aoqi@0 342 // jaxbAnnotation.add(ann);
aoqi@0 343 // }
aoqi@0 344 // }
aoqi@0 345 // return jaxbAnnotation;
aoqi@0 346 // }
aoqi@0 347
aoqi@0 348 private TypeMirror getSafeType(TypeMirror type) {
aoqi@0 349 return makeSafeVisitor.visit(type, builder.getProcessingEnvironment().getTypeUtils());
aoqi@0 350 }
aoqi@0 351
aoqi@0 352 private JType getType(TypeMirror typeMirror) {
aoqi@0 353 String type = typeMirror.toString();
aoqi@0 354 try {
aoqi@0 355 // System.out.println("typeName: "+typeName);
aoqi@0 356 return cm.parseType(type);
aoqi@0 357 // System.out.println("type: "+type);
aoqi@0 358 } catch (ClassNotFoundException e) {
aoqi@0 359 return cm.ref(type);
aoqi@0 360 }
aoqi@0 361 }
aoqi@0 362
aoqi@0 363 private void writeMembers(JDefinedClass cls, Collection<MemberInfo> members) {
aoqi@0 364 if (cls == null)
aoqi@0 365 return;
aoqi@0 366 for (MemberInfo memInfo : members) {
aoqi@0 367 JType type = getType(memInfo.getParamType());
aoqi@0 368 JFieldVar field = cls.field(JMod.PRIVATE, type, memInfo.getParamName());
aoqi@0 369 annotateParameterWithJaxbAnnotations(memInfo, field);
aoqi@0 370 }
aoqi@0 371 for (MemberInfo memInfo : members) {
aoqi@0 372 writeMember(cls, memInfo.getParamType(),
aoqi@0 373 memInfo.getParamName());
aoqi@0 374 }
aoqi@0 375 }
aoqi@0 376
aoqi@0 377 private void annotateParameterWithJaxbAnnotations(MemberInfo memInfo, JFieldVar field) {
aoqi@0 378 List<Annotation> jaxbAnnotations = memInfo.getJaxbAnnotations();
aoqi@0 379 for(Annotation ann : jaxbAnnotations) {
aoqi@0 380 if (ann instanceof XmlMimeType) {
aoqi@0 381 JAnnotationUse jaxbAnn = field.annotate(XmlMimeType.class);
aoqi@0 382 jaxbAnn.param("value", ((XmlMimeType)ann).value());
aoqi@0 383 } else if (ann instanceof XmlJavaTypeAdapter) {
aoqi@0 384 JAnnotationUse jaxbAnn = field.annotate(XmlJavaTypeAdapter.class);
aoqi@0 385 XmlJavaTypeAdapter ja = (XmlJavaTypeAdapter) ann;
aoqi@0 386 try {
aoqi@0 387 ja.value();
aoqi@0 388 throw new AssertionError();
aoqi@0 389 } catch (MirroredTypeException e) {
aoqi@0 390 jaxbAnn.param("value",getType(e.getTypeMirror()));
aoqi@0 391 }
aoqi@0 392 // XmlJavaTypeAdapter.type() is for package only. No need to copy.
aoqi@0 393 } else if (ann instanceof XmlAttachmentRef) {
aoqi@0 394 field.annotate(XmlAttachmentRef.class);
aoqi@0 395 } else if (ann instanceof XmlList){
aoqi@0 396 field.annotate(XmlList.class);
aoqi@0 397 } else if (ann instanceof XmlElement) {
aoqi@0 398 XmlElement elemAnn = (XmlElement)ann;
aoqi@0 399 JAnnotationUse jAnn = field.annotate(XmlElement.class);
aoqi@0 400 jAnn.param("name", elemAnn.name());
aoqi@0 401 jAnn.param("namespace", elemAnn.namespace());
aoqi@0 402 if (elemAnn.nillable()) {
aoqi@0 403 jAnn.param("nillable", true);
aoqi@0 404 }
aoqi@0 405 if (elemAnn.required()) {
aoqi@0 406 jAnn.param("required", true);
aoqi@0 407 }
aoqi@0 408 } else {
aoqi@0 409 throw new WebServiceException("SEI Parameter cannot have this JAXB annotation: " + ann);
aoqi@0 410 }
aoqi@0 411 }
aoqi@0 412 }
aoqi@0 413
aoqi@0 414 protected JDefinedClass getCMClass(String className, com.sun.codemodel.internal.ClassType type) {
aoqi@0 415 JDefinedClass cls;
aoqi@0 416 try {
aoqi@0 417 cls = cm._class(className, type);
aoqi@0 418 } catch (com.sun.codemodel.internal.JClassAlreadyExistsException e){
aoqi@0 419 cls = cm._getClass(className);
aoqi@0 420 }
aoqi@0 421 return cls;
aoqi@0 422 }
aoqi@0 423
aoqi@0 424 private boolean generateExceptionBean(TypeElement thrownDecl, String beanPackage) {
aoqi@0 425 if (!builder.isServiceException(thrownDecl.asType()))
aoqi@0 426 return false;
aoqi@0 427
aoqi@0 428 String exceptionName = ClassNameInfo.getName(thrownDecl.getQualifiedName().toString());
aoqi@0 429 if (processedExceptions.contains(exceptionName))
aoqi@0 430 return false;
aoqi@0 431 processedExceptions.add(exceptionName);
aoqi@0 432 WebFault webFault = thrownDecl.getAnnotation(WebFault.class);
aoqi@0 433 String className = beanPackage+ exceptionName + BEAN.getValue();
aoqi@0 434
aoqi@0 435 Collection<MemberInfo> members = ap_generator.collectExceptionBeanMembers(thrownDecl);
aoqi@0 436 boolean isWSDLException = isWSDLException(members, thrownDecl);
aoqi@0 437 String namespace = typeNamespace;
aoqi@0 438 String name = exceptionName;
aoqi@0 439 FaultInfo faultInfo;
aoqi@0 440 if (isWSDLException) {
aoqi@0 441 TypeMirror beanType = getFaultInfoMember(members).getParamType();
aoqi@0 442 faultInfo = new FaultInfo(TypeMonikerFactory.getTypeMoniker(beanType), true);
aoqi@0 443 namespace = webFault.targetNamespace().length()>0 ?
aoqi@0 444 webFault.targetNamespace() : namespace;
aoqi@0 445 name = webFault.name().length()>0 ?
aoqi@0 446 webFault.name() : name;
aoqi@0 447 faultInfo.setElementName(new QName(namespace, name));
aoqi@0 448 seiContext.addExceptionBeanEntry(thrownDecl.getQualifiedName(), faultInfo, builder);
aoqi@0 449 return false;
aoqi@0 450 }
aoqi@0 451 if (webFault != null) {
aoqi@0 452 namespace = webFault.targetNamespace().length()>0 ?
aoqi@0 453 webFault.targetNamespace() : namespace;
aoqi@0 454 name = webFault.name().length()>0 ?
aoqi@0 455 webFault.name() : name;
aoqi@0 456 className = webFault.faultBean().length()>0 ?
aoqi@0 457 webFault.faultBean() : className;
aoqi@0 458
aoqi@0 459 }
aoqi@0 460 JDefinedClass cls = getCMClass(className, CLASS);
aoqi@0 461 faultInfo = new FaultInfo(className, false);
aoqi@0 462
aoqi@0 463 if (duplicateName(className)) {
aoqi@0 464 builder.processError(WebserviceapMessages.WEBSERVICEAP_METHOD_EXCEPTION_BEAN_NAME_NOT_UNIQUE(
aoqi@0 465 typeElement.getQualifiedName(), thrownDecl.getQualifiedName()));
aoqi@0 466 }
aoqi@0 467
aoqi@0 468 boolean canOverWriteBean = builder.canOverWriteClass(className);
aoqi@0 469 if (!canOverWriteBean) {
aoqi@0 470 builder.log("Class " + className + " exists. Not overwriting.");
aoqi@0 471 seiContext.addExceptionBeanEntry(thrownDecl.getQualifiedName(), faultInfo, builder);
aoqi@0 472 return false;
aoqi@0 473 }
aoqi@0 474 if (seiContext.getExceptionBeanName(thrownDecl.getQualifiedName()) != null)
aoqi@0 475 return false;
aoqi@0 476
aoqi@0 477 //write class comment - JAXWS warning
aoqi@0 478 JDocComment comment = cls.javadoc();
aoqi@0 479 for (String doc : GeneratorBase.getJAXWSClassComment(ToolVersion.VERSION.MAJOR_VERSION)) {
aoqi@0 480 comment.add(doc);
aoqi@0 481 }
aoqi@0 482
aoqi@0 483 // XmlElement Declarations
aoqi@0 484 writeXmlElementDeclaration(cls, name, namespace);
aoqi@0 485
aoqi@0 486 // XmlType Declaration
aoqi@0 487 //members = sortMembers(members);
aoqi@0 488 XmlType xmlType = thrownDecl.getAnnotation(XmlType.class);
aoqi@0 489 String xmlTypeName = (xmlType != null && !xmlType.name().equals("##default")) ? xmlType.name() : exceptionName;
aoqi@0 490 String xmlTypeNamespace = (xmlType != null && !xmlType.namespace().equals("##default")) ? xmlType.namespace() : typeNamespace;
aoqi@0 491 writeXmlTypeDeclaration(cls, xmlTypeName, xmlTypeNamespace, members);
aoqi@0 492
aoqi@0 493 writeMembers(cls, members);
aoqi@0 494
aoqi@0 495 seiContext.addExceptionBeanEntry(thrownDecl.getQualifiedName(), faultInfo, builder);
aoqi@0 496 return true;
aoqi@0 497 }
aoqi@0 498
aoqi@0 499 protected boolean isWSDLException(Collection<MemberInfo> members, TypeElement thrownDecl) {
aoqi@0 500 WebFault webFault = thrownDecl.getAnnotation(WebFault.class);
aoqi@0 501 return webFault != null && members.size() == 2 && getFaultInfoMember(members) != null;
aoqi@0 502 }
aoqi@0 503
aoqi@0 504 /*
aoqi@0 505 * Returns the corresponding MemberInfo for getFaultInfo()
aoqi@0 506 * method of an exception. Returns null, if that method is not there.
aoqi@0 507 */
aoqi@0 508 private MemberInfo getFaultInfoMember(Collection<MemberInfo> members) {
aoqi@0 509 for(MemberInfo member : members) {
aoqi@0 510 if (member.getParamName().equals(FAULT_INFO.getValue())) {
aoqi@0 511 return member;
aoqi@0 512 }
aoqi@0 513 }
aoqi@0 514 return null;
aoqi@0 515 }
aoqi@0 516
aoqi@0 517 private void writeXmlElementDeclaration(JDefinedClass cls, String elementName, String namespaceUri) {
aoqi@0 518
aoqi@0 519 if (cls == null)
aoqi@0 520 return;
aoqi@0 521 JAnnotationUse xmlRootElementAnn = cls.annotate(XmlRootElement.class);
aoqi@0 522 xmlRootElementAnn.param("name", elementName);
aoqi@0 523 if (namespaceUri.length() > 0) {
aoqi@0 524 xmlRootElementAnn.param("namespace", namespaceUri);
aoqi@0 525 }
aoqi@0 526 JAnnotationUse xmlAccessorTypeAnn = cls.annotate(cm.ref(XmlAccessorType.class));
aoqi@0 527 xmlAccessorTypeAnn.param("value", XmlAccessType.FIELD);
aoqi@0 528 }
aoqi@0 529
aoqi@0 530 private void writeXmlTypeDeclaration(JDefinedClass cls, String typeName, String namespaceUri,
aoqi@0 531 Collection<MemberInfo> members) {
aoqi@0 532 if (cls == null)
aoqi@0 533 return;
aoqi@0 534 JAnnotationUse xmlTypeAnn = cls.annotate(cm.ref(XmlType.class));
aoqi@0 535 xmlTypeAnn.param("name", typeName);
aoqi@0 536 xmlTypeAnn.param("namespace", namespaceUri);
aoqi@0 537 if (members.size() > 1) {
aoqi@0 538 JAnnotationArrayMember paramArray = xmlTypeAnn.paramArray("propOrder");
aoqi@0 539 for (MemberInfo memInfo : members) {
aoqi@0 540 paramArray.param(memInfo.getParamName());
aoqi@0 541 }
aoqi@0 542 }
aoqi@0 543 }
aoqi@0 544
aoqi@0 545 private void writeMember(JDefinedClass cls, TypeMirror paramType,
aoqi@0 546 String paramName) {
aoqi@0 547
aoqi@0 548 if (cls == null)
aoqi@0 549 return;
aoqi@0 550
aoqi@0 551 String accessorName =BindingHelper.mangleNameToPropertyName(paramName);
aoqi@0 552 String getterPrefix = paramType.toString().equals("boolean")? "is" : "get";
aoqi@0 553 JType propType = getType(paramType);
aoqi@0 554 JMethod m = cls.method(JMod.PUBLIC, propType, getterPrefix+ accessorName);
aoqi@0 555 JDocComment methodDoc = m.javadoc();
aoqi@0 556 JCommentPart ret = methodDoc.addReturn();
aoqi@0 557 ret.add("returns "+propType.name());
aoqi@0 558 JBlock body = m.body();
aoqi@0 559 body._return( JExpr._this().ref(paramName) );
aoqi@0 560
aoqi@0 561 m = cls.method(JMod.PUBLIC, cm.VOID, "set"+accessorName);
aoqi@0 562 JVar param = m.param(propType, paramName);
aoqi@0 563 methodDoc = m.javadoc();
aoqi@0 564 JCommentPart part = methodDoc.addParam(paramName);
aoqi@0 565 part.add("the value for the "+ paramName+" property");
aoqi@0 566 body = m.body();
aoqi@0 567 body.assign( JExpr._this().ref(paramName), param );
aoqi@0 568 }
aoqi@0 569 }

mercurial