view flys-aft/src/main/java/de/intevation/db/Statements.java @ 4072:88f801888d85

load configuration an setup db connections. flys-aft/trunk@3401 c6561f87-3c4e-4783-a992-168aeb5c3f6f
author Sascha L. Teichmann <sascha.teichmann@intevation.de>
date Tue, 13 Dec 2011 12:33:36 +0000
parents a4e79e8e0aa0
children 2c70fae83d0c
line wrap: on
line source
package de.intevation.db;

import java.util.Properties;

import java.io.IOException;
import java.io.InputStream;

import org.apache.log4j.Logger;

import java.util.List;
import java.util.ArrayList;
import java.util.Map;
import java.util.HashMap;
import java.util.Enumeration;

import java.sql.Connection;
import java.sql.SQLException;
import java.sql.PreparedStatement;
import java.sql.Timestamp;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class Statements
{
    public static final Pattern VAR = Pattern.compile(":([a-zA-Z0-9_]+)");

    public static class SymbolicStatement {

        protected String statement;
        protected String compiled;
        protected Map<String, List<Integer>> positions;

        public class Instance {

            /** TODO: Support more types. */

            protected PreparedStatement stmnt;

            public Instance(Connection connection) throws SQLException {
                stmnt = connection.prepareStatement(compiled);
            }

            public void close() {
                try {
                    stmnt.close();
                }
                catch (SQLException sqle) {
                    log.error("cannot close statement", sqle);
                }
            }

            public void setInt(String key, int value)
            throws SQLException 
            {
                List<Integer> pos = positions.get(key.toLowerCase());
                if (pos != null) {
                    for (Integer p: pos) {
                        stmnt.setInt(p, value);
                    }
                }
            }

            public void setString(String key, String value)
            throws SQLException 
            {
                List<Integer> pos = positions.get(key.toLowerCase());
                if (pos != null) {
                    for (Integer p: pos) {
                        stmnt.setString(p, value);
                    }
                }
            }

            public void setObject(String key, Object value)
            throws SQLException 
            {
                List<Integer> pos = positions.get(key.toLowerCase());
                if (pos != null) {
                    for (Integer p: pos) {
                        stmnt.setObject(p, value);
                    }
                }
            }

            public void setTimestamp(String key, Timestamp value)
            throws SQLException 
            {
                List<Integer> pos = positions.get(key.toLowerCase());
                if (pos != null) {
                    for (Integer p: pos) {
                        stmnt.setTimestamp(p, value);
                    }
                }
            }

            public void setDouble(String key, int value)
            throws SQLException 
            {
                List<Integer> pos = positions.get(key.toLowerCase());
                if (pos != null) {
                    for (Integer p: pos) {
                        stmnt.setDouble(p, value);
                    }
                }
            }

            public void setNull(String key, int sqlType)
            throws SQLException 
            {
                List<Integer> pos = positions.get(key.toLowerCase());
                if (pos != null) {
                    for (Integer p: pos) {
                        stmnt.setNull(p, sqlType);
                    }
                }
            }

            public void set(Map<String, Object> map) throws SQLException {
                for (Map.Entry<String, Object> entry: map.entrySet()) {
                    setObject(entry.getKey(), entry.getValue());
                }
            }

        } // class Instance

        public SymbolicStatement(String statement) {
            this.statement = statement;
            compile();
        }

        public String getStatement() {
            return statement;
        }

        protected void compile() {
            positions = new HashMap<String, List<Integer>>();

            StringBuffer sb = new StringBuffer();
            Matcher m = VAR.matcher(statement);
            int index = 1;
            while (m.find()) {
                String key = m.group(1).toLowerCase();
                List<Integer> list = positions.get(key);
                if (list == null) {
                    list = new ArrayList<Integer>();
                    positions.put(key, list);
                }
                list.add(index++);
                m.appendReplacement(sb, "?");
            }
            m.appendTail(sb);
            compiled = sb.toString();
        }
    } // class SymbolicStatement


    private static Logger log = Logger.getLogger(Statements.class);

    public static final String RESOURCE_PATH = "/sql/";
    public static final String COMMON_PROPERTIES = "-common.properties";

    protected String type;
    protected String driver;

    protected Map<String, SymbolicStatement> statements;

    public Statements(String type, String driver) {
        this.type   = type;
        this.driver = driver;
    }

    public SymbolicStatement getStatement(String key) {
        if (statements == null) {
            statements = loadStatements();
        }
        return statements.get(key);
    }

    protected Map<String, SymbolicStatement> loadStatements() {
        Map<String, SymbolicStatement> statements =
            new HashMap<String, SymbolicStatement>();

        Properties properties = loadProperties();

        for (Enumeration e = properties.propertyNames(); e.hasMoreElements();) {
            String key = (String)e.nextElement();
            String value = properties.getProperty(key);
            SymbolicStatement symbolic = new SymbolicStatement(value);
            statements.put(key, symbolic);
        }

        return statements;
    }

    protected String driverToProperties() {
        return
            type + "-" + 
            driver.replace('.', '-').toLowerCase() + ".properties";
    }

    protected Properties loadCommon() {
        Properties common = new Properties();

        String path = RESOURCE_PATH + type + COMMON_PROPERTIES;

        InputStream in = Statements.class.getResourceAsStream(path);

        if (in != null) {
            try {
                common.load(in);
            }
            catch (IOException ioe) {
                log.error("cannot load defaults: " + path, ioe);
            }
            finally {
                try {
                    in.close();
                }
                catch (IOException ioe) {
                }
            }
        }
        else {
            log.warn("cannot find: " + path);
        }

        return common;
    }

    protected Properties loadProperties() {

        Properties common = loadCommon();

        Properties properties = new Properties(common);

        String path = RESOURCE_PATH + driverToProperties();

        InputStream in = Statements.class.getResourceAsStream(path);

        if (in != null) {
            try {
                properties.load(in);
            }
            catch (IOException ioe) {
                log.error("cannot load statements: " + path, ioe);
            }
            finally {
                try {
                    in.close();
                }
                catch (IOException ioe) {
                }
            }
        }
        else {
            log.warn("cannot find: " + path);
        }

        return properties;
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :

http://dive4elements.wald.intevation.org