Examples with TransactionAwareBufferedWriter org.springframework.batch.support.transaction.TransactionAwareBufferedWriter used on opensource projects

Example 1 with TransactionAwareBufferedWriter

use of org.springframework.batch.support.transaction.TransactionAwareBufferedWriter in project spring-batch by spring-projects.

the class StaxEventItemWriter method open.

/**
 * Helper method for opening output source at given file position
 */
private void open(long position) {
    File file;
    FileOutputStream os = null;
    FileChannel fileChannel = null;
    try {
        file = resource.getFile();
        FileUtils.setUpOutputFile(file, restarted, false, overwriteOutput);
        Assert.state(resource.exists(), "Output resource must exist");
        os = new FileOutputStream(file, true);
        fileChannel = os.getChannel();
        channel = os.getChannel();
        setPosition(position);
    } catch (IOException ioe) {
        throw new DataAccessResourceFailureException("Unable to write to file resource: [" + resource + "]", ioe);
    }
    XMLOutputFactory outputFactory = createXmlOutputFactory();
    if (outputFactory.isPropertySupported("com.ctc.wstx.automaticEndElements")) {
        // If the current XMLOutputFactory implementation is supplied by
        // Woodstox >= 3.2.9 we want to disable its
        // automatic end element feature (see:
        // https://jira.codehaus.org/browse/WSTX-165) per
        // https://jira.spring.io/browse/BATCH-761).
        outputFactory.setProperty("com.ctc.wstx.automaticEndElements", Boolean.FALSE);
    }
    if (outputFactory.isPropertySupported("com.ctc.wstx.outputValidateStructure")) {
        // On restart we don't write the root element so we have to disable
        // structural validation (see:
        // https://jira.spring.io/browse/BATCH-1681).
        outputFactory.setProperty("com.ctc.wstx.outputValidateStructure", Boolean.FALSE);
    }
    try {
        final FileChannel channel = fileChannel;
        if (transactional) {
            TransactionAwareBufferedWriter writer = new TransactionAwareBufferedWriter(channel, new Runnable() {

                @Override
                public void run() {
                    closeStream();
                }
            });
            writer.setEncoding(encoding);
            writer.setForceSync(forceSync);
            bufferedWriter = writer;
        } else {
            bufferedWriter = new BufferedWriter(new OutputStreamWriter(os, encoding));
        }
        delegateEventWriter = createXmlEventWriter(outputFactory, bufferedWriter);
        eventWriter = new NoStartEndDocumentStreamWriter(delegateEventWriter);
        initNamespaceContext(delegateEventWriter);
        if (!restarted) {
            startDocument(delegateEventWriter);
            if (forceSync) {
                channel.force(false);
            }
        }
    } catch (XMLStreamException xse) {
        throw new DataAccessResourceFailureException("Unable to write to file resource: [" + resource + "]", xse);
    } catch (UnsupportedEncodingException e) {
        throw new DataAccessResourceFailureException("Unable to write to file resource: [" + resource + "] with encoding=[" + encoding + "]", e);
    } catch (IOException e) {
        throw new DataAccessResourceFailureException("Unable to write to file resource: [" + resource + "]", e);
    }
}

Example 2 with TransactionAwareBufferedWriter

use of org.springframework.batch.support.transaction.TransactionAwareBufferedWriter in project spring-batch by spring-projects.

the class StaxEventItemWriterBuilderTests method testTransactional.

@Test
public void testTransactional() {
    StaxEventItemWriter<Foo> staxEventItemWriter = new StaxEventItemWriterBuilder<Foo>().name("fooWriter").resource(this.resource).marshaller(this.marshaller).transactional(true).forceSync(true).build();
    ExecutionContext executionContext = new ExecutionContext();
    staxEventItemWriter.open(executionContext);
    Object writer = ReflectionTestUtils.getField(staxEventItemWriter, "bufferedWriter");
    assertTrue(writer instanceof TransactionAwareBufferedWriter);
    assertTrue((Boolean) ReflectionTestUtils.getField(writer, "forceSync"));
}

Example 3 with TransactionAwareBufferedWriter

use of org.springframework.batch.support.transaction.TransactionAwareBufferedWriter in project spring-batch by spring-projects.

the class StaxEventItemWriter method getPosition.

/*
	 * Get the actual position in file channel. This method flushes any buffered
	 * data before position is read.
	 * 
	 * @return byte offset in file channel
	 */
private long getPosition() {
    long position;
    try {
        eventWriter.flush();
        position = channel.position();
        if (bufferedWriter instanceof TransactionAwareBufferedWriter) {
            position += ((TransactionAwareBufferedWriter) bufferedWriter).getBufferSize();
        }
    } catch (Exception e) {
        throw new DataAccessResourceFailureException("Unable to write to file resource: [" + resource + "]", e);
    }
    return position;
}