1 /*
2  * Copyright 2005-2010 the original author or authors.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.springframework.ws.server.endpoint.adapter.method.dom;
18
19 import javax.xml.transform.Source;
20 import javax.xml.transform.TransformerException;
21 import javax.xml.transform.dom.DOMSource;
22
23 import org.springframework.core.MethodParameter;
24 import org.springframework.ws.server.endpoint.adapter.method.AbstractPayloadSourceMethodProcessor;
25
26 import org.dom4j.Document;
27 import org.dom4j.Element;
28 import org.dom4j.io.DOMReader;
29 import org.dom4j.io.DocumentResult;
30 import org.dom4j.io.DocumentSource;
31
32 /**
33  * Implementation of {@link org.springframework.ws.server.endpoint.adapter.method.MethodArgumentResolver
34  * MethodArgumentResolver} and {@link org.springframework.ws.server.endpoint.adapter.method.MethodReturnValueHandler
35  * MethodReturnValueHandler} that supports dom4j {@linkplain Element elements}.
36  *
37  * @author Arjen Poutsma
38  * @since 2.0
39  */
40 public class Dom4jPayloadMethodProcessor extends AbstractPayloadSourceMethodProcessor {
41
42     @Override
43     protected boolean supportsRequestPayloadParameter(MethodParameter parameter) {
44         return supports(parameter);
45     }
46
47     @Override
48     protected Element resolveRequestPayloadArgument(MethodParameter parameter, Source requestPayload)
49             throws TransformerException {
50         if (requestPayload instanceof DOMSource) {
51             org.w3c.dom.Node node = ((DOMSource) requestPayload).getNode();
52             if (node.getNodeType() == org.w3c.dom.Node.DOCUMENT_NODE) {
53                 DOMReader domReader = new DOMReader();
54                 Document document = domReader.read((org.w3c.dom.Document) node);
55                 return document.getRootElement();
56             }
57         }
58         // we have no other option than to transform
59         DocumentResult dom4jResult = new DocumentResult();
60         transform(requestPayload, dom4jResult);
61         return dom4jResult.getDocument().getRootElement();
62     }
63
64     @Override
65     protected boolean supportsResponsePayloadReturnType(MethodParameter returnType) {
66         return supports(returnType);
67     }
68
69     @Override
70     protected Source createResponsePayload(MethodParameter returnType, Object returnValue) {
71         Element returnedElement = (Element) returnValue;
72         return new DocumentSource(returnedElement);
73     }
74
75     private boolean supports(MethodParameter parameter) {
76         return Element.class.equals(parameter.getParameterType());
77     }
78
79 }