summaryrefslogtreecommitdiffstats
path: root/sandbox/scottkurz/core-databinding/src/main/java/org/apache/tuscany/sca/core/databinding/wire/FaultTransformHelper.java
blob: 9c1a054d90c427449f1f6425af2285fa1f363a73 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
package org.apache.tuscany.sca.core.databinding.wire;

import java.lang.reflect.InvocationTargetException;
import java.util.HashMap;
import java.util.Map;

import javax.xml.namespace.QName;

import org.apache.tuscany.sca.databinding.Mediator;
import org.apache.tuscany.sca.interfacedef.DataType;
import org.apache.tuscany.sca.interfacedef.Operation;
import org.apache.tuscany.sca.interfacedef.impl.DataTypeImpl;
import org.apache.tuscany.sca.interfacedef.util.FaultException;
import org.apache.tuscany.sca.interfacedef.util.XMLType;
import org.apache.tuscany.sca.runtime.RuntimeWire;
import org.osoa.sca.ServiceRuntimeException;

public class FaultTransformHelper {
    
    private Mediator mediator;
    
    public FaultTransformHelper(Mediator mediator) {
        this.mediator = mediator;
    }
    
    public Object transformFault(Object result, Operation sourceOperation, Operation targetOperation, RuntimeWire wire) {

            // FIXME: How to match fault data to a fault type for the
            // operation?

            // If the result is from an InvocationTargetException look at
            // the actual cause.
            if (result instanceof InvocationTargetException) {
                result = ((InvocationTargetException)result).getCause();
            }
            DataType targetDataType = null;
            for (DataType exType : targetOperation.getFaultTypes()) {
                if (((Class)exType.getPhysical()).isInstance(result)) {
                    if (result instanceof FaultException) {
                        DataType faultType = (DataType)exType.getLogical();
                        if (((FaultException)result).isMatchingType(faultType.getLogical())) {
                            targetDataType = exType;
                            break;
                        }
                    } else {
                        targetDataType = exType;
                        break;
                    }
                }
            }

            /*
            if (targetDataType == null) {
                // Not a business exception
                return resultMsg;
            }
            */

            DataType targetFaultType = getFaultType(targetDataType);
            if (targetFaultType == null) {
                // No matching fault type, it's a system exception
                Throwable cause = (Throwable) result;
                throw new ServiceRuntimeException(cause);
            }

            // FIXME: How to match a source fault type to a target fault
            // type?
            DataType sourceDataType = null;
            DataType sourceFaultType = null;
            for (DataType exType : sourceOperation.getFaultTypes()) {
                DataType faultType = getFaultType(exType);
                // Match by the QName (XSD element) of the fault type
                if (faultType != null && typesMatch(targetFaultType.getLogical(), faultType.getLogical())) {
                    sourceDataType = exType;
                    sourceFaultType = faultType;
                    break;
                }
            }

            if (sourceFaultType == null) {
                // No matching fault type, it's a system exception
                Throwable cause = (Throwable) result;
                throw new ServiceRuntimeException(cause);
            }

            Map<String, Object> metadata = new HashMap<String, Object>();
            metadata.put("source.operation", targetOperation);
            metadata.put("target.operation", sourceOperation);
            // When is the 'wire' used?
            if (wire != null) {
                metadata.put("wire", wire);
            }
            
            Object newResult =
                transformException(result, targetDataType, sourceDataType, targetFaultType, sourceFaultType, metadata);
            
            return newResult;

    }
    
    /**
     * @param source The source exception
     * @param sourceExType The data type for the source exception
     * @param targetExType The data type for the target exception
     * @param sourceType The fault type for the source
     * @param targetType The fault type for the target
     * @return
     */
    private Object transformException(Object source,
                                      DataType sourceExType,
                                      DataType targetExType,
                                      DataType sourceType,
                                      DataType targetType,
                                      Map<String, Object> metadata) {
        
        if (sourceType == targetType || (sourceType != null && sourceType.equals(targetType))) {
            return source;
        }

        DataType<DataType> eSourceDataType =
            new DataTypeImpl<DataType>("idl:fault", sourceExType.getPhysical(), sourceType);
        DataType<DataType> eTargetDataType =
            new DataTypeImpl<DataType>("idl:fault", targetExType.getPhysical(), targetType);

        return mediator.mediate(source, eSourceDataType, eTargetDataType, metadata);
    }
    
    private DataType getFaultType(DataType exceptionType) {
        return exceptionType == null ? null : (DataType)exceptionType.getLogical();
    }

    private boolean typesMatch(Object first, Object second) {
        if (first.equals(second)) {
            return true;
        }
        if (first instanceof XMLType && second instanceof XMLType) {
            XMLType t1 = (XMLType)first;
            XMLType t2 = (XMLType)second;
            // TUSCANY-2113, we should compare element names only
            return matches(t1.getElementName(), t2.getElementName());
        }
        return false;
    }
    
    /**
     * @param qn1
     * @param qn2
     */
    private boolean matches(QName qn1, QName qn2) {
        if (qn1 == qn2) {
            return true;
        }
        if (qn1 == null || qn2 == null) {
            return false;
        }
        String ns1 = qn1.getNamespaceURI();
        String ns2 = qn2.getNamespaceURI();
        String e1 = qn1.getLocalPart();
        String e2 = qn2.getLocalPart();
        if (e1.equals(e2) && (ns1.equals(ns2) || ns1.equals(ns2 + "/") || ns2.equals(ns1 + "/"))) {
            // Tolerating the trailing / which is required by JAX-WS java package --> xml ns mapping
            return true;
        }
        return false;
    }
}