# DetailedAvailabilityNext14DaysParseByOther.java

    package cloud.wconnect.rateshop.service.revenue.report.detailedavailability;

    import cloud.wconnect.commonlib.cons.*;
    import cloud.wconnect.commonlib.exception.BadRequestException;
    import cloud.wconnect.commonlib.exception.InternalServerErrorException;
    import cloud.wconnect.commonlib.exception.ParseReportException;
    import cloud.wconnect.commonlib.pojo.bo.ReportDataBO;
    import cloud.wconnect.commonlib.pojo.dto.PropertyDTO;
    import cloud.wconnect.commonlib.util.CellUtils;
    import cloud.wconnect.commonlib.util.FileCheckUtils;
    import cloud.wconnect.commonlib.util.I18nMessageUtils;
    import cloud.wconnect.commonlib.util.UploadUtils;
    import cloud.wconnect.rateshop.cons.revenue.RevenueErrorFiledCons;
    import cloud.wconnect.rateshop.cons.revenue.RevenueImportCons;
    import cloud.wconnect.rateshop.entity.revenue.RevenueImportLog;
    import cloud.wconnect.rateshop.pojo.bo.revenue.RoomTypeRoomsBO;
    import cloud.wconnect.rateshop.pojo.dto.revenue.HandleImportReportDTO;
    import cloud.wconnect.rateshop.pojo.dto.revenue.ResolveExceptionDTO;
    import cloud.wconnect.rateshop.service.revenue.IRevenueRoomTypeRoomsService;
    import cloud.wconnect.rateshop.service.revenue.report.IImportRevenueReport;
    import cloud.wconnect.rateshop.service.revenue.report.IRevenueReportCommonService;
    import com.baomidou.dynamic.datasource.annotation.DS;
    import com.google.common.collect.Lists;
    import com.google.common.collect.Maps;
    import org.apache.commons.lang3.StringUtils;
    import org.apache.poi.ss.usermodel.*;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.beans.factory.annotation.Value;
    import org.springframework.stereotype.Service;
    import org.springframework.transaction.annotation.Transactional;
    import org.springframework.web.multipart.MultipartFile;

    import jakarta.annotation.Resource;
    import java.io.IOException;
    import java.io.InputStream;
    import java.math.BigDecimal;
    import java.time.LocalDate;
    import java.time.ZoneId;
    import java.time.format.DateTimeFormatter;
    import java.util.LinkedHashMap;
    import java.util.List;
    import java.util.Map;
    import java.util.Objects;

    /**
     * @author: yyt
     * 2025-02-12
     */
    @Service
    @DS("rate_shop")
    public class DetailedAvailabilityNext14DaysParseByOther implements IImportRevenueReport {

        @Resource
        private IRevenueReportCommonService revenueImportCommonService;
        @Resource
        private IRevenueRoomTypeRoomsService revenueRoomTypeRoomsService;
        @Autowired
        private I18nMessageUtils i18nMessageUtils;

        @Value("${file-max-size.import-report.revenue}")
        private Long importReportFileMaxSize;

        @Value("${aws-s3.third-party.key.revenue.detailed-availability-next-14-days}")
        private String s3Key;

        private static final String SOURCE = SystemPlatformEnum.OTHER.getDesc();

        private static final int FILE_TYPE = RevenueImportCons.DETAILED_AVAILABILITY_NEXT_14_DAYS;

        /**
         * 判断支持的平台
         *
         * @param handleImportReportDTO 处理报告
         * @return boolean
         * @author yyt
         * @date 2025/2/12
         */
        @Override
        public boolean supports(HandleImportReportDTO handleImportReportDTO) {
            return SOURCE.equals(handleImportReportDTO.getSource()) && FILE_TYPE == handleImportReportDTO.getReportType();
        }

        /**
         * 检查文件
         *
         * @param propertyDTO 资产
         * @param file        文件
         * @author yyt
         * @date 2025/2/12
         **/
        @Override
        public void checkFile(PropertyDTO propertyDTO, MultipartFile file) {
            // 限制文件大小
            FileCheckUtils.checkFileSize(file, importReportFileMaxSize);

            // 校验文件格式
            UploadUtils.checkFileFormat(file, Lists.newArrayList("xls", "xlsx"));

            try (InputStream in = file.getInputStream();
                 Workbook workbook = WorkbookFactory.create(in)) {
                Sheet sheet = workbook.getSheetAt(0);
                Row titleRow = sheet.getRow(0);
                if (Objects.isNull(titleRow)) {
                    throw new BadRequestException(CommonErrorI18nConstant.IMPORT_REPORT_NOT_REQUIRED,
                            BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                            "The imported report is not the one required for the current functionality");
                }

                List<String> roomTypes = Lists.newArrayListWithExpectedSize(titleRow.getLastCellNum());
                for (int j = 1; j < titleRow.getLastCellNum(); j++) {
                    String roomType = CellUtils.convertCellStr(titleRow.getCell(j));
                    if (StringUtils.isBlank(roomType)) {
                        continue;
                    }
                    if(roomTypes.contains(roomType)){
                        throw new BadRequestException(RevenueErrorFiledCons.ROOM_TYPE_REPEAT,
                                BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                                "The room type repeat");
                    }
                    roomTypes.add(roomType);
                }

                DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern(BaseConstant.DATE_FORMAT_YYYY_MM_DD);

                for (int i = 1; i <= sheet.getLastRowNum(); i++) {
                    Row row = sheet.getRow(i);
                    if (Objects.isNull(row)) {
                        continue;
                    }

                    Cell dateCell = row.getCell(0);
                    if (Objects.isNull(dateCell) || StringUtils.isBlank(dateCell.toString())) {
                        throw new BadRequestException(CommonErrorI18nConstant.CANNOT_EMPTY,
                                i18nMessageUtils.getErrorParamMapNeedTranslate(CommonFieldI18nConstant.DATE),
                                BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                                "The date must not be null");
                    } else {
                        try {
                            LocalDate.parse(CellUtils.convertCellStr(dateCell, BaseConstant.DATE_FORMAT_YYYY_MM_DD), dateTimeFormatter);
                        } catch (Exception e) {
                            LinkedHashMap<String, Boolean> map = new LinkedHashMap<>();
                            map.put(CommonFieldI18nConstant.DATE, true);
                            map.put(BaseConstant.DATE_FORMAT_YYYY_MM_DD,false);
                            throw new BadRequestException(CommonErrorI18nConstant.FORMAT_INCORRECT,
                                    i18nMessageUtils.getErrorParamMap(map),
                                    BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                                    "The date format error");
                        }
                    }

                    for (int j = 1; j < row.getLastCellNum(); j++) {
                        Cell roomsCell = row.getCell(j);
                        if (Objects.nonNull(roomsCell) && StringUtils.isNotBlank(roomsCell.toString())) {
                            BigDecimal rooms;
                            try {
                                rooms = new BigDecimal(CellUtils.convertCellStr(roomsCell));
                            } catch (Exception e) {
                                throw new BadRequestException(CommonErrorI18nConstant.MUST_NUMBER,
                                        i18nMessageUtils.getErrorParamMapNeedTranslate(RevenueErrorFiledCons.ROOMS),
                                        BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                                        "The rooms must be a number");
                            }
                            if (rooms.toBigInteger().toString().length() > 6) {
                                throw new BadRequestException(CommonErrorI18nConstant.NUMBER_OVERLENGTH,
                                        i18nMessageUtils.getErrorParamMapNeedTranslate(RevenueErrorFiledCons.ROOMS, 6, 0),
                                        BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                                        "The rooms exceed the limit");
                            }
                        }

                        String roomType = CellUtils.convertCellStr(titleRow.getCell(j));
                        if (StringUtils.isEmpty(roomType)) {
                            throw new BadRequestException(CommonErrorI18nConstant.CANNOT_EMPTY,
                                    i18nMessageUtils.getErrorParamMapNeedTranslate(RevenueErrorFiledCons.ROOM_TYPE),
                                    BaseErrorCodeEnum.DATA_VALIDATION_ERROR.getCode(),
                                    "The room type must not be empty");
                        }
                    }
                }

            } catch (BadRequestException e) {
                throw e;
            } catch (Exception e) {
                throw new InternalServerErrorException(BaseErrorCodeEnum.SYSTEM_ERROR.getI18nCode(),
                        BaseErrorCodeEnum.SYSTEM_ERROR.getCode(), "parse file failed", e);
            }
        }

        /**
         * 处理导入记录
         *
         * @param propertyDTO         资产
         * @param importLog           导入记录
         * @param resolveExceptionDTO 选择条件
         * @param file                文件
         * @param userId              操作人
         * @return 导入记录
         * @author yyt
         * @date 2025/2/12
         **/
        @Override
        public RevenueImportLog handleImportLog( PropertyDTO propertyDTO, RevenueImportLog importLog,
                                                ResolveExceptionDTO resolveExceptionDTO, MultipartFile file, int userId) {
            return revenueImportCommonService.handleImportLog(propertyDTO, importLog, resolveExceptionDTO, s3Key, file, userId);
        }

        /**
         * 处理文件
         *
         * @param propertyDTO 资产信息
         * @param file        文件
         * @param importLog   导入记录
         * @param userId      操作人
         * @author yyt
         * @date 2025/2/12
         **/
        @Override
        @Transactional(rollbackFor = Exception.class)
        public void handleFile(PropertyDTO propertyDTO, MultipartFile file, RevenueImportLog importLog, Integer userId) {
            boolean importSuccess = true;
            try (InputStream inputStream = file.getInputStream()) {
                ReportDataBO<RoomTypeRoomsBO> reportDataBO = parseFile(propertyDTO, inputStream, importLog.getStartDate(), importLog.getEndDate());
                if (reportDataBO.getHandleData()) {
                    revenueRoomTypeRoomsService.save(propertyDTO, importLog.getTheDate(), reportDataBO.getData(), userId);
                }
            } catch (Exception e) {
                importSuccess = false;
                throw new ParseReportException(BaseErrorCodeEnum.SYSTEM_ERROR.getI18nCode(),
                        BaseErrorCodeEnum.SYSTEM_ERROR.getCode(), "parse report error", e);
            } finally {
                revenueImportCommonService.updateImportStatus(importLog, importSuccess, userId);
            }
        }

        /**
         * 解析文件
         *
         * @param propertyDTO 资产
         * @param inputStream 文件
         * @param startDate   开始日期
         * @param endDate     结束日期
         * @return RoomTypeRoomsBO
         * @author yyt
         * @date 2024/8/16
         **/
        private ReportDataBO<RoomTypeRoomsBO> parseFile(PropertyDTO propertyDTO, InputStream inputStream, Long startDate, Long endDate) throws IOException {
            List<RoomTypeRoomsBO> roomTypeRoomsBOS = Lists.newArrayList();
            Workbook workbook = WorkbookFactory.create(inputStream);
            Sheet sheet = workbook.getSheetAt(0);

            Row titelRow = sheet.getRow(0);
            Map<String, Integer> indexByRoomType = Maps.newLinkedHashMapWithExpectedSize(titelRow.getLastCellNum());
            for (int i = 1; i < titelRow.getLastCellNum(); i++) {
                String roomType = CellUtils.convertCellStr(titelRow.getCell(i));
                if (StringUtils.isBlank(roomType)) continue;

                indexByRoomType.put(roomType, i);
            }

            // 报告中的数据在处理范围内
            boolean isInScope = false;

            ZoneId zoneId = ZoneId.of(propertyDTO.getTimezone());
            DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern(BaseConstant.DATE_FORMAT_YYYY_MM_DD);
            for (int i = 1; i <= sheet.getLastRowNum(); i++) {
                Row row = sheet.getRow(i);
                String date = CellUtils.convertCellStr(row.getCell(0), BaseConstant.DATE_FORMAT_YYYY_MM_DD);
                long targetDate = LocalDate.parse(date, dateTimeFormatter).atStartOfDay(zoneId).toEpochSecond();
                if (targetDate < startDate || targetDate > endDate) {
                    continue;
                } else {
                    isInScope = true;
                }

                indexByRoomType.forEach((roomType, index) -> {
                    String rooms = CellUtils.convertCellStr(row.getCell(index));
                    if (StringUtils.isNotBlank(rooms)) {
                        roomTypeRoomsBOS.add(new RoomTypeRoomsBO(targetDate, roomType, Integer.parseInt(rooms)));
                    }
                });
            }

            workbook.close();

            return new ReportDataBO<>(isInScope, roomTypeRoomsBOS);
        }

    }