com.complexible.stardog.ext.spring.DataImporter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of stardog-spring Show documentation
Stardog Spring bindings for using Stardog with the Springframework
There is a newer version: 6.1.0
/*
* 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;

import java.io.IOException;
import java.util.List;

import org.openrdf.rio.RDFFormat;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.core.io.Resource;

import com.complexible.stardog.StardogException;
import com.complexible.stardog.api.Connection;

/**
 * DataImporter
 * 
 * Given the support in Stardog to add/remove data in bulk via a stream,
 * this class provides a Spring Resource aware 
 * 
 * @author Clark and Parsia, LLC
 * @author Al Baker
 *
 */
public class DataImporter implements InitializingBean {

	final Logger log = LoggerFactory.getLogger(DataImporter.class);
	
	private SnarlTemplate snarlTemplate;

	private List inputFiles;
	
	private RDFFormat format;
	
	
	
	/**
	 * @return the format
	 */
	public RDFFormat getFormat() {
		return format;
	}

	/**
	 * @param format the format to set
	 */
	public void setFormat(RDFFormat format) {
		this.format = format;
	}

	/**
	 * @return the snarlTemplate
	 */
	public SnarlTemplate getSnarlTemplate() {
		return snarlTemplate;
	}

	/**
	 * @param snarlTemplate the snarlTemplate to set
	 */
	public void setSnarlTemplate(SnarlTemplate snarlTemplate) {
		this.snarlTemplate = snarlTemplate;
	}
	
	/**
	 * @return the inputFiles
	 */
	public List getInputFiles() {
		return inputFiles;
	}

	/**
	 * @param inputFiles the inputFiles to set
	 */
	public void setInputFiles(List inputFiles) {
		this.inputFiles = inputFiles;
	}

	/**
	 * inputfile
	 * 
	 * Ingests a file into Stardog, public API to be used after init time
	 * 
	 * @param format - RDFFormat enum
	 * @param resource - Spring Resource
	 * @return boolean
	 */
	public boolean inputFile(final RDFFormat format, final Resource resource) { 
		return snarlTemplate.execute(new ConnectionCallback() {
			@Override
			public Boolean doWithConnection(Connection connection) {
				try {
					connection.add().io().format(format).stream(resource.getInputStream());
				} catch (StardogException e) {
					log.error("Error with io adder to Stardog", e);
					return false;
				} catch (IOException e) {
					log.error("Error in IO to inputFile", e);
					return false;
				}		
				return true;
			} 
			
		});
	}

	/** 
	 * Loads any configured resources at init time 
	 * (non-Javadoc)
	 * @see org.springframework.beans.factory.InitializingBean#afterPropertiesSet()
	 */
	@Override
	public void afterPropertiesSet() throws Exception {
		if (inputFiles == null) { 
			return;
		}
		
		snarlTemplate.execute(new ConnectionCallback() {
			@Override
			public Boolean doWithConnection(Connection connection) {
				try {
					for (Resource entry : inputFiles) { 
						connection.add().io().format(format).stream(entry.getInputStream());
					}
				} catch (StardogException e) {
					log.error("Error with io adder to Stardog", e);
					return false;
				} catch (IOException e) {
					log.error("Error reading files for DataImporter initialization", e);
					return false;
				}		
				return true;
			} 
			
		});
	}
	
	
	
	
}