如何基于现有代码正确实现通用ModelMapper模型映射器并优化？

当前实现的问题 #

• 每次调用都得手动强制类型转换，既繁琐又丢失了编译时类型检查，一旦类型不匹配，运行时会抛出ClassCastException
• 返回值为Object类型，IDE无法提供类型提示，代码可读性差
• 未利用Java泛型的类型安全特性，无法在编译阶段提前发现类型不匹配的错误

核心优化：用泛型改造方法 #

把convertToType改成泛型方法，让编译器在编译阶段处理类型信息，彻底消除强制转换，同时保证类型安全。

优化后的代码 #

@Component
public class Mapper {
    private final ModelMapper modelMapper;

    @Autowired
    public Mapper(ModelMapper modelMapper) {
        this.modelMapper = modelMapper;
    }

    public <T> T convertToType(Object object, Class<T> type) {
        return modelMapper.map(object, type);
    }
}

使用示例 #

调用时无需再写强制转换，编译器会自动推断目标类型：

DepartmentDTO departmentDTO = modelMapper.convertToType(department.get(), DepartmentDTO.class);
Department department = modelMapper.convertToType(departmentDTO, Department.class);

额外优化建议 #

• 针对字段名不匹配的常用映射，可以提前在ModelMapper的Bean中配置自定义映射规则，避免运行时字段映射失败：

  @Bean
  public ModelMapper modelMapper() {
      ModelMapper modelMapper = new ModelMapper();
      // 示例：配置Department和DepartmentDTO的字段映射
      modelMapper.typeMap(Department.class, DepartmentDTO.class)
          .addMapping(Department::getDeptId, DepartmentDTO::setDepartmentId);
      return modelMapper;
  }
  

• 可以新增集合转换的重载方法，支持批量转换List这类集合：

  public <T> List<T> convertListToType(List<?> sourceList, Class<T> targetType) {
      return sourceList.stream()
          .map(item -> convertToType(item, targetType))
          .collect(Collectors.toList());
  }
  

内容的提问来源于stack exchange，提问作者Leobejj