我们现在操作的数据直接就使用了实体类,假如我们输出的用户对象需要包含用户拥有的权限列表,那么我们怎么封装呢?
直接封装进实体类中显然不行,因为实体类的职责就是提供与数据库表字段一一对应的映射关系,所以我们还需要封装出一层数据传输层 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); }