SpringBoot 入门：05 - 引入数据传输层

我们现在操作的数据直接就使用了实体类，假如我们输出的用户对象需要包含用户拥有的权限列表，那么我们怎么封装呢？

直接封装进实体类中显然不行，因为实体类的职责就是提供与数据库表字段一一对应的映射关系，所以我们还需要封装出一层数据传输层 DTO（Data Transfer Object），通过它暴露给外部，这样我们就能高度定制化我们需要输出的数据而不会影响到实体类。

在项目中创建 dto 目录，新建 UserDto.java 文件：

package com.foxescap.wxbox.dto;

import lombok.Data;

import java.time.LocalDateTime;

/**
 * @author xfly
 */
@Data
public class UserDto {
    private String username;

    private LocalDateTime expiredAt;

    private LocalDateTime createdAt;
    
    private List<RoleDto> roles;
}

结构看上去基本和对应的实体类非常相似，职责区别就是：

• 我不想暴露出去的字段可以直接不声明。

• 我可以加如其他的 DTO 作为其中的属性，如上所示。

对应我们的 Service 层就需要在拿到实体类对象后转换成 DTO 对象：

public UserDto findByUsername(String username) {
    var user = lambdaQuery().eq(User::getUsername, username).one();
    var userDto = new UserDto();
    userDto.setUsername(user.getUsername());
    userDto.setExpiredAt(user.getExpiredAt());
    userDto.setCreatedAt(user.getCreatedAt());

    return userDto;
}

这里你会发现，这个转换的过程特别难受，就像写 Getter 和 Setter 方法一样，写多了要吐了。

我会写吐别人也一样会写吐，然后就会有轮子造出来，modelmapper 就是解决这样需求的一个用的人比较多的轮子。

在 pom.xml 项目配置文件中的 dependencies 中加入 modelmapper 依赖：

<dependency>
  <groupId>org.modelmapper</groupId>
  <artifactId>modelmapper</artifactId>
  <version>2.3.7</version>
</dependency>

然后在 config 目录中的 WebMvcConfig.java 文件中注入这个对象：

@Bean
public ModelMapper modelMapper() {
    ModelMapper modelMapper = new ModelMapper();

    modelMapper.getConfiguration().setFullTypeMatchingRequired(true);
    modelMapper.getConfiguration().setMatchingStrategy(MatchingStrategies.STRICT);

    return modelMapper;
}

我们这个时候再来改造一下 findByUsername 方法：

private final ModelMapper modelMapper;

public UserService(ModelMapper modelMapper) {
    this.modelMapper = modelMapper;
}

public UserDto findByUsername(String username) {
    var user = lambdaQuery().eq(User::getUsername, username).one();

    return modelMapper.map(user, UserDto.class);
}