openecomp-be/api/openecomp-sdc-rest-webapp/vendor-software-products-rest/vendor-software-products-rest-services/src/main/java/org/openecomp/sdcrests/vsp/rest/mapping/MapImageEntityToImageDto.java

   1 package org.openecomp.sdcrests.vsp.rest.mapping;
   2
   3 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ImageEntity;
   4 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.ComponentData;
   5 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.Image;
   6 import org.openecomp.sdc.vendorsoftwareproduct.types.composition.ImageData;
   7 import org.openecomp.sdcrests.mapping.MappingBase;
   8 import org.openecomp.sdcrests.vendorsoftwareproducts.types.ImageDto;
   9
  10 public class MapImageEntityToImageDto extends MappingBase<ImageEntity, ImageDto> {
  11   @Override
  12   public void doMapping(ImageEntity source, ImageDto target) {
  13     target.setId(source.getId());
  14     Image image = source.getImageCompositionData();
  15
  16     if (image != null) {
  17       ImageData imageData = new ImageData(image.getFileName(), image.getDescription());
  18       new MapImageDataToImageDto().doMapping(imageData, target);
  19     }
  20   }
  21 }