ResultSetTableModel

 /*********************************************************************/
 /*(c) Copyright IBM Corp. 2003  All rights reserved.                 */
 /*                                                                   */
 /*This sample program is owned by International Business Machines    */
 /*Corporation or one of its subsidiaries ("IBM") and is copyrighted  */
 /*and licensed, not sold.                                            */
 /*                                                                   */
 /*You may copy, modify, and distribute this sample program in any    */
 /*form without payment to IBM,  for any purpose including developing,*/
 /*using, marketing or distributing programs that include or are      */
 /*derivative works of the sample program.                            */
 /*                                                                   */
 /*The sample program is provided to you on an "AS IS" basis, without */
 /*warranty of any kind.  IBM HEREBY  EXPRESSLY DISCLAIMS ALL         */
 /*WARRANTIES EITHER EXPRESS OR IMPLIED, INCLUDING, BUT NOT LIMITED TO*/
 /*THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTIC-*/
 /*ULAR PURPOSE. Some jurisdictions do not allow for the exclusion or */
 /*limitation of implied warranties, so the above limitations or      */
 /*exclusions may not apply to you.  IBM shall not be liable for any  */
 /*damages you suffer as a result of using, modifying or distributing */
 /*the sample program or its derivatives.                             */
 /*                                                                   */
 /*Each copy of any portion of this sample program or any derivative  */
 /*work,  must include a the above copyright notice and disclaimer of */
 /*warranty.                                                          */
 /*                                                                   */
 /*********************************************************************/

package com.ibm.DB2Excel;

/*********************************************************************/
  /******************************************************************/
  /* Stores the contents of a database table                         */
  /******************************************************************/
 import javax.swing.table.AbstractTableModel;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.*;
/**
 * Implements the TableModel interface so that a JTable can display the contents of ResultSet.
 **/
 public class ResultSetTableModel extends AbstractTableModel
{
    /**
     * number of rows in the ResultSet
     */
    private int rowCount;
     /**
     * number of columns in ResultSet
     */
    private int colCount;
     /**
     * Simple facade that hides the details of executing JDBC commands
     */
    private SQLFacade sqlFacade;
     /**
     * The names of the columns in the ResultSet
     */
    private List columnNames = new ArrayList();
     /**
     *  The class types of the columns being displayed.
     */
    private List columnTypes = new ArrayList();
     /**
     *  Transcribes the data in the ResultSet into a List of lists
     */
    private List result = new ArrayList();
     /**
     * Name of the database table being displayed
     */
    private String tableName;
     public ResultSetTableModel(SQLFacade facade) {
        this.sqlFacade = facade;
    }
 // ----------------------------------------------------------------------------------------- //
// methods of javax.swing.table.TableModel interface
// ----------------------------------------------------------------------------------------- //
    public int getRowCount() {
        return rowCount;
    }
     public int getColumnCount() {
        if(rowCount > 0)
            return colCount+1;
        else
            return 0;
    }
     public Object getValueAt(int rowIndex, int columnIndex) {
        if(columnIndex == 0)
            return new Integer(rowIndex);
        else
            return ((ArrayList) result.get(rowIndex)).get(columnIndex-1);
    }
     public String getColumnName(int colIndex) {
        if(colIndex == 0)
            return "Row";
        else
            return (String) columnNames.get(colIndex-1);
    }
     /**
     * returns the list of column names
     */
    public List getColumnNames() {
        return columnNames;
    }
     /**
     *  reloads the TableModel with the contents of specified tableName
     * @param tableName table with which to reload the TableModel
     * @exception SQLException if a database error occurs
     * @exception ClassNotFoundException
     */
    public void reloadTableModel(String tableName) throws SQLException, ClassNotFoundException {
        ResultSet rs = null;
        this.tableName = tableName;
        try {
            rs = sqlFacade.executeQuery("SELECT * from " + tableName);
            clearAll();
            updateColumnModel(rs.getMetaData());
            while (rs.next()) {
                ArrayList list = new ArrayList();
                for (int i = 1; i <= colCount; i++) {
                    Object o = rs.getObject(i);
                    list.add(o);
                }
                result.add(list);
                rowCount++;
            }
             fireTableStructureChanged();
        } finally {
            if (rs != null) {
                try {
                    rs.close();
                } catch (SQLException se) {
                }
            }
        }
    }
     private void updateColumnModel(ResultSetMetaData metaData) throws SQLException, ClassNotFoundException {
        colCount = metaData.getColumnCount();
        for (int i = 1; i <= colCount; i++) {
            String colName = metaData.getColumnLabel(i);
            String colClass = metaData.getColumnClassName(i);
            columnTypes.add(Class.forName(colClass));
            columnNames.add(colName);
        }
    }
     /**
     * clears the TableModel
     */
    private void clearAll() {
        for (int i = 0; i < result.size(); i++) {
            ArrayList list = (ArrayList) result.get(i);
            list.clear();
        }
        result.clear();
        columnNames.clear();
        columnTypes.clear();
        colCount = 0;
        rowCount = 0;
    }
     public List getRow(int i) {
        return (List) result.get(i);
    }
 }