对所有枚举使用相同的映射器

Question

我试图将所有枚举映射到相同的DTO类，我希望避免为每个枚举类型声明一个方法，或者为每个枚举字段使用@Map。

@Mapper
public interface EnumDtoMapper
{
    // approach one
    default <E extends Enum<E>> EnumDTO enumToEnumDTO(Enum<E> e)
    {
        if (e == null)
        {
            return null;
        }
        EnumDTO enumDTO = new EnumDTO();
        enumDTO.setName(e.name());
        return enumDTO;
    }

    //approach two: enum -> String -> EnumDTO
    default <E extends Enum<E>> String enumToString(Enum<E> e)
    {
        return e.name();
    }


    default EnumDTO stringToEumDTO(String name)
    {
        EnumDTO enumDTO = new EnumDTO();
        enumDTO.setName(name);
        return enumDTO;
    }
}

然后在我的其他地图上

@Mapper(componentModel = "cdi", injectionStrategy = InjectionStrategy.CONSTRUCTOR, uses = {
        EnumDtoMapper.class
})

但MapStruct仍在为每种枚举类型生成一种新的映射方法，而且实现方式不符合我的需要。

有没有办法用MapStruct做到这一点？还是太神奇了D

Answer 1

做了一些测试，如果没有组件模型设置为cdi，您所拥有的将生成您所期望的。但是有了它，您需要通配符输入枚举的通用类型，之后它应该可以工作。

如果EnuUtil看起来像这样：

class EnumUtil {

    static EnumDto enumToEnumDTO(Enum<?> e) {
        if ( e == null ) {
            return null;
        }
        EnumDto enumDTO = new EnumDto( e.name() );
        return enumDTO;
    }

    // approach two: enum -> String -> EnumDTO
    static String enumToString(Enum<?> e) {
        return e.name();
    }

    static EnumDto stringToEumDTO(String name) {
        return new EnumDto( name );
    }
}

使用以下映射器和类：

@Mapper( componentModel = ComponentModel.CDI,
        injectionStrategy = InjectionStrategy.CONSTRUCTOR,
        uses = {
            EnumUtil.class
        } )
public interface EnumMapper {

    @Mapping( target = "value", source = "enumValue" )
    DtoWithEnumDto map(SourceWithEnum source);
}

class SourceWithEnum {
    private MyEnum enumValue;
    // getter & setter
}

enum MyEnum {
    SOMETHING, ELSE
}

class EnumDto {
    private final String name;
    // constructor and getter
}

class DtoWithEnumDto {
    private EnumDto value;
    // constructor, getter and setter
}

您将得到以下生成的映射器：

@Generated(
    value = "org.mapstruct.ap.MappingProcessor",
    date = "2022-02-21T16:59:06+0100",
    comments = "version: 1.4.2.Final, compiler: Eclipse JDT (IDE) 1.4.50.v20210914-1429, environment: Java 17.0.1 (Azul Systems, Inc.)"
)
@ApplicationScoped
public class EnumMapperImpl implements EnumMapper {

    @Override
    public DtoWithEnumDto map(SourceWithEnum source) {
        if ( source == null ) {
            return null;
        }

        EnumDto value = null;

        value = EnumUtil.enumToEnumDTO( source.getEnumValue() );

        DtoWithEnumDto dtoWithEnumDto = new DtoWithEnumDto( value );

        return dtoWithEnumDto;
    }
}