com.complexible.stardog.ext.spring.batch.SnarlItemWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of stardog-spring-batch Show documentation

Stardog Spring bindings for using Stardog with the Springframework Batch subproject

The newest version!

/*
* Copyright (c) the original 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 com.complexible.stardog.ext.spring.batch;

import java.util.List;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.batch.item.ItemWriter;

import com.complexible.stardog.StardogException;
import com.complexible.stardog.api.Adder;
import com.complexible.stardog.api.Connection;
import com.complexible.stardog.ext.spring.DataSource;

/**
 * Implementation of the Spring Batch ItemWriter interface that supports the SNARL API
 * 
 * @author Al Baker
 * @author Clark & Parsia
 *
 */
public class SnarlItemWriter implements ItemWriter {

	final Logger log = LoggerFactory.getLogger(SnarlItemWriter.class);
	
	private DataSource dataSource;
	
	private BatchAdderCallback callback;
	
	@Override
	public void write(List items) throws Exception {
		Connection connection = dataSource.getConnection();
		Adder adder = null;
		try {
			connection.begin();
			adder = connection.add();
			callback.write(adder, items);
			connection.commit();
		} catch (StardogException e) {
			log.error("Error with SnarlItemWriter ", e);
			throw new RuntimeException(e);
		} finally { 
			adder = null;
			dataSource.releaseConnection(connection);
		}
		
	}

	/**
	 * @return the dataSource
	 */
	public DataSource getDataSource() {
		return dataSource;
	}

	/**
	 * @param dataSource the dataSource to set
	 */
	public void setDataSource(DataSource dataSource) {
		this.dataSource = dataSource;
	}

	/**
	 * @return the callback
	 */
	public BatchAdderCallback getCallback() {
		return callback;
	}

	/**
	 * @param callback the callback to set
	 */
	public void setCallback(BatchAdderCallback callback) {
		this.callback = callback;
	}

}