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
|
/*
* 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.notification;
import java.io.OutputStream;
import java.util.HashMap;
import org.apache.axiom.om.OMElement;
import org.apache.tuscany.sca.binding.notification.NotificationReferenceBindingProvider.SubscriberInfo;
import org.apache.tuscany.sca.binding.notification.encoding.Constants;
import org.apache.tuscany.sca.binding.notification.util.IOUtils;
import org.apache.tuscany.sca.binding.notification.util.IOUtils.IOUtilsException;
import org.apache.tuscany.sca.binding.notification.util.IOUtils.Writeable;
import org.apache.tuscany.sca.implementation.notification.ImmutableMessage;
import org.apache.tuscany.sca.interfacedef.Operation;
import org.apache.tuscany.sca.invocation.Invoker;
import org.apache.tuscany.sca.invocation.Message;
/**
* Turns invoke into remote message fan-out
*
* @version $Rev$ $Date$
*/
public class NotificationReferenceBindingInvoker implements Invoker {
private static final Message RESPONSE = new ImmutableMessage();
private Operation operation;
private NotificationReferenceBindingProvider notificationReferenceBindingProvider;
public NotificationReferenceBindingInvoker(Operation operation,
NotificationReferenceBindingProvider notificationReferenceBindingProvider) {
this.operation = operation;
this.notificationReferenceBindingProvider = notificationReferenceBindingProvider;
}
public Message invoke(Message msg) {
Object payload = msg.getBody();
if (payload == null) {
throw new RuntimeException("Message body is null");
}
Writeable writeable = null;
String incomingBrokerID = null;
if (payload.getClass().isArray()) {
Object[] bodyArray = (Object[])payload;
if (bodyArray.length == 3) {
writeable = getWriteableFromByteArray((byte[])bodyArray[1]);
incomingBrokerID = (String)bodyArray[2];
} else if (bodyArray.length == 1) {
writeable = getWriteableFromPayload(bodyArray[0]);
} else {
throw new RuntimeException("Invalid body array size");
}
} else {
writeable = getWriteableFromPayload(payload);
}
try {
for (SubscriberInfo subscriber : notificationReferenceBindingProvider.getSubscribers()) {
// check for each subscriber's broker id and skip if equal
if (incomingBrokerID != null && subscriber.brokerID != null && incomingBrokerID.equals(subscriber.brokerID)) {
continue;
}
HashMap<String, String> headers = new HashMap<String, String>();
headers.put(IOUtils.Notification_Operation, operation.getName());
String brokerID = notificationReferenceBindingProvider.getBrokerID();
if (brokerID != null) {
headers.put(Constants.Broker_ID, brokerID);
}
IOUtils.sendHttpRequest(subscriber.address, headers, writeable, null);
}
} catch(Exception e) {
e.printStackTrace();
throw new RuntimeException("Sender caught exception", e);
}
return RESPONSE;
}
private Writeable getWriteableFromPayload(Object payload) throws RuntimeException {
if (!(payload instanceof OMElement)) {
throw new RuntimeException("payload not OMElement");
}
final OMElement element = (OMElement)payload;
Writeable writeable = new Writeable() {
public void write(OutputStream os) throws IOUtilsException {
try {
element.serialize(os);
os.flush();
} catch(Exception e) {
throw new IOUtilsException(e);
}
}
};
return writeable;
}
private Writeable getWriteableFromByteArray(final byte[] payload) {
Writeable writeable = new Writeable() {
public void write(OutputStream os) throws IOUtilsException {
try {
os.write(payload);
os.flush();
} catch(Exception e) {
throw new IOUtilsException(e);
}
}
};
return writeable;
}
}
|