org.kuali.common.impex.spring.MpxSupplierConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kuali-impex-producer Show documentation
Kuali Impex SQL Producer
There is a newer version: 3.0.3
/**
 * Copyright 2011 The Kuali Foundation Licensed under the
 * Educational Community 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.osedu.org/licenses/ECL-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 org.kuali.common.impex.spring;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import javax.xml.bind.JAXBException;

import org.kuali.common.impex.data.SqlProducer;
import org.kuali.common.impex.data.impl.MpxLocationSupplier;
import org.kuali.common.impex.data.impl.mysql.MySqlProducer;
import org.kuali.common.impex.data.impl.oracle.OracleProducer;
import org.kuali.common.jdbc.spring.JdbcCommonConfig;
import org.kuali.common.jdbc.spring.JdbcDataSourceConfig;
import org.kuali.common.jdbc.supplier.LocationSupplierSourceBean;
import org.kuali.common.util.spring.SpringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.core.env.Environment;

@Configuration
@Import({ JdbcCommonConfig.class, BatchConfig.class, JdbcDataSourceConfig.class, XmlSchemaConfig.class })
public class MpxSupplierConfig {

    @Autowired
    Environment env;

    @Autowired
    JdbcCommonConfig jdbcCommonConfig;

    @Autowired
    JdbcDataSourceConfig dataSourceConfig;

    @Autowired
    BatchConfig batchConfig;

    @Autowired
    XmlSchemaConfig modelProviderConfig;

    private static final String DB_VENDOR_KEY = "db.vendor";

    @Bean
    Map vendorProducerMap() {
        Map results = new HashMap();

        results.put(OracleProducer.SUPPORTED_VENDOR, configureProducer(new OracleProducer()));
        results.put(MySqlProducer.SUPPORTED_VENDOR, configureProducer(new MySqlProducer()));

        return results;
    }

    private SqlProducer configureProducer(SqlProducer sqlProducer) {

        sqlProducer.setBatchDataSizeLimit(batchConfig.impexBatchSize());
        sqlProducer.setBatchRowCountLimit(batchConfig.impexBatchRows());

        return sqlProducer;
    }

    @Bean
    public SqlProducer impexProducer() {
        String vendor = SpringUtils.getProperty(env, DB_VENDOR_KEY);

        return vendorProducerMap().get(vendor);
    }

    @Bean
    public Map impexExtensionMappings() throws JAXBException, IOException {
        // This gets cloned for each .mpx file
        MpxLocationSupplier mls = new MpxLocationSupplier();
        mls.setSchema(modelProviderConfig.schema());
        mls.setProducer(impexProducer());

        // This hands out clones of MpxLocationSupplier, one for every .mpx file being parsed
        LocationSupplierSourceBean lssb = new LocationSupplierSourceBean();
        lssb.setSupplierClass(MpxLocationSupplier.class);
        lssb.setSupplierInstance(mls);

        // Add mpx as an extension JDBC can handle
        Map mappings = jdbcCommonConfig.jdbcExtensionMappings();
        mappings.put("mpx", lssb);
        return mappings;
    }

}