Add Builder for SynchronizedItemStreamReader
Reimplemented tests from the SynchronizedItemStreamReaderTests for the builder as a better part of valor. resolves BATCH-2595
This commit is contained in:
committed by
Michael Minella
parent
d512fef644
commit
4a3086fe4b
@@ -0,0 +1,61 @@
|
||||
/*
|
||||
* Copyright 2017 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.item.support.builder;
|
||||
|
||||
import org.springframework.batch.item.ItemStreamReader;
|
||||
import org.springframework.batch.item.support.SynchronizedItemStreamReader;
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
/**
|
||||
* Creates a fully qualified SynchronizedItemStreamReader.
|
||||
*
|
||||
* @author Glenn Renfro
|
||||
*
|
||||
* @since 4.0
|
||||
*/
|
||||
public class SynchronizedItemStreamReaderBuilder<T> {
|
||||
|
||||
private ItemStreamReader<T> delegate;
|
||||
|
||||
/**
|
||||
* The item stream reader to use as a delegate. Items are read from the delegate and
|
||||
* passed to the caller in
|
||||
* {@link org.springframework.batch.item.support.SynchronizedItemStreamReader#read()}.
|
||||
*
|
||||
* @param delegate the delegate to set
|
||||
* @return this instance for method chaining
|
||||
* @see SynchronizedItemStreamReader#setDelegate(ItemStreamReader)
|
||||
*/
|
||||
public SynchronizedItemStreamReaderBuilder delegate(ItemStreamReader<T> delegate) {
|
||||
this.delegate = delegate;
|
||||
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a fully constructed {@link SynchronizedItemStreamReader}.
|
||||
*
|
||||
* @return a new {@link SynchronizedItemStreamReader}
|
||||
*/
|
||||
public SynchronizedItemStreamReader<T> build() {
|
||||
Assert.notNull(this.delegate, "A delegate is required");
|
||||
|
||||
SynchronizedItemStreamReader<T> reader = new SynchronizedItemStreamReader<>();
|
||||
reader.setDelegate(this.delegate);
|
||||
return reader;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,146 @@
|
||||
/*
|
||||
* Copyright 2017 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.item.support.builder;
|
||||
|
||||
import java.util.HashSet;
|
||||
import java.util.Set;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import org.springframework.batch.item.ExecutionContext;
|
||||
import org.springframework.batch.item.ItemStreamReader;
|
||||
import org.springframework.batch.item.NonTransientResourceException;
|
||||
import org.springframework.batch.item.ParseException;
|
||||
import org.springframework.batch.item.support.AbstractItemStreamItemReader;
|
||||
import org.springframework.batch.item.support.SynchronizedItemStreamReader;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
/**
|
||||
* @author Glenn Renfro
|
||||
*/
|
||||
public class SynchronizedItemStreamReaderBuilderTests {
|
||||
|
||||
@Test
|
||||
public void testMultipleThreads() throws Exception {
|
||||
|
||||
// Initialized an ExecutionContext and a SynchronizedItemStreamReader to test.
|
||||
final ExecutionContext executionContext = new ExecutionContext();
|
||||
|
||||
final SynchronizedItemStreamReaderBuilderTests.TestItemReader testItemReader = new SynchronizedItemStreamReaderBuilderTests.TestItemReader();
|
||||
final SynchronizedItemStreamReader<Integer> synchronizedItemStreamReader = new SynchronizedItemStreamReaderBuilder<Integer>()
|
||||
.delegate(testItemReader).build();
|
||||
|
||||
// Open the ItemReader and make sure it's initialized properly.
|
||||
synchronizedItemStreamReader.open(executionContext);
|
||||
assertEquals(true,
|
||||
executionContext.get(SynchronizedItemStreamReaderBuilderTests.TestItemReader.HAS_BEEN_OPENED));
|
||||
assertFalse(testItemReader.isClosed());
|
||||
|
||||
/*
|
||||
* Set up SIZE threads that read from the reader and updates the execution
|
||||
* context.
|
||||
*/
|
||||
final Set<Integer> ecSet = new HashSet<Integer>();
|
||||
final int SIZE = 20;
|
||||
Thread[] threads = new Thread[SIZE];
|
||||
for (int i = 0; i < SIZE; i++) {
|
||||
threads[i] = new Thread() {
|
||||
public void run() {
|
||||
try {
|
||||
ecSet.add(synchronizedItemStreamReader.read());
|
||||
synchronizedItemStreamReader.update(executionContext);
|
||||
}
|
||||
catch (Exception ignore) {
|
||||
ignore.printStackTrace();
|
||||
}
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
// Start the threads and block until all threads are done.
|
||||
for (Thread thread : threads) {
|
||||
thread.run();
|
||||
}
|
||||
for (Thread thread : threads) {
|
||||
thread.join();
|
||||
}
|
||||
testItemReader.close();
|
||||
|
||||
/*
|
||||
* Ensure cleanup happens as expected: status variable is set correctly and
|
||||
* ExecutionContext variable is set properly. Lastly, the Set<Integer> should have
|
||||
* 1 to 20 which may not always be the case if the read is not synchronized.
|
||||
*/
|
||||
for (int i = 1; i <= SIZE; i++) {
|
||||
assertTrue(ecSet.contains(i));
|
||||
}
|
||||
assertTrue(testItemReader.isClosed());
|
||||
assertEquals(SIZE,
|
||||
executionContext.getInt(SynchronizedItemStreamReaderBuilderTests.TestItemReader.UPDATE_COUNT_KEY));
|
||||
}
|
||||
|
||||
/**
|
||||
* A simple class used to test the SynchronizedItemStreamReader. It simply returns the
|
||||
* number of times the read method has been called, manages some state variables and
|
||||
* updates an ExecutionContext.
|
||||
*
|
||||
* @author Matthew Ouyang
|
||||
*
|
||||
*/
|
||||
private class TestItemReader extends AbstractItemStreamItemReader<Integer> implements ItemStreamReader<Integer> {
|
||||
|
||||
private int cursor = 0;
|
||||
|
||||
private boolean isClosed = false;
|
||||
|
||||
public static final String HAS_BEEN_OPENED = "hasBeenOpened";
|
||||
|
||||
public static final String UPDATE_COUNT_KEY = "updateCount";
|
||||
|
||||
public Integer read() throws Exception, ParseException, NonTransientResourceException {
|
||||
cursor = cursor + 1;
|
||||
return cursor;
|
||||
}
|
||||
|
||||
public void close() {
|
||||
this.isClosed = true;
|
||||
}
|
||||
|
||||
public void open(ExecutionContext executionContext) {
|
||||
this.isClosed = false;
|
||||
executionContext.put(HAS_BEEN_OPENED, true);
|
||||
executionContext.remove(UPDATE_COUNT_KEY);
|
||||
}
|
||||
|
||||
public void update(ExecutionContext executionContext) {
|
||||
|
||||
if (!executionContext.containsKey(UPDATE_COUNT_KEY)) {
|
||||
executionContext.putInt(UPDATE_COUNT_KEY, 0);
|
||||
}
|
||||
|
||||
executionContext.putInt(UPDATE_COUNT_KEY, executionContext.getInt(UPDATE_COUNT_KEY) + 1);
|
||||
}
|
||||
|
||||
public boolean isClosed() {
|
||||
return this.isClosed;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user