cmssetupdb.java

内容管理

package com.opencms.boot;

import java.io.FileReader;
import java.io.LineNumberReader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Hashtable;
import java.util.Properties;
import java.util.StringTokenizer;
import java.util.Vector;

public class CmsSetupDb {

    private Connection m_con;
    private Vector m_errors;
    private String m_basePath;
    private boolean m_errorLogging = true;
    private final String C_OCSETUP_FOLDER = "WEB-INF/ocsetup/";

    public CmsSetupDb(String basePath) {
        m_errors = new Vector();
        m_basePath = basePath;
    }

    public void setConnection(String DbDriver, String DbConStr, String DbUser, String DbPwd) {
        try {
            Class.forName(DbDriver);
            m_con = DriverManager.getConnection(DbConStr, DbUser, DbPwd);
        } catch (SQLException e) {
            m_errors.addElement("Could no connect to database via: " + DbConStr + "\n" + e.toString() + "\n");
        } catch (ClassNotFoundException e) {
            m_errors.addElement("Error while loading driver: " + DbDriver + "\n" + e.toString() + "\n");
        }
    }

    public void closeConnection() {
        try {
            m_con.close();
        } catch (Exception e) {}
    }

    public void dropDatabase(String resourceBroker, Hashtable replacer) {
        String file = getScript(resourceBroker + ".dropdb");
        if (file != null) {
            m_errorLogging = true;
            parseScript(file, replacer);
        } else {
            m_errors.addElement("Could not open database drop script: " + resourceBroker + ".dropdb");
        }
    }

    public void createDatabase(String resourceBroker, Hashtable replacer) {
        String file = getScript(resourceBroker + ".createdb");
        if (file != null) {
            m_errorLogging = true;
            parseScript(file, replacer);
        } else {
            m_errors.addElement("No create database script found: " + resourceBroker + ".createdb \n");
        }
    }

    public void createTables(String resourceBroker) {
        String file = getScript(resourceBroker + ".createtables");
        if (file != null) {
            m_errorLogging = true;
            parseScript(file, null);
        } else {
            m_errors.addElement("No create tables script found: " + resourceBroker + ".createtables \n");
        }
    }

    public void dropTables(String resourceBroker) {
        String file = getScript(resourceBroker + ".droptables");
        if (file != null) {
            m_errorLogging = true;
            parseScript(file, null);
        } else {
            m_errors.addElement("No drop tables script found: " + resourceBroker + ".droptables \n");
        }
    }

    private void parseScript(String file, Hashtable replacers) {

        /* indicates if the setup script contains included files (oracle) */
        boolean includedFile = false;

        /* get and parse the setup script */
        try {
            LineNumberReader reader = new LineNumberReader(new FileReader(m_basePath + C_OCSETUP_FOLDER + file));
            String line = null;
            String statement = "";
            while (true) {
                line = reader.readLine();
                if (line == null)
                    break;

                StringTokenizer st = new StringTokenizer(line);

                while (st.hasMoreTokens()) {
                    String currentToken = st.nextToken();

                    /* comment! Skip rest of the line */
                    if (currentToken.startsWith("#")) {
                        break;
                    }

                    /* not to be executed */
                    if (currentToken.startsWith("prompt")) {
                        break;
                    }

                    /* there is an included file */
                    if (currentToken.startsWith("@")) {
                        /* cut of '@' */
                        currentToken = currentToken.substring(1);
                        includedFile = true;
                    }

                    /* add token to query */
                    statement += " " + currentToken;

                    /* query complete (terminated by ';') */
                    if (currentToken.endsWith(";")) {
                        /* cut of ';' at the end */
                        statement = statement.substring(0, (statement.length() - 1));

                        /* there is an included File. Get it and execute it */
                        if (includedFile) {
                            if (replacers != null) {
                                ExecuteStatement(replaceValues(getIncludedFile(statement), replacers));
                            } else {
                                ExecuteStatement(getIncludedFile(statement));
                            }
                            //reset
                            includedFile = false;
                        }
                        /* normal statement. Execute it */
                        else {
                            if (replacers != null) {
                                ExecuteStatement(replaceValues(statement, replacers));
                            } else {
                                ExecuteStatement(statement);
                            }
                        }
                        //reset
                        statement = "";
                    }
                }
                statement += " \n";
            }
            reader.close();
        } catch (Exception e) {
            if (m_errorLogging) {
                m_errors.addElement(e.toString() + "\n");
            }
        }
    }

    private String replaceValues(String source, Hashtable replacers) {
        StringTokenizer tokenizer = new StringTokenizer(source);
        String temp = "";

        while (tokenizer.hasMoreTokens()) {
            String token = tokenizer.nextToken();

            // replace identifier found
            if (token.startsWith("$$") && token.endsWith("$$")) {

                // look in the hashtable
                Object value = replacers.get(token);

                //found value
                if (value != null) {
                    token = value.toString();
                }
            }
            temp += token + " ";
        }
        return temp;
    }

    private String getScript(String key) {
        /* open properties, get the value */
        try {
            Properties properties = new Properties();
            properties.load(getClass().getClassLoader().getResourceAsStream("com/opencms/boot/dbsetup.properties"));
            String value = properties.getProperty(key);
            return value;
        } catch (Exception e) {
            if (m_errorLogging) {
                m_errors.addElement(e.toString() + " \n");
            }
            return null;
        }
    }

    private String getIncludedFile(String statement) {
        String file;
        statement = statement.trim();

        if (statement.startsWith("./")) {
            /* cut off */
            file = statement.substring(2);
        } else if (statement.startsWith(".")) {
            /* cut off */
            file = statement.substring(1);
        } else {
            file = statement;
        }

        /* read and return everything */
        try {
            LineNumberReader reader = new LineNumberReader(new FileReader(m_basePath + C_OCSETUP_FOLDER + file));
            String stat = "";
            String line = null;

            while (true) {
                line = reader.readLine();
                if (line == null)
                    break;
                line = line.trim();
                if (line.startsWith("--") || line.startsWith("/")) {} else {
                    stat += line + " \n";
                }
            }
            reader.close();
            return stat.trim();
        } catch (Exception e) {
            if (m_errorLogging) {
                m_errors.addElement(e.toString() + "\n");
            }
            return null;
        }
    }

    private void ExecuteStatement(String statement) {
        Statement stat;
        try {
            stat = m_con.createStatement();
            stat.execute(statement);
            stat.close();
        } catch (Exception e) {
            if (m_errorLogging) {
                m_errors.addElement(e.toString() + "\n");
            }
        }
    }

    public Vector getErrors() {
        return m_errors;
    }
    public void clearErrors() {
        m_errors.clear();
    }

    public boolean noErrors() {
        return m_errors.isEmpty();
    }

    protected void finalize() {
        try {
            m_con.close();
        } catch (SQLException e) {}
    }
}