BATCH-2005: Implement JSR RetryListener along side Spring Batch's RetryListener
This commit is contained in:
Chris Schaefer
committed by
Michael Minella
Michael Minella
parent
df53ea12b3
commit
e53533e9e3
@@ -0,0 +1,244 @@
|
||||
/*
|
||||
* Copyright 2013 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.springframework.batch.core.jsr.configuration.xml;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.util.Collection;
|
||||
import java.util.List;
|
||||
import javax.batch.api.chunk.ItemProcessor;
|
||||
import javax.batch.api.chunk.ItemReader;
|
||||
import javax.batch.api.chunk.ItemWriter;
|
||||
import javax.batch.api.chunk.listener.RetryProcessListener;
|
||||
import javax.batch.api.chunk.listener.RetryReadListener;
|
||||
import javax.batch.api.chunk.listener.RetryWriteListener;
|
||||
import javax.batch.operations.BatchRuntimeException;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.junit.Test;
|
||||
import org.springframework.batch.core.ExitStatus;
|
||||
import org.springframework.batch.core.Job;
|
||||
import org.springframework.batch.core.JobExecution;
|
||||
import org.springframework.batch.core.JobParameters;
|
||||
import org.springframework.batch.core.StepExecution;
|
||||
import org.springframework.batch.core.launch.JobLauncher;
|
||||
import org.springframework.context.ApplicationContext;
|
||||
import org.springframework.context.support.ClassPathXmlApplicationContext;
|
||||
import org.springframework.retry.RetryException;
|
||||
import org.springframework.util.Assert;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
/**
|
||||
* <p>
|
||||
* Test cases around JSR-352 retry listeners.
|
||||
* </p>
|
||||
*
|
||||
* @author Chris Schaefer
|
||||
* @since 3.0
|
||||
*/
|
||||
public class RetryListenerTests {
|
||||
private static final Log LOG = LogFactory.getLog(RetryListenerTests.class);
|
||||
|
||||
@Test
|
||||
public void testReadRetryExhausted() throws Exception {
|
||||
ApplicationContext context = new ClassPathXmlApplicationContext("org/springframework/batch/core/jsr/configuration/xml/RetryReadListenerExhausted.xml");
|
||||
|
||||
JobLauncher jobLauncher = context.getBean(JobLauncher.class);
|
||||
JobExecution jobExecution = jobLauncher.run(context.getBean(Job.class), new JobParameters());
|
||||
|
||||
List<Throwable> failureExceptions = jobExecution.getAllFailureExceptions();
|
||||
assertTrue("Expected 1 failure exceptions", failureExceptions.size() == 1);
|
||||
assertTrue("Failure exception must be of type RetryException", (failureExceptions.get(0) instanceof RetryException));
|
||||
assertTrue("Exception cause must be of type IllegalArgumentException", (failureExceptions.get(0).getCause() instanceof IllegalArgumentException));
|
||||
|
||||
assertEquals(ExitStatus.FAILED, jobExecution.getExitStatus());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReadRetryOnce() throws Exception {
|
||||
ApplicationContext context = new ClassPathXmlApplicationContext("org/springframework/batch/core/jsr/configuration/xml/RetryReadListenerRetryOnce.xml");
|
||||
|
||||
JobLauncher jobLauncher = context.getBean(JobLauncher.class);
|
||||
JobExecution jobExecution = jobLauncher.run(context.getBean(Job.class), new JobParameters());
|
||||
|
||||
Collection<StepExecution> stepExecutions = jobExecution.getStepExecutions();
|
||||
assertEquals(1, stepExecutions.size());
|
||||
|
||||
StepExecution stepExecution = stepExecutions.iterator().next();
|
||||
assertEquals(1, stepExecution.getCommitCount());
|
||||
assertEquals(2, stepExecution.getReadCount());
|
||||
|
||||
assertEquals(ExitStatus.COMPLETED, jobExecution.getExitStatus());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReadRetryExceptionInListener() throws Exception {
|
||||
ApplicationContext context = new ClassPathXmlApplicationContext("org/springframework/batch/core/jsr/configuration/xml/RetryReadListenerListenerException.xml");
|
||||
|
||||
JobLauncher jobLauncher = context.getBean(JobLauncher.class);
|
||||
JobExecution jobExecution = jobLauncher.run(context.getBean(Job.class), new JobParameters());
|
||||
|
||||
List<Throwable> failureExceptions = jobExecution.getAllFailureExceptions();
|
||||
assertTrue("Failure exceptions must equal one", failureExceptions.size() == 1);
|
||||
assertTrue("Failure exception must be of type RetryException", (failureExceptions.get(0) instanceof RetryException));
|
||||
assertTrue("Exception cause must be of type BatchRuntimeException", (failureExceptions.get(0).getCause() instanceof BatchRuntimeException));
|
||||
|
||||
assertEquals(ExitStatus.FAILED, jobExecution.getExitStatus());
|
||||
}
|
||||
|
||||
public static class ExceptionThrowingRetryReadListener implements RetryReadListener {
|
||||
@Override
|
||||
public void onRetryReadException(Exception ex) throws Exception {
|
||||
Assert.isInstanceOf(IllegalArgumentException.class, ex);
|
||||
throw new IllegalStateException();
|
||||
}
|
||||
}
|
||||
|
||||
public static class TestRetryReadListener implements RetryReadListener {
|
||||
@Override
|
||||
public void onRetryReadException(Exception ex) throws Exception {
|
||||
Assert.isInstanceOf(IllegalArgumentException.class, ex);
|
||||
}
|
||||
}
|
||||
|
||||
public static class TestRetryProcessListener implements RetryProcessListener {
|
||||
@Override
|
||||
public void onRetryProcessException(Object item, Exception ex) throws Exception {
|
||||
Assert.isInstanceOf(String.class, item);
|
||||
|
||||
String currentItem = (String) item;
|
||||
|
||||
Assert.isTrue("three".equals(currentItem));
|
||||
Assert.isInstanceOf(IllegalArgumentException.class, ex);
|
||||
}
|
||||
}
|
||||
|
||||
public static class TestRetryWriteListener implements RetryWriteListener {
|
||||
@Override
|
||||
public void onRetryWriteException(List<Object> items, Exception ex) throws Exception {
|
||||
Assert.isTrue(items.size() == 2, "Must be two items to write");
|
||||
Assert.isTrue(items.contains("three"), "Items must contain the string 'three'");
|
||||
Assert.isTrue(items.contains("one"), "Items must contain the string 'one'");
|
||||
Assert.isInstanceOf(IllegalArgumentException.class, ex);
|
||||
}
|
||||
}
|
||||
|
||||
public static class AlwaysFailItemReader implements ItemReader {
|
||||
@Override
|
||||
public void open(Serializable checkpoint) throws Exception {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() throws Exception {
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object readItem() throws Exception {
|
||||
throw new IllegalArgumentException();
|
||||
}
|
||||
|
||||
@Override
|
||||
public Serializable checkpointInfo() throws Exception {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
public static class FailOnceItemReader implements ItemReader {
|
||||
private int cnt;
|
||||
|
||||
@Override
|
||||
public void open(Serializable checkpoint) throws Exception {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() throws Exception {
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object readItem() throws Exception {
|
||||
if(cnt == 0) {
|
||||
cnt++;
|
||||
return "one";
|
||||
} else if (cnt == 1) {
|
||||
cnt++;
|
||||
throw new IllegalArgumentException();
|
||||
} else if (cnt == 2) {
|
||||
cnt++;
|
||||
return "three";
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Serializable checkpointInfo() throws Exception {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
public static class FailOnceItemProcessor implements ItemProcessor {
|
||||
private int cnt;
|
||||
|
||||
@Override
|
||||
public Object processItem(Object item) throws Exception {
|
||||
if(cnt == 0) {
|
||||
cnt++;
|
||||
return "one";
|
||||
} else if (cnt == 1) {
|
||||
cnt++;
|
||||
throw new IllegalArgumentException();
|
||||
} else if (cnt == 2) {
|
||||
cnt++;
|
||||
return "three";
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
public static class FailOnceItemWriter implements ItemWriter {
|
||||
private int cnt;
|
||||
|
||||
@Override
|
||||
public void open(Serializable checkpoint) throws Exception {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() throws Exception {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void writeItems(List<Object> items) throws Exception {
|
||||
for(Object item : items) {
|
||||
if(cnt == 0) {
|
||||
cnt++;
|
||||
LOG.info("one");
|
||||
} else if (cnt == 1) {
|
||||
cnt++;
|
||||
throw new IllegalArgumentException();
|
||||
} else if (cnt == 2) {
|
||||
cnt++;
|
||||
LOG.info("three");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public Serializable checkpointInfo() throws Exception {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user