From 7ae1a962680d65319909c616f3edfeff43b914a6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B7=AF=E6=98=8E=E6=85=A7?= <1191093413@qq.com> Date: Wed, 7 Aug 2024 15:14:06 +0800 Subject: [PATCH] =?UTF-8?q?feature:ftp=E6=96=87=E4=BB=B6=E5=AD=98=E5=82=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../baseservice/constant/ConfigurerInfo.java | 2 + order-management-order/pom.xml | 12 + .../config/FtpClientPooledObjectFactory.java | 88 +++++ .../modules/bespoke/config/FtpConfig.java | 92 +++++ .../modules/bespoke/config/FtpProperties.java | 20 + .../modules/bespoke/service/FtpProcessor.java | 30 ++ .../bespoke/service/OrderHandleService.java | 4 + .../service/impl/DefaultFtpProcessor.java | 361 ++++++++++++++++++ .../service/impl/OrderHandleServiceImpl.java | 76 +++- 9 files changed, 679 insertions(+), 6 deletions(-) create mode 100644 order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpClientPooledObjectFactory.java create mode 100644 order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpConfig.java create mode 100644 order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpProperties.java create mode 100644 order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/FtpProcessor.java create mode 100644 order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/impl/DefaultFtpProcessor.java diff --git a/order-management-base-service/src/main/java/com/dxhy/order/baseservice/constant/ConfigurerInfo.java b/order-management-base-service/src/main/java/com/dxhy/order/baseservice/constant/ConfigurerInfo.java index bc936b0..994043f 100644 --- a/order-management-base-service/src/main/java/com/dxhy/order/baseservice/constant/ConfigurerInfo.java +++ b/order-management-base-service/src/main/java/com/dxhy/order/baseservice/constant/ConfigurerInfo.java @@ -71,6 +71,8 @@ public class ConfigurerInfo { public static final String ERROR_CODE = "9999"; + + public static final String DEFAULT_FTP_PATH_ENCODING="ISO-8859-1"; /** * 查询数据相关 */ diff --git a/order-management-order/pom.xml b/order-management-order/pom.xml index 4fc8464..2f3097d 100644 --- a/order-management-order/pom.xml +++ b/order-management-order/pom.xml @@ -17,6 +17,18 @@ + + commons-net + commons-net + 3.6 + + + org.apache.commons + commons-pool2 + 2.7.0 + + + com.baomidou diff --git a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpClientPooledObjectFactory.java b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpClientPooledObjectFactory.java new file mode 100644 index 0000000..905a73c --- /dev/null +++ b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpClientPooledObjectFactory.java @@ -0,0 +1,88 @@ +package com.dxhy.order.consumer.modules.bespoke.config; + +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.net.ftp.FTPClient; +import org.apache.commons.pool2.PooledObject; +import org.apache.commons.pool2.PooledObjectFactory; +import org.apache.commons.pool2.impl.DefaultPooledObject; + +import java.io.IOException; + + +/** + * @Author Minco + * @Date 15:19 2021-02-19 + * @Description ftp工厂类 + *//***/ + +@Slf4j +public class FtpClientPooledObjectFactory implements PooledObjectFactory { + + private FtpProperties ftpProperties; + + public FtpClientPooledObjectFactory(FtpProperties ftpProperties) { + this.ftpProperties = ftpProperties; + } + + @Override + public PooledObject makeObject() throws Exception { + FTPClient ftpClient = new FTPClient(); + try { + ftpClient.connect(ftpProperties.getIp(), Integer.valueOf(ftpProperties.getPort())); + ftpClient.login(ftpProperties.getUsername(), ftpProperties.getPassword()); + log.info("连接ftp服务返回码：" + ftpClient.getReplyCode()); + ftpClient.setBufferSize(ftpProperties.getBufferSize()); + ftpClient.setControlEncoding(ftpProperties.getEncoding()); + ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE); + ftpClient.enterLocalPassiveMode(); + return new DefaultPooledObject<>(ftpClient); + } catch (Exception e) { + if (ftpClient.isAvailable()) { + ftpClient.disconnect(); + } + ftpClient = null; + log.error("建立ftp连接失败！", (Object) e.getStackTrace()); + throw new Exception("建立ftp连接失败！", e); + } + } + + @Override + public void destroyObject(PooledObject pooledObject) throws Exception { + FTPClient ftpClient = getObject(pooledObject); + if (null != ftpClient && ftpClient.isConnected()) { + ftpClient.disconnect(); + } + } + + @Override + public boolean validateObject(PooledObject pooledObject) { + FTPClient ftpClient = getObject(pooledObject); + if (null == ftpClient || !ftpClient.isConnected()) { + return false; + } + try { + ftpClient.changeWorkingDirectory("/"); + return true; + } catch (IOException e) { + log.error("验证ftp连接失败！", (Object) e.getStackTrace()); + return false; + } + } + + @Override + public void activateObject(PooledObject pooledObject) throws Exception { + + } + + @Override + public void passivateObject(PooledObject pooledObject) throws Exception { + + } + + private FTPClient getObject(PooledObject pooledObject) { + if (null == pooledObject || null == pooledObject.getObject()) { + return null; + } + return pooledObject.getObject(); + } +} diff --git a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpConfig.java b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpConfig.java new file mode 100644 index 0000000..bb538b3 --- /dev/null +++ b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpConfig.java @@ -0,0 +1,92 @@ +package com.dxhy.order.consumer.modules.bespoke.config; + + +import com.dxhy.order.consumer.modules.bespoke.service.FtpProcessor; +import com.dxhy.order.consumer.modules.bespoke.service.impl.DefaultFtpProcessor; +import org.apache.commons.net.ftp.FTPClient; +import org.apache.commons.pool2.ObjectPool; +import org.apache.commons.pool2.impl.GenericObjectPool; +import org.apache.commons.pool2.impl.GenericObjectPoolConfig; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; +import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; +import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; +import org.springframework.boot.context.properties.EnableConfigurationProperties; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import javax.annotation.PreDestroy; + + +@Configuration +@ConditionalOnClass({GenericObjectPool.class, FTPClient.class}) +@ConditionalOnProperty(prefix = "ftp", name = "isopen", havingValue = "true", matchIfMissing = true) +@EnableConfigurationProperties(FtpProperties.class) +public class FtpConfig { + + private Logger log = LoggerFactory.getLogger(FtpConfig.class); + + private final FtpProperties ftpProperties; + + @Autowired + public FtpConfig(FtpProperties ftpProperties) { + this.ftpProperties = ftpProperties; + } + + private ObjectPool pool; + + /** + * 预加载FTPClient连接到对象池中 + * + * @param initialSize + * @param maxIdle + */ + private void preLoadingFtpClient(Integer initialSize, int maxIdle) { + //如果初始化大小为null或者小于等于0，则不执行逻辑 + if (null == initialSize || initialSize <= 0) { + return; + } + int size = Math.min(initialSize, maxIdle); + try { + for (int i = 0; i < size; i++) { + pool.addObject(); + } + } catch (Exception e) { + log.error("预加载失败！", (Object) e.getStackTrace()); + } + } + + /** + * 销毁方法 + */ + @PreDestroy + public void destroy() { + if (null != pool) { + pool.close(); + log.info("销毁ftp客户端连接池。。。"); + } + } + + /** + * 判断不存在业务Service时初始化默认Bean到Spring + */ + @Bean + @ConditionalOnMissingBean(FtpProcessor.class) + public FtpProcessor ftpProcessor() { + log.info("没有找到ftp处理器，执行创建默认处理器"); + GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig(); + poolConfig.setTestOnBorrow(true); + poolConfig.setTestOnReturn(true); + poolConfig.setTestWhileIdle(true); + poolConfig.setMinEvictableIdleTimeMillis(6000); + poolConfig.setSoftMinEvictableIdleTimeMillis(50000); + poolConfig.setTimeBetweenEvictionRunsMillis(30000); + pool = new GenericObjectPool<>(new FtpClientPooledObjectFactory(ftpProperties), poolConfig); + preLoadingFtpClient(ftpProperties.getInitialSize(), poolConfig.getMaxIdle()); + DefaultFtpProcessor processor = new DefaultFtpProcessor(ftpProperties); + processor.setFtpClientPool(pool); + processor.setHasInit(true); + return processor; + } +} diff --git a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpProperties.java b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpProperties.java new file mode 100644 index 0000000..dbfe75c --- /dev/null +++ b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/config/FtpProperties.java @@ -0,0 +1,20 @@ +package com.dxhy.order.consumer.modules.bespoke.config; + +import lombok.Data; +import org.springframework.boot.context.properties.ConfigurationProperties; + +@Data +@ConfigurationProperties(prefix = "ftp") +public class FtpProperties { + + private String ip; + private String port; + private String username; + private String password; + private String basePath; + private Integer initialSize = 0; + private String encoding = "UTF-8"; + private Integer bufferSize = 4096; + private Integer retryCount = 3; + +} diff --git a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/FtpProcessor.java b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/FtpProcessor.java new file mode 100644 index 0000000..8ddfcdd --- /dev/null +++ b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/FtpProcessor.java @@ -0,0 +1,30 @@ +package com.dxhy.order.consumer.modules.bespoke.service; + +import org.apache.commons.net.ftp.FTPClient; + +import java.io.IOException; +import java.io.InputStream; +import java.util.List; + + +public interface FtpProcessor { + + boolean uploadFile(String path, String fileName, String originFileName); + + boolean uploadFile(String path, String fileName, InputStream inputStream); + + boolean downloadFile(String path, String fileName, String localPath); + + boolean deleteFile(String path, String fileName); + + boolean createDirectory(String remote, FTPClient ftpClient) throws IOException; + + boolean existFile(String path, FTPClient ftpClient) throws IOException; + + boolean makeDirectory(String directory, FTPClient ftpClient); + + List retrieveFileNames(String remotePath) throws IOException; + + + +} diff --git a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/OrderHandleService.java b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/OrderHandleService.java index 5e89290..c8b47b5 100644 --- a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/OrderHandleService.java +++ b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/OrderHandleService.java @@ -2,6 +2,7 @@ package com.dxhy.order.consumer.modules.bespoke.service; import com.dxhy.base.common.model.R; import com.dxhy.order.baseservice.exception.OrderReceiveException; +import com.dxhy.order.baseservice.model.OrderInvoiceInfo; import com.dxhy.order.consumer.modules.bespoke.model.req.ReqParam; import java.util.List; @@ -44,4 +45,7 @@ public interface OrderHandleService { * @throws OrderReceiveException */ int createOrderByCopyInvoice(String fpqqlsh, List shList, List entList,String fpzldm) throws OrderReceiveException; + + + String saveInvoiceFile(OrderInvoiceInfo orderInvoiceInfo); } diff --git a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/impl/DefaultFtpProcessor.java b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/impl/DefaultFtpProcessor.java new file mode 100644 index 0000000..b176b05 --- /dev/null +++ b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/impl/DefaultFtpProcessor.java @@ -0,0 +1,361 @@ +package com.dxhy.order.consumer.modules.bespoke.service.impl; + + +import com.dxhy.order.baseservice.constant.ConfigurerInfo; +import com.dxhy.order.consumer.modules.bespoke.config.FtpProperties; +import com.dxhy.order.consumer.modules.bespoke.service.FtpProcessor; +import lombok.Getter; +import lombok.Setter; +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.lang3.ArrayUtils; +import org.apache.commons.lang3.StringUtils; +import org.apache.commons.net.ftp.FTPClient; +import org.apache.commons.net.ftp.FTPFile; +import org.apache.commons.pool2.ObjectPool; +import org.springframework.util.Assert; +import java.io.*; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; +import java.util.Objects; +import java.util.stream.Collectors; + + +@Getter +@Setter +@Slf4j +public class DefaultFtpProcessor implements FtpProcessor { + + private final FtpProperties ftpProperties; + + //连接池初始化标志 + private boolean hasInit = false; + //连接池 + private ObjectPool ftpClientPool; + + public DefaultFtpProcessor(FtpProperties ftpProperties) { + this.ftpProperties = ftpProperties; + } + + /** + * 上传文件 + * + * @param path ftp服务器保存地址 + * @param fileName 上传到ftp的文件名 + * @param originFileName 等待上传的文件名（绝对地址或路径） + */ + @Override + public boolean uploadFile(String path, String fileName, String originFileName) { + boolean flag = false; + try { + InputStream inputStream = new FileInputStream(new File(originFileName)); + flag=uploadFile( path, fileName, inputStream); + } catch (Exception e) { + //log.error("上传文件出错！", (Object) e.getStackTrace()); + e.printStackTrace(); + } + return flag; + } + + /** + * 上传文件 + * + * @param path ftp服务器保存地址 + * @param fileName 上传到ftp的文件名 + * @param inputStream 文件流 + */ + @Override + public boolean uploadFile(String path, String fileName, InputStream inputStream){ + boolean flag = false; + FTPClient ftpClient = getFtpClient(); + try { + ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE); + createDirectory(path, ftpClient); + ftpClient.makeDirectory(path); + ftpClient.changeWorkingDirectory(path); + ftpClient.storeFile(fileName, inputStream); + flag = true; + } catch (Exception e) { + log.error("上传文件出错！:{}", e); + } finally { + if(null!=inputStream) { + try { + inputStream.close(); + } catch (IOException e) { + log.error("关流失败"); + } + } + releaseFtpClient(ftpClient); + } + return flag; + } + + /** + * 下载文件 + * + * @param path ftp服务器文件路径 + * @param fileName 文件名称 + * @param localPath 下载后的路径 + */ + @Override + public boolean downloadFile(String path, String fileName, String localPath) { + boolean flag = false; + OutputStream outputStream = null; + FTPClient ftpClient = getFtpClient(); + try { + ftpClient.changeWorkingDirectory(path); + FTPFile[] files = ftpClient.listFiles(); + for (FTPFile file : files) { + if (fileName.equalsIgnoreCase(file.getName())) { + File localFile = new File(localPath + "/" + file.getName()); + outputStream = new FileOutputStream(localFile); + ftpClient.retrieveFile(file.getName(), outputStream); + outputStream.close(); + } + } + flag = true; + } catch (IOException e) { + log.error("下载文件出错！", (Object) e.getStackTrace()); + } finally { + releaseFtpClient(ftpClient); + if (null != outputStream) { + try { + outputStream.close(); + } catch (IOException e) { + log.error("关闭输出流出错！", (Object) e.getStackTrace()); + } + } + } + return flag; + } + + /** + * 删除文件 + * + * @param path ftp文件路径 + * @param fileName 文件名 + */ + @Override + public boolean deleteFile(String path, String fileName) { + boolean flag = false; + FTPClient ftpClient = getFtpClient(); + try { + ftpClient.changeWorkingDirectory(path); + ftpClient.dele(fileName); + ftpClient.logout(); + flag = true; + } catch (IOException e) { + log.error("删除文件出错！", (Object) e.getStackTrace()); + } finally { + releaseFtpClient(ftpClient); + } + return flag; + } + + /** + * 创建多层目录，如果ftp服务器已存在该目录，则不创建，如果没有，则创建 + * + * @param remote 创建的目录 + * @param ftpClient + */ + @Override + public boolean createDirectory(String remote, FTPClient ftpClient) throws IOException { + String directory = remote + "/"; + //如果远程目录不存在，则递归创建远程目录 + if (!directory.equalsIgnoreCase("/") && !changeWorkingDirectory(directory, ftpClient)) { + int start = 0; + int end = 0; + if (directory.startsWith("/")) { + start = 1; + } + end = directory.indexOf("/", start); + String path = ""; + String paths = ""; + do { + String subDirectory = new String(remote.substring(start, end).getBytes(ftpProperties.getEncoding()), ConfigurerInfo.DEFAULT_FTP_PATH_ENCODING); + path = path + "/" + subDirectory; + if (!existFile(path, ftpClient)) { + if (makeDirectory(subDirectory, ftpClient)) { + changeWorkingDirectory(subDirectory, ftpClient); + } else { + log.warn("创建目录[" + subDirectory + "]失败"); + changeWorkingDirectory(subDirectory, ftpClient); + } + } else { + changeWorkingDirectory(subDirectory, ftpClient); + } + paths = paths + "/" + subDirectory; + start = end + 1; + end = directory.indexOf("/", start); + } while (end <= start); + } + return true; + } + + /** + * 判断ftp服务器的路径或文件是否存在 + * + * @param path + * @param ftpClient + */ + @Override + public boolean existFile(String path, FTPClient ftpClient) throws IOException { + boolean flag = false; + FTPFile[] files = ftpClient.listFiles(path); + if (files.length > 0) { + flag = true; + } + return flag; + } + + /** + * 创建目录 + * + * @param directory + * @param ftpClient + */ + @Override + public boolean makeDirectory(String directory, FTPClient ftpClient) { + boolean flag = true; + try { + flag = ftpClient.makeDirectory(directory); + if (flag) { + log.info("创建文件夹：" + directory); + } + } catch (IOException e) { + log.error("创建文件夹" + directory + "失败！", (Object) e.getStackTrace()); + } + return flag; + } + + /** + * 切换目录 + * + * @param directory 要切换的目录 + * @param ftpClient ftp客户端 + */ + public boolean changeWorkingDirectory(String directory, FTPClient ftpClient) { + boolean flag = true; + try { + flag = ftpClient.changeWorkingDirectory(directory); + if (flag) { + log.info("进入文件夹：" + directory); + } + } catch (IOException e) { + log.error("进入文件夹："+directory+"错误！", (Object) e.getStackTrace()); + } + return flag; + } + + /** + * 按行读取FTP文件 + * + * @param remoteFilePath ftp路径 + */ + public List readFileByLine(String remoteFilePath) throws IOException { + FTPClient ftpClient = getFtpClient(); + try (InputStream inputStream = ftpClient.retrieveFileStream(encodingPath(remoteFilePath)); + BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream))) { + return reader.lines() + .map(line -> StringUtils.trimToEmpty(line)) + .filter(line -> StringUtils.isNotEmpty(line)) + .collect(Collectors.toList()); + } finally { + ftpClient.completePendingCommand(); + releaseFtpClient(ftpClient); + } + } + + /** + * 获取指定路径下的ftp文件 + * + * @param remotePath 指定路径 + */ + public FTPFile[] retrieveFtpFiles(String remotePath) throws IOException { + FTPClient ftpClient = getFtpClient(); + try { + return ftpClient.listFiles(encodingPath(remotePath + "/"), file -> file != null && file.getSize() > 0); + } finally { + releaseFtpClient(ftpClient); + } + } + + /** + * 获取指定ftp路径下的文件名称 + * + * @param remotePath 指定ftp路径 + */ + @Override + public List retrieveFileNames(String remotePath) throws IOException { + FTPFile[] files = retrieveFtpFiles(remotePath); + if (ArrayUtils.isEmpty(files)) { + return new ArrayList<>(); + } + return Arrays.stream(files).filter(Objects::nonNull).map(FTPFile::getName).collect(Collectors.toList()); + } + + /** + * 获取编码后的文件路径 + */ + private String encodingPath(String path) throws UnsupportedEncodingException { + //在FTP协议中，规定文件名编码格式为ISO-8859-1，所以目录名或文件名需要转码 + return new String(path.replaceAll("//", "/").getBytes(ftpProperties.getEncoding()), ConfigurerInfo.DEFAULT_FTP_PATH_ENCODING); + } + + /** + * 获取ftp客户端 + */ + private FTPClient getFtpClient() { + checkFtpClientPoolAvailable(); + FTPClient ftpClient = null; + Exception exception = null; + //获取连接，做多尝试n次 + try { + for (int i = 0; i < ftpProperties.getRetryCount(); i++) { + ftpClient = ftpClientPool.borrowObject(); + ftpClient.enterLocalPassiveMode();//设置为被动模式 + ftpClient.changeWorkingDirectory("/"); + break; + } + } catch (Exception e) { + log.error("无法在连接池中获取ftp客户端！", (Object) e.getStackTrace()); + exception = e; + } + if (null == ftpClient) { + throw new RuntimeException("无法在连接池中获取ftp客户端", exception); + } + return ftpClient; + } + + /** + * 释放ftp客户端 + * + * @param ftpClient + */ + private void releaseFtpClient(FTPClient ftpClient) { + if (null != ftpClient) { + try { + //从ftp连接池中移除ftp客户端 + ftpClientPool.returnObject(ftpClient); + } catch (Exception e) { + try { + //判断客户端是否可用 + if (ftpClient.isAvailable()) { + //销毁连接 + ftpClient.disconnect(); + } + } catch (IOException ex) { + log.error("销毁ftp连接失败！", (Object) e.getStackTrace()); + } + log.error("从ftp连接池移除ftp客户端失败！", (Object) e.getStackTrace()); + } + } + } + + /** + * 检查ftp连接池是否可用 + */ + private void checkFtpClientPoolAvailable() { + Assert.state(hasInit, "ftp未启用或连接失败！"); + } +} diff --git a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/impl/OrderHandleServiceImpl.java b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/impl/OrderHandleServiceImpl.java index 727abfc..5f389d2 100644 --- a/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/impl/OrderHandleServiceImpl.java +++ b/order-management-order/src/main/java/com/dxhy/order/consumer/modules/bespoke/service/impl/OrderHandleServiceImpl.java @@ -1,35 +1,52 @@ package com.dxhy.order.consumer.modules.bespoke.service.impl; +import cn.hutool.core.date.DateUtil; +import cn.hutool.core.io.IoUtil; import cn.hutool.extra.spring.SpringUtil; import com.dxhy.base.common.constant.ConfigureConstant; +import com.dxhy.base.common.constant.OrderInfoEnum; import com.dxhy.base.common.model.OrderInfo; import com.dxhy.base.common.model.OrderItemInfo; import com.dxhy.base.common.model.OrderProcessInfo; import com.dxhy.base.common.model.R; import com.dxhy.base.common.model.qd.OrderInfoQd; import com.dxhy.base.common.utils.DistributedKeyMaker; +import com.dxhy.base.service.module.thirdservice.simsback.model.TaxEquipmentInfo; +import com.dxhy.base.service.module.thirdservice.simsback.service.SimsBackService; import com.dxhy.base.thirdservice.module.base.service.BaseService; import com.dxhy.order.baseservice.exception.OrderReceiveException; import com.dxhy.order.baseservice.model.OrderInvoiceInfo; import com.dxhy.order.baseservice.model.OrderOriginExtendInfo; import com.dxhy.order.consumer.dao.OrderOriginExtendInfoMapper; import com.dxhy.order.consumer.dao.OrderProcessInfoMapper; +import com.dxhy.order.consumer.modules.bespoke.config.FtpConfig; +import com.dxhy.order.consumer.modules.bespoke.config.FtpProperties; import com.dxhy.order.consumer.modules.bespoke.config.ServiceConfig; import com.dxhy.order.consumer.modules.bespoke.model.req.ReqParam; +import com.dxhy.order.consumer.modules.bespoke.service.FtpProcessor; import com.dxhy.order.consumer.modules.bespoke.service.OrderHandleService; import com.dxhy.order.consumer.modules.bespoke.service.OrderService; import com.dxhy.order.invoice.module.invoice.dao.OrderInfoMapper; import com.dxhy.order.invoice.module.invoice.dao.OrderInfoQdMapper; import com.dxhy.order.invoice.module.invoice.dao.OrderInvoiceInfoMapper; import com.dxhy.order.invoice.module.invoice.dao.OrderItemInfoMapper; +import com.dxhy.order.invoice.module.invoice.model.FileDownLoad; +import com.dxhy.order.invoice.module.invoice.service.OrderInvoiceInfoService; import lombok.SneakyThrows; import lombok.extern.slf4j.Slf4j; import org.apache.commons.collections4.CollectionUtils; import org.apache.commons.lang3.ObjectUtils; +import org.apache.commons.lang3.time.DateUtils; import org.springframework.stereotype.Service; + import javax.annotation.Resource; +import java.io.ByteArrayInputStream; +import java.io.IOException; +import java.nio.charset.StandardCharsets; import java.util.Date; import java.util.List; +import java.util.jar.Pack200; + @Slf4j @Service public class OrderHandleServiceImpl implements OrderHandleService { @@ -55,6 +72,18 @@ public class OrderHandleServiceImpl implements OrderHandleService { @Resource private BaseService baseService; + @Resource + private OrderInvoiceInfoService orderInvoiceInfoService; + + @Resource + private SimsBackService simsBackService; + + @Resource + private FtpProperties ftpProperties; + + @Resource + private FtpProcessor ftpProcessor; + @Override public R readOrder(ReqParam reqParam) { @@ -82,7 +111,7 @@ public class OrderHandleServiceImpl implements OrderHandleService { } @Override - public int createOrderByCopyInvoice(String fpqqlsh, List shList, List entList,String fpzldm) throws OrderReceiveException { + public int createOrderByCopyInvoice(String fpqqlsh, List shList, List entList, String fpzldm) throws OrderReceiveException { //查询发票信息 OrderInvoiceInfo orderInvoiceInfoParam = new OrderInvoiceInfo(); orderInvoiceInfoParam.setFpqqlsh(fpqqlsh); @@ -116,11 +145,11 @@ public class OrderHandleServiceImpl implements OrderHandleService { String orderProcessId = baseService.getGenerateShotKey(); String fpqqlshNew = DistributedKeyMaker.generateShotKey(); String ddh = orderInfo.getDdh().concat("copy"); - log.info("订单号：{}，长度{}",ddh,ddh.length()); - if (ddh.length()>50){ + log.info("订单号：{}，长度{}", ddh, ddh.length()); + if (ddh.length() > 50) { log.info("ddh长度大于50,截取后50位"); - ddh = ddh.substring(ddh.length()-50); - log.info("截取后50位ddh:"+ddh); + ddh = ddh.substring(ddh.length() - 50); + log.info("截取后50位ddh:" + ddh); } Date now = new Date(); //订单信息修改 @@ -189,7 +218,7 @@ public class OrderHandleServiceImpl implements OrderHandleService { OrderOriginExtendInfo orderOriginExtendInfoSelect = new OrderOriginExtendInfo(); orderOriginExtendInfoSelect.setOrderId(oldOrderId); List orderOriginExtendInfos = orderOriginExtendInfoMapper.queryOriginOrderByOrder(orderOriginExtendInfoSelect, shList); - if (CollectionUtils.isNotEmpty(orderOriginExtendInfos)){ + if (CollectionUtils.isNotEmpty(orderOriginExtendInfos)) { for (OrderOriginExtendInfo orderOriginExtendInfo : orderOriginExtendInfos) { OrderOriginExtendInfo orderOrginOrder = new OrderOriginExtendInfo(); orderOrginOrder.setCreateTime(now); @@ -219,4 +248,39 @@ public class OrderHandleServiceImpl implements OrderHandleService { } return ConfigureConstant.INT_1; } + + @Override + public String saveInvoiceFile(OrderInvoiceInfo orderInvoiceInfo) { + Date kprq = orderInvoiceInfo.getKprq(); + String xhfNsrsbh = orderInvoiceInfo.getXhfNsrsbh(); + String entId = orderInvoiceInfo.getEntId(); + //查询税控设备类型 + TaxEquipmentInfo taxEquipmentInfo = simsBackService.queryTaxEquipment(xhfNsrsbh, entId); + //获取文件 + List invoiceFile = orderInvoiceInfoService.getInvoiceFile("", orderInvoiceInfo, taxEquipmentInfo.getSksbCode(), OrderInfoEnum.FILE_TYPE_PDF.getKey()); + FileDownLoad fileDownLoad = invoiceFile.get(ConfigureConstant.INT_0); + //拼接路径 + StringBuilder path = new StringBuilder(ftpProperties.getBasePath()); + + path.append(ConfigureConstant.STRING_SLASH_LINE) + .append(orderInvoiceInfo.getXhfNsrsbh()); + if (ObjectUtils.isEmpty(kprq)) { + kprq = new Date(); + } + String format = DateUtil.format(kprq, ConfigureConstant.DATE_FORMAT_DATE_Y_M); + String[] split = format.split(ConfigureConstant.STRING_LINE); + path.append(ConfigureConstant.STRING_SLASH_LINE) + .append(split[ConfigureConstant.INT_0]) + .append(ConfigureConstant.STRING_SLASH_LINE) + .append(split[ConfigureConstant.INT_1]); + + ByteArrayInputStream utf8Stream = IoUtil.toUtf8Stream(fileDownLoad.getFileContent()); + + boolean flag = ftpProcessor.uploadFile(path.toString(), fileDownLoad.getFileName(), utf8Stream); + + if (flag){ + return path.append(ConfigureConstant.STRING_SLASH_LINE).append(fileDownLoad.getFileName()).toString(); + } + return ""; + } }