tbmodel.java

网上虚拟社区

import javax.swing.*;
import javax.swing.table.*;
import java.util.Vector;
import java.sql.*;

public class TbModel extends AbstractTableModel{
       private static final long serialVersionUID = 1L;

       private Vector data;   

       private Vector titles;

       Connection connect;

       Statement state;

       ResultSet rs;

       public TbModel()
       {
              data=new Vector();
              try
              {
                   Class.forName("sun.jdbc.odbc.JdbcOdbcDriver");
                   connect =DriverManager.getConnection( "jdbc:odbc:BankDB");
                   state=connect.createStatement();
                   rs=state.executeQuery("select * from Accounts");
                   while(rs.next())
                   {     
                          data.add(rs.getString("Num"));
                          data.add(rs.getDouble("Balance"));
                          data.add(rs.getString("Guest"));
                          data.add(rs.getString("Address"));
                          data.add(rs.getString("pin"));
                   }
               }
               catch(ClassNotFoundException e)
               {
                         System.out.println("JDBC driver not loaded:\n"+e);
                }
                catch(SQLException e)
                {
                         System.out.println("Connection failed:\n"+e);
                 }
                 titles = new Vector();
                 titles.add("Num");
                 titles.add("Balance");
                 titles.add("Guest");
                 titles.add("Address");
                 titles.add("pin");
       }
       public TbModel(Vector data,Vector titles)
       {
              this.data=data;
              this.titles = titles;
       }
       
       public int getRowCount() 
       {
              return data.size()/getColumnCount();
       }
       public int getColumnCount(){
              return  titles.size();
       }
       public Object getValueAt(int rowIndex, int columnIndex){     
              return data.get((rowIndex*getColumnCount())+columnIndex);
       }
       public String getColumnName(int columnIndex) {
              String colName = new String();
                if (columnIndex<=getColumnCount())
                 colName=(String)titles.get(columnIndex);
                return colName;

       }

}