Package org.springframework.batch.core.partition

Examples of org.springframework.batch.core.partition.StepExecutionSplitter

29
30
31
32
33
34
35
36
37
38
39
  public void testHandleNoReply() throws Exception {
    //execute with no default set
    messageChannelPartitionHandler = new MessageChannelPartitionHandler();
    //mock
    StepExecution masterStepExecution = mock(StepExecution.class);
    StepExecutionSplitter stepExecutionSplitter = mock(StepExecutionSplitter.class);
    MessagingOperations operations = mock(MessagingOperations.class);
    Message message = mock(Message.class);
    //when
    when(message.getPayload()).thenReturn(Collections.emptyList());
    when(operations.receive((PollableChannel) anyObject())).thenReturn(message);
View Full Code Here
51
52
53
54
55
56
57
58
59
60
61
  public void testHandleWithReplyChannel() throws Exception {
    //execute with no default set
    messageChannelPartitionHandler = new MessageChannelPartitionHandler();
    //mock
    StepExecution masterStepExecution = mock(StepExecution.class);
    StepExecutionSplitter stepExecutionSplitter = mock(StepExecutionSplitter.class);
    MessagingOperations operations = mock(MessagingOperations.class);
    Message message = mock(Message.class);
    PollableChannel replyChannel = mock(PollableChannel.class);
    //when
    when(message.getPayload()).thenReturn(Collections.emptyList());
View Full Code Here
36
37
38
39
40
41
42
43
44
45
46
  public void testNoPartitions() throws Exception {
    //execute with no default set
    messageChannelPartitionHandler = new MessageChannelPartitionHandler();
    //mock
    StepExecution masterStepExecution = mock(StepExecution.class);
    StepExecutionSplitter stepExecutionSplitter = mock(StepExecutionSplitter.class);

    //execute
    Collection<StepExecution> executions = messageChannelPartitionHandler.handle(stepExecutionSplitter, masterStepExecution);
    //verify
    assertNull(executions);
View Full Code Here
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
  public void testHandleNoReply() throws Exception {
    //execute with no default set
    messageChannelPartitionHandler = new MessageChannelPartitionHandler();
    //mock
    StepExecution masterStepExecution = mock(StepExecution.class);
    StepExecutionSplitter stepExecutionSplitter = mock(StepExecutionSplitter.class);
    MessagingTemplate operations = mock(MessagingTemplate.class);
    Message message = mock(Message.class);
    //when
    HashSet<StepExecution> stepExecutions = new HashSet<StepExecution>();
    stepExecutions.add(new StepExecution("step1", new JobExecution(5l)));
    when(stepExecutionSplitter.split((StepExecution) anyObject(), eq(1))).thenReturn(stepExecutions);
    when(message.getPayload()).thenReturn(Collections.emptyList());
    when(operations.receive((PollableChannel) anyObject())).thenReturn(message);
    //set
    messageChannelPartitionHandler.setMessagingOperations(operations);
View Full Code Here
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
  public void testHandleWithReplyChannel() throws Exception {
    //execute with no default set
    messageChannelPartitionHandler = new MessageChannelPartitionHandler();
    //mock
    StepExecution masterStepExecution = mock(StepExecution.class);
    StepExecutionSplitter stepExecutionSplitter = mock(StepExecutionSplitter.class);
    MessagingTemplate operations = mock(MessagingTemplate.class);
    Message message = mock(Message.class);
    PollableChannel replyChannel = mock(PollableChannel.class);
    //when
    HashSet<StepExecution> stepExecutions = new HashSet<StepExecution>();
    stepExecutions.add(new StepExecution("step1", new JobExecution(5l)));
    when(stepExecutionSplitter.split((StepExecution) anyObject(), eq(1))).thenReturn(stepExecutions);
    when(message.getPayload()).thenReturn(Collections.emptyList());
    when(operations.receive(replyChannel)).thenReturn(message);
    //set
    messageChannelPartitionHandler.setMessagingOperations(operations);
    messageChannelPartitionHandler.setReplyChannel(replyChannel);
View Full Code Here
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
  public void messageReceiveTimeout() throws Exception {
    //execute with no default set
    messageChannelPartitionHandler = new MessageChannelPartitionHandler();
    //mock
    StepExecution masterStepExecution = mock(StepExecution.class);
    StepExecutionSplitter stepExecutionSplitter = mock(StepExecutionSplitter.class);
    MessagingTemplate operations = mock(MessagingTemplate.class);
    Message message = mock(Message.class);
    //when
    HashSet<StepExecution> stepExecutions = new HashSet<StepExecution>();
    stepExecutions.add(new StepExecution("step1", new JobExecution(5l)));
    when(stepExecutionSplitter.split((StepExecution) anyObject(), eq(1))).thenReturn(stepExecutions);
    when(message.getPayload()).thenReturn(Collections.emptyList());
    //set
    messageChannelPartitionHandler.setMessagingOperations(operations);

    //execute
View Full Code Here
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
          partitionStepExecutions = applyPartitionPlan(stepExecution, plan, true);
        }

      } else {
        StepExecutionSplitter stepSplitter = new JsrStepExecutionSplitter(jobRepository, allowStartIfComplete, stepExecution.getStepName(), true);
        partitionStepExecutions = stepSplitter.split(stepExecution, partitions);
      }
    } else {
      if(mapper != null) {
        PartitionPlan plan = mapper.mapPartitions();
        partitionStepExecutions = applyPartitionPlan(stepExecution, plan, true);
      } else {
        StepExecutionSplitter stepSplitter = new JsrStepExecutionSplitter(jobRepository, allowStartIfComplete, stepExecution.getStepName(), true);
        partitionStepExecutions = stepSplitter.split(stepExecution, partitions);
      }
    }
    return partitionStepExecutions;
  }
View Full Code Here
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
    return partitionStepExecutions;
  }

  private Set<StepExecution> applyPartitionPlan(StepExecution stepExecution,
      PartitionPlan plan, boolean restoreState) throws JobExecutionException {
    StepExecutionSplitter stepSplitter;
    Set<StepExecution> partitionStepExecutions;
    if(plan.getThreads() > 0) {
      threads = plan.getThreads();
    } else if(plan.getPartitions() > 0) {
      threads = plan.getPartitions();
    } else {
      throw new IllegalArgumentException("Either a number of threads or partitions are required");
    }

    stepExecution.getExecutionContext().put("partitionPlanState", new PartitionPlanState(plan));

    stepSplitter = new JsrStepExecutionSplitter(jobRepository, allowStartIfComplete, stepExecution.getStepName(), restoreState);
    partitionStepExecutions = stepSplitter.split(stepExecution, plan.getPartitions());
    registerPartitionProperties(partitionStepExecutions, plan);
    return partitionStepExecutions;
  }
View Full Code Here
TOP

Related Classes of org.springframework.batch.core.partition.StepExecutionSplitter

  • org.springframework.batch.core.jsr.partition.JsrPartitionHandler
  • org.springframework.batch.integration.partition.MessageChannelPartitionHandlerTest
  • org.springframework.batch.integration.partition.MessageChannelPartitionHandlerTests
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.