clienttreereader.java

用java开发的一个企业的DRP系统源码

package com.bjsxt.drp.basedata;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import com.bjsxt.drp.util.DB;

public class ClientTreeReader {
	
	//树的html字符串
	private StringBuffer sbfTree = new StringBuffer();
	
	/**
	 * 读取树
	 *
	 */
	public void read() {
		Connection conn = null;
		try {
			conn = DB.getConn();
			read(conn, 0, 0);
		}catch(SQLException e) {
			e.printStackTrace();
		}finally {
			DB.closeConn(conn);
		}
	}
	
	/**
	 * 递归读取
	 * @param conn 数据库连接
	 * @param id 
	 * @param level 
	 * @throws SQLException
	 */
	private void read(Connection conn, int id, int level) 
	throws SQLException {
		level++;
		String sql = "select * from t_client where pid=?";
		PreparedStatement pstmt = null;
		ResultSet rs = null;
		try {
			pstmt = conn.prepareStatement(sql);
			pstmt.setInt(1, id);
			rs = pstmt.executeQuery();
			while (rs.next()) {
				sbfTree.append("<div>");
				sbfTree.append("\n");
				for (int i = 0; i < level - 1; i++) {
					sbfTree.append("<img src=\"../images/white.gif\">");
				}
				if (rs.getString("is_leaf").equals("N")) {
					sbfTree.append("<img alt=\"展开\" style=\"cursor:hand;\" onClick=\"display('" + rs.getInt("id") + "');\" id=\"img" + rs.getInt("id") + "\" src=\"../images/plus.gif\">");
					sbfTree.append("\n");
					sbfTree.append("<img id=\"im" + rs.getInt("id") + "\" src=\"../images/closedfold.gif\">");
					sbfTree.append("\n");
					sbfTree.append("<a href=\"client_node_crud.jsp?id=" + rs.getInt("id") + "\" target=\"clientDispAreaFrame\">" + rs.getString("name") + "</a>");
					sbfTree.append("\n");
					sbfTree.append("<div style=\"display:none;\" id=\"div" + rs.getInt("id") + "\">");
					sbfTree.append("\n");
					read(conn, rs.getInt("id"), level);
					sbfTree.append("</div>");
					sbfTree.append("\n");
				}else {
					sbfTree.append("<img src=\"../images/minus.gif\">");
					sbfTree.append("\n");
					sbfTree.append("<img src=\"../images/openfold.gif\">");
					sbfTree.append("\n");
					if (rs.getString("is_client").equals("Y")) {
						sbfTree.append("<a href=\"client_crud.jsp?id=" + rs.getInt("id") +  "\" target=\"clientDispAreaFrame\">" + rs.getString("name") + "</a>");
					}else {
						sbfTree.append("<a href=\"client_node_crud.jsp?id=" + rs.getInt("id") +  "\" target=\"clientDispAreaFrame\">" + rs.getString("name") + "</a>");						
					}
					sbfTree.append("\n");
				}
				sbfTree.append("</div>");
				sbfTree.append("\n");
			}
		}finally {
			DB.closeRs(rs);
			DB.closeStmt(pstmt);
		}
	}
	
	public String getTreeString() {
		return sbfTree.toString();
	}
}