ResultsDecoratorSQL.java Source code

Introduction

Here is the source code for ResultsDecoratorSQL.java

Source

/*
 * 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 ResultsDecoratorSQL extends ResultsDecorator {
    ResultsDecoratorSQL(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();
}