/** Callback invoked by the protocol stack to deliver a message batch */
public void up(MessageBatch batch) {
if (stats) {
received_msgs += batch.size();
received_bytes += batch.length();
}
// discard local messages (sent by myself to me)
if (discard_own_messages
&& local_addr != null
&& batch.sender() != null
&& local_addr.equals(batch.sender())) return;
for (Message msg : batch) {
if (up_handler != null) {
try {
up_handler.up(new Event(Event.MSG, msg));
} catch (Throwable t) {
log.error(Util.getMessage("UpHandlerFailure"), t);
}
} else if (receiver != null) {
try {
receiver.receive(msg);
} catch (Throwable t) {
log.error(Util.getMessage("ReceiverFailure"), t);
}
}
}
}
@ManagedOperation(description = "Disconnects the channel if connected")
public synchronized void disconnect() {
switch (state) {
case OPEN:
case CLOSED:
return;
case CONNECTING:
case CONNECTED:
if (cluster_name != null) {
// Send down a DISCONNECT event, which travels down to the GMS, where a response is
// returned
try {
down(new Event(Event.DISCONNECT, local_addr)); // DISCONNECT is handled by each layer
} catch (Throwable t) {
log.error(Util.getMessage("DisconnectFailure"), local_addr, t);
}
}
state = State.OPEN;
stopStack(true, false);
notifyChannelDisconnected(this);
init(); // sets local_addr=null; changed March 18 2003 (bela) -- prevented successful
// rejoining
break;
default:
throw new IllegalStateException("state " + state + " unknown");
}
}
public Map<String, String> handleProbe(String... keys) {
Map<String, String> map = new HashMap<>(3);
for (String key : keys) {
if (key.startsWith("jmx")) {
handleJmx(map, key);
continue;
}
if (key.startsWith("reset-stats")) {
resetAllStats();
continue;
}
if (key.startsWith("invoke") || key.startsWith("op")) {
int index = key.indexOf("=");
if (index != -1) {
try {
handleOperation(map, key.substring(index + 1));
} catch (Throwable throwable) {
log.error(
Util.getMessage("OperationInvocationFailure"),
key.substring(index + 1),
throwable);
}
}
}
}
return map;
}
protected void stopStack(boolean stop, boolean destroy) {
if (prot_stack != null) {
try {
if (stop) prot_stack.stopStack(cluster_name);
if (destroy) prot_stack.destroy();
} catch (Exception e) {
log.error(Util.getMessage("StackDestroyFailure"), e);
}
TP transport = prot_stack.getTransport();
if (transport != null) transport.unregisterProbeHandler(probe_handler);
}
}
/**
* Invokes an operation and puts the return value into map
*
* @param map
* @param operation Protocol.OperationName[args], e.g. STABLE.foo[arg1 arg2 arg3]
*/
protected void handleOperation(Map<String, String> map, String operation) throws Exception {
int index = operation.indexOf(".");
if (index == -1)
throw new IllegalArgumentException(
"operation " + operation + " is missing the protocol name");
String prot_name = operation.substring(0, index);
Protocol prot = prot_stack.findProtocol(prot_name);
if (prot == null) return; // less drastic than throwing an exception...
int args_index = operation.indexOf("[");
String method_name;
if (args_index != -1) method_name = operation.substring(index + 1, args_index).trim();
else method_name = operation.substring(index + 1).trim();
String[] args = null;
if (args_index != -1) {
int end_index = operation.indexOf("]");
if (end_index == -1) throw new IllegalArgumentException("] not found");
List<String> str_args =
Util.parseCommaDelimitedStrings(operation.substring(args_index + 1, end_index));
Object[] strings = str_args.toArray();
args = new String[strings.length];
for (int i = 0; i < strings.length; i++) args[i] = (String) strings[i];
}
Method method = MethodCall.findMethod(prot.getClass(), method_name, args);
if (method == null) {
log.warn(
Util.getMessage("MethodNotFound"),
local_addr,
prot.getClass().getSimpleName(),
method_name);
return;
}
MethodCall call = new MethodCall(method);
Object[] converted_args = null;
if (args != null) {
converted_args = new Object[args.length];
Class<?>[] types = method.getParameterTypes();
for (int i = 0; i < args.length; i++)
converted_args[i] = MethodCall.convert(args[i], types[i]);
}
Object retval = call.invoke(prot, converted_args);
if (retval != null) map.put(prot_name + "." + method_name, retval.toString());
}
protected void handleJmx(Map<String, String> map, String input) {
Map<String, Object> tmp_stats;
int index = input.indexOf("=");
if (index > -1) {
List<String> list = null;
String protocol_name = input.substring(index + 1);
index = protocol_name.indexOf(".");
if (index > -1) {
String rest = protocol_name;
protocol_name = protocol_name.substring(0, index);
String attrs = rest.substring(index + 1); // e.g. "num_sent,msgs,num_received_msgs"
list = Util.parseStringList(attrs, ",");
// check if there are any attribute-sets in the list
for (Iterator<String> it = list.iterator(); it.hasNext(); ) {
String tmp = it.next();
index = tmp.indexOf("=");
if (index != -1) {
String attrname = tmp.substring(0, index);
String attrvalue = tmp.substring(index + 1);
Protocol prot = prot_stack.findProtocol(protocol_name);
Field field = prot != null ? Util.getField(prot.getClass(), attrname) : null;
if (field != null) {
Object value = MethodCall.convert(attrvalue, field.getType());
if (value != null) prot.setValue(attrname, value);
} else {
// try to find a setter for X, e.g. x(type-of-x) or setX(type-of-x)
ResourceDMBean.Accessor setter =
ResourceDMBean.findSetter(
prot, attrname); // Util.getSetter(prot.getClass(), attrname);
if (setter != null) {
try {
Class<?> type =
setter instanceof ResourceDMBean.FieldAccessor
? ((ResourceDMBean.FieldAccessor) setter).getField().getType()
: setter instanceof ResourceDMBean.MethodAccessor
? ((ResourceDMBean.MethodAccessor) setter)
.getMethod()
.getParameterTypes()[0]
.getClass()
: null;
Object converted_value = MethodCall.convert(attrvalue, type);
setter.invoke(converted_value);
} catch (Exception e) {
log.error("unable to invoke %s() on %s: %s", setter, protocol_name, e);
}
} else log.warn(Util.getMessage("FieldNotFound"), attrname, protocol_name);
}
it.remove();
}
}
}
tmp_stats = dumpStats(protocol_name, list);
if (tmp_stats != null) {
for (Map.Entry<String, Object> entry : tmp_stats.entrySet()) {
Map<String, Object> tmp_map = (Map<String, Object>) entry.getValue();
String key = entry.getKey();
map.put(key, tmp_map != null ? tmp_map.toString() : null);
}
}
} else {
tmp_stats = dumpStats();
if (tmp_stats != null) {
for (Map.Entry<String, Object> entry : tmp_stats.entrySet()) {
Map<String, Object> tmp_map = (Map<String, Object>) entry.getValue();
String key = entry.getKey();
map.put(key, tmp_map != null ? tmp_map.toString() : null);
}
}
}
}