1ae012fde7563dc3882b8f526d9503c8a74200c6
[mir.git] / source / mir / storage / RecordUpdater.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.util.*;\r
34 \r
35 /**\r
36  * Class to generate update statements\r
37  */\r
38 public class RecordUpdater extends StatementGenerator{\r
39   public RecordUpdater(String aTableName, String aRecordId) {\r
40     tableName = aTableName;\r
41     recordId = aRecordId;\r
42   }\r
43 \r
44   /**\r
45    * Assigns a value to a string typed field\r
46    */\r
47   public void assignString(String aFieldName, String aValue) {\r
48     assignObject(aFieldName, aValue);\r
49   }\r
50 \r
51   /**\r
52    * Assigns a value to a date/time typed field\r
53    */\r
54   public void assignDateTime(String aFieldName, Date aDate) {\r
55     assignObject(aFieldName, new java.sql.Timestamp(aDate.getTime()));\r
56   }\r
57 \r
58   /**\r
59    * Assigns a verbatim value to a field. Use with case: no quoting/escaping\r
60    * will be performed\r
61    **/\r
62   public void assignVerbatim(String aFieldName, String aValue) {\r
63     if (firstAssignment) {\r
64       appendQuery("update " + tableName + " set ");\r
65       firstAssignment = false;\r
66     }\r
67     else {\r
68       appendQuery(",");\r
69     }\r
70 \r
71     appendQuery(CRLF);\r
72     appendQuery("  " + aFieldName + " = " + aValue);\r
73   }\r
74 \r
75   /**\r
76    * Executes the statement. Returns then number of records\r
77    * changed\r
78    */\r
79   public int execute(Connection aConnection) throws DatabaseFailure {\r
80     appendQuery(CRLF);\r
81     appendQuery("where id = ?");\r
82     appendParameter(recordId);\r
83 \r
84     return executeWithModifiedCount(aConnection);\r
85   }\r
86 \r
87   private void assignObject(String aFieldName, Object anObject) {\r
88     appendParameter(anObject);\r
89     assignVerbatim(aFieldName, "?");\r
90   }\r
91 \r
92   private String tableName;\r
93   private static final String CRLF = "\n\r";\r
94   private String recordId;\r
95   private boolean firstAssignment = true;\r
96 }