categoryroottreenode.java

来自「基于Struts的客户资料管理系统」· Java 代码 · 共 70 行

package net.robin.crm.ui;


import java.util.Iterator;
import java.util.List;

import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.MutableTreeNode;

import net.robin.crm.service.CRMService;
import net.robin.crm.service.Category;
import net.robin.crm.service.Company;
import net.robin.crm.service.ConfigException;
import net.robin.crm.service.InstanceExeption;
import net.robin.crm.service.ServiceException;

public class CategoryRootTreeNode extends DefaultMutableTreeNode {

	CRMService service = null;
	
	public CategoryRootTreeNode() {
		super("分类");
	
		init();
	}

	private void init() {
		
		try {
			service = ServiceHandeler.getInstance().getService();
			
			List list = service.findCategories();
			
			for (Iterator i = list.iterator(); i.hasNext();) {
				Category category = (Category) i.next();
				add(createCategoryNode(category, service));
			}
			

		} catch (ConfigException e) {
			// TODO 自动生成 catch 块
			e.printStackTrace();
		} catch (InstanceExeption e) {
			// TODO 自动生成 catch 块
			e.printStackTrace();
		} catch (ServiceException e) {
			// TODO 自动生成 catch 块
			e.printStackTrace();
		}
		
		
	}

	private MutableTreeNode createCategoryNode(Category category, CRMService service) throws ServiceException {
		DefaultMutableTreeNode node = new DefaultMutableTreeNode(category);
		
		List companys =  service.findCompanies( category );
		
		for (Iterator i = companys.iterator(); i.hasNext();) {
			Company company = (Company) i.next();
			
			node.add(new DefaultMutableTreeNode(company, false));
			
		}
		return node;
	}
	

}