All Downloads are FREE. Search and download functionalities are using the official Maven repository.
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.
io.elastic.sailor.impl.AmqpServiceImpl Maven / Gradle / Ivy
package io.elastic.sailor.impl;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.google.inject.name.Named;
import com.rabbitmq.client.AMQP;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import io.elastic.api.Message;
import io.elastic.api.Module;
import io.elastic.sailor.*;
import org.slf4j.LoggerFactory;
import javax.json.Json;
import javax.json.JsonObject;
import javax.json.JsonObjectBuilder;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.net.URI;
import java.util.Map;
@Singleton
public class AmqpServiceImpl implements AmqpService {
private static final org.slf4j.Logger logger = LoggerFactory.getLogger(AmqpServiceImpl.class);
private Connection amqp;
private Channel subscribeChannel;
private Channel publishChannel;
private String amqpUri;
private String subscribeExchangeName;
private String publishExchangeName;
private String dataRoutingKey;
private String errorRoutingKey;
private String reboundRoutingKey;
private String snapshotRoutingKey;
private Integer prefetchCount;
private CryptoServiceImpl cipher;
private MessageProcessor messageProcessor;
private String consumerTag;
@Inject
public AmqpServiceImpl(CryptoServiceImpl cipher) {
this.cipher = cipher;
}
@Inject
public void setAmqpUri(
@Named(Constants.ENV_VAR_AMQP_URI) String amqpUri) {
this.amqpUri = amqpUri;
}
@Inject
public void setSubscribeExchangeName(
@Named(Constants.ENV_VAR_LISTEN_MESSAGES_ON) String subscribeExchangeName) {
this.subscribeExchangeName = subscribeExchangeName;
}
@Inject
public void setPublishExchangeName(
@Named(Constants.ENV_VAR_PUBLISH_MESSAGES_TO) String publishExchangeName) {
this.publishExchangeName = publishExchangeName;
}
@Inject
public void setDataRoutingKey(
@Named(Constants.ENV_VAR_DATA_ROUTING_KEY) String dataRoutingKey) {
this.dataRoutingKey = dataRoutingKey;
}
@Inject
public void setErrorRoutingKey(
@Named(Constants.ENV_VAR_ERROR_ROUTING_KEY) String errorRoutingKey) {
this.errorRoutingKey = errorRoutingKey;
}
@Inject
public void setReboundRoutingKey(
@Named(Constants.ENV_VAR_REBOUND_ROUTING_KEY) String reboundRoutingKey) {
this.reboundRoutingKey = reboundRoutingKey;
}
@Inject
public void setSnapshotRoutingKey(
@Named(Constants.ENV_VAR_SNAPSHOT_ROUTING_KEY) String snapshotRoutingKey) {
this.snapshotRoutingKey = snapshotRoutingKey;
}
@Inject
public void setMessageProcessor(MessageProcessor messageProcessor) {
this.messageProcessor = messageProcessor;
}
@Inject
public void setPrefetchCount(
@Named(Constants.ENV_VAR_RABBITMQ_PREFETCH_SAILOR) Integer prefetchCount) {
this.prefetchCount = prefetchCount;
}
public void connect() {
openConnection(this.amqpUri);
openPublishChannel();
openSubscribeChannel();
}
public void disconnect() {
logger.info("About to disconnect from AMQP");
try {
subscribeChannel.close();
} catch (IOException e) {
logger.info("Subscription channel is already closed: " + e);
}
try {
publishChannel.close();
} catch (IOException e) {
logger.info("Publish channel is already closed: " + e);
}
try {
amqp.close();
} catch (IOException e) {
logger.info("AMQP connection is already closed: " + e);
}
logger.info("Successfully disconnected from AMQP");
}
public void subscribeConsumer(final Module module) {
final MessageConsumer consumer = new MessageConsumer(subscribeChannel, cipher, this.messageProcessor, module);
try {
consumerTag = subscribeChannel.basicConsume(this.subscribeExchangeName, consumer);
} catch (IOException e) {
throw new RuntimeException(e);
}
logger.info("Subscribed consumer. Waiting for messages to arrive ...");
}
public void cancelConsumer() {
if (consumerTag != null) {
logger.info("Canceling consumer {}", consumerTag);
try {
subscribeChannel.basicCancel(consumerTag);
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
public void ack(Long deliveryTag) {
try {
logger.info(String.format("Message #%s ack", deliveryTag));
subscribeChannel.basicAck(deliveryTag, false);
} catch (Exception e) {
throw new RuntimeException(e);
}
}
public void reject(Long deliveryTag) {
try {
logger.info(String.format("Message #%s reject", deliveryTag));
subscribeChannel.basicReject(deliveryTag, false);
} catch (Exception e) {
throw new RuntimeException(e);
}
}
public void sendData(byte[] payload, AMQP.BasicProperties options) {
sendToExchange(this.dataRoutingKey, payload, options);
}
public void sendHttpReply(byte[] payload, AMQP.BasicProperties options) {
final Map headers = options.getHeaders();
final Object routingKey = headers.get("reply_to");
if (routingKey == null) {
throw new RuntimeException(
"Component emitted 'httpReply' event but 'reply_to' was not found in AMQP headers");
}
sendToExchange(routingKey.toString(), payload, options);
}
public void sendSnapshot(byte[] payload, AMQP.BasicProperties options) {
sendToExchange(this.snapshotRoutingKey, payload, options);
}
public void sendError(Throwable e, AMQP.BasicProperties options, Message originalMessage) {
final StringWriter writer = new StringWriter();
e.printStackTrace(new PrintWriter(writer));
final JsonObjectBuilder builder = Json.createObjectBuilder()
.add("name", e.getClass().getName())
.add("stack", writer.toString());
if (e.getMessage() != null) {
builder.add("message", e.getMessage());
}
final JsonObject error = builder.build();
final JsonObjectBuilder payloadBuilder = Json.createObjectBuilder()
.add("error", cipher.encryptJsonObject(error));
if (originalMessage != null) {
payloadBuilder.add("errorInput", cipher.encryptMessage(originalMessage));
}
final JsonObject payload = payloadBuilder.build();
byte[] errorPayload = payload.toString().getBytes();
sendToExchange(this.errorRoutingKey, errorPayload, options);
}
public void sendRebound(byte[] payload, AMQP.BasicProperties options) {
sendToExchange(this.reboundRoutingKey, payload, options);
}
private AmqpServiceImpl openConnection(String uri) {
try {
if (amqp == null) {
ConnectionFactory factory = new ConnectionFactory();
factory.setUri(new URI(uri));
amqp = factory.newConnection();
logger.info("Connected to AMQP");
}
return this;
} catch (Exception e) {
throw new RuntimeException(e);
}
}
private AmqpServiceImpl openPublishChannel() {
try {
if (publishChannel == null) {
publishChannel = amqp.createChannel();
logger.info("Opened publish channel");
}
return this;
} catch (IOException e) {
throw new RuntimeException(e);
}
}
private AmqpServiceImpl openSubscribeChannel() {
try {
if (subscribeChannel == null) {
subscribeChannel = amqp.createChannel();
subscribeChannel.basicQos(this.prefetchCount);
logger.info("Opened subscribe channel");
}
return this;
} catch (IOException e) {
throw new RuntimeException(e);
}
}
private void sendToExchange(String routingKey, byte[] payload, AMQP.BasicProperties options) {
logger.info("Pushing to exchange={}, routingKey={}",this.publishExchangeName, routingKey);
logger.info("Message headers: {}",options.getHeaders());
try {
publishChannel.basicPublish(this.publishExchangeName, routingKey, options, payload);
} catch (IOException e) {
throw new RuntimeException(String.format("Failed to publish message to exchange %s", publishExchangeName), e);
}
logger.info("Successfully published data to {}", this.publishExchangeName);
}
@Override
protected void finalize() throws Throwable {
disconnect();
super.finalize();
}
public void setSubscribeChannel(Channel subscribeChannel) {
this.subscribeChannel = subscribeChannel;
}
public void setPublishChannel(Channel publishChannel) {
this.publishChannel = publishChannel;
}
}
