testtreemodel.java

来自「jsf+spring+ibatis 的一个程序,主要用来介绍JSF程序.」· Java 代码 · 共 39 行

package com.wxd.common.popedom.presentation;

import java.util.Collection;
import java.util.Iterator;

import javax.faces.context.FacesContext;

import org.apache.myfaces.custom.tree.model.TreeModel;
import org.apache.myfaces.custom.tree.model.DefaultTreeModel;
import org.apache.myfaces.custom.tree.DefaultMutableTreeNode;

import com.wxd.common.popedom.persistence.entity.Module;
import com.wxd.common.popedom.service.ServiceFactory;

public class TestTreeModel {
    private TreeModel treeModel;

    public TreeModel getTreeModel() {
        if (treeModel == null) {
            DefaultMutableTreeNode root = new DefaultMutableTreeNode("腾建中青网络家园");
            Collection<Module> modules=ServiceFactory.getModuleService(FacesContext.getCurrentInstance()).getAllModules();
            Iterator<Module> it=modules.iterator();
            while(it.hasNext()){
            	Module module=it.next();
            	DefaultMutableTreeNode a = new DefaultMutableTreeNode(module.getModuleName().toString());
            	a.setUserObject(module);
            	root.insert(a);
            }
            treeModel = new DefaultTreeModel(root);
        }
        return treeModel;
    }

    public void setTreeModel(TreeModel treeModel) {
        this.treeModel = treeModel;
    }

}