@Test public void testResequencingWithCapacity() throws InterruptedException { this.resequencer.setReleaseStrategy(new SequenceSizeReleaseStrategy(true)); // INT-3846 this.resequencer.setMessageStore(new SimpleMessageStore(3, 2)); QueueChannel replyChannel = new QueueChannel(); Message<?> message1 = createMessage("123", "ABC", 4, 4, replyChannel); Message<?> message2 = createMessage("456", "ABC", 4, 2, replyChannel); Message<?> message3 = createMessage("789", "ABC", 4, 1, replyChannel); this.resequencer.handleMessage(message1); this.resequencer.handleMessage(message2); try { this.resequencer.handleMessage(message3); fail("Expected exception"); } catch (MessagingException e) { assertThat(e.getMessage(), containsString("out of capacity (2) for group 'ABC'")); } }
@Test public void testBasicResequencing() throws InterruptedException { QueueChannel replyChannel = new QueueChannel(); Message<?> message1 = createMessage("123", "ABC", 3, 3, replyChannel); Message<?> message2 = createMessage("456", "ABC", 3, 1, replyChannel); Message<?> message3 = createMessage("789", "ABC", 3, 2, replyChannel); this.resequencer.handleMessage(message1); this.resequencer.handleMessage(message3); this.resequencer.handleMessage(message2); Message<?> reply1 = replyChannel.receive(0); Message<?> reply2 = replyChannel.receive(0); Message<?> reply3 = replyChannel.receive(0); assertNotNull(reply1); assertThat(new IntegrationMessageHeaderAccessor(reply1).getSequenceNumber(), is(1)); assertNotNull(reply2); assertThat(new IntegrationMessageHeaderAccessor(reply2).getSequenceNumber(), is(2)); assertNotNull(reply3); assertThat(new IntegrationMessageHeaderAccessor(reply3).getSequenceNumber(), is(3)); }
@Test public void testRemovalOfBarrierWhenLastMessageOfSequenceArrives() { QueueChannel replyChannel = new QueueChannel(); String correlationId = "ABC"; Message<?> message1 = createMessage("123", correlationId, 1, 1, replyChannel); resequencer.handleMessage(message1); assertEquals(0, store.getMessageGroup(correlationId).size()); }
@Test public void testResequencingWithDuplicateMessages() { QueueChannel replyChannel = new QueueChannel(); Message<?> message1 = createMessage("123", "ABC", 3, 3, replyChannel); Message<?> message2 = createMessage("456", "ABC", 3, 1, replyChannel); Message<?> message3 = createMessage("789", "ABC", 3, 2, replyChannel); this.resequencer.handleMessage(message1); this.resequencer.handleMessage(message3); this.resequencer.handleMessage(message3); this.resequencer.handleMessage(message2); Message<?> reply1 = replyChannel.receive(0); Message<?> reply2 = replyChannel.receive(0); Message<?> reply3 = replyChannel.receive(0); assertNotNull(reply1); assertEquals(1, new IntegrationMessageHeaderAccessor(reply1).getSequenceNumber()); assertNotNull(reply2); assertEquals(2, new IntegrationMessageHeaderAccessor(reply2).getSequenceNumber()); assertNotNull(reply3); assertEquals(3, new IntegrationMessageHeaderAccessor(reply3).getSequenceNumber()); }
@Test public void testBasicResequencingA() throws InterruptedException { SequenceSizeReleaseStrategy releaseStrategy = new SequenceSizeReleaseStrategy(); releaseStrategy.setReleasePartialSequences(true); this.resequencer = new ResequencingMessageHandler(processor, store, null, releaseStrategy); this.resequencer.setBeanFactory(mock(BeanFactory.class)); this.resequencer.afterPropertiesSet(); QueueChannel replyChannel = new QueueChannel(); Message<?> message1 = createMessage("123", "ABC", 3, 1, replyChannel); Message<?> message3 = createMessage("789", "ABC", 3, 3, replyChannel); this.resequencer.handleMessage(message3); assertNull(replyChannel.receive(0)); this.resequencer.handleMessage(message1); assertNotNull(replyChannel.receive(0)); assertNull(replyChannel.receive(0)); }
@Test public void testResequencingWithWrongSequenceSizeAndNumber() throws InterruptedException { QueueChannel discardChannel = new QueueChannel(); Message<?> message1 = createMessage("123", "ABC", 2, 4, null); this.resequencer.setSendPartialResultOnExpiry(false); this.resequencer.setDiscardChannel(discardChannel); this.resequencer.handleMessage(message1); // this.resequencer.discardBarrier(this.resequencer.barriers.get("ABC")); Message<?> reply1 = discardChannel.receive(0); // No message has been received - the message has been rejected. assertNull(reply1); }
@Test public void testResequencingWithDiscard() throws InterruptedException { QueueChannel discardChannel = new QueueChannel(); Message<?> message1 = createMessage("123", "ABC", 4, 2, null); Message<?> message2 = createMessage("456", "ABC", 4, 1, null); Message<?> message3 = createMessage("789", "ABC", 4, 4, null); this.resequencer.setSendPartialResultOnExpiry(false); this.resequencer.setDiscardChannel(discardChannel); this.resequencer.handleMessage(message1); this.resequencer.handleMessage(message2); assertEquals(1, store.expireMessageGroups(-10000)); Message<?> reply1 = discardChannel.receive(0); Message<?> reply2 = discardChannel.receive(0); Message<?> reply3 = discardChannel.receive(0); // only messages 1 and 2 should have been received by now assertNotNull(reply1); assertNotNull(reply2); assertNull(reply3); ArrayList<Integer> sequence = new ArrayList<>( Arrays.asList(new IntegrationMessageHeaderAccessor(reply1).getSequenceNumber(), new IntegrationMessageHeaderAccessor(reply2).getSequenceNumber())); Collections.sort(sequence); assertEquals("[1, 2]", sequence.toString()); // Once a group is expired, late messages are discarded immediately by default this.resequencer.handleMessage(message3); reply3 = discardChannel.receive(0); assertNotNull(reply3); }
@Test public void testResequencingWithDifferentSequenceSizes() throws InterruptedException { QueueChannel discardChannel = new QueueChannel(); Message<?> message1 = createMessage("123", "ABC", 4, 2, null); Message<?> message2 = createMessage("456", "ABC", 5, 1, null); this.resequencer.setSendPartialResultOnExpiry(false); this.resequencer.setDiscardChannel(discardChannel); this.resequencer.setReleasePartialSequences(true); // force SequenceSizeReleaseStrategy this.resequencer.handleMessage(message1); this.resequencer.handleMessage(message2); // this.resequencer.discardBarrier(this.resequencer.barriers.get("ABC")); Message<?> discard1 = discardChannel.receive(0); Message<?> discard2 = discardChannel.receive(0); // message2 has been discarded because it came in with the wrong sequence size assertNotNull(discard1); assertEquals(1, new IntegrationMessageHeaderAccessor(discard1).getSequenceNumber()); assertNull(discard2); }
@Test public void testTimeoutDefaultExpiry() throws InterruptedException { this.resequencer.setGroupTimeoutExpression(new SpelExpressionParser().parseExpression("100")); ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler(); taskScheduler.afterPropertiesSet(); this.resequencer.setTaskScheduler(taskScheduler); QueueChannel discardChannel = new QueueChannel(); this.resequencer.setDiscardChannel(discardChannel); QueueChannel replyChannel = new QueueChannel(); this.resequencer.setOutputChannel(replyChannel); Message<?> message3 = createMessage("789", "ABC", 3, 3, null); Message<?> message2 = createMessage("456", "ABC", 3, 2, null); this.resequencer.handleMessage(message3); this.resequencer.handleMessage(message2); Message<?> out1 = replyChannel.receive(10); assertNull(out1); out1 = discardChannel.receive(10000); assertNotNull(out1); Message<?> out2 = discardChannel.receive(10); assertNotNull(out2); Message<?> message1 = createMessage("123", "ABC", 3, 1, null); this.resequencer.handleMessage(message1); Message<?> out3 = discardChannel.receive(0); assertNotNull(out3); }
@Test public void testResequencingWithPartialSequenceAndComparator() throws InterruptedException { this.resequencer.setReleaseStrategy(new SequenceSizeReleaseStrategy(true)); QueueChannel replyChannel = new QueueChannel(); Message<?> message1 = createMessage("456", "ABC", 4, 2, replyChannel); Message<?> message2 = createMessage("123", "ABC", 4, 1, replyChannel); Message<?> message3 = createMessage("XYZ", "ABC", 4, 4, replyChannel); Message<?> message4 = createMessage("789", "ABC", 4, 3, replyChannel); this.resequencer.handleMessage(message1); this.resequencer.handleMessage(message2); this.resequencer.handleMessage(message3); Message<?> reply1 = replyChannel.receive(0); Message<?> reply2 = replyChannel.receive(0); Message<?> reply3 = replyChannel.receive(0); // only messages 1 and 2 should have been received by now assertNotNull(reply1); assertEquals(1, new IntegrationMessageHeaderAccessor(reply1).getSequenceNumber()); assertNotNull(reply2); assertEquals(2, new IntegrationMessageHeaderAccessor(reply2).getSequenceNumber()); assertNull(reply3); // when sending the last message, the whole sequence must have been sent this.resequencer.handleMessage(message4); reply3 = replyChannel.receive(0); Message<?> reply4 = replyChannel.receive(0); assertNotNull(reply3); assertEquals(3, new IntegrationMessageHeaderAccessor(reply3).getSequenceNumber()); assertNotNull(reply4); assertEquals(4, new IntegrationMessageHeaderAccessor(reply4).getSequenceNumber()); }
@Test public void testBasicUnboundedResequencing() throws InterruptedException { SequenceSizeReleaseStrategy releaseStrategy = new SequenceSizeReleaseStrategy(); releaseStrategy.setReleasePartialSequences(true); this.resequencer = new ResequencingMessageHandler(processor, store, null, releaseStrategy); QueueChannel replyChannel = new QueueChannel(); this.resequencer.setCorrelationStrategy(message -> "A"); this.resequencer.setBeanFactory(mock(BeanFactory.class)); this.resequencer.afterPropertiesSet(); //Message<?> message0 = MessageBuilder.withPayload("0").setSequenceNumber(0).build(); Message<?> message1 = MessageBuilder.withPayload("1").setSequenceNumber(1).setReplyChannel(replyChannel).build(); Message<?> message2 = MessageBuilder.withPayload("2").setSequenceNumber(2).setReplyChannel(replyChannel).build(); Message<?> message3 = MessageBuilder.withPayload("3").setSequenceNumber(3).setReplyChannel(replyChannel).build(); Message<?> message4 = MessageBuilder.withPayload("4").setSequenceNumber(4).setReplyChannel(replyChannel).build(); Message<?> message5 = MessageBuilder.withPayload("5").setSequenceNumber(5).setReplyChannel(replyChannel).build(); this.resequencer.handleMessage(message3); assertNull(replyChannel.receive(0)); this.resequencer.handleMessage(message1); assertNotNull(replyChannel.receive(0)); this.resequencer.handleMessage(message2); assertNotNull(replyChannel.receive(0)); assertNotNull(replyChannel.receive(0)); assertNull(replyChannel.receive(0)); this.resequencer.handleMessage(message5); assertNull(replyChannel.receive(0)); this.resequencer.handleMessage(message4); assertNotNull(replyChannel.receive(0)); }
Message<?> message3 = createMessage("789", "ABC", 4, 4, replyChannel); Message<?> message4 = createMessage("XYZ", "ABC", 4, 3, replyChannel); this.resequencer.handleMessage(message1); this.resequencer.handleMessage(message2); this.resequencer.handleMessage(message3); Message<?> reply1 = replyChannel.receive(0); Message<?> reply2 = replyChannel.receive(0); assertNull(reply3); this.resequencer.handleMessage(message4); reply1 = replyChannel.receive(0); reply2 = replyChannel.receive(0);
Message<?> message3 = createMessage("789", "ABC", 4, 1, replyChannel); // release 2 after this one Message<?> message4 = createMessage("XYZ", "ABC", 4, 3, replyChannel); this.resequencer.handleMessage(message1); this.resequencer.handleMessage(message2); this.resequencer.handleMessage(message3); Message<?> reply1 = replyChannel.receive(0); Message<?> reply2 = replyChannel.receive(0); assertNull(reply3); this.resequencer.handleMessage(message4); reply3 = replyChannel.receive(0); Message<?> reply4 = replyChannel.receive(0);