sascha@4073: package de.intevation.db;
sascha@4073: 
sascha@4073: import java.util.List;
sascha@4073: import java.util.ArrayList;
sascha@4073: import java.util.Map;
sascha@4073: import java.util.HashMap;
sascha@4073: 
sascha@4073: import java.util.regex.Matcher;
sascha@4073: import java.util.regex.Pattern;
sascha@4073: 
sascha@4073: import java.sql.Connection;
sascha@4073: import java.sql.SQLException;
sascha@4073: import java.sql.PreparedStatement;
sascha@4073: import java.sql.Timestamp;
sascha@4073: 
sascha@4073: import org.apache.log4j.Logger;
sascha@4073: 
sascha@4073: public class SymbolicStatement {
sascha@4073: 
sascha@4073:     private static Logger log = Logger.getLogger(SymbolicStatement.class);
sascha@4073: 
sascha@4073:     public static final Pattern VAR = Pattern.compile(":([a-zA-Z0-9_]+)");
sascha@4073: 
sascha@4073:     protected String statement;
sascha@4073:     protected String compiled;
sascha@4073:     protected Map<String, List<Integer>> positions;
sascha@4073: 
sascha@4073:     public class Instance {
sascha@4073: 
sascha@4073:         /** TODO: Support more types. */
sascha@4073: 
sascha@4073:         protected PreparedStatement stmnt;
sascha@4073: 
sascha@4073:         public Instance(Connection connection) throws SQLException {
sascha@4073:             stmnt = connection.prepareStatement(compiled);
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void close() {
sascha@4073:             try {
sascha@4073:                 stmnt.close();
sascha@4073:             }
sascha@4073:             catch (SQLException sqle) {
sascha@4073:                 log.error("cannot close statement", sqle);
sascha@4073:             }
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void setInt(String key, int value)
sascha@4073:         throws SQLException 
sascha@4073:         {
sascha@4073:             List<Integer> pos = positions.get(key.toLowerCase());
sascha@4073:             if (pos != null) {
sascha@4073:                 for (Integer p: pos) {
sascha@4073:                     stmnt.setInt(p, value);
sascha@4073:                 }
sascha@4073:             }
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void setString(String key, String value)
sascha@4073:         throws SQLException 
sascha@4073:         {
sascha@4073:             List<Integer> pos = positions.get(key.toLowerCase());
sascha@4073:             if (pos != null) {
sascha@4073:                 for (Integer p: pos) {
sascha@4073:                     stmnt.setString(p, value);
sascha@4073:                 }
sascha@4073:             }
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void setObject(String key, Object value)
sascha@4073:         throws SQLException 
sascha@4073:         {
sascha@4073:             List<Integer> pos = positions.get(key.toLowerCase());
sascha@4073:             if (pos != null) {
sascha@4073:                 for (Integer p: pos) {
sascha@4073:                     stmnt.setObject(p, value);
sascha@4073:                 }
sascha@4073:             }
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void setTimestamp(String key, Timestamp value)
sascha@4073:         throws SQLException 
sascha@4073:         {
sascha@4073:             List<Integer> pos = positions.get(key.toLowerCase());
sascha@4073:             if (pos != null) {
sascha@4073:                 for (Integer p: pos) {
sascha@4073:                     stmnt.setTimestamp(p, value);
sascha@4073:                 }
sascha@4073:             }
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void setDouble(String key, int value)
sascha@4073:         throws SQLException 
sascha@4073:         {
sascha@4073:             List<Integer> pos = positions.get(key.toLowerCase());
sascha@4073:             if (pos != null) {
sascha@4073:                 for (Integer p: pos) {
sascha@4073:                     stmnt.setDouble(p, value);
sascha@4073:                 }
sascha@4073:             }
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void setNull(String key, int sqlType)
sascha@4073:         throws SQLException 
sascha@4073:         {
sascha@4073:             List<Integer> pos = positions.get(key.toLowerCase());
sascha@4073:             if (pos != null) {
sascha@4073:                 for (Integer p: pos) {
sascha@4073:                     stmnt.setNull(p, sqlType);
sascha@4073:                 }
sascha@4073:             }
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void set(Map<String, Object> map) throws SQLException {
sascha@4073:             for (Map.Entry<String, Object> entry: map.entrySet()) {
sascha@4073:                 setObject(entry.getKey(), entry.getValue());
sascha@4073:             }
sascha@4073:         }
sascha@4073: 
sascha@4073:         public void clearParameters() throws SQLException {
sascha@4073:             stmnt.clearParameters();
sascha@4073:         }
sascha@4073: 
sascha@4073:     } // class Instance
sascha@4073: 
sascha@4073:     public SymbolicStatement(String statement) {
sascha@4073:         this.statement = statement;
sascha@4073:         compile();
sascha@4073:     }
sascha@4073: 
sascha@4073:     public String getStatement() {
sascha@4073:         return statement;
sascha@4073:     }
sascha@4073: 
sascha@4073:     protected void compile() {
sascha@4073:         positions = new HashMap<String, List<Integer>>();
sascha@4073: 
sascha@4073:         StringBuffer sb = new StringBuffer();
sascha@4073:         Matcher m = VAR.matcher(statement);
sascha@4073:         int index = 1;
sascha@4073:         while (m.find()) {
sascha@4073:             String key = m.group(1).toLowerCase();
sascha@4073:             List<Integer> list = positions.get(key);
sascha@4073:             if (list == null) {
sascha@4073:                 list = new ArrayList<Integer>();
sascha@4073:                 positions.put(key, list);
sascha@4073:             }
sascha@4073:             list.add(index++);
sascha@4073:             m.appendReplacement(sb, "?");
sascha@4073:         }
sascha@4073:         m.appendTail(sb);
sascha@4073:         compiled = sb.toString();
sascha@4073:     }
sascha@4073: } // class SymbolicStatement