org.craftercms.studio.impl.v1.dal.DataSourceInitializerImpl.java Source code

Introduction

Here is the source code for org.craftercms.studio.impl.v1.dal.DataSourceInitializerImpl.java

Source

/*
 * Crafter Studio Web-content authoring solution
 * Copyright (C) 2007-2016 Crafter Software Corporation.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

package org.craftercms.studio.impl.v1.dal;

import org.apache.ibatis.jdbc.ScriptRunner;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.craftercms.studio.api.v1.dal.DataSourceInitializer;
import org.craftercms.studio.api.v1.log.Logger;
import org.craftercms.studio.api.v1.log.LoggerFactory;
import org.craftercms.studio.api.v1.util.StudioConfiguration;

import java.io.*;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Map;

import static org.craftercms.studio.api.v1.util.StudioConfiguration.DB_INITIALIZER_ENABLED;
import static org.craftercms.studio.api.v1.util.StudioConfiguration.DB_PLATFORM;

public class DataSourceInitializerImpl implements DataSourceInitializer {

    private final static Logger logger = LoggerFactory.getLogger(DataSourceInitializerImpl.class);

    @Override
    public void initDataSource() {
        if (isEnabled()) {
            String scriptPath = getScriptPath();
            SqlSession session = sqlSessionFactory.openSession();
            Connection conn = session.getConnection();

            ScriptRunner sr = new ScriptRunner(conn);

            sr.setDelimiter(delimiter);
            InputStream is = getClass().getClassLoader().getResourceAsStream(scriptPath);
            Reader reader = new InputStreamReader(is);
            sr.runScript(reader);
            try {
                conn.close();
            } catch (SQLException e) {
                logger.error("Error while closing connection with database", e);
            }
        }
    }

    public boolean isEnabled() {
        boolean toReturn = Boolean.parseBoolean(studioConfiguration.getProperty(DB_INITIALIZER_ENABLED));
        return toReturn;
    }

    private String getScriptPath() {
        String pathToScript = vendorScriptsMapping.get(getVendor());
        return pathToScript;
    }

    public String getVendor() {
        return studioConfiguration.getProperty(DB_PLATFORM);
    }

    public Map<String, String> getVendorScriptsMapping() {
        return vendorScriptsMapping;
    }

    @Override
    public void setVendorScriptsMapping(Map<String, String> vendorScriptsMapping) {
        this.vendorScriptsMapping = vendorScriptsMapping;
    }

    public SqlSessionFactory getSqlSessionFactory() {
        return sqlSessionFactory;
    }

    public void setSqlSessionFactory(SqlSessionFactory sqlSessionFactory) {
        this.sqlSessionFactory = sqlSessionFactory;
    }

    public String getDelimiter() {
        return delimiter;
    }

    public void setDelimiter(String delimiter) {
        this.delimiter = delimiter;
    }

    public StudioConfiguration getStudioConfiguration() {
        return studioConfiguration;
    }

    public void setStudioConfiguration(StudioConfiguration studioConfiguration) {
        this.studioConfiguration = studioConfiguration;
    }

    protected Map<String, String> vendorScriptsMapping;
    protected SqlSessionFactory sqlSessionFactory;
    protected String delimiter;
    protected StudioConfiguration studioConfiguration;
}