BuildingTreeMaker

package com.zx.services;

import java.util.Iterator;
import java.util.List;

import com.zx.pojos.UserBean;
import com.zx.views.tree.TreeNode;

public class BuildingTreeMaker
{

public static TreeNode makerTree ( )
{
UserInfoDao dao = new UserInfoDao();
int cnt = dao.getTotalCount();
UserBean[] users = dao.getUsers(0, cnt);
TreeNode p = new TreeNode("", "");
if ( users == null || users.length == 0 ) return p;
for ( int i = 0 ; i < users.length ; i++ )
{
UserBean user = users[i];
String buildname = user.getBuilding();
TreeNode node = p.findChild(buildname);
TreeNode houseNoNode = new TreeNode(String.valueOf(user.getId()),
user.getHouseno());
if ( node == null )
{
TreeNode buildingNode = new TreeNode(String.valueOf(user
.getId()), user.getBuilding());
p.addChild(buildingNode);
buildingNode.addChild(houseNoNode);
}
else
{
node.addChild(houseNoNode);
}
}
return p;
}

public static void print ( TreeNode tn )
{
if ( tn.isLeaf() )
{
System.out.println(" " + tn.getKey() + "-" + tn.getName());
}
else
{
//System.out.println(tn.getKey() + "-" + tn.getName());
List<TreeNode> children = tn.getChildren();
for ( Iterator<TreeNode> it = children.iterator() ; it.hasNext() ; )
{
print(it.next());
}
}
}

public static void main ( String[] a )
{
TreeNode t = BuildingTreeMaker.makerTree();
BuildingTreeMaker.print(t);
}

}