如何在 Spring Boot 中使用 Mapstruct 映射父级和子级？

Question

我有父级（产品）和子级（书籍、家具），并且希望将产品实体映射到产品 DTO。如您所见，产品被映射并存储在数据库中的单个表中。如何映射具有子项额外详细信息的父项产品？

我看过这个，这个和这个来得到一些想法，但没有运气

实体

@Entity
@Table(name = "product")
@Inheritance(strategy = InheritanceType.SINGLE_TABLE)
public class Product {
  @Id
  private long id;
  private String productName;
}

@Entity
@DiscriminatorValue("Book")
public class Book extends Product { 
  private String author;
  ...
}
@Entity
@DiscriminatorValue("Furniture")
public class Furniture extends Product {
  String color;
  ...
}

数据传输组织

public class ProductDto {
  private long id;
  private String productName;
  ...
}

public class BookDto extends ProductDto {
  private String author;
  ...
}
public class FurnitureDto extends ProductDto {
   String color;
   ... 
}

映射器

@Mapper(uses = {BookMapper.class,FurnitureMapper.class})
public interface ProductMapper {
    
    ProductDto productToProductDto(Product product);
    Product productDtoToProduct(ProductDto productDto);
}

@Mapper
public interface BookMapper {
    BookDto bookToBookDto(Book book);
    Book bookDtoToBook(BookDto bookDto);
}

@Mapper
public interface FurnitureMapper {
    FurnitureDto furnitureToFurnitureDto(Furniture furniture);
    Furniture furnitureDtoToFurniture(FurnitureDto furnitureDto);
}

服务

@Service
public class ProductServiceImpl implements ProductService {

    @Autowired
    ProductRepository productRepository;
    @Autowired
    ProductMapper productMapper;

    @Override
    public List<ProductDto> getAllProducts() {
        List<ProductDto> listOfProducts = new ArrayList<>();
        productRepository.findAll().forEach(i -> 
        listOfProducts.add(productMapper.productToProductDto(i)));
        return listOfProducts;
    }
}

已编辑

将产品实体映射到产品 dto 后，我得到以下结果。它不绑定数据，也不包含其子属性。上面的映射器部分正确吗？

[
    {
        "id": 0,
        "productName": null
    },
    {
        "id": 0,
        "productName": null
    },
    ...
]

但结果应该如下所示：

[
    {
        "id": 11,
        "productName": ABC,
        "author":"James"
    },
    {
        "id": 22,
        "productName": XYZ,
        "color":"Oak"
    },
    ...
]

Answer 1

长话短说

没有干净的方法可以做到这一点。原因在于Java的编译时方法选择。但是有一种使用访问者模式的比较干净的方法。

为什么它不起作用

当您迭代包含不同类型（产品、书籍、家具）的实体列表时，您需要为每种类型调用不同的映射方法（即不同的 MapStruct 映射器）。

除非您遵循instanceofAmir 建议的方式并显式选择映射器，否则您需要使用方法重载来为每个实体类调用不同的映射方法。问题是 Java 将在编译时选择重载方法，此时编译器只能看到Product对象列表（存储库方法返回的对象）。MapStruct、Spring 或您自己的自定义代码是否尝试执行此操作并不重要。这也是为什么 yourProductMapper总是被调用：它是编译时唯一可见的类型。

使用访客模式

由于我们需要手动选择正确的映射器，因此我们可以选择哪种方式更干净或更易于维护。这绝对是有主见的。

我的建议是按以下方式使用访问者模式（实际上是它的变体）：

为需要映射的实体引入一个新接口：

public interface MappableEntity {

    public ProductDto map(EntityMapper mapper);
}

您的实体将需要实现此接口，例如：

public class Book extends Product implements MappableEntity {
//...
@Override
    public ProductDto map(EntityMapper mapper) {
        return mapper.map(this);//This is the magic part. We choose which method to call because the parameter is this, which is a Book!
    }
}

EntityMapper是访问者接口：

public interface EntityMapper {

    ProductDto map(Product entity);

    BookDto map(Book entity);

    FurnitureDto map(Furniture entity);

    // Add your next entity here
}

最后，您需要 MasterMapper：

// Not a class name I'm proud of
public class MasterMapper implements EntityMapper {

    // Inject your mappers here

    @Override
    public ProductDto map(Product product) {
        ProductMapper productMapper = Mappers.getMapper(ProductMapper.class);
        return productMapper.map(product);
    }

    @Override
    public BookDto map(Book product) {
        BookMapper productMapper = Mappers.getMapper(BookMapper.class);
        return productMapper.map(product);
    }

    @Override
    public FurnitureDto map(Furniture product) {
        FurnitureMapper productMapper = Mappers.getMapper(FurnitureMapper.class);
        return productMapper.map(product);
    }

    // Add your next mapper here

}

您的服务方法将如下所示：

MasterMapper mm = new MasterMapper();
List<Product> listOfEntities = productRepository.findAll();
List<ProductDto> listOfProducts = new ArrayList<>(listOfEntities.size());
listOfEntities.forEach(i -> {
        if (i instanceof MappableEntity) {
            MappableEntity me = i;
            ProductDto dto = me.map(mm);
            listOfProducts.add(dto);
        } else {
            // Throw an AssertionError during development (who would run server VMs with -ea ?!?!)
            assert false : "Can't properly map " + i.getClass() + " as it's not implementing MappableEntity";
            // Use default mapper as a fallback
            final ProductDto defaultDto = Mappers.getMapper(ProductMapper.class).map(i);
            listOfProducts.add(defaultDto);
        }
     });
    return listOfProducts;

您可以安全地忽略Mappers.getMapper()调用：由于该问题与 Spring 无关，为了简单起见，我使用 MapStruct 的工厂在GitHub上创建了一个工作示例。您只需使用 CDI 注入映射器即可。