com.complexible.stardog.ext.spring.DataImporter.java Source code

Introduction

Here is the source code for com.complexible.stardog.ext.spring.DataImporter.java

Source

/*
* 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<Resource> 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<Resource> getInputFiles() {
        return inputFiles;
    }

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

    /**
     * <code>inputfile</code>
     * 
     * 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<Boolean>() {
            @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<Boolean>() {
            @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;
            }

        });
    }

}