MapStruct指定子映射

Question

这是我的例子。

public class Company {
    ....
    private String companyName;
    ....
}

public class Employee {
    ....
    private String name;
    ....
}

public class EmployeeDto {
    ....
    private String name;
    private String companyName;
    ....
}

@Mapper(componentModel = "spring")
public interface EmployeeDtoMapper {

    @Mapping(target = "name", source = "source1.name")
    @Mapping(target = "companyName", source = "source2.companyName")
    EmployeeDto toDto(Employee source1, Company source2);

    List<EmployeeDto> toDtos(List<Employee> sources, @Context Company source2);

}

我希望toDto方法使用toDto将Employee映射到EmployeeDto，但mapstruct会生成employeeDtoToEmployeeDto方法。我该怎么修？

谢谢

Answer 1

Mapstruct仅允许基于1个源和1个目标选择子映射。因此有了@Context注释。这将有效地忽略该参数，只将其传递给子映射。。

但是...你可以这样解决你的问题...

@Mapper(componentModel = "spring")
public interface EmployeeDtoMapper {

    @Mapping(target = "name", source = "source1.name")
    EmployeeDto toDto(Employee source1, @Context Company source2);

    @AfterMapping
    default afterToDto(@MappingTarget EmployeeDto target, @Context Company source2) {
       target.setCompanyName( source2.getCompanyName();
    }

    List<EmployeeDto> toDtos(List<Employee> sources, @Context Company source2);

}

注意：如果您在mapping source2中有多个参数，并且您也喜欢使用MapStruct，那么您可以定义一个新的接口方法mappingEmploydTodo（Company source），并从默认方法调用它。