|
/*
* Copyright (c) Ian F. Darwin, http://www.darwinsys.com/, 1996-2002.
* All rights reserved. Software written by Ian F. Darwin and others.
* $Id: LICENSE,v 1.8 2004/02/09 03:33:38 ian Exp $
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS''
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*
* Java, the Duke mascot, and all variants of Sun's Java "steaming coffee
* cup" logo are trademarks of Sun Microsystems. Sun's, and James Gosling's,
* pioneering role in inventing and promulgating (and standardizing) the Java
* language and environment is gratefully acknowledged.
*
* The pioneering role of Dennis Ritchie and Bjarne Stroustrup, of AT&T, for
* inventing predecessor languages C and C++ is also gratefully acknowledged.
*/
import java.io.IOException;
import java.io.PrintWriter;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Types;
/**
* Print an SQL ResultSet in SQL-import format. TODO: check all escaped
* characters needed! Test on PGSQL and DB2 at least...
*
* @version $Id: ResultsDecoratorSQL.java,v 1.2 2004/03/26 02:39:33 ian Exp $
*/
public class Main extends ResultsDecorator {
Main(ResultsDecoratorPrinter out) {
super(out);
}
public void write(ResultSet rs) throws IOException, SQLException {
ResultSetMetaData md = rs.getMetaData();
// This assumes you're not using a Join!!
String tableName = md.getTableName(1);
int cols = md.getColumnCount();
StringBuffer sb = new StringBuffer("insert into ").append(tableName)
.append("(");
for (int i = 1; i <= cols; i++) {
sb.append(md.getColumnName(i));
if (i != cols) {
sb.append(", ");
}
}
sb.append(") values (");
String insertCommand = sb.toString();
while (rs.next()) {
println(insertCommand);
for (int i = 1; i <= cols; i++) {
String tmp = rs.getString(i);
if (rs.wasNull()) {
print("null");
} else {
int type = md.getColumnType(i);
// Don't quote numeric types; quote all others for now.
switch (type) {
case Types.BIGINT:
case Types.DECIMAL:
case Types.DOUBLE:
case Types.FLOAT:
case Types.INTEGER:
print(tmp);
break;
default:
tmp = tmp.replaceAll("'", "''");
print("'" + tmp + "'");
}
}
if (i != cols) {
print(", ");
}
}
println(");");
}
}
void write(int rowCount) throws IOException {
println("RowCount: " + rowCount);
}
/*
* (non-Javadoc)
*
* @see ResultsDecorator#getName()
*/
String getName() {
return "SQL";
}
}
/**
* Base class for a series of ResultSet printers.
*
* @version $Id: ResultsDecorator.java,v 1.2 2004/03/26 02:39:33 ian Exp $
*/
abstract class ResultsDecorator {
ResultsDecoratorPrinter parent;
ResultsDecorator(ResultsDecoratorPrinter wr) {
this.parent = wr;
}
/** Print the name of this Decorator's output format */
abstract String getName();
/** Print the contents of a ResultSet */
abstract void write(ResultSet rs) throws IOException, SQLException;
/** Print the results of an operation as a Count */
abstract void write(int rowCount) throws IOException;
void println(String line) throws IOException {
parent.println(line);
}
void println() throws IOException {
parent.println();
}
void print(String lineSeg) throws IOException {
parent.print(lineSeg);
}
}
/**
* Callback so that ResultsDecorator can call invoker to handle redirections
* etc.
*
* @version $Id: ResultsDecoratorPrinter.java,v 1.1 2004/03/26 02:39:33 ian Exp $
*/
interface ResultsDecoratorPrinter {
void print(String line) throws IOException;
void println(String line) throws IOException;
void println() throws IOException;
PrintWriter getPrintWriter();
}
|
