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);

}

我希望toDtos方法使用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);

}

请注意，如果在映射source2中有多个参数，并且还希望使用MapStruct，则可以定义一个新的接口方法映射EmployDTO toDo(Company source)并从默认方法中调用它。

MapStruct指定子映射

1 个答案: