treemodel.java

来自「这是我在北大青鸟accp4.0 第一学期 的java 项目<员工管理系统&」· Java 代码 · 共 86 行

package com.accp.model;

import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.sql.ResultSet;

import javax.swing.JTable;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;

import com.accp.db.DBManager;

/**
 * 树模型.
 * @author Administrator
 *
 */
public class TreeModel {
	ResultSet rsPar;
	ResultSet rsSub;
	DBManager db;
	
	public static TreeModel getTreeMod(){
		return new TreeModel();
	}
	/**
	 * 得到类别的树模型
	 * @return root
	 */
	public DefaultTreeModel getTreeModel(){
		db=new DBManager();
		String sqlPar="select DISTINCT catename from category";
		
		DefaultMutableTreeNode root=new DefaultMutableTreeNode("类别管理");
		try {
			rsPar=db.getRs(sqlPar);
			while(rsPar.next()){
				
				String bigName=rsPar.getString(1).trim();				
				DefaultMutableTreeNode bigNames=new DefaultMutableTreeNode(bigName);
				
				String sqlSub="select subcatename from category where catename='"+bigName+"'";
				rsSub=db.getRs(sqlSub);
				
				while(rsSub.next()){
					bigNames.add(new DefaultMutableTreeNode(rsSub.getString(1).trim()));
				}
				root.add(bigNames);
			}
		} catch (Exception e) {
			e.printStackTrace();
		}	
		
		
		return new DefaultTreeModel(root);
	}

	public DefaultTreeModel getTreeModel2(){
		db=new DBManager();
		String sqlPar="select DISTINCT catename from category";
		
		DefaultMutableTreeNode root=new DefaultMutableTreeNode("类别管理");
		try {
			rsPar=db.getRs(sqlPar);
			while(rsPar.next()){
				
				String bigName=rsPar.getString(1).trim();				
				DefaultMutableTreeNode bigNames=new DefaultMutableTreeNode(bigName);
				
				String sqlSub="select subcatename from category where catename='"+bigName+"'";
				rsSub=db.getRs(sqlSub);
				
				while(rsSub.next()){
					bigNames.add(new DefaultMutableTreeNode(rsSub.getString(1).trim()));
				}
				root.add(bigNames);
			}
		} catch (Exception e) {
			e.printStackTrace();
		}	
		
		
		return new DefaultTreeModel(root);
	}
}