JPA

JPA 单表双向自关联、转dto、jackson序列化循环引用问题

November 19, 2020

首先看实体,Organization

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package com.example.organization.domain.repo.po;

import lombok.Getter;
import lombok.Setter;
import org.hibernate.annotations.GenericGenerator;

import javax.persistence.*;
import java.util.Objects;
import java.util.Set;

@Entity
@Table(name = "organization")
@Getter
@Setter
public class OrganizationPo {
    @Id
    @GenericGenerator(name = "idGenerator", strategy = "uuid")
    @GeneratedValue(generator = "idGenerator")
    @Column(name = "org_id")
    private String orgId;
@Column(name = "org_name")
private String orgName;
 
@ManyToOne
@JoinColumn(name = "parent_id", referencedColumnName = "org_id")
private OrganizationPo parent;
 
@OneToMany(mappedBy = "parent")
private Set<OrganizationPo> children;
 
@Override
public boolean equals(Object o) {
    if (this == o) {
        return true;
    }
    if (o == null || getClass() != o.getClass()) {
        return false;
    }
    OrganizationPo that = (OrganizationPo) o;
    return orgId.equals(that.orgId);
}
 
@Override
public int hashCode() {
    return Objects.hash(orgId);
	}
}

这是一个双向自关联的实体。表数据如下:

对于 @OneToMany 默认fetch策略是 Lazy,也就是在实际访问到 children 属性时才会发出数据库请求,比如你打了断点调试,查看 children 属性时,或者 @ResponseBody 序列化到前端时才会发出新的查询。

注意不要使用 lombok 的 @Data,会有带上所有字段的 toString() 方法,此时debug的时候会因为调用 toString() 而导致循环引用从而抛出 stackoverflowError,使用默认的 toString() 就好,默认的 toString() 采用的是 hashCode 的16进制。

接下来需要做 po(persistence object)到do(domain object)、do到dto(data transfer object) 的转换,此时依旧会出现循环引用、堆栈溢出的问题。do实体如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package com.example.organization.domain.aggregate;

import lombok.Getter;
import lombok.Setter;

import java.util.Objects;
import java.util.Set;

@Getter
@Setter
public class Organization {
    private String orgId;
    private String orgName;
    private Organization parent;
    private Set<Organization> children;
@Override
public boolean equals(Object o) {
    if (this == o) {
        return true;
    }
    if (o == null || getClass() != o.getClass()) {
        return false;
    }
    Organization that = (Organization) o;
    return orgId.equals(that.orgId);
}
 
@Override
public int hashCode() {
    return Objects.hash(orgId);
	}
}

我采用的 bean mapping 工具是 mapstruct ,各方面都较优的一个映射工具,它提供的解决方案是:

CycleAvoidingMappingContext 如下:

1
2
3
4
5
6
7
8
9
10
11
12
public class CycleAvoidingMappingContext {
    private Map<Object, Object> knownInstances = new IdentityHashMap<Object, Object>();
@BeforeMapping
public <T> T getMappedInstance(Object source, @TargetType Class<T> targetType) {
    return (T) knownInstances.get(source);
}
 
@BeforeMapping
public void storeMappedInstance(Object source, @MappingTarget Object target) {
    knownInstances.put(source, target);
	}
}

使用方式:

1
2
3
private static Organization toDo(OrganizationPo po) {
        return OrganizationMapper.INSTANCE.toDo(po, new CycleAvoidingMappingContext());
    }

原理就是以已经映射过的 po 为 key,do 为 value,存到一个 map 里。在递归设值的过程中,再遇到同一个 po,就可以直接从 map 中取出对应的 do,从而避免了无限循环。

最后到了序列化的阶段,此时仍然有循环引用的问题。springboot 默认采用的是 jackson 框架,jackson 提供了一个注解可以解决此问题:

意思就是对于相同的对象,直接使用 orgId 属性表示。最后假如执行 dao.findById("1"),返回的结果如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
{
  "orgId": "1",
  "orgName": "研发部",
  "children": [
    {
      "orgId": "2",
      "orgName": "河北项目组",
      "parent": "1",
      "children": [
        {
          "orgId": "4",
          "orgName": "北京小分队",
          "parent": "2",
          "children": [
            {
              "orgId": "5",
              "orgName": "销售组",
              "parent": "4",
              "children": [
                          
			]
        }
      ]
    }
  ]
},
{
  "orgId": "3",
  "orgName": "福建项目组",
  "parent": "1",
  "children": [
    
  		]
	}
  ]
}

可以看到解决循环引用的思路都是想办法切断循环引用,一个是返回已有的对象,一个是用其他属性表示,都可以终止递归调用。