treetag.java

来自「人力资源管理系统」· Java 代码 · 共 59 行

package com.accphr.taglib;

import java.util.Iterator;

import javax.servlet.http.HttpSession;
import javax.servlet.jsp.JspException;
import javax.servlet.jsp.tagext.TagSupport;

import com.accphr.entity.Rights;
import com.accphr.entity.Users;

/**
 * 自定义标签定义类
 * @author zhuohai
 *
 */
public class TreeTag extends TagSupport {

	@Override
	public int doEndTag() throws JspException {
		//获取当前登录用户
		HttpSession session=this.pageContext.getSession();
		Users us=(Users)session.getAttribute("loginuser");
		
		//创建树形菜单
		StringBuffer b=new StringBuffer();
		b.append("<a href='javascript: tree.openAll();'>打开所有</a> | <a href='javascript: tree.closeAll();'>关闭所有</a>");
		b.append("<script>\r\n");
		b.append("tree=new dTree('tree');\r\n");
		b.append("tree.add(0,-1,'青鸟人力资源管理系统','','青鸟人力资源管理系统');");
		//循环添加权限功能到树形菜单
		for(Iterator it=us.getRole().getRightses().iterator();it.hasNext();){
			Rights r=(Rights)it.next();
			b.append("tree.add(");
			b.append(r.getRightId()+",");
			b.append(r.getRightParentId()+",");
			b.append("'"+r.getRightText()+"',");
			b.append("'"+r.getRightUrl()+"',");
			b.append("'"+r.getRightTitle()+"',");
			b.append("'mainFrame'");
			b.append(");\r\n");
			
		}
		b.append("document.write(tree);");
		b.append("</script>\r\n");


		//输出到浏览器
		try {
			this.pageContext.getOut().print(b.toString());
		} catch (Exception e) {
			e.printStackTrace();
		}
		
		return this.EVAL_PAGE;
	}

}