Add message tracking information in VMRouter by rogin · Pull Request #49 · OpenIntegrationEngine/engine

rogin · 2025-04-19T03:29:24Z

Enhance the VMRouter using @tonygermano's existing code template.

Initial PR with @jonbartels feedback is here.

Note that I removed the boolean property that toggled its functionality, so it's always on now.

pacmano1 · 2025-04-24T16:58:41Z

Revised nextgenhealthcare/connect#5293, closed issue and renamed appending (fixed in OIE fork) to issue.

Signed-off-by: Richard Ogin <rogin@users.noreply.github.com>

Copilot

Pull request overview

This PR enhances the VMRouter class to automatically track message routing chains, similar to how Channel Writer connectors work. The enhancement adds source channel IDs and message IDs to the sourceMap of routed messages, making it easier to trace message flow through multiple channels when using VMRouter programmatically instead of Channel Writer connectors.

Key Changes:

Added a new VMRouter constructor that accepts tracking information (channelId, messageId, sourceMap)
Introduced new overloaded routing methods that accept sourceMap and destinationSet parameters for enhanced control
Created a TrackingEnhancer inner class that enriches source maps with routing chain information

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 10 comments.

File	Description
server/src/com/mirth/connect/server/util/javascript/JavaScriptScopeUtil.java	Adds `addRouterEnhancement` method to create VMRouter instances with tracking capabilities and integrates them into all JavaScript execution scopes (preprocessor, postprocessor, filter/transformer, etc.). Also adds an unused `doGetContext` method.
server/src/com/mirth/connect/server/userutil/VMRouter.java	Extends VMRouter with new constructor, overloaded routing methods for enhanced control, TrackingEnhancer inner class for automatic source tracking, createRawMessage helper method, and improved Javadoc with @link references.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copilot · 2025-12-12T18:38:46Z

+ public RawMessage createRawMessage(Object message, Map<String, Object> sourceMap,
+ Collection<Number> destinationSet) {
+ if (trackingEnhancer != null) {
+ sourceMap = trackingEnhancer.enrich(sourceMap);
+ }
+
+ if (message instanceof byte[]) {
+ return new RawMessage((byte[]) message, destinationSet, sourceMap);
 } else {
- return new com.mirth.connect.donkey.model.message.RawMessage(message.getRawData(), message.getDestinationMetaDataIds(), message.getSourceMap());
+ return new RawMessage(message.toString(), destinationSet, sourceMap);
+ }
+ }


The createRawMessage method doesn't validate that the message parameter is non-null before calling message.toString() on line 282. If a null message is passed, this will throw a NullPointerException. Consider adding validation at the beginning of this method to throw a more meaningful exception or handle the null case appropriately.

Copilot · 2025-12-12T18:38:46Z

+ * Adds additional message tracking data.
+ * 
+ * TrackingEnhancer


The Javadoc comment "TrackingEnhancer" on line 289 is redundant as it simply restates the class name. Consider either removing this line or expanding it to provide a more meaningful description of what the TrackingEnhancer does, such as "Inner class that enriches source maps with message tracking information for routing chains."
Suggested change
* Addsadditionalmessagetrackingdata.
*
* TrackingEnhancer
* Innerclassthatenrichessourcemapswithmessagetrackinginformationforroutingchains.
* Usedtoaddchannelandmessageidentifierstothesourcemapfortrackingpurposes.

Copilot · 2025-12-12T18:38:47Z

+ List<String> sourceChannelIds = lookupAsList("sourceChannelIds", "sourceChannelId");
+ List<String> sourceMessageIds = lookupAsList("sourceMessageIds", "sourceMessageId");
+
+ HashMap<String, Object> newSourceMap = new HashMap<String, Object>(messageSourceMap);


The HashMap is constructed with an initial capacity based on messageSourceMap size, but then 4 additional entries are added. To avoid potential resizing, consider using new HashMap<>(messageSourceMap.size() + 4) or similar to ensure adequate initial capacity.
Suggested change
HashMap<String, Object> newSourceMap = newHashMap<String, Object>(messageSourceMap);
HashMap<String, Object> newSourceMap = newHashMap<String, Object>(messageSourceMap.size() + 4);
newSourceMap.putAll(messageSourceMap);

Copilot · 2025-12-12T18:38:47Z

+ ((Collection<?>) primaryValue).stream().map(i -> i.toString()).forEach(result::add);
+ } else if (primaryValue instanceof Object[]) {
+ Arrays.stream((Object[]) primaryValue).map(i -> i.toString()).forEach(result::add);


The lambda expression i -> i.toString() can be simplified to a method reference: Object::toString. Similarly on line 357. This improves code readability and is a common Java best practice.
Suggested change
((Collection<?>) primaryValue).stream().map(i -> i.toString()).forEach(result::add);
} elseif (primaryValueinstanceofObject[]) {
Arrays.stream((Object[]) primaryValue).map(i -> i.toString()).forEach(result::add);
((Collection<?>) primaryValue).stream().map(Object::toString).forEach(result::add);
} elseif (primaryValueinstanceofObject[]) {
Arrays.stream((Object[]) primaryValue).map(Object::toString).forEach(result::add);

Copilot · 2025-12-12T18:38:48Z

+ /**
+ * Enrich the given source map with additional message tracking data.
+ * 
+ * @param messageSourceMap


The @PARAM documentation for messageSourceMap is missing. While parameters in private methods don't strictly require documentation, it would improve maintainability to document what this parameter represents (e.g., "the source map provided by the caller, may be null").
Suggested change
* @parammessageSourceMap
* @parammessageSourceMapthesourcemapprovidedbythecaller, maybenull

Copilot · 2025-12-12T18:38:48Z

+ /**
+ * Look up a value from the environment's {@link SourceMap}
+ * 
+ * @param key


The @PARAM documentation for key is missing. Adding a brief description (e.g., "the key to look up in the source map") would improve code documentation consistency.
Suggested change
* @paramkey
* @paramkeythekeytolookupinthesourcemap

Copilot · 2025-12-12T18:38:48Z

+ public Context doGetContext(ContextFactory contextFactory) {
+ return JavaScriptScopeUtil.getContext(contextFactory);
+ }
+


This public method doGetContext appears to be unused throughout the codebase. It's a non-static wrapper around the static getContext method but serves no clear purpose in this PR. Consider removing it unless there's a specific planned use case.
Suggested change
publicContextdoGetContext(ContextFactorycontextFactory) {
returnJavaScriptScopeUtil.getContext(contextFactory);
}

Copilot · 2025-12-12T18:38:49Z

+ * 
+ * TrackingEnhancer
+ */
+ private class TrackingEnhancer {


TrackingEnhancer should be made static, since the enclosing instance is not used.
Suggested change
privateclassTrackingEnhancer {
privatestaticclassTrackingEnhancer {

Copilot · 2025-12-12T18:38:49Z

+ * 
+ * @see #routeMessageByChannelId(String, Object, Map, Collection)
+ */
+ public Response routeMessageByChannelId(String channelId, Object message) {


Method VMRouter.routeMessageByChannelId(..) could be confused with overloaded method routeMessageByChannelId, since dispatch depends on static types.

Copilot · 2025-12-12T18:38:49Z

+ * @return The {@link Response} object returned by the channel, if its source
+ * connector is configured to return one.
 */
 public Response routeMessageByChannelId(String channelId, RawMessage rawMessage) {


Method VMRouter.routeMessageByChannelId(..) could be confused with overloaded method routeMessageByChannelId, since dispatch depends on static types.

ssrowe · 2026-03-11T18:30:43Z

@rogin this is slated for our next release. Can you take a look at the Copilot feedback above, pull in the changes that you think make sense and rebase? We can then get some additional reviews on this and get it merged.

jonbartels approved these changes Apr 20, 2025
View reviewed changes

rogin force-pushed the enhance-vmrouter branch from 854ad08 to 99a0e5eCompare April 21, 2025 02:36

rogin mentioned this pull request Apr 24, 2025
[IDEA] router.routeMessage should populate sourceMap "sourceChannelID" like a channel writer destination (fixed in OIE fork) nextgenhealthcare/connect#5293
Closed

Add msg tracking info in VMRouter
feeeb74
Signed-off-by: Richard Ogin <rogin@users.noreply.github.com>

jonbartels force-pushed the enhance-vmrouter branch from 99a0e5e to feeeb74Compare April 28, 2025 14:21

jonbartels added this to the Technical Preview 2 milestone May 6, 2025

kpalang requested review from a team, gibson9583, kayyagari, pacmano1 and ssrowe and removed request for a teamSeptember 8, 2025 15:25

jonbartels requested review from a team, gibson9583, kayyagari, kpalang, pacmano1, ssrowe and tonygermano and removed request for a team, gibson9583, kayyagari, pacmano1 and ssrowe September 8, 2025 15:26

pacmano1 requested a review from Copilot December 12, 2025 18:30

Copilotstarted reviewing on behalf of pacmano1 December 12, 2025 18:30 View session

CopilotAI reviewed Dec 12, 2025
View reviewed changes

jonbartels modified the milestones: Next Release, Next Release + 1 Mar 20, 2026

Navigation Menu

Uh oh!

Add message tracking information in VMRouter - #49

Add message tracking information in VMRouter#49
rogin wants to merge 1 commit into
OpenIntegrationEngine:mainfrom
rogin:enhance-vmrouter

rogin commented Apr 19, 2025

Uh oh!

pacmano1 commented Apr 24, 2025

Uh oh!

CopilotAI left a comment

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

CopilotAIDec 12, 2025

Uh oh!

ssrowe commented Mar 11, 2026•
edited
Loading

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

5 participants

- * Addsadditionalmessagetrackingdata.
- *
- * TrackingEnhancer
+ * Innerclassthatenrichessourcemapswithmessagetrackinginformationforroutingchains.
+ * Usedtoaddchannelandmessageidentifierstothesourcemapfortrackingpurposes.

	HashMap<String, Object> newSourceMap = newHashMap<String, Object>(messageSourceMap);
	HashMap<String, Object> newSourceMap = newHashMap<String, Object>(messageSourceMap.size() + 4);
	newSourceMap.putAll(messageSourceMap);

- ((Collection<?>) primaryValue).stream().map(i -> i.toString()).forEach(result::add);
- } elseif (primaryValueinstanceofObject[]) {
-Arrays.stream((Object[]) primaryValue).map(i -> i.toString()).forEach(result::add);
+ ((Collection<?>) primaryValue).stream().map(Object::toString).forEach(result::add);
+ } elseif (primaryValueinstanceofObject[]) {
+Arrays.stream((Object[]) primaryValue).map(Object::toString).forEach(result::add);

	* @parammessageSourceMap
	* @parammessageSourceMapthesourcemapprovidedbythecaller, maybenull

-publicContextdoGetContext(ContextFactorycontextFactory) {
-returnJavaScriptScopeUtil.getContext(contextFactory);
- }

	privateclassTrackingEnhancer {
	privatestaticclassTrackingEnhancer {

Uh oh!

Conversation

rogin commented Apr 19, 2025

Uh oh!

pacmano1 commented Apr 24, 2025

Uh oh!

CopilotAI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

CopilotAIDec 12, 2025

Choose a reason for hiding this comment

Uh oh!

ssrowe commented Mar 11, 2026• edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

5 participants

ssrowe commented Mar 11, 2026•
edited
Loading