@Component
@Slf4j
public class ReadShareFilesTask {

    @Resource
    InspectEquipmentInfoService inspectEquipmentInfoService;

    /**
     * 現(xiàn)在暫定每天執(zhí)行
     */
    @Scheduled(cron = "0 0 12 * * ?")
    public void executeReadShareFiles() {
        LocalDateTime dateTime = LocalDateTime.now().minusDays(1);
        List<InspectEquipmentInfo> equipmentInfos = inspectEquipmentInfoService.lambdaQuery().eq(InspectEquipmentInfo::getEquipmentType, EquipmentTypeEnum.FILE_SHARING.getCode()).list();
        equipmentInfos.forEach(equipmentInfo -> {
            try {
                log.info("開(kāi)始讀取設(shè)備:{}, 設(shè)備名稱:{}, 設(shè)備ip:{}", equipmentInfo.getEquipmentCode(), equipmentInfo.getEquipmentName(), equipmentInfo.getEquipmentAddr());
                FileReadStrategy fileReadStrategy = ClassExecuteServiceFactory.getFileReadStrategy(equipmentInfo.getEquipmentCode());
                ValidationUtils.checkNotNull(fileReadStrategy, "策略未找到");
                fileReadStrategy.listFile(equipmentInfo.getEquipmentAddr(),dateTime);
            } catch (Exception e) {
                log.error("讀取設(shè)備:{}, 設(shè)備名稱:{}, 設(shè)備ip:{} 失敗", equipmentInfo.getEquipmentCode(), equipmentInfo.getEquipmentName(), equipmentInfo.getEquipmentAddr(), e);
            }
        });
    }
}

@Component
public class ClassExecuteServiceFactory implements ApplicationContextAware {

    private final static Map<String, FileReadStrategy> CLASS_CODE_ABSTRACT_CLASS_HANDLE_MAP = new HashMap<>();

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        Map<String, FileReadStrategy> types = applicationContext.getBeansOfType(FileReadStrategy.class);
        types.values().forEach(e -> CLASS_CODE_ABSTRACT_CLASS_HANDLE_MAP.putIfAbsent(e.getCode(), e));
    }

    public static FileReadStrategy getFileReadStrategy(String code) {
        return CLASS_CODE_ABSTRACT_CLASS_HANDLE_MAP.get(code);
    }
}

public interface FileReadStrategy {

    String getCode();

    void listFile(String ip, LocalDateTime dateTime);

    void handleFileRead(InputStream inputStream, InspectEquipmentInfo info);
}

@Component
@Slf4j
public class SpectrometerFileReadStrategy implements FileReadStrategy {

    @Resource
    private InspectEquipmentInfoService equipmentInfoService;

    @Resource
    private InspectEquipmentRecordService recordService;

    @Resource
    private ReceiveEquipmentApi receiveEquipmentApi;

    @Override
    public String getCode() {
        return ShareFilesAddressEnum.CODE.getCode();
    }

    @Override
    public void listFile(String ip, LocalDateTime dateTime) {
        InspectEquipmentInfo equipmentInfo = equipmentInfoService.lambdaQuery().eq(InspectEquipmentInfo::getEquipmentAddr, ip).one();
        if (ObjectUtil.isEmpty(equipmentInfo)) {
            log.error("該設(shè)備未配置到設(shè)備表當(dāng)中，設(shè)備地址 {}", ip);
            return;
        }
        SmbFileReader.listSmbFile(equipmentInfo, dateTime);
    }

    @Override
    public void handleFileRead(InputStream inputStream, InspectEquipmentInfo info) {
        try {
            if (inputStream == null) {
                log.error("無(wú)法找到ICP光譜儀CSV文件");
                return;
            }
            //根據(jù)文件流解析數(shù)據(jù)
      }

@Slf4j
public class SmbFileReader {

    public static void listSmbFile(InspectEquipmentInfo info, LocalDateTime dateTime) {
        String path = info.getPath();
        String ip = info.getEquipmentAddr();
        String user = info.getUser();
        String pass = info.getPass();
        String suffix = info.getSuffix();
        String code = info.getEquipmentCode();
        SMBClient client = new SMBClient();
        try (Connection connection = client.connect(ip)) {
            AuthenticationContext ac = new AuthenticationContext(user, pass.toCharArray(), null);
            Session session = connection.authenticate(ac);
            try (DiskShare share = (DiskShare) session.connectShare(path)) {
                for (FileIdBothDirectoryInformation f : share.list("", suffix)) {
                    FileTime changeTime = f.getLastWriteTime();
                    long windowsTimestamp = changeTime.toEpochMillis();
                    Instant instant = Instant.ofEpochMilli(windowsTimestamp);
                    LocalDateTime localDateTime = LocalDateTime.ofInstant(instant, ZoneId.systemDefault());
                    if (dateTime.isAfter(localDateTime)) {
                        log.info("File : {},Change Time : {}, 跳過(guò)", f.getFileName(), localDateTime);
                        continue;
                    }
                    log.info("File : {},Change Time : {}", f.getFileName(), localDateTime);
                    String fileUrl = f.getFileName();
                    File smbFileRead = share.openFile(fileUrl, EnumSet.of(AccessMask.GENERIC_READ), null, SMB2ShareAccess.ALL, SMB2CreateDisposition.FILE_OPEN, null);
                    InputStream in = smbFileRead.getInputStream();
                    ClassExecuteServiceFactory.getFileReadStrategy(code).handleFileRead(in, info);
                    log.info("File : {} read success", fileUrl);
                }
            }
        } catch (Exception e) {
            log.error("Error reading file from SMB: {}", e.getMessage(), e);
        }

    }

}