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
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

package org.openecomp.sdcrests.vsp.rest.mapping;

import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ImageEntity;
import org.openecomp.sdc.vendorsoftwareproduct.types.composition.Image;
import org.openecomp.sdc.vendorsoftwareproduct.types.composition.ImageData;
import org.openecomp.sdcrests.mapping.MappingBase;
import org.openecomp.sdcrests.vendorsoftwareproducts.types.ImageDto;

public class MapImageEntityToImageDto extends MappingBase<ImageEntity, ImageDto> {
  @Override
  public void doMapping(ImageEntity source, ImageDto target) {
    target.setId(source.getId());
    Image image = source.getImageCompositionData();

    if (image != null) {
      ImageData imageData = new ImageData(image.getFileName(), image.getDescription());
      new MapImageDataToImageDto().doMapping(imageData, target);
    }
  }
}