Mapstruct：在对象列表中自动增加对象变量

Question

我有一些这样的 POJO：

@Data
public class A {
    List<B> bList;
}

@Data
public class B {
    int id;
    int x;
}

@Data
public class C {
    List<D> dList;
}

@Data
public class D {
    int x;
}

现在我想使用 MapStruct 将 C 类映射到 A 类，这样 A 类中 bList 中的 id 值会自动递增。我正在使用 Mapstruct 1.1.0.Final。

为了将List<D> 映射到List<B>，我可以为class B 和class D 定义一个映射器，POJO 将被自动映射，但找不到一种方法可以将 POJO 与自动增量一起映射在一个领域。

我的 POJO 很大。 Map Struct 中有没有一种方法可以自动增加列表中的变量？

Answer 1

1.1.0.Final 有 2 个选项，1.2.0.Beta2 有另外一个选项。

将D 映射到B 时的第一个选项

您可以在D 和B 之间使用@AfterMapping，并在那里使用ThreadLocal。

public class AutoIncrementorThreadLocal {

    @AfterMapping
    public /*static*/ void autoIncrementId(D source, @MappingTarget B target) {
        //here you use the ThreadLocal and do something like getAndIncrement
    }
}

在您的映射器中，您必须说您使用的是AutoIncrementorThreadLocal。方法也可以是静态的，MapStruct 也可以调用静态方法。

将C 映射到A 时的第二个选项

public class AutoIncrementorBulk {

    @AfterMapping
    public /*static*/ void autoIncrementId(C source, @MappingTarget A target) {
        //here you iterate the list and increment the ids
    }
}

在您的映射器中，您必须说您使用的是AutoIncrementorBulk。方法也可以是静态的，MapStruct 也可以调用静态方法。

1.2.0.Beta2 的第三个选项

您可以使用新的@Context 并在上面添加@AfterMapping。

public class AutoIncrementorContext {

    private int counter = 0;
    @AfterMapping
    public void autoIncrementId(D source, @MappingTarget B target) {
        //here you use your counter, because you are only using this instance during one mapping
    }
}

您的映射器需要如下所示：

@Mapper
public interface MyMapper {

    C map(A source, @Context AutoIncrementorContext context);

    D map(B source, @Context AutoIncrementorContext context);
}

当你要调用你的映射器时，你需要这样称呼它：

myMapper.map(a, new AutoIncrementorContext());

第一种和第三种方法相似，唯一的区别是您不必将ThreadLocal(s) 与第三种方法一起使用。