MapStruct中处理原始类型默认值映射为null的技术方案

背景介绍

在使用MapStruct进行对象映射时，开发人员经常会遇到原始类型(primitive type)与包装类型(wrapper type)之间的映射问题。特别是在处理Thrift等RPC框架生成的DTO时，原始类型字段总是带有默认值(如int默认为0)，而我们希望在某些业务场景下将这些默认值映射为包装类型的null值。

问题分析

以汽车对象映射为例，我们有以下两个类：

public class Car {
    private int numberOfSeats;  // 原始int类型，默认值为0
}

public class CarDTO {
    private Integer seats;      // 包装类型，可接受null值

当Car.numberOfSeats为0时，我们希望CarDTO.seats被设置为null而不是Integer(0)。

解决方案

MapStruct提供了几种方式来处理这种映射需求：

1. 使用条件映射(Conditional Mapping)

可以通过@Condition注解定义一个条件方法，控制何时进行属性映射：

@Mapper
public interface CarMapper {
    
    @Mapping(target = "seats", source = "numberOfSeats", conditionExpression = "java(source.getNumberOfSeats() != 0)")
    CarDto toDto(Car source);
}

2. 使用自定义条件方法

更优雅的方式是定义一个可重用的条件方法：

@Mapper
public interface CarMapper {

    @Mapping(target = "seats", source = "numberOfSeats", conditionQualifiedByName = "hasSeats")
    CarDto map(Car car);

    @Named("hasSeats")
    @Condition
    default boolean hasSeats(int seats) {
        return seats != 0;
    }
}

这种方法可以轻松扩展到其他原始类型字段，只需定义相应的条件方法即可。

3. 源对象添加判断方法

如果可以在源对象上添加方法，更简洁的方式是：

public class Car {
    private int numberOfSeats;
    
    public boolean hasNumberOfSeats() {
        return numberOfSeats != 0;
    }
}

然后在映射接口中：

@Mapper
public interface CarMapper {
    CarDto toDto(Car car);
}

MapStruct会自动检测并使用hasXxx()方法作为条件判断。

高级应用

对于需要处理多种原始类型的情况，可以创建通用的条件方法：

@Mapper
public interface GenericMapper {
    
    @Condition
    default boolean isNotDefault(int value) {
        return value != 0;
    }
    
    @Condition
    default boolean isNotDefault(long value) {
        return value != 0L;
    }
    
    @Condition
    default boolean isNotDefault(double value) {
        return Double.compare(value, 0.0) != 0;
    }
    // 其他原始类型...
}

这些方法会自动应用于所有相应类型的属性映射。

最佳实践

1. 一致性：在整个项目中保持相同的默认值处理策略
2. 明确性：在映射接口中添加注释说明默认值处理逻辑
3. 可测试性：为条件方法编写单元测试
4. 性能考虑：对于性能敏感场景，评估条件检查的开销

总结

MapStruct提供了灵活的方式来处理原始类型到包装类型的映射，特别是默认值到null的转换。通过合理使用条件映射，可以保持代码的简洁性和可维护性，同时满足业务需求。选择哪种方案取决于具体项目的需求和约束条件。