[学习笔记]小型论坛功能——实现树状结构的主题贴显示的类[2]

 //article.java

//存放贴子信息

package com.bbs.model;

/**
 *
 * @author he
 * 用来建立树状的主题显示贴的各属性
 * 本类个个变量都是数据库中bbs.sql表中的每一列
 */
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Date;

public class Article {
 private int id;        //本贴id 
 private int pid;       //父贴id
 private int rootid;    //根贴id
 private String title;  //标题
 private String cont;   //内容
 private Date pdate;    //发贴日期
 private boolean isLeaf;//是否是叶子结点 true[后面还有跟贴] 　false[无跟贴]
 private int grade;     //贴子级别

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public int getPid() {
        return pid;
    }

    public void setPid(int pid) {
        this.pid = pid;
    }

    public int getRootid() {
        return rootid;
    }

    public void setRootid(int rootid) {
        this.rootid = rootid;
    }

    public String getTitle() {
        return title;
    }

    public void setTitle(String title) {
        this.title = title;
    }

    public String getCont() {
        return cont;
    }

    public void setCont(String cont) {
        this.cont = cont;
    }

    public Date getPdate() {
        return pdate;
    }

    public void setPdate(Date pdate) {
        this.pdate = pdate;
    }

    public boolean isIsLeaf() {
        return isLeaf;
    }

    public void setIsLeaf(boolean isLeaf) {
        this.isLeaf = isLeaf;
    }

    public int getGrade() {
        return grade;
    }

    public void setGrade(int grade) {
        this.grade = grade;
    }
   
      //.....对from和resultSet
  //封装
  public void initFormRs(ResultSet rs){
   try{
   setId(rs.getInt("id"));
   setPid(rs.getInt("pid"));
   setRootid(rs.getInt("rootid"));
   setTitle(rs.getString("title"));
   setCont(rs.getString("cont")); //首页不用内容
   setPdate(rs.getTimestamp("pdate"));
   setIsLeaf(rs.getInt("isLeaf") == 0?true:false);
   setGrade(0);
   }catch(SQLException e){
    e.printStackTrace();
   }
  }
 
}

 

//arcticManager.java

管理贴子（显示树状的类）

package com.bbs.model;

import com.bbs.util.DB;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 *
 * @author he
 * 用来完成对Article内属性的操作。
 * 实现如何树状显示的方法。
 *
 */
public class ArticleManager {
   
    /**
     * 显示级别树
     *
     *
     * private void tree(List<Article> articles,
     *                          Connection conn,int id,int grade)
     * 参数说明：
     * articles 为Article的对象
     * id       表示帖子id号
     * grade    id对应的级别
     *
     * 思路为：
     * [1]列出pid=参数 mid 的结果集[用SQL语句查]
     * [2]将结果集内所有的元组放入到Article对象中
     * [3] 判断结果集每一行中的 isleaf列的值
     *          if  是0
     *              则表示并非叶子，就需要查找数据表中
     *              其pid=当前id[设为nid]的元组
     *              再调用此方法，将nid传给参数id,grade+1传给参数grade
     *              重复执行[1][2]
     *        else 
     *              什么也不用做
     */
  public static void showTree(List<Article> article,Connection conn,int id,int grade){
       String sql = "select * from article where pid= " + id; //id为int类型不用'
          
            Statement stmt = null;
            ResultSet rs = null;
      try {
             conn = DB.getConnection();
            stmt = DB.getStmt(conn);
            rs = DB.executeQuery(stmt, sql);
            while (rs.next()) {
                Article a=new Article();
                a.initFormRs(rs);
                a.setGrade(grade);
                article.add(a);
                if(!a.isIsLeaf()){
                   showTree(article,conn,a.getId(),grade+1);
                }
            }
        } catch (SQLException ex) {
            Logger.getLogger(ArticleManager.class.getName()).log(Level.SEVERE, null, ex);
        }finally{
        DB.close(null,stmt,rs);
        }
  }
  /**
   * 测试方法的正确性
   */
 
 public  static void main(String args[]){
  List<Article> list = new ArrayList<Article>();
  Connection conn=DB.getConnection();
  ArticleManager.showTree(list, conn, 0, 0);
  DB.close(conn,null, null);
  for(int i=0;i<list.size();i++){
     String ss=list.get(i).getCont();
     System.out.println(ss);

  }
 }
}