How to use

onSkipInWrite

method

in

org.springframework.batch.core.listener.MulticasterBatchListener

/**
 * Test method for
 * {@link org.springframework.batch.core.listener.MulticasterBatchListener#onSkipInWrite(java.lang.Object, java.lang.Throwable)}
 * .
 */
@Test
public void testOnSkipInWrite() {
  multicast.register(new SkipListenerSupport<Object,Object>() {
    @Override
    public void onSkipInWrite(Object item, Throwable t) {
      count++;
      super.onSkipInWrite(item, t);
    }
  });
  multicast.onSkipInWrite(null, new RuntimeException("foo"));
  assertEquals(1, count);
}

/**
 * Test method for
 * {@link org.springframework.batch.core.listener.MulticasterBatchListener#onSkipInWrite(java.lang.Object, java.lang.Throwable)}
 * .
 */
@Test
public void testOnSkipInWriteFails() {
  multicast.register(new SkipListenerSupport<Object,Object>() {
    @Override
    public void onSkipInWrite(Object item, Throwable t) {
      count++;
      throw new RuntimeException("foo");
    }
  });
  try {
    multicast.onSkipInWrite(null, new RuntimeException("bar"));
    fail("Expected RuntimeException");
  }
  catch (RuntimeException e) {
    // expected
    String message = e.getMessage();
    assertEquals("Wrong message: " + message, "foo", message);
  }
  assertEquals(1, count);
}

private void callSkipListeners(final Chunk<I> inputs, final Chunk<O> outputs) {
  for (SkipWrapper<I> wrapper : inputs.getSkips()) {
    I item = wrapper.getItem();
    if (item == null) {
      continue;
    }
    Throwable e = wrapper.getException();
    callProcessSkipListener(item, e);
  }
  for (SkipWrapper<O> wrapper : outputs.getSkips()) {
    Throwable e = wrapper.getException();
    try {
      getListener().onSkipInWrite(wrapper.getItem(), e);
    }
    catch (RuntimeException ex) {
      throw new SkipListenerFailedException("Fatal exception in SkipListener.", ex, e);
    }
  }
  // Clear skips if we are possibly going to process this chunk again
  outputs.clearSkips();
  inputs.clearSkips();
}

  @Override
  @SuppressWarnings({ "unchecked", "rawtypes" })
  public Object doWithRetry(RetryContext context) throws Exception {
    chunkMonitor.setChunkSize(chunk.size());
    try {
      doPersist(contribution, chunk);
    }
    catch (Exception e) {
      if (shouldSkip(skipPolicy, e, contribution.getStepSkipCount())) {
        // Per section 9.2.7 of JSR-352, the SkipListener receives all the items within the chunk 						 
        ((MulticasterBatchListener) getListener()).onSkipInWrite(chunk.getItems(), e);
      } else {
        getListener().onRetryWriteException((List<Object>) chunk.getItems(), e);
        if (rollbackClassifier.classify(e)) {
          throw e;
        }
      }
      /*
       * If the exception is marked as no-rollback, we need to
       * override that, otherwise there's no way to write the
       * rest of the chunk or to honour the skip listener
       * contract.
       */
      throw new ForceRollbackForWriteSkipException(
          "Force rollback on skippable exception so that skipped item can be located.", e);
    }
    contribution.incrementWriteCount(chunk.size());
    return null;
  }
};

private void callSkipListeners(final Chunk<I> inputs, final Chunk<O> outputs) {
  for (SkipWrapper<I> wrapper : inputs.getSkips()) {
    I item = wrapper.getItem();
    if (item == null) {
      continue;
    }
    Throwable e = wrapper.getException();
    callProcessSkipListener(item, e);
  }
  for (SkipWrapper<O> wrapper : outputs.getSkips()) {
    Throwable e = wrapper.getException();
    try {
      getListener().onSkipInWrite(wrapper.getItem(), e);
    }
    catch (RuntimeException ex) {
      throw new SkipListenerFailedException("Fatal exception in SkipListener.", ex, e);
    }
  }
  // Clear skips if we are possibly going to process this chunk again
  outputs.clearSkips();
  inputs.clearSkips();
}

private void callSkipListeners(final Chunk<I> inputs, final Chunk<O> outputs) {
  for (SkipWrapper<I> wrapper : inputs.getSkips()) {
    I item = wrapper.getItem();
    if (item == null) {
      continue;
    }
    Throwable e = wrapper.getException();
    callProcessSkipListener(item, e);
  }
  for (SkipWrapper<O> wrapper : outputs.getSkips()) {
    Throwable e = wrapper.getException();
    try {
      getListener().onSkipInWrite(wrapper.getItem(), e);
    }
    catch (RuntimeException ex) {
      throw new SkipListenerFailedException("Fatal exception in SkipListener.", ex, e);
    }
  }
  // Clear skips if we are possibly going to process this chunk again
  outputs.clearSkips();
  inputs.clearSkips();
}

private void callSkipListeners(final Chunk<I> inputs, final Chunk<O> outputs) {
  for (SkipWrapper<I> wrapper : inputs.getSkips()) {
    I item = wrapper.getItem();
    if (item == null) {
      continue;
    }
    Throwable e = wrapper.getException();
    callProcessSkipListener(item, e);
  }
  for (SkipWrapper<O> wrapper : outputs.getSkips()) {
    Throwable e = wrapper.getException();
    try {
      getListener().onSkipInWrite(wrapper.getItem(), e);
    }
    catch (RuntimeException ex) {
      throw new SkipListenerFailedException("Fatal exception in SkipListener.", ex, e);
    }
  }
  // Clear skips if we are possibly going to process this chunk again
  outputs.clearSkips();
  inputs.clearSkips();
}

private void callSkipListeners(final Chunk<I> inputs, final Chunk<O> outputs) {
  for (SkipWrapper<I> wrapper : inputs.getSkips()) {
    I item = wrapper.getItem();
    if (item == null) {
      continue;
    }
    Throwable e = wrapper.getException();
    callProcessSkipListener(item, e);
  }
  for (SkipWrapper<O> wrapper : outputs.getSkips()) {
    Throwable e = wrapper.getException();
    try {
      getListener().onSkipInWrite(wrapper.getItem(), e);
    }
    catch (RuntimeException ex) {
      throw new SkipListenerFailedException("Fatal exception in SkipListener.", ex, e);
    }
  }
  // Clear skips if we are possibly going to process this chunk again
  outputs.clearSkips();
  inputs.clearSkips();
}

  @Override
  @SuppressWarnings({ "unchecked", "rawtypes" })
  public Object doWithRetry(RetryContext context) throws Exception {
    chunkMonitor.setChunkSize(chunk.size());
    try {
      doPersist(contribution, chunk);
    }
    catch (Exception e) {
      if (shouldSkip(skipPolicy, e, contribution.getStepSkipCount())) {
        // Per section 9.2.7 of JSR-352, the SkipListener receives all the items within the chunk 						 
        ((MulticasterBatchListener) getListener()).onSkipInWrite(chunk.getItems(), e);
      } else {
        getListener().onRetryWriteException((List<Object>) chunk.getItems(), e);
        if (rollbackClassifier.classify(e)) {
          throw e;
        }
      }
      /*
       * If the exception is marked as no-rollback, we need to
       * override that, otherwise there's no way to write the
       * rest of the chunk or to honour the skip listener
       * contract.
       */
      throw new ForceRollbackForWriteSkipException(
          "Force rollback on skippable exception so that skipped item can be located.", e);
    }
    contribution.incrementWriteCount(chunk.size());
    return null;
  }
};

  @Override
  @SuppressWarnings({ "unchecked", "rawtypes" })
  public Object doWithRetry(RetryContext context) throws Exception {
    chunkMonitor.setChunkSize(chunk.size());
    try {
      doPersist(contribution, chunk);
    }
    catch (Exception e) {
      if (shouldSkip(skipPolicy, e, contribution.getStepSkipCount())) {
        // Per section 9.2.7 of JSR-352, the SkipListener receives all the items within the chunk 						 
        ((MulticasterBatchListener) getListener()).onSkipInWrite(chunk.getItems(), e);
      } else {
        getListener().onRetryWriteException((List<Object>) chunk.getItems(), e);
        if (rollbackClassifier.classify(e)) {
          throw e;
        }
      }
      /*
       * If the exception is marked as no-rollback, we need to
       * override that, otherwise there's no way to write the
       * rest of the chunk or to honour the skip listener
       * contract.
       */
      throw new ForceRollbackForWriteSkipException(
          "Force rollback on skippable exception so that skipped item can be located.", e);
    }
    contribution.incrementWriteCount(chunk.size());
    return null;
  }
};

  @Override
  @SuppressWarnings({ "unchecked", "rawtypes" })
  public Object doWithRetry(RetryContext context) throws Exception {
    chunkMonitor.setChunkSize(chunk.size());
    try {
      doPersist(contribution, chunk);
    }
    catch (Exception e) {
      if (shouldSkip(skipPolicy, e, contribution.getStepSkipCount())) {
        // Per section 9.2.7 of JSR-352, the SkipListener receives all the items within the chunk 						 
        ((MulticasterBatchListener) getListener()).onSkipInWrite(chunk.getItems(), e);
      } else {
        getListener().onRetryWriteException((List<Object>) chunk.getItems(), e);
        if (rollbackClassifier.classify(e)) {
          throw e;
        }
      }
      /*
       * If the exception is marked as no-rollback, we need to
       * override that, otherwise there's no way to write the
       * rest of the chunk or to honour the skip listener
       * contract.
       */
      throw new ForceRollbackForWriteSkipException(
          "Force rollback on skippable exception so that skipped item can be located.", e);
    }
    contribution.incrementWriteCount(chunk.size());
    return null;
  }
};