/** * Returns a fully constructed {@link ClassifierCompositeItemWriter}. * * @return a new {@link ClassifierCompositeItemWriter} */ public ClassifierCompositeItemWriter<T> build() { Assert.notNull(classifier, "A classifier is required."); ClassifierCompositeItemWriter<T> writer = new ClassifierCompositeItemWriter<>(); writer.setClassifier(this.classifier); return writer; }
@Test public void testWrite() throws Exception { Map<String, ItemWriter<? super String>> map = new HashMap<>(); ItemWriter<String> fooWriter = new ItemWriter<String>() { @Override public void write(List<? extends String> items) throws Exception { foos.addAll(items); } }; ItemWriter<String> defaultWriter = new ItemWriter<String>() { @Override public void write(List<? extends String> items) throws Exception { defaults.addAll(items); } }; map.put("foo", fooWriter ); map.put("*", defaultWriter); writer.setClassifier(new PatternMatchingClassifier<>(map)); writer.write(Arrays.asList("foo", "foo", "one", "two", "three")); assertEquals("[foo, foo]", foos.toString()); assertEquals("[one, two, three]", defaults.toString()); }
@Test public void testWrite() throws Exception { Map<String, ItemWriter<? super String>> map = new HashMap<>(); ItemWriter<String> fooWriter = items -> foos.addAll(items); ItemWriter<String> defaultWriter = items -> defaults.addAll(items); map.put("foo", fooWriter); map.put("*", defaultWriter); ClassifierCompositeItemWriter<String> writer = new ClassifierCompositeItemWriterBuilder<String>() .classifier(new PatternMatchingClassifier<>(map)).build(); writer.write(Arrays.asList("foo", "foo", "one", "two", "three")); assertEquals("[foo, foo]", foos.toString()); assertEquals("[one, two, three]", defaults.toString()); }
@Test public void testSetNullClassifier() throws Exception { try { ClassifierCompositeItemWriter<String> writer = new ClassifierCompositeItemWriter<>(); writer.setClassifier(null); fail("A classifier is required."); } catch (IllegalArgumentException iae) { assertEquals("Message returned from exception did not match expected result.", "A classifier is required.", iae.getMessage()); } } }
/** * Returns a fully constructed {@link ClassifierCompositeItemWriter}. * * @return a new {@link ClassifierCompositeItemWriter} */ public ClassifierCompositeItemWriter<T> build() { Assert.notNull(classifier, "A classifier is required."); ClassifierCompositeItemWriter<T> writer = new ClassifierCompositeItemWriter<>(); writer.setClassifier(this.classifier); return writer; }
@Bean public ClassifierCompositeItemWriter<Customer> itemWriter() throws Exception { ClassifierCompositeItemWriter<Customer> itemWriter = new ClassifierCompositeItemWriter<>(); itemWriter.setClassifier(new CustomerClassifier(xmlItemWriter(), jsonItemWriter())); return itemWriter; }