private void setAnswers() throws ObjectStoreException { doAnswer(invocation -> { countOfEventGroups++; eventGroupWasSaved = true; return null; }).when(partitionableObjectStore).store(any(), any(), any()); doAnswer(invocation -> { countOfEventGroups--; return null; }).when(partitionableObjectStore).remove(any(String.class), any(String.class)); } }
private EventCorrelator createEventCorrelator() throws Exception { registerIntoMockContext(mockMuleContext, OBJECT_STORE_MANAGER, mockObjectStoreManager); memoryObjectStore.store(TEST_GROUP_ID, mockEventGroup, "prefix.eventGroups"); when(mockEventGroup.getGroupId()).thenReturn(TEST_GROUP_ID); when(mockEventGroup.getMessageCollectionEvent()).thenReturn(mock(CoreEvent.class)); when(mockFlowConstruct.getName()).thenReturn("flowName"); return new EventCorrelator(mockEventCorrelatorCallback, mockTimeoutMessageProcessor, mockMuleContext, mockFlowConstruct, memoryObjectStore, "prefix", mockProcessedGroups); }
private void ensurePartitionIsCleared(boolean isPersistent) throws ObjectStoreException, InitialisationException { try { ObjectStorePartition<Serializable> store = createStorePartition(TEST_PARTITION_NAME, isPersistent); store.getBaseStore().store(TEST_KEY, TEST_VALUE, TEST_PARTITION_NAME); assertThat(store.allKeys().size(), is(1)); storeManager.disposeStore(TEST_PARTITION_NAME); assertThat(store.allKeys().size(), is(0)); } finally { storeManager.dispose(); } }
@Override protected void doStore(String key, T value, String partitionName) throws ObjectStoreException { wrapped.store(key, value, partitionName); }
@Override protected void doStore(String key, T value) throws ObjectStoreException { partitionedObjectStore.store(key, value, partitionName); }
private void setAnswers() throws ObjectStoreException { doAnswer(invocation -> { countOfEventGroups++; eventGroupWasSaved = true; return null; }).when(partitionableObjectStore).store(any(), any(), any()); doAnswer(invocation -> { countOfEventGroups--; return null; }).when(partitionableObjectStore).remove(any(String.class), any(String.class)); } }
protected EventGroup addEventGroup(EventGroup group) throws ObjectStoreException { try { correlatorStore.store((String) group.getGroupId(), group, getEventGroupsPartitionKey()); return group; } catch (ObjectAlreadyExistsException e) { return getEventGroup((String) group.getGroupId()); } }
/** * Add the given event to this group. * * @param event the event to add * @throws ObjectStoreException */ public void addEvent(CoreEvent event) throws ObjectStoreException { synchronized (this) { event = CoreEvent.builder(event).addVariable(MULE_ARRIVAL_ORDER_PROPERTY, ++arrivalOrderCounter).build(); // Using both event ID and CorrelationSequence since in certain instances // when an event is split up, the same event IDs are used. String key = getEventKey(event); eventsObjectStore.store(key, event, eventsPartitionKey); } }
private EventCorrelator createEventCorrelator() throws Exception { registerIntoMockContext(mockMuleContext, OBJECT_STORE_MANAGER, mockObjectStoreManager); memoryObjectStore.store(TEST_GROUP_ID, mockEventGroup, "prefix.eventGroups"); when(mockEventGroup.getGroupId()).thenReturn(TEST_GROUP_ID); when(mockEventGroup.getMessageCollectionEvent()).thenReturn(mock(CoreEvent.class)); when(mockFlowConstruct.getName()).thenReturn("flowName"); return new EventCorrelator(mockEventCorrelatorCallback, mockTimeoutMessageProcessor, mockMuleContext, mockFlowConstruct, memoryObjectStore, "prefix", mockProcessedGroups); }
private void ensurePartitionIsCleared(boolean isPersistent) throws ObjectStoreException, InitialisationException { try { ObjectStorePartition<Serializable> store = createStorePartition(TEST_PARTITION_NAME, isPersistent); store.getBaseStore().store(TEST_KEY, TEST_VALUE, TEST_PARTITION_NAME); assertThat(store.allKeys().size(), is(1)); storeManager.disposeStore(TEST_PARTITION_NAME); assertThat(store.allKeys().size(), is(0)); } finally { storeManager.dispose(); } }
newEvent); correlatorStore.store((String) group.getGroupId(), group.getCreated(), getExpiredAndDispatchedPartitionKey()); } else { logger.warn(MessageFormat.format("Discarding group {0}", group.getGroupId()));