1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
|
/*************************************************************************
*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* Copyright 2008 by Sun Microsystems, Inc.
*
* OpenOffice.org - a multi-platform office productivity suite
*
* $RCSfile: SQLExecution.java,v $
* $Revision: 1.7 $
*
* This file is part of OpenOffice.org.
*
* OpenOffice.org is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 3
* only, as published by the Free Software Foundation.
*
* OpenOffice.org is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License version 3 for more details
* (a copy is included in the LICENSE file that accompanied this code).
*
* You should have received a copy of the GNU Lesser General Public License
* version 3 along with OpenOffice.org. If not, see
* <http://www.openoffice.org/license.html>
* for a copy of the LGPLv3 License.
*
************************************************************************/
package stats;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.Statement;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.StringTokenizer;
import java.util.Vector;
/**
*
*/
public class SQLExecution {
protected Connection mConnection = null;
protected Statement mStatement = null;
protected String mJdbcClass = null;
protected String mDbURL = null;
protected String mUser = null;
protected String mPassword = null;
protected boolean m_bConnectionOpen = false;
protected boolean m_bDebug = false;
/** Creates a new instance of SQLExecution
* @param jdbcClass The jdbc class for the connection.
* @param dbUrl The url of the database.
* @param user The user for connecting the database.
* @param password The password of throws user.
*/
public SQLExecution(String jdbcClass, String dbUrl, String user, String password) {
mJdbcClass = jdbcClass;
mUser = user;
mPassword = password;
mDbURL = dbUrl;
}
/** Creates a new instance of SQLExecution with additional debug output.
* @param jdbcClass The jdbc class for the connection.
* @param dbUrl The url of the database.
* @param user The user for connecting the database.
* @param password The password of throws user.
* @param debug Write debug information, if true.
*/
public SQLExecution(String jdbcClass, String dbUrl, String user, String password, boolean debug) {
mJdbcClass = jdbcClass;
mUser = user;
mPassword = password;
mDbURL = dbUrl;
m_bDebug = debug;
}
/**
* Open a connection to the DataBase
* @return True, if no error occured.
*/
public boolean openConnection() {
if(m_bConnectionOpen) return true;
try {
Class.forName(mJdbcClass);
} catch (ClassNotFoundException e) {
System.err.println("Couldn't find jdbc driver : " + e.getMessage());
return false;
}
try {
// establish database connection
mConnection = DriverManager.getConnection(
mDbURL, mUser, mPassword);
mStatement = mConnection.createStatement();
}
catch(java.sql.SQLException e) {
System.err.println("Couldn't establish a connection: " + e.getMessage());
return false;
}
m_bConnectionOpen = true;
return true;
}
/**
* Close the connection to the DataBase
* @return True, if no error occured.
*/
public boolean closeConnection() {
if (!m_bConnectionOpen) return true;
try {
// close database connection
mStatement.close();
mConnection.close();
}
catch(java.sql.SQLException e) {
System.err.println("Couldn't close the connection: " + e.getMessage());
return false;
}
m_bConnectionOpen = false;
return true;
}
/**
* Execute an sql command.
* @param command The command to execute.
* @param sqlInput Input values for the command.
* @param sqlOutput The results of the command are put in this Hashtable.
* @return True, if no error occured.
*/
public boolean executeSQLCommand(String command, Hashtable sqlInput, Hashtable sqlOutput)
throws IllegalArgumentException {
return executeSQLCommand(command, sqlInput, sqlOutput, false);
}
/**
* Execute an sql command.
* @param command The command to execute.
* @param sqlInput Input values for the command.
* @param sqlOutput The results of the command are put in this Hashtable.
* @param mergeOutputIntoInput The output of the result is put into the
* sqlInput Hashtable.
* @return True, if no error occured.
*/
public boolean executeSQLCommand(String command, Hashtable sqlInput, Hashtable sqlOutput, boolean mergeOutputIntoInput)
throws IllegalArgumentException {
if (sqlOutput == null) {
sqlOutput = new Hashtable();
// this has to be true, so the user of this method gets a return
mergeOutputIntoInput = true;
if (sqlInput == null) {
System.out.println("sqlInput and sqlOutput are null: cannot return the results of the sql command.");
return false;
}
}
Vector sqlCommand = new Vector();
sqlCommand.add("");
boolean update = false;
// synchronize all "$varname" occurences in the command string with
// values from input
StringTokenizer token = new StringTokenizer(command, " ");
while (token.hasMoreTokens()) {
String originalKey = token.nextToken();
// search for keys, beginning with "$"
int index = originalKey.indexOf('$');
if (index != -1) {
// found key
String pre = "";
pre = originalKey.substring(0,index);
// generate key: remove "$"
String key = originalKey.substring(index+1);
String post = "";
// remove any endings the key might have
while (key.endsWith(",") || key.endsWith("\'") ||
key.endsWith(";") || key.endsWith(")") ||
key.endsWith("\"")) {
post = key.substring(key.length()-1) + post;
key = key.substring(0, key.length()-1);
}
// look for key in the Hashtable
if (sqlInput.containsKey(key)) {
// is there a value for the key?
Object in = sqlInput.get(key);
if (in instanceof String[]) {
// value is a String[]
String[]vals = (String[])in;
if (vals.length != sqlCommand.size() && sqlCommand.size() > 1) {
// size of this array and previous array(s) does not match
throw new IllegalArgumentException("The key '" + key + "' in command \n'"
+ command + "'\n has not the same value count as the keys before.");
}
// build up the commands
boolean addNewVals = (sqlCommand.size() == 1);
for (int i=0; i<vals.length; i++) {
String value = checkForQuotationMarks(vals[i]);
// add the values
if (addNewVals && i!=0) {
// all values until now were of type String, not String[], so now new values have to be added.
sqlCommand.add(i, (String)sqlCommand.get(0) + " " + pre + value + post);
}
else {
// we already have vals.length commands (or are at the first command), so just add.
sqlCommand.set(i, (String)sqlCommand.get(i) + " " + pre + value + post);
}
}
}
else {
// value is a String: no other possibility
String value = checkForQuotationMarks((String)sqlInput.get(key));
for (int i=0; i<sqlCommand.size(); i++) {
sqlCommand.set(i, (String)sqlCommand.get(i) + " " + pre + value + post);
}
}
}
else {
// no input value found
throw new IllegalArgumentException ("The key '" + key + "' in command \n'"
+ command + "'\n does not exist in the input values.");
}
}
else {
// token is not a key, just add it
for (int i=0; i<sqlCommand.size(); i++)
sqlCommand.set(i, (String)sqlCommand.get(i) + " " + originalKey);
if (originalKey.equalsIgnoreCase("update") ||
originalKey.equalsIgnoreCase("delete") ||
originalKey.equalsIgnoreCase("insert")) {
update = true;
}
}
}
for (int i=0;i<sqlCommand.size(); i++) {
execute((String)sqlCommand.get(i), sqlOutput, update);
// merge output with input
if (!update && mergeOutputIntoInput) {
Enumeration keys = sqlOutput.keys();
while(keys.hasMoreElements()) {
String key = (String)keys.nextElement();
String[]val = (String[])sqlOutput.get(key);
if (val != null && val.length != 0) {
if (val.length == 1)
sqlInput.put(key, val[0]);
else
sqlInput.put(key, val);
}
}
}
}
if (!update && sqlOutput == null)
return false;
return true;
}
/**
* Execute any SQL command.
* @param command The command.
* @param update If true, it is a update/alter command instead of an select
* command
* @return A Hashtable with the result.
*/
private void execute(String command, Hashtable output, boolean update) {
if (m_bDebug)
System.out.println("Debug - SQLExecution - execute Command: " + command);
try {
if (update) {
// make an update
mStatement.executeUpdate(command);
}
else {
// make a select: collect the result
ResultSet sqlResult = mStatement.executeQuery(command);
ResultSetMetaData sqlRSMeta = sqlResult.getMetaData();
int columnCount = sqlRSMeta.getColumnCount();
String[] columnNames = new String[columnCount];
int countRows = 0;
boolean goThroughRowsTheFirstTime = true;
for(int i=1; i<=columnCount; i++) {
columnNames[i-1] = sqlRSMeta.getColumnName(i);
// initialize output
Vector v = new Vector();
sqlResult.beforeFirst();
while (sqlResult.next()) {
String value = sqlResult.getString(i);
v.add(value);
// the first time: count rows
if (goThroughRowsTheFirstTime)
countRows++;
}
// rows are counted
if (goThroughRowsTheFirstTime)
goThroughRowsTheFirstTime = false;
// put result in output Hashtable
String[]s = new String[countRows];
s = (String[])v.toArray(s);
output.put(columnNames[i-1], s);
if (m_bDebug) {
if (i == 1) {
System.out.print("Debug - SQLExecution - Command returns: ");
System.out.print("row: " + columnNames[i-1] + " vals: ");
}
for (int j=0; j<s.length; j++)
System.out.print(s[j] + " ");
if (i == columnCount - 1)
System.out.println();
}
}
}
}
catch (java.sql.SQLException e) {
e.printStackTrace();
}
}
/**
* Replace <"> with <''> in the value Strings, or the command will fail.
* @param checkString The String that is checked: a part of the command
* @return The String, cleared of all quotation marks.
*/
private String checkForQuotationMarks(String checkString) {
String returnString = checkString;
int quotIndex = 0;
while ((quotIndex = returnString.indexOf('\"')) != -1) {
String firstHalf = returnString.substring(0, quotIndex);
String secondHalf = returnString.substring(quotIndex+1);
returnString = firstHalf + "\'\'" + secondHalf;
}
return returnString;
}
}
|