datamap.java~1~

javaGUI编写的人事管理系统(毕业学生答辩可参考)

package project;

import java.util.*;
import java.sql.ResultSet;
import java.sql.*;

/**
 * <p>Title: </p>
 *
 * <p>Description: </p>
 *
 * <p>Copyright: Copyright (c) 2006</p>
 *
 * <p>Company: </p>
 *
 * @author not attributable
 * @version 1.0
 */
public class DataMap
    extends HashMap {
  private ResultSet rs = null;
  private Object[][] Data = null;
  private ArrayList[] col_list = null;
  private String[] col_name = null;
  private int col_num = 0;

  public DataMap(ResultSet rs) {
    this.rs = rs;
    if(this.rs != null){
      rs_to_DataMap();
    }
  }

  private void rs_to_DataMap() {

    try {
      //得到列名和列的数目,存放到 col_name[]中 和 col_num 中
      ResultSetMetaData rsmd = this.rs.getMetaData();
      col_num = rsmd.getColumnCount();
      col_name = new String[col_num];
      for (int i = 0; i < col_num; i++) {
        col_name[i] = rsmd.getColumnName(i + 1);
      }
      //创建和列数相等的ArrayList数组，用于存放每一列的数据
      col_list = new ArrayList[col_num];

      for(int i = 0; i<col_num; i++){
        col_list[i] = new ArrayList();
      }


      //开始向ArrayList数组添加数据
      while (this.rs.next()) {
        for (int i = 0; i < col_num; i++) {
          col_list[i].add(this.rs.getObject(i + 1).toString());
        }
      }

      //把得到的ArrayList数组添加到datamap中去
      for (int i = 0; i < col_num; i++) {
        this.put(col_name[i], col_list[i]);
      }

    }
    catch (SQLException ex) {
      ex.printStackTrace();
    }

  }

//返回列数
public int get_DataMap_column(){
  return this.col_num;
}

//返回行数
public int get_DataMap_row(){
  if(this.col_num > 0){
    return col_list[0].size();
  }else{
    return 0;
  }
}

public String[] get_Column_Title(){
  return this.col_name;
}

//把Datamap的对象变为二维一个对象数组，并返回
public Object[][] get_DataMap_Array(){
  int col = this.get_DataMap_column();
  int row = this.get_DataMap_row();
  Data = new Object[row][col];
  ArrayList temp = null;
  for(int i = 0; i < col; i++){
    temp = (ArrayList)this.get(col_name[i]);
    for(int j = 0; j < row; j++){
      Data[j][i] = temp.get(j);
    }
  }
  return Data;
}
}