利用mybatis的collection来实现递归查询,id、pid结构递归的另外一种便捷实现

递归在程序设计与实现中非常重要,比如我们的部门,包含子部门,子部门又有子部门,再比如,我们的菜单,有子菜单,子菜单可能还要包含子菜单。当然还有人员的管理中,上线发展下线,下线再发展下线,等等,这些都是非常常见的需要用到递归场景。
通常的做法上,我们可以通过Java来写递归,通过定义递归函数来实现。递归函数通过多次执行相同的语句,通过id,pid这样索引,最后将数据组织成我们期待的样子。因为需要递归的数据量一般不会特别巨大,层级也是有限的,所以性能和效率上也并无明显问题,大不了可以提前缓存,加快读取的速度。都是可以解决的。今天介绍的方法,另辟蹊径,更简洁,更优雅,更容易理解。

来源是B站某视频。我是学习后转述方法。

1.试验结构与数据准备

我们利用id与pid的设计在数据和结构上承载这种需要递归的逻辑关系。

CREATE TABLE `t_department` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `depName` varchar(64) DEFAULT NULL,
  `parentId` int(11) DEFAULT NULL,
  `depPath` varchar(255) NOT NULL,
  `enabled` tinyint(4) DEFAULT '1',
  `isParent` tinyint(4) DEFAULT '0',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=14 DEFAULT CHARSET=utf8mb4;

/*试验数据 */

insert  into `t_department`(`id`,`depName`,`parentId`,`depPath`,`enabled`,`isParent`) values (1,'股东会',-1,'.1',1,1),(2,'董事会',1,'.1.2',1,1),(3,'总办',2,'.1.2.3',1,1),(4,'财务部',3,'.1.2.3.4',1,0),(5,'市场部',3,'.1.2.3.5',1,1),(6,'华东市场部',5,'.1.2.3.5.6',1,0),(7,'华南市场部',5,'.1.2.3.5.7',1,0),(8,'上海市场部',5,'.1.2.3.5.8',1,0),(9,'西北市场部',5,'.1.2.3.5.9',1,0),(10,'贵阳市场',10,'.10',1,1),(11,'乌当区市场',10,'.10.11',1,0),(12,'技术部',3,'.1.2.3.12',1,0),(13,'运维部',3,'.1.2.3.13',1,0);

2.准备表结构对应的java类。

/**
 * t_department
 * @author jy@cjlu.edu.cn
 */
@Data
@Accessors(chain = true)
@ApiModel(value = "Department", description = "部门管理")
@Table(name = "t_department")
public class Department implements Serializable {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    @ApiModelProperty(value = "id")
    private Integer id;
    @ApiModelProperty(value = "部门名称")
    private String depname;
    @ApiModelProperty(value = "父id")
    private Integer parentid;
    @ApiModelProperty(value = "路径")
    private String deppath;
    @ApiModelProperty(value = "是否启用")
    private Byte enabled;
    @ApiModelProperty(value = "是否是父节点")
    private Byte isparent;

    @Transient
    @ApiModelProperty(value = "子节点")
    List<Department> childrens;
    private static final long serialVersionUID = 1L;
}

这里特别注意的是我们增加了一个childrens用来装配子节点信息,这里就是一般递归的重点。

3.mybatis的xml文件的配置。

这里重点是我们利用resultMap,而不是直接用resultType来承载查询结果的数据。

核心的要点是对于resultMap的定义,我们通过collection方式来对childrens进行赋值,这里我们再次调用dao中定义的getAllDepartments函数,传递id作为入口参数。getAllDepartments函数将重新执行该xml配置的getAllDepartments的select语句。这样就实现了递归。
实现上非常简洁。以下是xml的定义。

<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.mtttt.yeb.dao.DepartmentDao">
    <resultMap id="BaseResultMap" type="cn.mtttt.yeb.pojo.Department">
        <id column="id" jdbcType="INTEGER" property="id"/>
        <result column="depName" jdbcType="VARCHAR" property="depname"/>
        <result column="parentId" jdbcType="INTEGER" property="parentid"/>
        <result column="depPath" jdbcType="VARCHAR" property="deppath"/>
        <result column="enabled" jdbcType="TINYINT" property="enabled"/>
        <result column="isParent" jdbcType="TINYINT" property="isparent"/>
    </resultMap>
    <sql id="Base_Column_List">        
id,depName,parentId,depPath,enabled,isParent
    </sql>
    <resultMap id="departMents" type="cn.mtttt.yeb.pojo.Department" extends="BaseResultMap">
       <collection property="childrens" ofType="cn.mtttt.yeb.pojo.Department" select="cn.mtttt.yeb.dao.DepartmentDao.getAllDepartments" column="id"/>
    </resultMap>
    <select id="getAllDepartments" resultMap="departMents">
        select
        <include refid="Base_Column_List"/>
        from t_department
        where parentId= #{parentId}
    </select>
</mapper>

以上xml配置文件的内容中有些涉及mybatis的基础使用问题,我就不在这里详细加以说明了,相信有些基础的小伙伴一眼就能看出来那些地方是需要在自己的实验中更换掉的。根据自己的爱好,具体可以使用mybatis或者通用mapper或者mybatis plus等实现。

4.实现的效果

我们使用一个测试来跑一下。
@Test
public void test3(){
List<Department> allDepartments = departmentDao.getAllDepartments(-1);
System.out.println(JSON.toJSONString(allDepartments));
}
最后呈现的效果是每一个节点,包括这个节点的子节点,通过一个函数完整构造出来。。

    "childrens": [{
        "childrens": [{
            "childrens": [{
                "childrens": [],
                "depname": "财务部",
                "deppath": ".1.2.3.4",
                "enabled": 1,
                "id": 4,
                "isparent": 0,
                "parentid": 3
            }, {
                "childrens": [{
                    "childrens": [],
                    "depname": "华东市场部",
                    "deppath": ".1.2.3.5.6",
                    "enabled": 1,
                    "id": 6,
                    "isparent": 0,
                    "parentid": 5
                }, {
                    "childrens": [],
                    "depname": "华南市场部",
                    "deppath": ".1.2.3.5.7",
                    "enabled": 1,
                    "id": 7,
                    "isparent": 0,
                    "parentid": 5
                }, {
                    "childrens": [],
                    "depname": "上海市场部",
                    "deppath": ".1.2.3.5.8",
                    "enabled": 1,
                    "id": 8,
                    "isparent": 0,
                    "parentid": 5
                }, {
                    "childrens": [],
                    "depname": "西北市场部",
                    "deppath": ".1.2.3.5.9",
                    "enabled": 1,
                    "id": 9,
                    "isparent": 0,
                    "parentid": 5
                }],
                "depname": "市场部",
                "deppath": ".1.2.3.5",
                "enabled": 1,
                "id": 5,
                "isparent": 1,
                "parentid": 3
            }, {
                "childrens": [],
                "depname": "技术部",
                "deppath": ".1.2.3.12",
                "enabled": 1,
                "id": 12,
                "isparent": 0,
                "parentid": 3
            }, {
                "childrens": [],
                "depname": "运维部",
                "deppath": ".1.2.3.13",
                "enabled": 1,
                "id": 13,
                "isparent": 0,
                "parentid": 3
            }],
            "depname": "总办",
            "deppath": ".1.2.3",
            "enabled": 1,
            "id": 3,
            "isparent": 1,
            "parentid": 2
        }],
        "depname": "董事会",
        "deppath": ".1.2",
        "enabled": 1,
        "id": 2,
        "isparent": 1,
        "parentid": 1
    }],
    "depname": "股东会",
    "deppath": ".1",
    "enabled": 1,
    "id": 1,
    "isparent": 1,
    "parentid": -1
}]

至此我们实现了利用mybatis自身一对多配置collection功能,通过重复调用dao层的函数达到递归的目的。在代码层面非常简洁。而且易于理解。
从数据库这边做递归,我觉得目前这个方法最为简洁,所以在此记录,并做分享。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,294评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,493评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,790评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,595评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,718评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,906评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,053评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,797评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,250评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,570评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,711评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,388评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,018评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,796评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,023评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,461评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,595评论 2 350

推荐阅读更多精彩内容