--- /dev/null
+/*\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 java.sql.Connection;\r
+import java.sql.PreparedStatement;\r
+import java.sql.ResultSet;\r
+import java.util.Date;\r
+\r
+/**\r
+ * Class to generate insert statements\r
+ */\r
+public class RecordInserter extends StatementGenerator {\r
+\r
+ public RecordInserter(String aTableName, String aSequenceName) {\r
+ tableName = aTableName;\r
+ sequenceName = aSequenceName;\r
+ }\r
+\r
+ /**\r
+ * Assigns a value to a string typed field\r
+ */\r
+ public void assignString(String aFieldName, String aValue) {\r
+ assignObject(aFieldName, aValue);\r
+ }\r
+\r
+ /**\r
+ * Assigns a value to a date/time typed field\r
+ */\r
+ public void assignDate(String aFieldName, Date aDate) {\r
+ assignObject(aFieldName, new java.sql.Date(aDate.getTime()));\r
+ }\r
+\r
+ public void assignVerbatim(String aFieldName, String aText) {\r
+ if (firstAssignment) {\r
+ appendQuery("insert into " + tableName + "(");\r
+ firstAssignment = false;\r
+ }\r
+ else {\r
+ appendQuery(",");\r
+ values.append(",");\r
+ }\r
+\r
+ values.append(aText);\r
+\r
+ appendQuery(aFieldName);\r
+ }\r
+\r
+ /**\r
+ * Executes the statement. Returns the id of the insterted record.\r
+ */\r
+ public String execute(Connection aConnection) throws DatabaseFailure {\r
+ appendQuery(")");\r
+ appendQuery(CRLF);\r
+ appendQuery("values (");\r
+ appendQuery(values.toString());\r
+ appendQuery(")");\r
+\r
+ try {\r
+ int modified = executeWithModifiedCount(aConnection);\r
+ if (modified!=1) {\r
+ throw new Exception("modified count != 1 after insert");\r
+ }\r
+\r
+ PreparedStatement statement = aConnection.prepareStatement("select currval('" + sequenceName + "')");\r
+ ResultSet rs = statement.executeQuery();\r
+ rs.next();\r
+ return rs.getString(1);\r
+ }\r
+ catch (Throwable e) {\r
+ throw new DatabaseFailure(e);\r
+ }\r
+ }\r
+\r
+ private void assignObject(String aFieldName, Object anObject) {\r
+ appendParameter(anObject);\r
+\r
+ assignVerbatim(aFieldName, "?");\r
+ }\r
+\r
+ private String tableName;\r
+ private static final String CRLF = "\n\r";\r
+ private boolean firstAssignment = true;\r
+ private String sequenceName;\r
+ private StringBuffer values = new StringBuffer();\r
+}
\ No newline at end of file