merged 1.1 branch into head

[mir.git] / source / mir / storage / StatementGenerator.java

diff --git a/source/mir/storage/StatementGenerator.java b/source/mir/storage/StatementGenerator.java
new file mode 100755 (executable)
index 0000000..1eb4036
--- /dev/null
+++ b/source/mir/storage/StatementGenerator.java
@@ -0,0 +1,147 @@
+/*\r
+ * Copyright (C) 2001-2006 The Mir-coders group\r
+ *\r
+ * This file is part of Mir.\r
+ *\r
+ * Mir is free software; you can redistribute it and/or modify\r
+ * it under the terms of the GNU General Public License as published by\r
+ * the Free Software Foundation; either version 2 of the License, or\r
+ * (at your option) any later version.\r
+ *\r
+ * Mir is distributed in the hope that it will be useful,\r
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
+ * GNU General Public License for more details.\r
+ *\r
+ * You should have received a copy of the GNU General Public License\r
+ * along with Mir; if not, write to the Free Software\r
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\r
+ *\r
+ * In addition, as a special exception, The Mir-coders gives permission to link\r
+ * the code of this program with  any library licensed under the Apache Software License,\r
+ * and distribute linked combinations including the two.  You must obey the\r
+ * GNU General Public License in all respects for all of the code used other than\r
+ * the above mentioned libraries.  If you modify this file, you may extend this\r
+ * exception to your version of the file, but you are not obligated to do so.\r
+ * If you do not wish to do so, delete this exception statement from your version.\r
+ */\r
+package mir.storage;\r
+\r
+import mir.log.LoggerWrapper;\r
+\r
+import java.sql.Connection;\r
+import java.sql.PreparedStatement;\r
+import java.sql.ResultSet;\r
+import java.util.ArrayList;\r
+import java.util.Iterator;\r
+import java.util.List;\r
+\r
+/**\r
+ * Base class for (sql) statement generators\r
+ */\r
+public class StatementGenerator {\r
+  private static LoggerWrapper logger = new LoggerWrapper("Database.Statements");\r
+\r
+  public StatementGenerator() {\r
+    query = new StringBuffer();\r
+    parameters = new ArrayList();\r
+  }\r
+\r
+  protected void appendQuery(String aPart) {\r
+    query.append(aPart);\r
+  }\r
+\r
+  protected void appendParameter(Object aParameter) {\r
+    parameters.add(aParameter);\r
+  }\r
+\r
+  /**\r
+    * Executes the statement. Returns the number of modified records.\r
+    */\r
+  protected int executeWithModifiedCount(Connection aConnection) throws DatabaseFailure {\r
+    long start = System.currentTimeMillis();\r
+\r
+    try {\r
+      aConnection.setAutoCommit(false);\r
+      logQueryBefore(query.toString());\r
+\r
+      PreparedStatement statement = aConnection.prepareStatement(query.toString());\r
+\r
+      int index = 1;\r
+      Iterator i = parameters.iterator();\r
+\r
+      while (i.hasNext()) {\r
+        statement.setObject(index, i.next());\r
+        index++;\r
+      }\r
+\r
+      int result = statement.executeUpdate();\r
+\r
+      logQueryAfter(query.toString(), System.currentTimeMillis() - start);\r
+\r
+      return result;\r
+    }\r
+    catch (Throwable e) {\r
+      logQueryError(query.toString(), System.currentTimeMillis() - start, e);\r
+      System.out.println(e.toString());\r
+\r
+      throw new DatabaseFailure(e);\r
+    }\r
+    finally {\r
+      try {\r
+        aConnection.setAutoCommit(true);\r
+      }\r
+      catch (Exception e) {\r
+      }\r
+    }\r
+  }\r
+\r
+  /**\r
+    * Executes the statement. Returns a resultset.\r
+    */\r
+  protected ResultSet executeWithResultSet(Connection aConnection) throws DatabaseFailure {\r
+    long start = System.currentTimeMillis();\r
+\r
+    try {\r
+      logQueryBefore(query.toString());\r
+\r
+      PreparedStatement statement = aConnection.prepareStatement(query.toString());\r
+\r
+      int index = 1;\r
+      Iterator i = parameters.iterator();\r
+\r
+      while (i.hasNext()) {\r
+        statement.setObject(index, i.next());\r
+        index++;\r
+      }\r
+\r
+      ResultSet result = statement.executeQuery();\r
+\r
+      logQueryAfter(query.toString(), System.currentTimeMillis() - start);\r
+\r
+      return result;\r
+    }\r
+    catch (Throwable e) {\r
+      logQueryError(query.toString(), System.currentTimeMillis() - start, e);\r
+      System.out.println(e.toString());\r
+\r
+      throw new DatabaseFailure(e);\r
+    }\r
+  }\r
+\r
+  private void logQueryBefore(String aQuery) {\r
+    logger.debug("about to perform QUERY " + aQuery);\r
+  }\r
+\r
+  private void logQueryAfter(String aQuery, long aTime) {\r
+    logger.info("QUERY " + aQuery + " took " + aTime + "ms.");\r
+  }\r
+\r
+  private void logQueryError(String aQuery, long aTime, Throwable anException) {\r
+    logger.error("QUERY " + aQuery + " took " + aTime + "ms, but threw exception " + anException.toString());\r
+  }\r
+\r
+\r
+  private StringBuffer query;\r
+  private List parameters;\r
+}\r