根据数据库中取记录自定义一棵树结构

在servlet/jsp开发中，我们可能也会涉及到树结构的定义问题，比如从数据库中取出记录并定义一个XML树用于论坛结构的显示等。
因为是servlet，使用JAVA SWING中提供的JTree并不现实。

其实一棵实现基本功能的树并不复杂，所以我们自已动手。

首先，在数据库的结构一般是这样的几个字段不能少：

int id   节点id
int pid  父节点id
String title 显示字串
......

(其实前两个节点维持着树的结构是最重要的，后面你想怎么加都行了......)

然后我们来定义一个树节点类，代码如下：

/**
* 自定义的树结点
* @author:sharetop([email protected])
*/
import java.util.*;

class CTreeNode
{
  int id;       /*点ID号，对应数据库相应字段*/
 String title=null; 
 String href=null;
 
 CTreeNode parent=null; /*父节点对象*/
 Vector allChild;    /*保存了全部子节点*/
 
 /**
 *构造
 */
 CTreeNode()
 {
  allChild=new Vector();
 }
 
 /**
 *加子节点
 */
 void addChild(CTreeNode child)
 {
  allChild.add(child);
 }

 /**
 * 根据ID递归查找子节点，返回节点对象
 */
 CTreeNode findChild(int id)
 {
  if( id==this.id ) return this;
  for( Enumeration e = allChild.elements();e.hasMoreElements();)
  {
   CTreeNode temp = (CTreeNode)e.nextElement();
   if( temp.id==id ) return temp;
   CTreeNode tmp = temp.findChild(id);
   if( tmp!=null )return tmp;
  }
  return null;
 }
 
 /**
 * 生成一个XML字串
 */
 String create()
 {
  StringBuffer result= new StringBuffer();
  result.append("<node title=\"");
  result.append(title);
  result.append("\" href=\"");
  result.append(href);
  result.append("\" >");
  if( allChild.size()!=0 ){
    for( Enumeration e = allChild.elements();e.hasMoreElements();) {
     CTreeNode temp = (CTreeNode)e.nextElement();
     result.append(temp.create());
    }
  }
  result.append("</node>");
  return result.toString();
 }
 
}

 

 

好了，然后就是我们在servlet中如何使用这个类了：

在我们的TestServlet.java中加一个方法：

private String getXMLResult()
{
 StringBuffer result = new StringBuffer();
 
 /*这里是树根*/
 CTreeNode root = new CTreeNode();
 root.id=0;
 root.title="根目录";
 root.href="";
 
 try{
  Connection con = dbpool.getConnection();
  PreparedStatement ps = con.prepareStatement("select * from block order by id");
  ResultSet rs = ps.executeQuery();
  
  while( rs.next() ){
   /*每一条记录对应一个节点，关键是找到它在树上所处的位置*/
   CTreeNode node = new CTreeNode();
   node.id=rs.getInt("id");
   node.title=rs.getString("title");
   node.parent=root.findChild(rs.getInt("pid"));
   node.href=node.title+".html";
   node.parent.addChild(node);
  }    
  rs.close();
  ps.close();
  con.close();
  dbpool.freeConnection(con);
 }
 catch( Exception ex )
 {
  System.err.println("Error in Update - SQLBean : \r\n");
  ex.printStackTrace(System.err);
  return "";
 }
 return root.create();
 }

就是这么简单，TestServlet的doGet方法我也给在下面吧

  public void doGet(HttpServletRequest req,HttpServletResponse resp) throws ServletException, IOException
  {
 resp.setContentType("text/xml;charset=gb2312");
 PrintWriter out = resp.getWriter(); 
         
  out.println("<?xml version=\"1.0\" encoding=\"gb2312\" ?>");
 
 out.println(getResult());  
   
  out.flush();
  out.close();
 }

有兴趣可以测试一下。

这只是一个基本的树结构，如果要让它符合你的要求，你需要对它进行扩充，欢迎与我讨论 [email protected]

 

本文地址：http://com.8s8s.com/it/it18481.htm