MapStruct用法


 

1 MapStruct配置

        MapStuct的使用非常簡單,把對應的jar包引入即可。

<properties>
    <mapstruct.version>1.3.1.Final</mapstruct.version>
</properties>
<dependency>
    <groupId>org.mapstruct</groupId>
    <artifactId>mapstruct</artifactId>
    <version>${mapstruct.version}</version>
</dependency>
<dependency>
    <groupId>org.mapstruct</groupId>
    <artifactId>mapstruct-processor</artifactId>
    <version>${mapstruct.version}</version>
</dependency>

2 原理&性能 

2.1 實現原理    

        對象拷貝工具實現上一般分為2種:

       (1) 在運行時,通過反射調用set/get方法或者直接對成員變量進行賦值。

       (2)在編譯期,生成調用get/set方法進行賦值的代碼,生成對應的class文件。

       MapStrut屬於第二種,在編譯期間消耗少許的時間,換取運行時的高性能。

       接口聲明:

@Mapper
public interface ProductAssembler { SkuDTO toDTO(Sku sku); }

 

        編輯生成的class反編譯 

public class ProductAssemblerImpl implements ProductAssembler {
    @Override
    public SkuDTO toDTO(Sku sku) { if ( sku == null ) { return null; } SkuDTO skuDTO = new SkuDTO(); skuDTO.setSkuId( sku.getSkuId() ); return skuDTO; } }  

2.2 性能對比

        

3 使用方法

        使用@Mapper注解,聲明映射器,可以是接口,或者抽象類。

        使用@Mapping注解,實現靈活的字段映射,定制映射的規則。

3.1 轉換器的檢索

        在聲明好轉換接口之后,MapStruct提供幾種方式獲取生成的Mapper映射器。

3.1.1 使用Mappers工廠獲取

        可以通過提供的Mappers工廠類,獲取指定的類型。

@Mapper
public interface Assembler { //使用工廠方法獲取Mapper實例 Assembler INSTANCE = Mappers.getMapper(Assembler.class); ProductDTO toDTO(Product product); }

3.1.2 通過依賴注入的方式獲取

  MapStuct同時支持和其他框架結合,通過依賴注入的方式獲取Mapper實例。目前支持spring和cdi。

@Mapper(componentModel = "spring")
public interface Assembler { ProductDTO toDTO(Product product); }
@Component 
public class AssemblerImpl implements Assembler { @Override public ProductDTO toDTO(Product product) { if ( product == null ) { return null; } ProductDTO productDTO = new ProductDTO(); productDTO.setProductId( product.getProductId() ); return productDTO; } }

3.2 簡單映射

3.2.1 基本映射

        對於同名同屬性的字段,無需特別聲明指定,自動轉換。

        對於不同名相同屬性的字段,可以使用Mapping注解指定。

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Product {
    private String productId;
    private String name;
}
Product
@Data
@NoArgsConstructor
@AllArgsConstructor
public class ProductDTO implements Serializable {
    private static final long serialVersionUID = -6780322740093464581L;
    private String productId;
    private String productName;
}
ProductDTO

   定義映射器:

@Mapper(componentModel = "spring")
public interface Assembler { @Mapping(source = "name", target = "productName") ProductDTO toDTO(Product product); }

   生成的映射器試實現:

@Component
public class AssemblerImpl implements Assembler { @Override public ProductDTO toDTO(Product product) { if ( product == null ) { return null; } ProductDTO productDTO = new ProductDTO(); productDTO.setProductName( product.getName() ); //不同字段名映射  productDTO.setProductId( product.getProductId() ); //相同映射名自動轉換 return productDTO; } }

3.2.2 多源參數映射

  支持把多個參數映射成一個類型,使用@Mapping指定即可。

@Mapper(componentModel = "spring")
public interface Demo6Assembler {
    @Mapping(target = "productId", source = "product.productId")
    @Mapping(target = "desc", source = "detail.desc")
    ProductDTO toDetailDTO(Product product, ProductDetail detail);
}

3.2.3 更新對象

  映射時除了生成新的新對象外,還支持現存對象的更新:

@Mapper(componentModel = "spring")
public interface Demo6Assembler {
    @Mapping(target = "desc", source = "desc")
    void updateDTO(@MappingTarget ProductDTO productDTO, ProductDetail detail);
}

3.3 數據類型轉換

3.3.1 對於基礎數據類型會進行自動隱式的轉換

        如int、long、String,Integer、Long等。

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Product {
    private String productId;
    private Long price;
}
Product
@Data
@NoArgsConstructor
@AllArgsConstructor
public class ProductDTO implements Serializable {
    private static final long serialVersionUID = -6780322740093464581L;
    private int productId;
    private String price;
}
ProductDTO

   定義映射器:

@Mapper(componentModel = "spring")
public interface Assembler { ProductDTO toDTO(Product product); }

   生成的映射代碼:

@Component
public class AssemblerImpl implements Assembler { @Override public ProductDTO toDTO(Product product) { if ( product == null ) { return null; } ProductDTO productDTO = new ProductDTO(); if ( product.getProductId() != null ) { //String自動轉int  productDTO.setProductId( Integer.parseInt( product.getProductId() ) ); } if ( product.getPrice() != null ) { //Long轉String  productDTO.setPrice( String.valueOf( product.getPrice() ) ); } return productDTO; } }

 3.3.2 指定轉換格式

  某些類型的轉換,我們可以指定具體轉換的格式。

  (1)對於基本數據類型與String之間的轉換,可以使用 numberFormat 指定轉換格式,使用的是java.text.DecimalFormat 實現。

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Product {
    private String productId;
    private BigDecimal price;
    private String stock;
}
Product
@Data
@NoArgsConstructor
@AllArgsConstructor
public class ProductDTO implements Serializable {
    private static final long serialVersionUID = -6780322740093464581L;
    private String productId;
    private String price;
    private Integer stock;
}
ProductDTO

 

  映射器定義:

@Mapper(componentModel = "spring")
public interface Demo3Assembler {
    @Mapping(target = "price", numberFormat = "#.00元")   //BigDecimal轉換成字符串
    @Mapping(target = "stock", numberFormat = "#個")      //字符串轉換成int
    ProductDTO toDTO(Product product);
}

  實現代碼:

@Component
public class Demo3AssemblerImpl implements Demo3Assembler {
    @Override
    public ProductDTO toDTO(Product product) {
        if ( product == null ) {
            return null;
        }
        ProductDTO productDTO = new ProductDTO();
        productDTO.setProductId( product.getProductId() );
        if ( product.getPrice() != null ) {
            //BigDecimal格式化成字符串
            productDTO.setPrice( createDecimalFormat( "#.00元" ).format( product.getPrice() ) );
        }
        try {
            if ( product.getStock() != null ) {
                 //字符串格式化為int
                productDTO.setStock( new DecimalFormat( "#個" ).parse( product.getStock() ).intValue() );
            }
        }
        catch ( ParseException e ) {
            throw new RuntimeException( e );
        }
        return productDTO;
    }
    private DecimalFormat createDecimalFormat( String numberFormat ) {
        DecimalFormat df = new DecimalFormat( numberFormat );
        df.setParseBigDecimal( true );
        return df;
    }
}

  測試代碼:

@Test
public void test2() {
    com.gotten.study.mapstruct.demo3.Product  product = new com.gotten.study.mapstruct.demo3.Product ();
    product.setProductId("P001");
    product.setPrice(new BigDecimal("100"));
    product.setStock("1個");
    com.gotten.study.mapstruct.demo3.ProductDTO productDTO = demo3Assembler.toDTO(product);
    System.out.println("productDTO:" + JSON.toJSONString(productDTO));
}


productDTO:{"price":"100.00元","productId":"P001","stock":1}
test

 

  (2)Date和String之間的轉換,可以通過dateFormat指定轉換格式,使用的是SimpleDateFormat的實現。

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Product {
    private String productId;
    private Date saleTime;
    private String validTime;
}
Product
@Data
@NoArgsConstructor
@AllArgsConstructor
public class ProductDTO implements Serializable {
    private static final long serialVersionUID = -6780322740093464581L;
    private String productId;
    private String saleTime;
    private Date validTime;
}
ProductDTO

  定義映射器:

@Mapper(componentModel = "spring")
public interface Demo4Assembler {
    @Mapping(target = "saleTime", dateFormat = "yyyy-MM-dd HH:mm:ss")  //Date轉換成String
    @Mapping(target = "validTime", dateFormat = "yyyy-MM-dd HH:mm")    //String轉換成Date
    ProductDTO toDTO(Product product);
}

  實現代碼:

@Component
public class Demo4AssemblerImpl implements Demo4Assembler {
    @Override
    public ProductDTO toDTO(Product product) {
        if ( product == null ) {
            return null;
        }
        ProductDTO productDTO = new ProductDTO();
        productDTO.setProductId( product.getProductId() );
        if ( product.getSaleTime() != null ) {
            productDTO.setSaleTime( new SimpleDateFormat( "yyyy-MM-dd HH:mm:ss" ).format( product.getSaleTime() ) ); //轉換成String
        }
        try {
            if ( product.getValidTime() != null ) {
                productDTO.setValidTime( new SimpleDateFormat( "yyyy-MM-dd HH:mm" ).parse( product.getValidTime() ) ); //轉換成Date
            }
        }
        catch ( ParseException e ) {
            throw new RuntimeException( e );
        }
        return productDTO;
    }
}

3.3.3 屬性為復雜對象的映射

        (1)如果是相同類型的對象引用,不會創建新的對象,直接把對象的引用從源對象賦值給目標對象。

        (2)如果類型相同,但是是集合類的引用,會創建一個新的集合,集合里面的所有引用進行拷貝。

@Override
    public ProductDTO toDTO(Product product) {
        if ( product == null ) {
            return null;
        }
        ProductDTO productDTO = new ProductDTO();
        productDTO.setProductId( product.getProductId() );
        List<Sku> list = product.getSkuList();
        if ( list != null ) {
            productDTO.setSkuList( new ArrayList<Sku>( list ) ); //創建新的集合,並對所有元素進行拷貝
        }

        return productDTO;
    }

  

  (3)對象的類型不同,會檢查映射器中是否存在對應的映射方法,如果存在,直接使用,否則會嘗試自動創建子映射方法。

@Data
@AllArgsConstructor
@NoArgsConstructor
public class Product {
    private String productId;
    private ProductDetail productDetail;
}
Product
@Data
@AllArgsConstructor
@NoArgsConstructor
public class ProductDetail {
    private String id;
}
ProductDetail
@Data
@AllArgsConstructor
@NoArgsConstructor
public class ProductDTO implements Serializable {
    private static final long serialVersionUID = 2184784038009791692L;
    private String productId;
    private ProductDetailDTO productDetail;
}
ProductDTO
@Data
@AllArgsConstructor
@NoArgsConstructor
public class ProductDetailDTO {
    private String detailId;
}
ProductDetailDTO

  定義映射器:

@Mapper(componentModel = "spring")
public interface Demo6Assembler {
ProductDTO toDTO(Product product);

@Mapping(target = "detailId", source = "id")
ProductDetailDTO toDetailDTO(ProductDetail detail);
}

   生成代碼:

@Component
public class Demo6AssemblerImpl implements Demo6Assembler {

    @Override
    public ProductDTO toDTO(Product product) {
        if ( product == null ) {
            return null;
        }
        ProductDTO productDTO = new ProductDTO();
        productDTO.setProductId( product.getProductId() );
        productDTO.setProductDetail( toDetailDTO( product.getProductDetail() ) ); //查找使用存在的轉換方法
        return productDTO;
    }
    @Override
    public ProductDetailDTO toDetailDTO(ProductDetail detail) {
        if ( detail == null ) {
            return null;
        }
        ProductDetailDTO productDetailDTO = new ProductDetailDTO();
        productDetailDTO.setDetailId( detail.getId() );
        return productDetailDTO;
    }
}

 

  (4)多層bean之間的轉換

    @Mapping注解支持跨層級的屬性轉換,屬性可以在不同層級之間切換。

@Data
@AllArgsConstructor
@NoArgsConstructor
public class Product {
    private String productId;
    private ProductDetail productDetail;
}
Product
@Data
@AllArgsConstructor
@NoArgsConstructor
public class ProductDetail {
    private String id;
}
ProductDetail
@Data
@AllArgsConstructor
@NoArgsConstructor
public class ProductDTO implements Serializable {
    private static final long serialVersionUID = 2184784038009791692L;
    private String productId;
    private ProductDetailDTO productDetail;
}
ProductDTO
@Data
@AllArgsConstructor
@NoArgsConstructor
public class ProductDetailDTO {
    private String productId;
    private String detailId;
}
ProductDetailDTO

  定義映射器:

@Mapper(componentModel = "spring")
public interface Demo7Assembler {
    @Mapping(target = "productDetail.detailId", source = "productDetail.id") //聲明productDetail下的屬性轉換規則
    @Mapping(target = "productDetail.productId", source = "productId") //跨層級的屬性轉換,把product層級的productId放到productDetail層級
    ProductDTO toDTO(Product product);
}

  生成代碼:

@Component
public class Demo7AssemblerImpl implements Demo7Assembler {

    @Override
    public ProductDTO toDTO(Product product) {
        if ( product == null ) {
            return null;
        }
        ProductDTO productDTO = new ProductDTO();
        if ( product.getProductDetail() != null ) {
            if ( productDTO.getProductDetail() == null ) {
                productDTO.setProductDetail( new ProductDetailDTO() );
            }
            productDetailToProductDetailDTO( product.getProductDetail(), productDTO.getProductDetail() );
        }
        if ( productDTO.getProductDetail() == null ) {
            productDTO.setProductDetail( new ProductDetailDTO() );
        }
        productToProductDetailDTO( product, productDTO.getProductDetail() );
        productDTO.setProductId( product.getProductId() );
        return productDTO;
    }

  //detail的轉換方法
    protected void productDetailToProductDetailDTO(ProductDetail productDetail, ProductDetailDTO mappingTarget) {
        if ( productDetail == null ) {
            return;
        }
        mappingTarget.setDetailId( productDetail.getId() );
    }

  //product轉成detail(更新處理)
    protected void productToProductDetailDTO(Product product, ProductDetailDTO mappingTarget) {
        if ( product == null ) {
            return;
        }
        mappingTarget.setProductId( product.getProductId() );
    }
}
Demo7AssemblerImpl

 

3.3.4 自定義轉換器

  MapStruct支持自定義轉換器,實現類型之間的轉換自定義的規則。

  一個自定義映射器可以定義多個映射方法,匹配時,是以方法的入參和出參進行匹配的。如果綁定的映射中,存在多個相同的入參和出參方法,將會報錯。

  如果多個入參或者出參方法存在繼承關系,將會匹配最具體的那一個方法。

@Data
@AllArgsConstructor
@NoArgsConstructor
public class Product {
    private String productId;
    private List<String> images;
}

@Data
@AllArgsConstructor
@NoArgsConstructor
public class ProductDTO implements Serializable {
    private static final long serialVersionUID = 2184784038009791692L;
    private String productId;
    private String images;
}
bean

   定義映射器:

@Component
public class ImageFormater {
    public String format(List<String> images) {
        return String.join(",", images);
    }
}

 

  綁定轉換器:

@Mapper(componentModel = "spring", uses = ImageFormater.class)
public interface Demo8Assembler {
    ProductDTO toDTO(Product product);
}

 

  映射器實現:

@Component
public class Demo8AssemblerImpl implements Demo8Assembler {

    @Autowired
    private ImageFormater imageFormater;

    @Override
    public ProductDTO toDTO(Product product) {
        if ( product == null ) {
            return null;
        }

        ProductDTO productDTO = new ProductDTO();
        productDTO.setProductId( product.getProductId() );
     //調用自定義的映射器進行映射,把list轉成string
        productDTO.setImages( imageFormater.format( product.getImages() ) );
        return productDTO;
    }
}

 3.3.5 使用限定符限定使用轉換方法

  自定義轉換器時,存在多個相同入參和出參的方法,MapStruct無法匹配使用哪個映射方法。這時可以使用限定符綁定每個屬性轉換時使用的轉換方法。

  (1)限定符使用自定義注解實現。

    聲明限定符:

import org.mapstruct.Qualifier;
//映射器上的限定符
@Qualifier //標記為限定符
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.CLASS)
public @interface Formators {
}
//映射方法上的限定符
@Qualifier //標記為限定符
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.CLASS)
public @interface FormatImages {
}

//映射方法上的限定符
@Qualifier //標記為限定符
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.CLASS)
public @interface FormatDetails {
}

    綁定限定符到映射器的方法上面:

@Component
@Formators //綁定限定符
public class CusFormater {
    @FormatImages //綁定限定符
    public String formatImages(List<String> images) {
        return String.join(",", images);
    }
    @FormatDetails //綁定限定符
    public String formatDetails(List<String> images) {
        return String.join(",", images);
    }
}

  映射時,綁定限定符,定位映射方法:

@Mapper(componentModel = "spring", uses = CusFormater.class)
public interface Demo9Assembler {
    @Mapping(target = "images", qualifiedBy = FormatImages.class) //轉換指定限定符,定位具體的映射方法
    @Mapping(target = "details", qualifiedBy = FormatDetails.class)//轉換指定限定符,定位具體的映射方法
    ProductDTO toDTO(Product product);
}

   生成代碼:

@Component
public class Demo9AssemblerImpl implements Demo9Assembler {
    @Autowired
    private CusFormater cusFormater;

    @Override
    public ProductDTO toDTO(Product product) {
        if ( product == null ) {
            return null;
        }
        ProductDTO productDTO = new ProductDTO();
        productDTO.setProductId( product.getProductId() );
        productDTO.setImages( cusFormater.formatImages( product.getImages() ) ); //定位方法
        productDTO.setDetails( cusFormater.formatDetails( product.getDetails() ) );
        return productDTO;
    }
} 

  (2)基於named注解實現(推薦)

    除了使用自定義注解的方法,還可以使用@Named注解實現限定符的綁定。

@Component
@Named("CusFormater")
public class CusFormater {
    //綁定限定符
    @Named("formatImages")
    public String formatImages(List<String> images) {
        return String.join(",", images);
    }
    //綁定限定符
    @Named("formatDetails")
    public String formatDetails(List<String> images) {
        return String.join(",", images);
    }
}

  使用時綁定:

@Mapper(componentModel = "spring", uses = CusFormater.class)
public interface Demo10Assembler {
    @Mapping(target = "images", qualifiedByName = "formatImages") //轉換指定限定符,定位具體的映射方法
    @Mapping(target = "details", qualifiedByName = "formatDetails")//轉換指定限定符,定位具體的映射方法
    ProductDTO toDTO(Product product);
}

 

3.4 Map的映射

  可以使用@MapMapping實現對key和value的分別映射:

@Mapper(componentModel = "spring")
public interface Demo11Assembler { @MapMapping(valueDateFormat = "yyyy-MM-dd HH:mm:ss") Map<String, String> toDTO(Map<Long, Date> map); }

3.5 枚舉值之間的轉換

  MapStruct可以在多個枚舉值之間轉換,使用@ValueMapping注解。

public enum E1 {
    E1_1,
    E1_2,
    E1_3,
    ;
}

public enum E2 { E2_1, E2_2, E2_3, ; }
@Mapper(componentModel = "spring")
public interface Demo11Assembler { @ValueMapping(target = "E1_1", source = "E2_1") @ValueMapping(target = "E1_2", source = "E2_2") @ValueMapping(target = MappingConstants.NULL, source = "E2_3") //轉換成null  E1 toDTO(E2 e2); }

  生成代碼:

@Component
public class Demo11AssemblerImpl implements Demo11Assembler { @Override public E1 toDTO(E2 e2) { if ( e2 == null ) { return null; } E1 e1; switch ( e2 ) { case E2_1: e1 = E1.E1_1; break; case E2_2: e1 = E1.E1_2; break; case E2_3: e1 = null; break; default: throw new IllegalArgumentException( "Unexpected enum constant: " + e2 ); } return e1; } }

3.6 定制Bean生成

  使用MapStruct可以使用對象工廠來創建bean,同時也可以更新bean。

  定義對象工廠:

public class DTOFactory {
    public ProductDTO createDTO() { ProductDTO productDTO = new ProductDTO(); productDTO.setStock(0); return productDTO; } }

  使用對象工廠:

@Mapper(componentModel = "spring", uses = DTOFactory.class) //指定使用的對象工廠
public interface Demo13Assembler {
    ProductDTO toDTO(Product product);
}

  生成代碼:

@Component
public class Demo13AssemblerImpl implements Demo13Assembler { @Autowired private DTOFactory dTOFactory; @Override public ProductDTO toDTO(Product product) { if ( product == null ) { return null; } ProductDTO productDTO = dTOFactory.createDTO(); //使用對象工廠創建對象 productDTO.setProductId( product.getProductId() ); return productDTO; } }

 

3.7 缺省值和常量

  MapStruct允許設置缺省值和常量,同時缺省值允許使用表達式。

  注意:使用缺省值,源字段必須存在,否則缺省值不生效,否則應該使用常量。

@Mapper(componentModel = "spring", imports = UUID.class)
public interface Demo15Assembler {
@Mapping(target = "productId", source = "productId", defaultValue = "0") //當product的productId為null,設置為0
@Mapping(target = "random", source = "random", defaultExpression = "java(UUID.randomUUID().toString())") //缺省設置隨機數
@Mapping(target = "stock", constant = "0") //固定設置為0
@Mapping(target = "createTime", dateFormat = "yyyy-MM-dd", constant = "2020-05-30") //固定格式化設置為2020-05-30,
ProductDTO toDTO(Product product);
}

  

@Component
public class Demo15AssemblerImpl implements Demo15Assembler {

@Override
public ProductDTO toDTO(Product product) {
if ( product == null ) {
return null;
}
ProductDTO productDTO = new ProductDTO();

if ( product.getProductId() != null ) {
productDTO.setRandom( product.getProductId() );
}
else {
productDTO.setRandom( UUID.randomUUID().toString() );
}
if ( product.getProductId() != null ) {
productDTO.setProductId( product.getProductId() );
}
else {
productDTO.setProductId( "0" );
}

productDTO.setStock( 0 );
try {
productDTO.setCreateTime( new SimpleDateFormat( "yyyy-MM-dd" ).parse( "2020-05-30" ) );
}
catch ( ParseException e ) {
throw new RuntimeException( e );
}

return productDTO;
}
}

3.8 存在繼承關系的結果處理

  當返回的結果類型存在繼承關系時,可以使用 @BeanMapping注解指定真實返回的結果類型。

@Mapper(componentModel = "spring")
public interface Demo17Assembler { @BeanMapping(resultType = DogDTO.class) //指定返回的結果類型  Animal toDTO(Dog dog); }

  

@Component
public class Demo17AssemblerImpl implements Demo17Assembler { @Override public Animal toDTO(Dog dog) { if ( dog == null ) { return null; } DogDTO animal = new DogDTO(); animal.setId( dog.getId() ); return animal; } }

3.9 映射關系繼承

  MapStruct允許對映射關系進行繼承,使用@InheritConfiguration標記當前方法繼承其他映射方法的映射關系。會自動查找相同類型映射源、映射目標的方法進行繼承,如果存在多個相同類型的方法,則需要手工指定。

@Mapper(componentModel = "spring")
public interface Demo18Assembler {
@Mapping(target = "productId", source = "id")
@Mapping(target = "detail", source = "detail1")
ProductDTO toDTO(Product product);

@Mapping(target = "productId", source = "id2")
@Mapping(target = "detail", source = "detail2")
ProductDTO toDTO2(Product product);

@InheritConfiguration(name = "toDTO") //對toDTO的映射關系進行繼承
@Mapping(target = "detail", source = "detail2") //對繼承的關系進行重寫
void update(@MappingTarget ProductDTO productDTO, Product product);

}

        除了正向繼承規則外,還可以進行規則逆向繼承,從被繼承方法的目標對象映射到源對象。

@Mapper(componentModel = "spring")
public interface Demo18Assembler {
@Mapping(target = "productId", source = "id")
@Mapping(target = "detail", source = "detail1")
ProductDTO toDTO(Product product);

@Mapping(target = "productId", source = "id2")
@Mapping(target = "detail", source = "detail2")
ProductDTO toDTO2(Product product);

@InheritInverseConfiguration(name = "toDTO") //對toDTO的映射關系進行逆繼承
@Mapping(target = "detail2", source = "detail") //對逆向繼承的關系進行重寫
Product toEntity(ProductDTO dto);
}

 3.10 復雜映射的實現

  有時候我們除了普通映射外,還需要進行一些復雜的映射,如把多個字段計算映射成一個字段,或者借用一些工具進行映射的計算等。MapStruct提供了集中方式實現。

3.10.1 使用java表達式進行映射

  對於復雜的映射,允許使用java表達式實現字段的映射。

  注意要導入使用到的類。

@Mapper(componentModel = "spring", imports = DecimalUtils.class) //導入java表達式使用的類
public interface Demo16Assembler {
    @Mapping(target = "price", expression = "java(product.getPrice1() + product.getPrice2())") //直接相加
    @Mapping(target = "price2", expression = "java(DecimalUtils.add(product.getPrice1(), product.getPrice2()))") //使用工具類處理
    ProductDTO toDTO(Product product);
}

   生成的映射代碼:

@Component
public class Demo16AssemblerImpl implements Demo16Assembler {

    @Override
    public ProductDTO toDTO(Product product) {
        if ( product == null ) {
            return null;
        }
        ProductDTO productDTO = new ProductDTO();
        productDTO.setProductId( product.getProductId() );
        productDTO.setPrice( product.getPrice1() + product.getPrice2() );
        productDTO.setPrice2( DecimalUtils.add(product.getPrice1(), product.getPrice2()) );
        return productDTO;
    }
}

 3.10.2 使用裝飾器進行映射

  MapStruct允許使用裝飾器進行一些復雜映射,同時可以支持和Spring結合。

  定義一個映射器,同時聲明綁定裝飾器:

@Mapper(componentModel = "spring")
@DecoratedWith(Demo18AssemblerDecorator.class) //聲明綁定裝飾器
public interface Demo18Assembler {
    ProductDTO toDTO(Product product);
}

  定義裝飾器:

public abstract class Demo18AssemblerDecorator implements Demo18Assembler {
    @Autowired
    @Qualifier("delegate") //注入mapStruct生成的轉換器,原始的轉換器注入spring時,會使用delegate裝飾符
    private Demo18Assembler assembler;

    //可以獲取spring的bean進行操作
    @Autowired
    private StringUtils stringUtils;

    @Override
    public ProductDTO toDTO(Product product) {
        //調用MapStruct進行轉換
        ProductDTO productDTO = assembler.toDTO(product);
        //自定義操作
        stringUtils.join(product.getName(), "-", product.getTitle());
        return productDTO;
    }
}

  生成裝飾器代碼:

@Component
@Primary //Primary修飾,方便使用時直接使用autowired注入
public class Demo18AssemblerImpl extends Demo18AssemblerDecorator implements Demo18Assembler {
}

 3.10.3 使用前后置處理實現復雜映射

  使用@BeforeMapping和@AfterMapping注解可以指定映射過程的的回調方法,進行一些前置或者后置的操作。

  前置回調方法的執行時機是在映射方法開始時,后置方法是在映射完成return之前。

  回調方法可以直接定義在映射器內:

@Mapper(componentModel = "spring")
public interface Demo19Assembler {

    ProductDTO toDTO(Product product);

    @BeforeMapping //前置執行
    default ProductDTO toDTOBefore(Product product) {
        ProductDTO productDTO = new ProductDTO();
        productDTO.setSales(9999);
        return productDTO;
    }

    @AfterMapping //后置執行
    default void toDTOAfter(Product product, @MappingTarget ProductDTO productDTO) {
        productDTO.setViewName(product.getName() + "-" + product.getTitle());
    }
}

  生成的實現代碼如下:

@Component
public class Demo19AssemblerImpl implements Demo19Assembler {

    @Override
    public ProductDTO toDTO(Product product) {
        ProductDTO target = toDTOBefore( product ); //前置
        if ( target != null ) {
            return target;
        }

        if ( product == null ) {
            return null;
        }

        ProductDTO productDTO = new ProductDTO();

        productDTO.setProductId( product.getProductId() );

        toDTOAfter( product, productDTO ); //后置

        return productDTO;
    }
}

  回調方法與映射的方法的匹配規則:

  (1)映射方法和回調方法沒有強綁定的關系,是依靠參數類型來匹配映射方法與回調方法的。映射方法的所有入參和出參類型,能覆蓋回調方法的入參,就會調用對應的回調方法,當要注意,如果回調方法的入參是映射方法的出參類型,回調方法中需要用@MappingTarget 指定,否則不會調用。

  (2)回調方法是void或者返回映射方法的出參類型才能匹配,但要注意,如果返回的是映射方法的出參類型,如果執行時返回不為null,則映射方法直接返回回調方法執行結果,不會往后執行。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM