读取文件夹生成目录树

// model
import java.util.Set;

public class Node {

// 标识符
private int id;

// 节点名称
private String name;

// 层次
private int level;

// 是否叶子节点
private boolean leaf;

// 父节点 * --- 1
private Node parent; // 一的一端

// 子节点 1 --- *
private Set<Node> children; // 多的一端

public int getId() {
return id;
}

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

public boolean isLeaf() {
return leaf;
}

public void setLeaf(boolean leaf) {
this.leaf = leaf;
}

public int getLevel() {
return level;
}

public void setLevel(int level) {
this.level = level;
}

public String getName() {
return name;
}

public void setName(String name) {
this.name = name;
}

public Node getParent() {
return parent;
}

public void setParent(Node parent) {
this.parent = parent;
}

public void setChildren(Set<Node> children) {
this.children = children;
}

public Set<Node> getChildren() {
return children;
}
}
// Node.hbm.xml
<hibernate-mapping>
<class name="com.bjsxt.hibernate.Node" table="t_node">
<id name="id">
<generator class="native"/>
</id>
<property name="name"/>
<property name="level"/>
<property name="leaf"/>
<many-to-one name="parent" column="pid"/>
<set name="children" lazy="extra" inverse="true">
<key column="pid"/>
<one-to-many class="com.bjsxt.hibernate.Node"/>
</set>
</class>
</hibernate-mapping>
import junit.framework.TestCase;

public class NodeManagerTest extends TestCase {

public void testCreateTree() {
NodeManager.getInstance().createTree("D:\\hibernate_basemapping\\");
}

public void testPrintTree() {
NodeManager.getInstance().printTree(1);
}
}


import java.io.File;
import java.util.Iterator;
import java.util.Set;

import org.hibernate.Session;

public class NodeManager {

private static NodeManager nodeManager;

private NodeManager() {
}

public static synchronized NodeManager getInstance() {
if (nodeManager == null) {
nodeManager = new NodeManager();
}

return nodeManager;
}

// 创建树型结构
public void createTree(String dir) {
Session session = HibernateUtils.getSession();

try {
session.beginTransaction();

File root = new File(dir); // 创建文件目录
saveTree(root, session, null, 0);

session.getTransaction().commit();
} catch (RuntimeException e) {
e.printStackTrace();
session.getTransaction().rollback();
} finally {
HibernateUtils.closeSession(session);
}
}

// 递归创建一棵树
private void saveTree(File file, Session session, Node parent, int level) {

if (file == null || !file.exists()) {
return;
}

boolean isLeaf = file.isFile();

Node node = new Node();
node.setName(file.getName());
node.setLevel(level);
node.setParent(parent);
node.setLeaf(isLeaf);
session.save(node);

File[] subs = file.listFiles();
if (subs != null && subs.length > 0) {
for (int i = 0; i < subs.length; i++) {
saveTree(subs[i], session, node, level + 1);
}
}
}

public void printTree(int id) {
Session session = HibernateUtils.getSession();

try {
session.beginTransaction();

Node root = (Node) session.load(Node.class, id);
printNode(root);

session.getTransaction().commit();
} catch (RuntimeException e) {
e.printStackTrace();
session.getTransaction().rollback();
} finally {
HibernateUtils.closeSession(session);
}
}

private void printNode(Node node) {

if (node == null) {
return;
}
int level = node.getLevel();
if (level > 0) {
for (int i = 0; i < level; i++) {
System.out.print(" |");
}
System.out.print("--");
}

System.out.println(node.getName()
+ (node.isLeaf() ? "" : "[" + node.getChildren().size() + "]"));

Set<Node> children = node.getChildren();
for (Iterator<Node> iter = children.iterator(); iter.hasNext();) {
Node child = iter.next();
printNode(child);
}
}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值