blob: 29eb24bc61ff8efc7ca4d85f9a3a624d264eacac (
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
|
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.tuscany.sca.binding.axis2;
import java.lang.reflect.InvocationTargetException;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.soap.SOAPEnvelope;
import org.apache.axis2.AxisFault;
import org.apache.axis2.Constants;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.engine.AxisEngine;
import org.apache.axis2.util.Utils;
import org.apache.tuscany.sca.binding.axis2.Axis2ServiceBindingProvider.InvocationContext;
import org.apache.tuscany.sca.invocation.Invoker;
import org.apache.tuscany.sca.invocation.Message;
public class Axis2ServiceCallbackTargetInvoker implements Invoker {
private Axis2ServiceBindingProvider service;
protected static final OMElement RESPONSE = null;
public Axis2ServiceCallbackTargetInvoker(Axis2ServiceBindingProvider service) {
this.service = service;
}
private Object invokeTarget(final Object payload, Object correlationId) throws InvocationTargetException {
try {
// Use current correlation id as index to retrieve inv context
InvocationContext invCtx = service.retrieveMapping(correlationId);
MessageContext outMC = Utils.createOutMessageContext(invCtx.inMessageContext);
outMC.getOperationContext().addMessageContext(outMC);
OMElement responseOM;
if (payload != null && !payload.getClass().isArray()) {
responseOM = (OMElement) payload;
} else {
responseOM = (OMElement) ((Object[]) payload)[0];
}
SOAPEnvelope soapEnvelope = invCtx.soapFactory.getDefaultEnvelope();
soapEnvelope.getBody().addChild(responseOM);
outMC.setEnvelope(soapEnvelope);
outMC.getOperationContext().setProperty(Constants.RESPONSE_WRITTEN, Constants.VALUE_TRUE);
AxisEngine engine =
new AxisEngine(
invCtx.inMessageContext.getOperationContext().getServiceContext().getConfigurationContext());
engine.send(outMC);
invCtx.doneSignal.countDown();
service.removeMapping(correlationId);
} catch (AxisFault e) {
throw new InvocationTargetException(e);
} catch (Throwable t) {
throw new InvocationTargetException(t);
}
return RESPONSE;
}
public Message invoke(Message msg) {
try {
Object correlationId = msg.getCorrelationID();
if (correlationId == null) {
throw new RuntimeException("Missing correlation id");
}
Object resp = invokeTarget(msg.getBody(), correlationId);
msg.setBody(resp);
} catch (InvocationTargetException e) {
msg.setFaultBody(e.getCause());
} catch (Throwable e) {
msg.setFaultBody(e);
}
return msg;
}
public Axis2ServiceCallbackTargetInvoker clone() throws CloneNotSupportedException {
try {
return (Axis2ServiceCallbackTargetInvoker) super.clone();
} catch (CloneNotSupportedException e) {
// will not happen
return null;
}
}
}
|