source/mir/storage/RecordInserter.java

   1 /*\r
   2  * Copyright (C) 2005 The Mir-coders group\r
   3  *\r
   4  * This file is part of Mir.\r
   5  *\r
   6  * Mir is free software; you can redistribute it and/or modify\r
   7  * it under the terms of the GNU General Public License as published by\r
   8  * the Free Software Foundation; either version 2 of the License, or\r
   9  * (at your option) any later version.\r
  10  *\r
  11  * Mir is distributed in the hope that it will be useful,\r
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  14  * GNU General Public License for more details.\r
  15  *\r
  16  * You should have received a copy of the GNU General Public License\r
  17  * along with Mir; if not, write to the Free Software\r
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\r
  19  *\r
  20  * In addition, as a special exception, The Mir-coders gives permission to link\r
  21  * the code of this program with  any library licensed under the Apache Software License,\r
  22  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library\r
  23  * (or with modified versions of the above that use the same license as the above),\r
  24  * and distribute linked combinations including the two.  You must obey the\r
  25  * GNU General Public License in all respects for all of the code used other than\r
  26  * the above mentioned libraries.  If you modify this file, you may extend this\r
  27  * exception to your version of the file, but you are not obligated to do so.\r
  28  * If you do not wish to do so, delete this exception statement from your version.\r
  29  */\r
  30 package mir.storage;\r
  31 \r
  32 import java.sql.Connection;\r
  33 import java.sql.PreparedStatement;\r
  34 import java.sql.ResultSet;\r
  35 import java.sql.SQLException;\r
  36 import java.util.Date;\r
  37 \r
  38 /**\r
  39  * Class to generate insert statements\r
  40  */\r
  41 public class RecordInserter extends StatementGenerator {\r
  42 \r
  43   public RecordInserter(String aTableName, String aSequenceName) {\r
  44     tableName = aTableName;\r
  45     sequenceName = aSequenceName;\r
  46   }\r
  47 \r
  48   /**\r
  49    * Assigns a value to a string typed field\r
  50    */\r
  51   public void assignString(String aFieldName, String aValue) {\r
  52     assignObject(aFieldName, aValue);\r
  53   }\r
  54 \r
  55   /**\r
  56    * Assigns a value to a date/time typed field\r
  57    */\r
  58   public void assignDate(String aFieldName, Date aDate) {\r
  59     assignObject(aFieldName, new java.sql.Date(aDate.getTime()));\r
  60   }\r
  61 \r
  62   public void assignVerbatim(String aFieldName, String aText) {\r
  63     if (firstAssignment) {\r
  64       appendQuery("insert into " + tableName + "(");\r
  65       firstAssignment = false;\r
  66     }\r
  67     else {\r
  68       appendQuery(",");\r
  69       values.append(",");\r
  70     }\r
  71 \r
  72     values.append(aText);\r
  73 \r
  74     appendQuery(aFieldName);\r
  75   }\r
  76 \r
  77   /**\r
  78    * Executes the statement. Returns the id of the insterted record.\r
  79    */\r
  80   public String execute(Connection aConnection) throws DatabaseFailure {\r
  81     appendQuery(")");\r
  82     appendQuery(CRLF);\r
  83     appendQuery("values (");\r
  84     appendQuery(values.toString());\r
  85     appendQuery(")");\r
  86 \r
  87     try {\r
  88       int modified = executeWithModifiedCount(aConnection);\r
  89       if (modified!=1) {\r
  90         throw new Exception("modified count != 1 after insert");\r
  91       }\r
  92 \r
  93       PreparedStatement statement = aConnection.prepareStatement("select currval('" + sequenceName + "')");\r
  94       ResultSet rs = statement.executeQuery();\r
  95       rs.next();\r
  96       return rs.getString(1);\r
  97     }\r
  98     catch (Throwable e) {\r
  99       throw new DatabaseFailure(e);\r
 100     }\r
 101   }\r
 102 \r
 103   private void assignObject(String aFieldName, Object anObject) {\r
 104     appendParameter(anObject);\r
 105 \r
 106     assignVerbatim(aFieldName, "?");\r
 107   }\r
 108 \r
 109   private String tableName;\r
 110   private static final String CRLF = "\n\r";\r
 111   private boolean firstAssignment = true;\r
 112   private String sequenceName;\r
 113   private StringBuffer values = new StringBuffer();\r
 114 }