Apache/nifi
1
0
Fork 0
mirror of https://github.com/apache/nifi.git synced 2025-02-06 01:58:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

NIFI-12478 Return Message Type as body for JMS Object Messages (#8131)

Browse Source
This commit is contained in:
David Handermann 2023-12-07 15:41:34 -06:00 committed by GitHub
parent b9dbcab160
commit f73888e7dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 9 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nifi-nar-bundles/nifi-jms-bundle/nifi-jms-processors/src/main/java/org/apache/nifi/jms/processors/JMSConsumer.java
View File

@ -38,6 +38,7 @@ import javax.jms.StreamMessage;
import javax.jms.TextMessage;
import javax.jms.Topic;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;
@ -184,7 +185,8 @@ class JMSConsumer extends JMSWorker {
messageBody = MessageBodyToBytesConverter.toBytes((BytesMessage) message);
} else if (message instanceof ObjectMessage) {
messageType = ObjectMessage.class.getSimpleName();
messageBody = MessageBodyToBytesConverter.toBytes((ObjectMessage) message);
// Return Message Type as body to avoid unsupported class references
messageBody = messageType.getBytes(StandardCharsets.UTF_8);
} else if (message instanceof StreamMessage) {
messageType = StreamMessage.class.getSimpleName();
messageBody = MessageBodyToBytesConverter.toBytes((StreamMessage) message);

15
nifi-nar-bundles/nifi-jms-bundle/nifi-jms-processors/src/main/java/org/apache/nifi/jms/processors/MessageBodyToBytesConverter.java
View File

@ -31,13 +31,11 @@ import javax.jms.BytesMessage;
import javax.jms.JMSException;
import javax.jms.MapMessage;
import javax.jms.MessageEOFException;
import javax.jms.ObjectMessage;
import javax.jms.StreamMessage;
import javax.jms.TextMessage;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.SerializationUtils;
/**
*
@ -88,19 +86,6 @@ abstract class MessageBodyToBytesConverter {
}
}
/**
*
* @param message instance of {@link ObjectMessage}
* @return byte array representing the {@link ObjectMessage}
*/
public static byte[] toBytes(ObjectMessage message) {
try {
return SerializationUtils.serialize(message.getObject());
} catch (Exception e) {
throw new MessageConversionException("Failed to convert " + ObjectMessage.class.getSimpleName() + " to byte[]", e);
}
}
/**
* @param message instance of {@link StreamMessage}
* @return byte array representing the {@link StreamMessage}

13
nifi-nar-bundles/nifi-jms-bundle/nifi-jms-processors/src/test/java/org/apache/nifi/jms/processors/JMSPublisherConsumerIT.java
View File

@ -18,7 +18,6 @@ package org.apache.nifi.jms.processors;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.apache.commons.lang3.SerializationUtils;
import org.apache.nifi.jms.processors.JMSConsumer.JMSResponse;
import org.apache.nifi.logging.ComponentLog;
import org.junit.jupiter.api.Test;
@ -70,9 +69,9 @@ public class JMSPublisherConsumerIT {
};
Consumer<JMSResponse> responseChecker = response -> {
assertEquals(
"stringAsObject",
SerializationUtils.deserialize(response.getMessageBody())
assertArrayEquals(
ObjectMessage.class.getSimpleName().getBytes(StandardCharsets.UTF_8),
response.getMessageBody()
);
};
@ -447,7 +446,7 @@ public class JMSPublisherConsumerIT {
}
callbackInvoked.set(true);
assertEquals("1", new String(response.getMessageBody()));
assertEquals("2", new String(response.getMessageBody()));
acknowledge(response);
});
}
@ -464,7 +463,7 @@ public class JMSPublisherConsumerIT {
}
callbackInvoked.set(true);
assertEquals("2", new String(response.getMessageBody()));
assertEquals("1", new String(response.getMessageBody()));
throw new RuntimeException("intentional to avoid explicit ack");
});
}
@ -483,7 +482,7 @@ public class JMSPublisherConsumerIT {
}
callbackInvoked.set(true);
assertEquals("2", new String(response.getMessageBody()));
assertEquals("1", new String(response.getMessageBody()));
acknowledge(response);
});
}

1
pom.xml
View File

@ -1166,7 +1166,6 @@
!ITDeleteAzureBlobStorage_v12,
!ITMoveAzureDataLakeStorage,
!AzureGraphUserGroupProviderIT,
!JMSPublisherConsumerIT#validateMessageRedeliveryWhenNotAcked,
!GremlinClientServiceYamlSettingsAndBytecodeIT,
!GremlinClientServiceControllerSettingsIT,
!ITestConsumeEmail#validateUrl,