com.univocity.app.utils.DatabaseImpl.java Source code

Introduction

Here is the source code for com.univocity.app.utils.DatabaseImpl.java

Source

/*******************************************************************************
 * Copyright (c) 2014 uniVocity Software Pty Ltd. All rights reserved.
 * This file is subject to the terms and conditions defined in file
 * 'LICENSE.txt', which is part of this source code package.
 ******************************************************************************/
package com.univocity.app.utils;

import java.io.*;
import java.util.*;

import javax.sql.*;

import org.springframework.jdbc.core.*;
import org.springframework.jdbc.datasource.*;

class DatabaseImpl implements Database {

    private final JdbcTemplate jdbcTemplate;
    private final Set<String> tableNames = new TreeSet<String>();

    public DatabaseImpl(DataSource dataSource, File dirWithCreateTableScripts) {
        this.jdbcTemplate = new JdbcTemplate(dataSource);
        createTables(dirWithCreateTableScripts);
    }

    public DatabaseImpl(String databaseName, File dirWithCreateTableScripts) {
        try {
            Class.forName("org.hsqldb.jdbcDriver");
            DataSource dataSource = new SingleConnectionDataSource("jdbc:hsqldb:mem:" + databaseName, "sa", "",
                    true);
            this.jdbcTemplate = new JdbcTemplate(dataSource);
        } catch (Exception ex) {
            throw new IllegalStateException("Error creating database " + databaseName, ex);
        }

        createTables(dirWithCreateTableScripts);
    }

    private void createTables(File dirWithCreateTableScripts) {
        Map<String, String> scripts = new HashMap<String, String>();
        for (File scriptFile : dirWithCreateTableScripts.listFiles()) {
            if (scriptFile.isDirectory()) {
                continue;
            }
            String name = scriptFile.getName();
            String script = readFile(scriptFile);
            scripts.put(name.toLowerCase(), script);
        }

        createTables(scripts);
    }

    private void createTables(Map<String, String> scripts) {
        String scriptOrder = scripts.get("script_order");
        String[] order = scriptOrder.split(",");

        for (String tableName : order) {
            tableName = tableName.trim().toLowerCase();
            tableNames.add(tableName);

            String scriptName = tableName + ".tbl";

            String createTableScript = scripts.get(scriptName);
            try {
                jdbcTemplate.execute(createTableScript);
            } catch (Exception e) {
                throw new IllegalArgumentException("Error creating table with script " + scriptName, e);
            }
        }
    }

    private String readFile(File file) {
        StringBuilder out = new StringBuilder();
        try {
            BufferedReader in = new BufferedReader(new FileReader(file));
            String str;
            while ((str = in.readLine()) != null) {
                out.append(str).append('\n');
            }
            in.close();
        } catch (IOException e) {
            throw new IllegalStateException("Error reading file " + file.getAbsolutePath(), e);
        }
        return out.toString();
    }

    @Override
    public JdbcTemplate getJdbcTemplate() {
        return jdbcTemplate;
    }

    @Override
    public DataSource getDataSource() {
        return jdbcTemplate.getDataSource();
    }

    @Override
    public Set<String> getTableNames() {
        return Collections.unmodifiableSet(this.tableNames);
    }
}