最近工作中要根据数据库查询出来的数据生成一棵XML树,虽然以前也了解一些别的XML的解析方法,但整体感觉dom4j在生成XML树方面更快捷简单。下面贴上部分代码,以供以后参考。
//对oracle树查询出来的结果进行处理生成XML树
private String getAllDepartments(String deptid){
Connection conn = dbPool.getConnection();
if(conn==null){
return "";
}
PreparedStatement ps = null;
ResultSet rs = null;
try{
String sql = "select so.orgid,so.parentdeptid,so.orgname,so.orgfullname,so.orgno from sys_orginfo so where so.orgusable!=-1 "+
"start with so.orgid=16 connect by prior so.orgid=so.parentdeptid order siblings by so.orgorder";
ps = conn.prepareStatement(sql);
rs = ps.executeQuery();
Document document=DocumentHelper.createDocument();
Element departsInfo=DocumentHelper.createElement("root");
document.setRootElement(departsInfo);
Element data = DocumentHelper.createElement("data");
document.getRootElement().add(data);
while(rs.next()){
int oid=rs.getInt("ORGID");
int pid=rs.getInt("PARENTDEPTID");
Element depart=DocumentHelper.createElement("org");
Element orgid=depart.addElement("orgid");
Element orgName=depart.addElement("orgname");
@SuppressWarnings("unused")
Element departs=depart.addElement("departs");
orgid.setText(oid+"");
orgName.setText(rs.getString("ORGNAME")+"");
if(oid==Integer.valueOf(deptid)){
departsInfo.element("data").add(depart);
}else{
makeTree(data,pid,depart);
}
}
departsInfo.addAttribute("code", "1");
departsInfo.addAttribute("message", "OK");
return toXML(document, "utf-8");
}catch(SQLException e){
LogUtil.errorRecord(e);
return "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\" ?><root code=\"-1\"/>";
}finally {
dbPool.dbClose(ps, rs);
if (conn != null) {
dbPool.closeConnection();
}
}
}
//生成XML树的递归算法
private static void makeTree(Element element,int pid,Element depart) {
List<?> elements = element.elements();
if(elements.size()>0) {
for (Iterator<?> it = elements.iterator(); it.hasNext();) {
Element elem = (Element) it.next();
if(elem.elementText("orgid").equals(pid+"")){
elem.element("departs").add(depart);
return;
}else if(elem.element("departs").nodeCount()>0){
makeTree(elem.element("departs"),pid,depart);
}
}
}
}
//解决XML转字符串中文可能出现乱码问题
public static String toXML(Document xmlDoc, String encoding)
{
OutputFormat format = OutputFormat.createPrettyPrint();
format.setEncoding(encoding);
StringWriter sw = new StringWriter();
XMLWriter xw = new XMLWriter(sw, format);
try {
xw.write(xmlDoc);
xw.flush();
xw.close();
} catch (IOException e) {
LogUtil.errorRecord(e);
if(xmlDoc.getRootElement().element("code")!=null){
return "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\" ?><root><code><value>-1</value></code></root>";
}else{
return "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\" ?><root code=\"-1\"/>";
}
}
return sw.toString();
}
中间主要用到方法:
createElement创建节点
addElement在指定节点上创建新节点
add将已创建好的节点添加到指定节点上
elem.element("departs")取elem节点下的departs节点
addAttribute为节点添加属性
值得注意的是如果使用documen.t.asXML()方法将document转换为字符串的时候如果有中文可能出现乱码导致XML错误,在IE里面访问错误提示是"缺少所需的空白区"得使用后面提供的toXML进行转码处理后就不会有乱码问题了。
以上的查询结构部门下面可能有子级部门,子级部门下面可能还有更多的子级部门,整个结果是一棵树,通过以上递归方法,可以实现一棵结构清晰的多层XML树。