private Map<String, String> sendDirectMessage(
final ProcessContext context,
final ProcessSession session,
final FlowFile flowFile,
final ProcessorLog logger,
Message message,
Recipient recipient) {
String groupAttribute = context.getProperty(GROUP_ATTRIBUTE_NAME).getValue();
String groupSufixAttribute =
context.getProperty(GROUP_SUFIX).isSet()
? context.getProperty(GROUP_SUFIX).evaluateAttributeExpressions(flowFile).getValue()
: "";
String groupSubjectAttribute = context.getProperty(GROUP_SUBJECT_ATTRIBUTE_NAME).getValue();
String participantsAttribute = context.getProperty(PARTICIPANTS_ATTRIBUTE_NAME).getValue();
String groupName =
message.getHeader().getRelatedConversationId() != null
? message.getHeader().getRelatedConversationId()
: UUID.randomUUID().toString();
groupName += groupSufixAttribute;
String groupSubject =
message.getHeader().getSubject() != null
? message.getHeader().getSubject()
: "Private Message";
String referenceAttributeName = context.getProperty(REFERENCE_ATTRIBUTE_NAME).getValue();
Map<String, String> generatedReferences = new HashMap<>();
// create references
generatedReferences.put(recipient.getRecipientId(), UUID.randomUUID().toString());
FlowFile clone = session.clone(flowFile);
session.getProvenanceReporter().clone(flowFile, clone);
clone = session.putAttribute(clone, groupAttribute, groupName);
clone = session.putAttribute(clone, groupSubjectAttribute, groupSubject);
clone =
session.putAttribute(
clone,
referenceAttributeName,
generatedReferences.values().stream().collect(Collectors.joining(",")));
clone =
session.putAttribute(
clone,
participantsAttribute,
recipient.getDeliveryAddress().getPhysicalAddress().getAddress());
session.getProvenanceReporter().modifyAttributes(clone);
logger.debug("Routing message {} to {}.", new Object[] {clone, REL_DIRECT_MESSAGE});
session.transfer(clone, REL_DIRECT_MESSAGE);
session.getProvenanceReporter().route(clone, REL_DIRECT_MESSAGE);
return generatedReferences;
}
@Override
public void onTrigger(final ProcessContext context, final ProcessSession session) {
FlowFile flowFile = session.get();
if (flowFile == null) {
return;
}
final ProcessorLog logger = getLogger();
final ObjectHolder<Throwable> errorHolder = new ObjectHolder<>(null);
final ObjectHolder<MessageWrapper> messageWrapperHolder = new ObjectHolder<>(null);
session.read(
flowFile,
(final InputStream rawIn) -> {
try {
messageWrapperHolder.set(MessageSerializer.deserializeMessageWrapper(rawIn));
} catch (MessageSerializationException ex) {
errorHolder.set(
new RuntimeException(
"Error deserializing FlowFile content into a MessageWrapper instance. Routing to FAILURE",
ex));
}
});
if (errorHolder.get() != null) {
UCSCreateException.routeFlowFileToException(
context,
session,
logger,
flowFile,
REL_FAILURE,
null,
"Error in message deserialization: " + errorHolder.get().getCause() != null
? errorHolder.get().getCause().getMessage()
: errorHolder.get().getMessage(),
ExceptionType.InvalidMessage,
null,
null);
return;
}
Message message = messageWrapperHolder.get().getMessage();
// resolve the sender. We couldn't resolved it before because we needed
// the specific serviceId.
UCSController ucsService =
context.getProperty(UCS_CONTROLLER_SERVICE).asControllerService(UCSController.class);
UserContactInfo uci =
ucsService.resolveUserContactInfo(
message.getHeader().getSender().getPhysicalAddress().getAddress());
if (uci == null) {
UCSCreateException.routeFlowFileToException(
context,
session,
logger,
flowFile,
REL_FAILURE,
null,
"Unknown User: "******"Unknown Service "
+ context.getProperty(SERVICE_ID).getValue()
+ " for User: "******"GROUP:").
Map<Boolean, List<Recipient>> chatRecipients =
message
.getHeader()
.getRecipientsList()
.stream()
.filter(
r ->
r.getDeliveryAddress() != null
&& r.getDeliveryAddress().getPhysicalAddress() != null)
.filter(
r ->
context
.getProperty(SERVICE_ID)
.getValue()
.equals(r.getDeliveryAddress().getPhysicalAddress().getServiceId()))
.collect(
Collectors.groupingBy(
r ->
r.getDeliveryAddress()
.getPhysicalAddress()
.getAddress()
.startsWith("GROUP:")));
Map<String, String> generatedReferences = new HashMap<>();
if (chatRecipients.containsKey(Boolean.TRUE)) {
generatedReferences.putAll(
this.sendMessagesToPreStablishedGroups(
context, session, flowFile, logger, chatRecipients.get(Boolean.TRUE)));
}
if (chatRecipients.containsKey(Boolean.FALSE)) {
List<Recipient> recipients = chatRecipients.get(Boolean.FALSE);
if (recipients.size() == 1) {
generatedReferences.putAll(
this.sendDirectMessage(context, session, flowFile, logger, message, recipients.get(0)));
} else {
generatedReferences.putAll(
this.sendMessageToDynamicGroup(
context, session, flowFile, logger, message, recipients));
}
}
logger.debug("Removing original FlowFile");
session.remove(flowFile);
// keep track of the generated references
// TODO: is this check correct/enough?
if (message.getHeader().isReceiptNotification()) {
logger.debug(
"The message has ReceiptNotification flag enabled -> We are persisting its references.");
generatedReferences
.entrySet()
.stream()
.forEach(
(gr) -> {
ucsService.saveMessageReference(message, gr.getKey(), gr.getValue());
});
} else {
logger.debug(
"The message doesn't have ReceiptNotification flag enabled -> We are not persisting its references.");
}
}
private Map<String, String> sendMessageToDynamicGroup(
final ProcessContext context,
final ProcessSession session,
final FlowFile flowFile,
final ProcessorLog logger,
Message message,
List<Recipient> recipients) {
String groupAttribute = context.getProperty(GROUP_ATTRIBUTE_NAME).getValue();
String groupSufixAttribute =
context.getProperty(GROUP_SUFIX).isSet()
? context.getProperty(GROUP_SUFIX).evaluateAttributeExpressions(flowFile).getValue()
: "";
String groupSubjectAttribute = context.getProperty(GROUP_SUBJECT_ATTRIBUTE_NAME).getValue();
String participantsAttribute = context.getProperty(PARTICIPANTS_ATTRIBUTE_NAME).getValue();
String groupName =
message.getHeader().getRelatedConversationId() != null
? message.getHeader().getRelatedConversationId()
: UUID.randomUUID().toString();
groupName += groupSufixAttribute;
String groupSubject =
message.getHeader().getSubject() != null
? message.getHeader().getSubject()
: "Group Message";
String referenceAttributeName = context.getProperty(REFERENCE_ATTRIBUTE_NAME).getValue();
Map<String, String> generatedReferences = new HashMap<>();
// create references
recipients
.stream()
.forEach(r -> generatedReferences.put(r.getRecipientId(), UUID.randomUUID().toString()));
// convert the recipients insto a JSON array.
JsonArray participants = new JsonArray();
recipients
.stream()
.map(
r -> {
JsonObject jo = new JsonObject();
jo.addProperty(
"participant", r.getDeliveryAddress().getPhysicalAddress().getAddress());
return jo;
})
.forEach(participants::add);
FlowFile clone = session.clone(flowFile);
session.getProvenanceReporter().clone(flowFile, clone);
clone = session.putAttribute(clone, groupAttribute, groupName);
clone = session.putAttribute(clone, groupSubjectAttribute, groupSubject);
clone =
session.putAttribute(
clone,
referenceAttributeName,
generatedReferences.values().stream().collect(Collectors.joining(",")));
clone = session.putAttribute(clone, participantsAttribute, new Gson().toJson(participants));
session.getProvenanceReporter().modifyAttributes(clone);
logger.debug("Routing message {} to {}.", new Object[] {clone, REL_DYNAMIC_GROUP});
session.transfer(clone, REL_DYNAMIC_GROUP);
session.getProvenanceReporter().route(clone, REL_DYNAMIC_GROUP);
return generatedReferences;
}
public String getMesageSummary(Message msg) {
StringBuffer buf = new StringBuffer();
buf.append(msg.getHeader().getMessageType().name());
buf.append(" message");
return buf.toString();
}
