com.arextest.web.model.mapper.PlanMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arex-web-model Show documentation
The newest version!
package com.arextest.web.model.mapper;

import com.arextest.web.model.contract.contracts.ReportInitialRequestType;
import com.arextest.web.model.contract.contracts.common.PlanStatistic;
import com.arextest.web.model.contract.contracts.replay.UpdateReportInfoRequestType;
import com.arextest.web.model.dao.mongodb.ReportPlanStatisticCollection;
import com.arextest.web.model.dto.ReportPlanStatisticDto;
import com.arextest.web.model.enums.ReplayStatusType;
import java.util.Objects;
import org.mapstruct.Mapper;
import org.mapstruct.Mapping;
import org.mapstruct.Mappings;
import org.mapstruct.Named;
import org.mapstruct.factory.Mappers;

@Mapper
public interface PlanMapper {

  PlanMapper INSTANCE = Mappers.getMapper(PlanMapper.class);

  @Mappings({@Mapping(target = "appId", source = "application.appId"),
      @Mapping(target = "appName", source = "application.appName"),
      @Mapping(target = "sourceEnv", source = "hostEnv.sourceEnv"),
      @Mapping(target = "targetEnv", source = "hostEnv.targetEnv"),
      @Mapping(target = "sourceHost", source = "hostEnv.sourceHost"),
      @Mapping(target = "targetHost", source = "hostEnv.targetHost"),
      @Mapping(target = "caseSourceType", source = "caseSourceEnv.caseSourceType"),
      @Mapping(target = "caseTags", source = "caseSourceEnv.caseTags"),
      @Mapping(target = "caseStartTime", source = "caseSourceEnv.caseStartTime"),
      @Mapping(target = "caseEndTime", source = "caseSourceEnv.caseEndTime"),
      @Mapping(target = "targetImageId", source = "targetImage.targetImageId"),
      @Mapping(target = "targetImageName", source = "targetImage.targetImageName"),
      @Mapping(target = "coreVersion", source = "version.coreVersion"),
      @Mapping(target = "extVersion", source = "version.extVersion"),
      @Mapping(target = "caseRecordVersion", source = "version.caseRecordVersion"),})
  ReportPlanStatisticDto dtoFromContract(ReportInitialRequestType contract);

  @Mapping(target = "appName", expression = "java(\"unknown app name\".equals(dao.getAppName()) ? dao.getAppId() : dao.getAppName())")
  ReportPlanStatisticDto dtoFromDao(ReportPlanStatisticCollection dao);

  @Mapping(target = "errorMessage", source = ".", qualifiedByName = "convertErrorMessage")
  PlanStatistic contractFromDto(ReportPlanStatisticDto dto);

  ReportPlanStatisticDto dtoFromContract(UpdateReportInfoRequestType contract);


  @Named("convertErrorMessage")
  default String convertErrorMessage(ReportPlanStatisticDto dto) {
    if (dto != null && (Objects.equals(dto.getStatus(), ReplayStatusType.FAIL_INTERRUPTED)
      || Objects.equals(dto.getStatus(), ReplayStatusType.CANCELLED))) {
      return dto.getErrorMessage();
    }
    return null;
  }
}