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.soap.server.endpoint;
18
19 import javax.xml.namespace.QName;
20
21 import org.springframework.util.StringUtils;
22 import org.springframework.ws.soap.server.endpoint.annotation.FaultCode;
23 import org.springframework.ws.soap.server.endpoint.annotation.SoapFault;
24
25 /**
26  * Implementation of the {@link org.springframework.ws.server.EndpointExceptionResolver} interface that uses the {@link
27  * SoapFault} annotation to map exceptions to SOAP Faults.
28  *
29  * @author Arjen Poutsma
30  * @since 1.0.0
31  */
32 public class SoapFaultAnnotationExceptionResolver extends AbstractSoapFaultDefinitionExceptionResolver {
33
34     @Override
35     protected final SoapFaultDefinition getFaultDefinition(Object endpoint, Exception ex) {
36         SoapFault faultAnnotation = ex.getClass().getAnnotation(SoapFault.class);
37         if (faultAnnotation != null) {
38             SoapFaultDefinition definition = new SoapFaultDefinition();
39             if (faultAnnotation.faultCode() != FaultCode.CUSTOM) {
40                 definition.setFaultCode(faultAnnotation.faultCode().value());
41             }
42             else if (StringUtils.hasLength(faultAnnotation.customFaultCode())) {
43                 definition.setFaultCode(QName.valueOf(faultAnnotation.customFaultCode()));
44             }
45             definition.setFaultStringOrReason(faultAnnotation.faultStringOrReason());
46             definition.setLocale(StringUtils.parseLocaleString(faultAnnotation.locale()));
47             return definition;
48         }
49         else {
50             return null;
51         }
52     }
53 }
54