使用MapStruct的Spring数据和JPA一对多

Question

配置和配置头之间有一对多关系。以下是配置映射程序：

@Mapper(componentModel = "spring", uses = {UserMapper.class, ConfigHeadersMapper.class})
public interface ConfigMapper extends EntityMapper<ConfigDTO, Config> {

    @Mapping(source = "user.id", target = "userId")
    ConfigDTO toDto(Config config);

    @Mapping(source = "userId", target = "user")
    @Mapping(target = "messages", ignore = true)
    Config toEntity(ConfigDTO configDTO);

    default Config fromId(Long id) {
        if (id == null) {
            return null;
        }
        Config config = new Config();
        config.setId(id);
        return config;
    }
}

以下是ConfigHeadersMapper：

@Mapper(componentModel = "spring", uses = {ConfigMapper.class})
public interface ConfigHeadersMapper extends EntityMapper<ConfigHeadersDTO, ConfigHeaders> {

    @Mapping(source = "config.id", target = "configId")
    ConfigHeadersDTO toDto(ConfigHeaders configHeaders);

    @Mapping(source = "configId", target = "config")
    ConfigHeaders toEntity(ConfigHeadersDTO configHeadersDTO);

    default ConfigHeaders fromId(Long id) {
        if (id == null) {
            return null;
        }
        ConfigHeaders configHeaders = new ConfigHeaders();
        configHeaders.setId(id);
        return configHeaders;
    }
}

当我试图保存一个新实体时（html" target="_blank">配置的ID等于null）

final Config config = configMapper.toEntity(configDTO);
Config newConfig = configRepository.save(config);

保存配置和ConfigHeaders，但ConfigHeaders处的配置id（FK）为空。

所以我尝试了以下代码：

final Config config = configMapper.toEntity(configDTO);
config.getHeaders().stream().forEach(header -> header.setConfig(config));
Config newConfig = configRepository.save(config);

并使用自动生成的父ID（config_ID）保存子项（ConfigHeaders）。

你能告诉我Map结构有什么问题吗？我对这个工具很陌生，我不相信这是正确的解决方案。我认为以前的代码现在只是一种解决方法。

实际上，我已经检查了已生成的MapStruct实现代码，并注意到它正确设置了父ID（对于新的ID为null），但它没有设置与父实体的向后关系。我如何通过MapStruct实现这一点？

提前谢谢你

Answer 1

你也可以使用@Context。这也让你有可能使用EntityManager做一些事情。看看这里的例子。

Answer 2

正如您所注意到的，为了让JPA正确执行保存，您需要在ConfigHeader中设置指向配置的链接。

有两种方法可以实现这一点：

第一个选项：

您可以使用Collection MappingStrategy#ADDER_PREFERRED（在此处查看更多信息）。为此，您需要在Config中添加如下内容：

public void addHeader(ConfigHeader header) {
    this.headers.add(header);
    header.setConfig(this);
}

第二种选择：

您可以在ConfigMapper中使用@AfterMap并在那里的所有标头上设置配置。

@AfterMapping
default void linkHeaders(@MappingTarget Config config) {
    config.getHeaders().stream().forEach(header -> header.setConfig(config));
}

第三个选项由@sjaak在https://stackoverflow.com/a/48974119/1115491:

可以使用mapstruct jpa子父示例中所示的@Context属性。这与第一个选项具有相同的性能，因为您不必在头上迭代两次。另一个好处是，当无法向实体添加内容时，可以使用它。

它看起来像：

public class ConfigContext {
    private Config config;
    @BeforeMapping
    public void setConfig(@MappingTarget Config config) {
       this.config = config;
    }

    @AfterMapping
    public void establishRelation(@MappingTarget ConfigHeader header) {
        header.setConfig( header );
    }
}

您还需要将方法调整为实体方法，以便它们也包括上下文。