跟着汤阳光同志做一个OA系统（七）：树形处理（递归）

显示树形部门练习

package com.xsy.oa.test;

 

import java.util.ArrayList;

import java.util.Collection;

import java.util.LinkedHashSet;

import java.util.List;

import java.util.Set;

 

import org.junit.Test;

 

import com.xsy.oa.domain.Department;;

 

/**

 * 说明：不能使用多层循环的方式，因为需要能支持任意层。

 */

publicclass TreeViewPractice {

 

    /**

     * 练习一：打印所有顶层部门及其子孙部门的信息（名称）提示：假设有一个打印部门树的信息的方法

     *

     * 要求打印如下效果：

     *

     * <pre>

     * 市场部

     * 宣传部

     * 业务部

     * 业务一部

     * 业务二部

     * 开发部

     * 开发一部

     * 开发二部

     * </pre>

     */

    @Test

    publicvoid printAllDepts_1() {

       List<Department> topList = findTopLevelDepartmentList();

 

       // // 方式一

       // for (Department top :topList) {

       // showTree(top);

       // }

 

       // 方式二

       showTreeList(topList);

    }

 

    /**

     * 显示一颗部门树中所有节点的信息

     *

     * @param top

     *            树的顶点（根节点）

     */

    privatevoid showTree(Department top) {

       // 顶点

       System.out.println(top.getName());

       // 子树

       for (Department child : top.getChildren()) {

           showTree(child);

       }

    }

 

    /**

     * 显示多颗树的所有节点的信息

     *

     * @param topList

     */

    privatevoid showTreeList(Collection<Department> topList) {

       for (Department top : topList) {

           // 顶点

           System.out.println(top.getName());

           // 子树

           showTreeList(top.getChildren());

       }

    }

 

    /**

     * 练习二：打印所有顶层部门及其子孙部门的信息（名称），用不同的缩进表示层次（使用全角空格）。<br>

     * 子部门的名称前比上级部门多一个空格，最顶层部门的名字前没有空格。提示：假设有一个打印部门集合中所有部门信息的方法

     *

     * 要求打印如下效果：

     *

     * <pre>

     * ┣市场部

     *    ┣宣传部

     *    ┣业务部

     *       ┣业务一部

     *       ┣业务二部

     * ┣开发部

     *    ┣开发一部

     *    ┣开发二部

     * </pre>

     */

    @Test

    publicvoid printAllDepts_2() {

       List<Department> topList = findTopLevelDepartmentList();

 

       showTreeList_2(topList, "┣");

    }

 

    // 显示树

    privatevoid showTreeList_2(Collection<Department> topList,String prefix) {

       for (Department top : topList) {

           // 顶点

           System.out.println(prefix + top.getName());

           // 子树

           showTreeList_2(top.getChildren(), "    " + prefix);

       }

    }

 

    /**

     * 结构如下：

     *

     * <pre>

     * ┣市场部

     *    ┣宣传部

     *    ┣业务部

     *       ┣业务一部

     *       ┣业务二部

     * ┣开发部

     *    ┣开发一部

     *    ┣开发二部

     * </pre>

     *

     * @return所有最顶层的部门的列表

     */

    publicstatic List<Department> findTopLevelDepartmentList() {

       Department dept_1_1 = new Department();

       dept_1_1.setId(new Long(11));

       dept_1_1.setName("宣传部");

 

       Department dept_1_2 = new Department();

       dept_1_2.setId(new Long(12));

       dept_1_2.setName("业务部");

 

       Department dept_1_2_1 = new Department();

       dept_1_2_1.setId(new Long(121));

       dept_1_2_1.setName("业务一部");

 

       Department dept_1_2_2 = new Department();

       dept_1_2_2.setId(new Long(122));

       dept_1_2_2.setName("业务二部");

 

       dept_1_2_1.setParent(dept_1_2);

       dept_1_2_2.setParent(dept_1_2);

       Set<Department> children_0 = new LinkedHashSet<Department>();

       children_0.add(dept_1_2_1);

       children_0.add(dept_1_2_2);

       dept_1_2.setChildren(children_0);

 

       // ================================

 

       Department dept_1 = new Department();

       dept_1.setId(new Long(1));

       dept_1.setName("市场部");

 

       dept_1_1.setParent(dept_1);

       dept_1_2.setParent(dept_1);

       Set<Department> children_1 = new LinkedHashSet<Department>();

       children_1.add(dept_1_1);

       children_1.add(dept_1_2);

       dept_1.setChildren(children_1);

 

       // ---

 

       Department dept_2_1 = new Department();

       dept_2_1.setId(new Long(21));

       dept_2_1.setName("开发一部");

 

       Department dept_2_2 = new Department();

       dept_2_2.setId((new Long(22)));

       dept_2_2.setName("开发二部");

 

       Department dept_2 = new Department();

       dept_2.setId(new Long(2));

       dept_2.setName("开发部");

 

       dept_2_1.setParent(dept_2);

       dept_2_2.setParent(dept_2);

       Set<Department> children_2 = new LinkedHashSet<Department>();

       children_2.add(dept_2_1);

       children_2.add(dept_2_2);

       dept_2.setChildren(children_2);

 

       // ---

 

       List<Department> depts = new ArrayList<Department>();

       depts.add(dept_1);

       depts.add(dept_2);

       return depts;

    }

 

}

 

 

在Action中准备好经过修改名字的部门

publicclass DepartmentUtils {

 

    /**

     * 遍历部门树，把所有的部门遍历出来放到同一个集合中返回，并且其中所有部门的名称都修改了，以表示层次。

     *

     * @param topList，几个顶级部门

     * @return

     */

    publicstatic List<Department>getAllDepartments(List<Department> topList) {

       List<Department> list = new ArrayList<Department>();

       walkDepartmentTreeList(topList, "┣", list);

       return list;

    }

 

    /**

     * 遍历部门树，把遍历出的部门信息放到指定的集合中

     *

     * @param topList

     */

    privatestaticvoid walkDepartmentTreeList(Collection<Department>topList, String prefix, List<Department> list) {

       for (Department top : topList) {

           // 顶点

           Department copy = new Department(); // 使用副本，因为原对象在Session中

           copy.setId(top.getId());

           copy.setName(prefix + top.getName());

           list.add(copy); // 把副本添加到同一个集合中

 

           // 子树

           walkDepartmentTreeList(top.getChildren(), "" + prefix, list); // 使用全角的空格

       }

    }