Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
package com.jetdrone.vertx.mods.stomp;
import org.vertx.java.busmods.BusModBase;
import org.vertx.java.core.Handler;
import org.vertx.java.core.eventbus.Message;
import org.vertx.java.core.json.JsonObject;
import java.util.Random;
import java.util.UUID;
public class StompClientBusMod extends BusModBase implements Handler> {
private static final Random RANDOM = new Random();
private StompClient stompClient;
private StompSubscriptions stompSubscriptions;
private String baseAddress;
private static String generateID() {
return new UUID(RANDOM.nextLong(), RANDOM.nextLong()).toString();
}
private static String getRequiredField(String name, Message msg) throws StompFrameException {
String field = msg.body().getString(name);
if (field == null) {
throw new StompFrameException("Field: <" + name + "> is required!");
}
return field;
}
@Override
public void start() {
super.start();
String host = getOptionalStringConfig("host", "localhost");
int port = getOptionalIntConfig("port", 61613);
String login = getOptionalStringConfig("login", null);
String passcode = getOptionalStringConfig("passcode", null);
stompSubscriptions = new StompSubscriptions();
stompClient = new StompClient(vertx, logger, host, port, login, passcode, stompSubscriptions);
stompClient.connect(null);
baseAddress = getOptionalStringConfig("address", "vertx.mod-stomp-io");
eb.registerHandler(baseAddress, this);
}
@Override
public void stop() {
stompClient.send(new Frame("DISCONNECT"), true, new Handler() {
@Override
public void handle(Frame event) {
// NOOP
}
});
}
@Override
public void handle(final Message message) {
String command = message.body().getString("command");
if (command == null) {
sendError(message, "command must be specified");
return;
}
final Frame frame = new Frame(command);
JsonObject headers;
try {
switch (command) {
case "connect":
case "disconnect":
sendError(message, "This command is not allowed, CONNECT/DISCONNECT is handled by the module");
break;
case "send":
frame.headers.put("destination", getRequiredField("destination", message));
frame.headers.put("transaction", message.body().getString("transaction"));
// add user defined headers
headers = message.body().getObject("headers");
if (headers != null) {
for (String header : headers.getFieldNames()) {
frame.headers.put(header, headers.getString(header));
}
}
Object body = message.body().getField("body");
if (body != null) {
if (body instanceof String) {
frame.body = (String) body;
frame.headers.put("content-type", "text/plain");
}
else if (body instanceof JsonObject) {
frame.body = ((JsonObject) body).encode();
frame.headers.put("content-type", "application/json");
}
}
boolean sync = message.body().getBoolean("sync", false);
if (sync) {
// if receipt is present a RECEIPT frame is returned, else nothing is
frame.headers.put("receipt", generateID());
stompClient.send(frame, false, new Handler() {
@Override
public void handle(Frame frame) {
// should also include the id in the response
sendOK(message, new JsonObject().putString("receipt", frame.headers.get("receipt-id")));
}
});
} else {
stompClient.send(frame, true, new Handler() {
@Override
public void handle(Frame event) {
sendOK(message, null);
}
});
}
break;
case "subscribe":
// for convenience if the `id` header is not set, we create a new one for this client
// that will be returned to be able to unsubscribe this subscription
final String subscribeId = message.body().getString("id", generateID());
String destination = getRequiredField("destination", message);
frame.headers.put("id", subscribeId);
frame.headers.put("destination", destination);
frame.headers.put("ack", message.body().getString("ack", "auto"));
// add user defined headers
headers = message.body().getObject("headers");
if (headers != null) {
for (String header : headers.getFieldNames()) {
frame.headers.put(header, headers.getString(header));
}
}
// compose the listening address as base + destination
final String vertxChannel = baseAddress + destination;
stompSubscriptions.registerSubscribeHandler(subscribeId, new Handler() {
@Override
public void handle(Frame reply) {
JsonObject replyMessage = new JsonObject();
replyMessage.putString("status", "ok");
replyMessage.putObject("value", reply.toJSON());
eb.send(vertxChannel, replyMessage);
}
});
stompClient.send(frame, true, new Handler() {
@Override
public void handle(Frame event) {
sendOK(message, new JsonObject().putString("id", subscribeId));
}
});
break;
case "unsubscribe":
final String unsubscribeId = getRequiredField("id", message);
frame.headers.put("id", unsubscribeId);
stompClient.send(frame, true, new Handler() {
@Override
public void handle(Frame event) {
stompSubscriptions.unregisterSubscribeHandler(unsubscribeId);
sendOK(message);
}
});
break;
case "ack":
case "nack":
frame.headers.put("id", getRequiredField("id", message));
frame.headers.put("transaction", message.body().getString("transaction"));
stompClient.send(frame, true, new Handler() {
@Override
public void handle(Frame event) {
sendOK(message, null);
}
});
break;
case "begin":
// for convenience if the `transaction` header is not set, we create a new one for this client
// that will be returned to be able to commit/abort/send this tx
frame.headers.put("transaction", message.body().getString("transaction", generateID()));
stompClient.send(frame, true, new Handler() {
@Override
public void handle(Frame event) {
sendOK(message, new JsonObject().putString("transaction", frame.headers.get("transaction")));
}
});
break;
case "commit":
case "abort":
frame.headers.put("transaction", getRequiredField("transaction", message));
stompClient.send(frame, true, new Handler() {
@Override
public void handle(Frame event) {
sendOK(message, null);
}
});
break;
default:
sendError(message, "Invalid command: " + command);
}
} catch (StompFrameException ex) {
ex.printStackTrace();
sendError(message, ex.getMessage());
}
}
}
