package shkd.utils; import java.io.File; import java.io.IOException; import java.io.InputStream; import java.net.HttpURLConnection; import java.net.URL; import java.nio.file.Files; import java.util.*; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.google.common.io.ByteStreams; import kd.bos.cache.CacheFactory; import kd.bos.cache.TempFileCache; import kd.bos.context.RequestContext; import kd.bos.entity.MainEntityType; import kd.bos.fileservice.FileItem; import kd.bos.fileservice.FileService; import kd.bos.fileservice.FileServiceFactory; import kd.bos.logging.Log; import kd.bos.logging.LogFactory; import kd.bos.servicehelper.AttachmentServiceHelper; import kd.bos.servicehelper.MetadataServiceHelper; import kd.bos.util.FileNameUtils; public class AttachmentFileUtil { private static final Log logger = LogFactory.getLog(AttachmentFileUtil.class); /**将第三方提供的某个在线文件上传至星瀚对应单据附件上 * @param entity 目标单据实体标识 * @param pk 目标数据id * @param fileUrl 待存入文件的url * @param fileName 待存入文件名称 * @param fileSize 待存入文件大小 * @param attachKey 附件面板标识 * @throws IOException */ public static boolean saveUrlFile2Attchment(String entity,Object pk,String fileUrl,String fileName,String fileSize,String attachKey){ List> attachments = new ArrayList<>(); Map attachItem = new HashMap<>(); //文件名称 attachItem.put("name", fileName); //将url文件转换成输入流 InputStream inputStream = null; HttpURLConnection conn = null; String tempUrl = null; try { URL url = new URL(fileUrl); conn = (HttpURLConnection)url.openConnection(); //设置超时间为30秒 conn.setConnectTimeout(30000); //防止屏蔽程序抓取而返回403错误 conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)"); //得到输入流 inputStream = conn.getInputStream(); //文件大小 attachItem.put("size", fileSize); //获取临时文件缓存 TempFileCache cache = CacheFactory.getCommonCacheFactory().getTempFileCache(); //将文件流存入临时文件缓存（拷贝完成）（最后一个参数为缓存有效期，600秒） tempUrl = cache.saveAsFullUrl(fileName, inputStream, 600); } catch (Exception e) { e.printStackTrace(); return false; } finally { //关闭流 try { inputStream.close(); } catch (IOException e) { throw new RuntimeException(e); } //关闭连接 conn.disconnect(); } //获取文件的缓存路径 // tempUrl = EncreptSessionUtils.encryptSession(tempUrl); // //获取域名前缀 // String address = RequestContext.get().getClientFullContextPath(); // if (!address.endsWith("/")) { // address = address + "/"; // } // //拼接url路径 // String tempUrl3 = address + tempUrl; //修改时间 attachItem.put("lastModified",System.currentTimeMillis()); //获取appId MainEntityType dataEntityType = MetadataServiceHelper.getDataEntityType(entity); String appId = dataEntityType.getAppId(); //将文件缓存中的附件文件上传到正式文件服务器 String actUrl = AttachmentServiceHelper.saveTempToFileService(tempUrl,appId,entity, pk, (String)attachItem.get("name")); TempFileCache cache = CacheFactory.getCommonCacheFactory().getTempFileCache(); InputStream in = cache.getInputStream(tempUrl); FileService service = FileServiceFactory.getAttachmentFileService(); RequestContext requestContext = RequestContext.get(); String uuid = UUID.randomUUID().toString().replace("-", ""); // 生成文件路径-上传附件时远程服务器需要存储文件的位置 String pathParam = FileNameUtils.getAttachmentFileName(requestContext.getTenantId(), requestContext.getAccountId(), uuid, fileName); FileItem fileItem = new FileItem(fileName, pathParam, in); // cache.remove(url); // 上传附件到文件服务器 String downUrl =service.upload(fileItem); //将新文件的物理路径存入map attachItem.put("url", actUrl); attachments.add(attachItem); //维护单据和附件的关系（非文件上传） AttachmentServiceHelper.upload(entity, pk, attachKey, attachments); return true; } /**将文件服务器上某个目录下的所有文件上传至星瀚对应单据附件上 * @param appId 目标单据所属应用标识 * @param entity 目标单据实体标识 * @param pk 目标数据id * @param attachKey 附件面板标识 * @param dirpath 文件目录路径 * @param filenames 文件名称集合 * @author yuxueliang */ public static boolean saveDirFile2Attchment(String appId, String entity, Object pk, String dirpath, String attachKey, JSONArray filenames) throws Exception { logger.info("传入的文件夹路径"+dirpath); List> attachments = new ArrayList<>(1); Map attachItem; String fileName; JSONObject destFile; String tempUrl;//星瀚临时文件url String actUrl;//星瀚正式文件url //获取临时文件缓存 TempFileCache cache = CacheFactory.getCommonCacheFactory().getTempFileCache(); InputStream inputs; FileService service = FileServiceFactory.getAttachmentFileService(); byte[] filecontent; //遍历目标目录，得到所有文件 for (int i = 0; i < filenames.size(); i++) { destFile = filenames.getJSONObject(i); attachments.clear(); // service = FileServiceFactory.getAttachmentFileService(); // cache = CacheFactory.getCommonCacheFactory().getTempFileCache(); fileName = destFile.getString("fileName");//文件名称可能涉及字符集转义，需要注意 attachItem = new HashMap<>(); //附件编号 如果不设置，在前台删除某一个文件时，会把所有编号为空的都删掉 attachItem.put("uid", "nas"+System.currentTimeMillis()); //文件名称 attachItem.put("name", fileName); logger.info("文件名称"+fileName); inputs = service.getInputStream(dirpath+"/"+fileName); // inputs = service.getInputStream("E:\\dobews\\dobe_cosmic6\\replay_pid13348.log"); // logger.info("文件大小"+inputs.available());//Linux服务器上此方法不能返回文件大小，都是0 //文件大小 filecontent = ByteStreams.toByteArray(inputs); attachItem.put("size", filecontent.length); logger.info("文件大小"+filecontent.length); if(filecontent.length == 0){ attachItem.put("size", 60000); } //将文件流存入临时文件缓存（拷贝完成）（最后一个参数为缓存有效期，600秒） tempUrl = cache.saveAsUrl(fileName, filecontent, 600);//此方法执行后inputs会被自动关闭，后续不能再对inputs操作 //修改时间 attachItem.put("lastModified",System.currentTimeMillis()); logger.info("文件临时路径"+tempUrl); //将文件缓存中的附件文件上传到正式文件服务器 actUrl = AttachmentServiceHelper.saveTempToFileService(tempUrl,appId,entity,pk,fileName); logger.info("文件正式路径"+actUrl); //将新文件的物理路径存入map attachItem.put("url", actUrl); attachments.add(attachItem); //维护单据和附件的关系（非文件上传） AttachmentServiceHelper.upload(entity, pk, attachKey, attachments); } return true; } /**将文件服务器上某个目录下的所有文件上传至星瀚对应单据附件上 * @param appId 目标单据所属应用标识 * @param entity 目标单据实体标识 * @param pk 目标数据id * @param attachKey 附件面板标识 * @param dirpath 文件目录路径 * @throws IOException */ public static boolean saveDirFile2Attchment(String appId,String entity,Object pk,String dirpath,String attachKey) throws IOException { File currentDir = new File("."); logger.info("当前系统路径"+currentDir.getAbsolutePath()); logger.info("传入的路径"+dirpath); File dirfile = new File(dirpath); if(!dirfile.isDirectory()){ //传入的路径不是目录或者目录不存在 logger.info("不是目录或者目录不存在"+dirpath); return false; } File[] files = dirfile.listFiles(); if(files.length == 0){ //目录下没有文件，不需要处理 logger.info("目录下没有文件，不需要处理"+dirpath); return false; } List> attachments = new ArrayList<>(); Map attachItem; String fileName; File destFile; String tempUrl;//星瀚临时文件url String actUrl;//星瀚正式文件url //获取临时文件缓存 TempFileCache cache = CacheFactory.getCommonCacheFactory().getTempFileCache(); //遍历目标目录，得到所有文件 for (int i = 0; i < files.length; i++) { destFile = files[i]; fileName = destFile.getName(); attachItem = new HashMap<>(); //文件名称 attachItem.put("name", fileName); //文件大小 attachItem.put("size", destFile.length()); //将文件流存入临时文件缓存（拷贝完成）（最后一个参数为缓存有效期，600秒） tempUrl = cache.saveAsFullUrl(fileName, Files.readAllBytes(destFile.toPath()), 900); //修改时间 attachItem.put("lastModified",System.currentTimeMillis()); //将文件缓存中的附件文件上传到正式文件服务器 actUrl = AttachmentServiceHelper.saveTempToFileService(tempUrl,appId,entity,pk,fileName); //如下代码功能与saveTempToFileService实际功能重复，先注释掉，待调试确认 // FileService service = FileServiceFactory.getAttachmentFileService(); // RequestContext requestContext = RequestContext.get(); // String uuid = UUID.randomUUID().toString().replace("-", ""); // // 生成文件路径-上传附件时远程服务器需要存储文件的位置 // String pathParam = FileNameUtils.getAttachmentFileName(requestContext.getTenantId(), // requestContext.getAccountId(), uuid, fileName); // FileItem fileItem = new FileItem(fileName, pathParam, cache.getInputStream(tempUrl)); // // 上传附件到文件服务器 // service.upload(fileItem); //将新文件的物理路径存入map attachItem.put("url", actUrl); attachments.add(attachItem); } //维护单据和附件的关系（非文件上传） AttachmentServiceHelper.upload(entity, pk, attachKey, attachments); return true; } /**将星瀚nfs上某个目录下的所有文件关联至星瀚对应单据附件上，不需要文件上传 * @param appId 目标单据所属应用标识 * @param entity 目标单据实体标识 * @param pk 目标数据id * @param attachKey 附件面板标识 * @param dirpath 文件目录路径 * @throws IOException */ public static boolean realtionDirFile2Attchment(String appId,String entity,Object pk,String dirpath,String attachKey) { // File[] files = new File(dirpath).listFiles(); // if(files.length == 0){ // //目录下没有文件，不需要处理 // return false; // } List> attachments = new ArrayList<>(); Map attachItem; String fileName = "1.docx"; // File destFile; //遍历目标目录，得到所有文件 // for (int i = 0; i < files.length; i++) { // destFile = files[i]; // fileName = destFile.getName(); attachItem = new HashMap<>(); //文件名称 attachItem.put("name", fileName); //文件大小 attachItem.put("size", 1); //修改时间 attachItem.put("lastModified",System.currentTimeMillis()); TempFileCache cache = CacheFactory.getCommonCacheFactory().getTempFileCache(); FileService service = FileServiceFactory.getAttachmentFileService(); // service.download("/oa/uat/ht-202502001-nj/232906839800890085/1.docx"); String tempUrl = cache.saveAsUrl(fileName, service.getInputStream("/oa/uat/ht-202502001-nj/232906839800890085/1.docx"), 900); String actUrl = AttachmentServiceHelper.saveTempToFileService(tempUrl,appId,entity,pk,fileName); //将文件的物理路径存入map attachItem.put("url", actUrl); attachments.add(attachItem); // } // long filesize = service.getFileSize("/oa/uat/ht-202502001-nj/232906839800890085/1.docx"); // service.download(); // logger.info("文件大小"+filesize); //维护单据和附件的关系（非文件上传） AttachmentServiceHelper.upload(entity, pk, attachKey, attachments); return true; } /**将第三方提供的某个在线文件上传至星瀚对应单据附件上 * @param entity 目标单据实体标识 * @param pk 目标数据id * @param fileUrl 待存入文件的url * @param fileName 待存入文件名称 * @param fileSize 待存入文件大小 * @param attachKey 附件面板标识 * @throws IOException */ public static boolean saveUrlFile2Attch(String entity,Object pk,String fileUrl,String fileName,String fileSize,String attachKey) throws IOException{ List> attachments = new ArrayList<>(); Map attachItem = new HashMap<>(); //文件名称 attachItem.put("name", fileName); //将url文件转换成输入流 InputStream inputStream = null; HttpURLConnection conn = null; String tempUrl = null; try { URL url = new URL(fileUrl); conn = (HttpURLConnection)url.openConnection(); //设置超时间为30秒 conn.setConnectTimeout(30000); //防止屏蔽程序抓取而返回403错误 conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)"); //得到输入流 inputStream = conn.getInputStream(); //文件大小 attachItem.put("size", fileSize); //获取临时文件缓存 TempFileCache cache = CacheFactory.getCommonCacheFactory().getTempFileCache(); //将文件流存入临时文件缓存（拷贝完成）（最后一个参数为缓存有效期，600秒） tempUrl = cache.saveAsFullUrl(fileName, inputStream, 600); } catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); } finally { //关闭流 try { inputStream.close(); } catch (IOException e) { throw new RuntimeException(e); } //关闭连接 conn.disconnect(); } //修改时间 attachItem.put("lastModified",System.currentTimeMillis()); //获取appId MainEntityType dataEntityType = MetadataServiceHelper.getDataEntityType(entity); String appId = dataEntityType.getAppId(); //将文件缓存中的附件文件上传到正式文件服务器 String actUrl = AttachmentServiceHelper.saveTempToFileService(tempUrl,appId,entity,pk,fileName); //将新文件的物理路径存入map attachItem.put("url", actUrl); attachments.add(attachItem); //维护单据和附件的关系（非文件上传） AttachmentServiceHelper.upload(entity, pk, attachKey, attachments); return true; } }