01: /**
02: * Licensed to the Apache Software Foundation (ASF) under one
03: * or more contributor license agreements. See the NOTICE file
04: * distributed with this work for additional information
05: * regarding copyright ownership. The ASF licenses this file
06: * to you under the Apache License, Version 2.0 (the
07: * "License"); you may not use this file except in compliance
08: * with the License. You may obtain a copy of the License at
09: *
10: * http://www.apache.org/licenses/LICENSE-2.0
11: *
12: * Unless required by applicable law or agreed to in writing,
13: * software distributed under the License is distributed on an
14: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15: * KIND, either express or implied. See the License for the
16: * specific language governing permissions and limitations
17: * under the License.
18: */package org.apache.cxf.tools.wsdlto.frontend.jaxws.processor.internal.annotator;
19:
20: import javax.jws.soap.SOAPBinding;
21: import org.apache.cxf.tools.common.model.Annotator;
22: import org.apache.cxf.tools.common.model.JavaAnnotatable;
23: import org.apache.cxf.tools.common.model.JavaAnnotation;
24: import org.apache.cxf.tools.common.model.JavaMethod;
25:
26: public class WebResultAnnotator implements Annotator {
27:
28: public void annotate(JavaAnnotatable ja) {
29: JavaMethod method = null;
30: if (ja instanceof JavaMethod) {
31: method = (JavaMethod) ja;
32: } else {
33: throw new RuntimeException(
34: "WebResult can only annotate JavaMethod");
35: }
36:
37: if (method.isOneWay()) {
38: JavaAnnotation oneWayAnnotation = new JavaAnnotation(
39: "Oneway");
40: method.addAnnotation("Oneway", oneWayAnnotation);
41: method.getInterface().addImport("javax.jws.Oneway");
42: return;
43: }
44:
45: if ("void".equals(method.getReturn().getType())) {
46: return;
47: }
48: JavaAnnotation resultAnnotation = new JavaAnnotation(
49: "WebResult");
50: String targetNamespace = method.getReturn()
51: .getTargetNamespace();
52: String name = "return";
53:
54: if (method.getSoapStyle() == SOAPBinding.Style.DOCUMENT
55: && !method.isWrapperStyle()) {
56: name = method.getName() + "Response";
57: }
58:
59: if (method.getSoapStyle() == SOAPBinding.Style.RPC) {
60: name = method.getReturn().getName();
61: targetNamespace = method.getInterface().getNamespace();
62:
63: }
64: if (method.getSoapStyle() == SOAPBinding.Style.DOCUMENT) {
65: if (method.getReturn().getQName() != null) {
66: name = method.getReturn().getQName().getLocalPart();
67: }
68: targetNamespace = method.getReturn().getTargetNamespace();
69: }
70:
71: resultAnnotation.addArgument("name", name);
72: resultAnnotation.addArgIgnoreEmtpy("targetNamespace",
73: targetNamespace, "\"");
74:
75: if (method.getSoapStyle() == SOAPBinding.Style.RPC
76: || (method.getSoapStyle() == SOAPBinding.Style.DOCUMENT && !method
77: .isWrapperStyle())) {
78: resultAnnotation.addArgument("partName", method.getReturn()
79: .getName());
80: }
81:
82: method.addAnnotation("WebResult", resultAnnotation);
83: method.getInterface().addImport("javax.jws.WebResult");
84: }
85: }
|