diff --git a/health-consultation/health-consultation-biz/src/main/java/com/renkang/consultation/controller/ConDoctorController.java b/health-consultation/health-consultation-biz/src/main/java/com/renkang/consultation/controller/ConDoctorController.java index 89eb667..7308873 100644 --- a/health-consultation/health-consultation-biz/src/main/java/com/renkang/consultation/controller/ConDoctorController.java +++ b/health-consultation/health-consultation-biz/src/main/java/com/renkang/consultation/controller/ConDoctorController.java @@ -6,35 +6,40 @@ import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.ObjectMapper; import com.renkang.consultation.async.AsyncExportService; import com.renkang.consultation.constant.ConsultationConstants; import com.renkang.consultation.dto.AccountDTO; import com.renkang.consultation.dto.ConDoctorDTO; import com.renkang.consultation.dto.DoctorAudioStatusDTO; import com.renkang.consultation.dto.DoctorJumpHolidayDTO; -import com.renkang.consultation.entity.ConCostLog; -import com.renkang.consultation.entity.ConDoctor; -import com.renkang.consultation.entity.ConDoctorExAndUser; -import com.renkang.consultation.entity.CondepartmentId; +import com.renkang.consultation.entity.*; +import com.renkang.consultation.mapper.ConResourceMapper; import com.renkang.consultation.mapper.ESConDoctorMapper; import com.renkang.consultation.service.IConDepartmentService; import com.renkang.consultation.service.IConDoctorService; import com.renkang.consultation.service.IConResourceService; -import com.renkang.consultation.vo.AccountVO; -import com.renkang.consultation.vo.ConDoctorExportVO; -import com.renkang.consultation.vo.ConDoctorVO; -import com.renkang.consultation.vo.ConServiceVO; +import com.renkang.consultation.util.PicturesUtils; +import com.renkang.consultation.vo.*; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Parameter; import io.swagger.v3.oas.annotations.tags.Tag; import lombok.extern.slf4j.Slf4j; +import org.apache.poi.ooxml.POIXMLDocumentPart; +import org.apache.poi.ss.usermodel.*; +import org.apache.poi.xssf.usermodel.*; import org.apache.shiro.authz.annotation.RequiresPermissions; import org.jeecg.common.api.vo.Result; import org.jeecg.common.aspect.annotation.AutoLog; import org.jeecg.common.constant.CommonConstant; import org.jeecg.common.system.api.ISysBaseAPI; import org.jeecg.common.system.base.controller.JeecgController; +import org.jeecg.common.system.util.DictUtil; +import org.jeecg.common.system.vo.DictModel; import org.jeecg.common.util.CheckPasswordUtil; +import org.jeecg.common.util.MyUploadUtil; +import org.jeecg.global.GlobalUtils; import org.jeecg.util.RSAEncryptUtils; import org.jeecgframework.poi.excel.ExcelImportUtil; import org.jeecgframework.poi.excel.entity.ImportParams; @@ -44,13 +49,13 @@ import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.annotation.Resource; -import java.io.ByteArrayInputStream; -import java.io.IOException; +import java.io.*; import java.math.BigDecimal; -import java.util.ArrayList; -import java.util.Arrays; -import java.util.Date; -import java.util.List; +import java.net.HttpURLConnection; +import java.net.URL; +import java.nio.file.Files; +import java.nio.file.Paths; +import java.util.*; /** * @Description: con_doctor @@ -72,6 +77,10 @@ public class ConDoctorController extends JeecgController pictures = picturesUtils.getPicturesWithRow(fileData); + + try (InputStream is = file.getInputStream(); + XSSFWorkbook workbook = new XSSFWorkbook(is)) { + + XSSFSheet sheet = workbook.getSheetAt(0); + + for (int rowIndex = 2; rowIndex <= sheet.getLastRowNum(); rowIndex++) { + Row row = sheet.getRow(rowIndex); + if (row == null) { + continue; + } + + ConDoctorExAndUser doctorExAndUser = new ConDoctorExAndUser(); + ConDoctorDoctorVO doctorVO = new ConDoctorDoctorVO(); + ConDoctorUserVO userVO = new ConDoctorUserVO(); + + String name = getCellStringValue(row.getCell(0)); + doctorVO.setDoctorName(name); + String account = getCellStringValue(row.getCell(1)); + userVO.setUsername(name); + String password = getCellStringValue(row.getCell(2)); + String g = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCmZfR/bA9X3vp86y1aEpvwzXJYKRRF1fLau2+05/ZtaITLpV8bhkmSf3neSy/Q9gAdvG75Fr73E+GWE+K5b0BpvIS1jDGo319+PpZR39SaZTKZ27XFXrosmJTZutN79t819HS1VseleunHAFgMVufE9U5jP6LGzl/wbkSy01GhzwIDAQAB"; + String cipherText = RSAEncryptUtils.encrypt1(password, g); + + String key = RSAEncryptUtils.decrypt1(cipherText, CommonConstant.PRIVATE_KEY); + String s = CheckPasswordUtil.checkPasswordRule(key); + if (StrUtil.isNotBlank(s)) { + return null; + } + userVO.setPassword(cipherText); + + System.out.println("第" + (rowIndex + 1) + "行数据：姓名=" + name + ", 账号=" + account + ", 密码=" + password); + + // 根据行号直接获取对应图片 + PicturesUtils.PicturePosition picPos = pictures.get(rowIndex); + if (picPos != null) { + XSSFPictureData xssfPictureData = picPos.getPictureData(); + String fileName = xssfPictureData.getPackagePart().getPartName().getName(); + System.out.println("图片文件名：" + fileName); + + File imgFile = new File("D:\\" + fileName); + File dir = imgFile.getParentFile(); + if (!dir.exists()) { + dir.mkdirs(); + } + try { + Files.write(Paths.get(imgFile.getPath()), xssfPictureData.getData()); + // 调用之前写好的上传接口，传入token + String url = uploadImage(imgFile, MyUploadUtil.TEMP_BIZ, token,urlPath); + userVO.setAvatar(url); + } catch (IOException e) { + e.printStackTrace(); + } + + String resourceName = getCellStringValue(row.getCell(4)); + doctorVO.setResourceName(resourceName); + List conResources = conResourceMapper.selectByResourceName(resourceName); + if (conResources != null && !conResources.isEmpty()) { + ConResource conResource = conResources.get(0); + if (conResource != null) { + doctorVO.setResourceId(conResource.getId()); + } + } + + String zc = getCellStringValue(row.getCell(6)); + List models = dictUtil.queryDictItemListByCode("z_doct_lev"); + DictModel dictModel = models.stream().filter(item -> item.getText().equals(zc)).findFirst().orElse(null); + if (dictModel != null) { + doctorVO.setDoctorJob(dictModel.getValue()); + } + + doctorVO.setGoodAt(getCellStringValue(row.getCell(7))); + doctorVO.setExperience(getCellStringValue(row.getCell(8))); + String sex = getCellStringValue(row.getCell(9)); + userVO.setSex(Integer.valueOf(sex)); + } else { + System.out.println("第" + (rowIndex + 1) + "行无对应图片"); + } + doctorExAndUser.setConDoctor(doctorVO); + doctorExAndUser.setSysUserModel(userVO); + conDoctorService.saveDoctorUser(doctorExAndUser); + + System.out.println(rowIndex); + } + return "处理成功"; + } + } catch (Exception e) { + e.printStackTrace(); + return "处理失败：" + e.getMessage(); + } + } + + /** + * 上传图片文件到指定接口，返回响应字符串（一般是json）。 + * @param file 头像文件 + * @param bizValue biz字段值，示例中固定为 "temp" + * @param token x-access-token 请求头值 + * @return 服务端响应内容 + * @throws Exception 异常抛出 + */ + private String uploadImage(File file, String bizValue, String token,String urlPath) throws Exception { + String boundary = "----WebKitFormBoundaryGEKNMBfLirBXKE2d"; + String LINE_FEED = "\r\n"; + String charset = "UTF-8"; + + URL url = new URL(urlPath); + HttpURLConnection conn = (HttpURLConnection) url.openConnection(); + conn.setUseCaches(false); + conn.setDoOutput(true); + conn.setDoInput(true); + conn.setRequestMethod("POST"); + conn.setRequestProperty("Content-Type", "multipart/form-data; boundary=" + boundary); + conn.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/141.0.0.0 Safari/537.36"); + conn.setRequestProperty("x-access-token", token); + conn.setRequestProperty("x-requested-with", "XMLHttpRequest"); + + try ( + OutputStream outputStream = conn.getOutputStream(); + PrintWriter writer = new PrintWriter(new OutputStreamWriter(outputStream, charset), true); + ) { + // 写 biz 字段 + writer.append("--").append(boundary).append(LINE_FEED); + writer.append("Content-Disposition: form-data; name=\"biz\"").append(LINE_FEED); + writer.append(LINE_FEED); + writer.append(bizValue).append(LINE_FEED); + writer.flush(); + + // 写文件字段 + String fileName = file.getName(); + String mimeType = Files.probeContentType(file.toPath()); + if (mimeType == null) { + mimeType = "application/octet-stream"; + } + writer.append("--").append(boundary).append(LINE_FEED); + writer.append("Content-Disposition: form-data; name=\"file\"; filename=\"").append(fileName).append("\"").append(LINE_FEED); + writer.append("Content-Type: ").append(mimeType).append(LINE_FEED); + writer.append(LINE_FEED); + writer.flush(); + + try (FileInputStream inputStream = new FileInputStream(file)) { + byte[] buffer = new byte[4096]; + int bytesRead; + while ((bytesRead = inputStream.read(buffer)) != -1) { + outputStream.write(buffer, 0, bytesRead); + } + outputStream.flush(); + } + + writer.append(LINE_FEED); + writer.flush(); + + // 结尾 + writer.append("--").append(boundary).append("--").append(LINE_FEED); + writer.flush(); + } + + int status = conn.getResponseCode(); + InputStream responseStream = (status >= 200 && status < 300) ? conn.getInputStream() : conn.getErrorStream(); + + try (BufferedReader reader = new BufferedReader(new InputStreamReader(responseStream, charset))) { + StringBuilder response = new StringBuilder(); + String line; + while ((line = reader.readLine()) != null) { + response.append(line); + } + + // 解析json，提取result.url + ObjectMapper mapper = new ObjectMapper(); + JsonNode rootNode = mapper.readTree(response.toString()); + if (rootNode.path("success").asBoolean(false)) { + JsonNode urlNode = rootNode.path("result").path("url"); + if (!urlNode.isMissingNode()) { + return urlNode.asText(); + } + } + // 如果失败或无url，返回完整响应或空 + return ""; + } finally { + conn.disconnect(); + } + } + + + + private String getCellStringValue(Cell cell) { + if (cell == null) { + return ""; + } + cell.setCellType(CellType.STRING); + return cell.getStringCellValue().trim(); + } + + public static void main(String[] args) { + PicturesUtils picturesUtils = new PicturesUtils(); + byte[] fileData = picturesUtils.getFileStream(new File("C:\\Users\\Huawei\\Desktop\\专家库.xlsm")); + Map pictures = picturesUtils.getPicturesWithRow(fileData); + pictures.forEach((row, picturePosition) -> { + System.out.println("图片所在行号（从0开始）：" + row); + XSSFPictureData pictureData = picturePosition.getPictureData(); + String fileName = pictureData.getPackagePart().getPartName().getName(); + System.out.println("文件名：" + fileName); + // 保存路径，注意转义反斜杠 + File file = new File("D:\\" + fileName); + File dir = file.getParentFile(); + if (!dir.exists()) { + dir.mkdirs(); + } + try { + Files.write(Paths.get(file.getPath()), pictureData.getData()); + System.out.println("图片保存成功，路径：" + file.getPath()); + } catch (IOException e) { + e.printStackTrace(); + } + }); + } + +// public static void main(String[] args) { +// String filePath = "C:\\Users\\Huawei\\Desktop\\专家库.xlsm"; +// +// try (FileInputStream fis = new FileInputStream(filePath); +// XSSFWorkbook workbook = new XSSFWorkbook(fis)) { +// +// XSSFSheet sheet = workbook.getSheetAt(0); +// +// XSSFDrawing drawing = sheet.getDrawingPatriarch(); +// if (drawing == null) { +// System.out.println("该sheet没有图片"); +// return; +// } +// +// List shapes = drawing.getShapes(); +// for (XSSFShape shape : shapes) { +// if (shape instanceof XSSFPicture) { +// XSSFPicture picture = (XSSFPicture) shape; +// XSSFClientAnchor anchor = picture.getPreferredSize(); +// +// int row = anchor.getRow1(); // 图片所在的起始行号（从0开始） +// int col = anchor.getCol1(); // 图片所在的起始列号 +// +// String fileName = picture.getPictureData().getPackagePart().getPartName().getName(); +// System.out.println("图片对应行：" + (row + 1) + "，列：" + (col + 1) + "，文件名：" + fileName); +// } +// } +// +// } catch (Exception e) { +// e.printStackTrace(); +// } // } } diff --git a/health-consultation/health-consultation-biz/src/main/java/com/renkang/consultation/util/PicturesUtils.java b/health-consultation/health-consultation-biz/src/main/java/com/renkang/consultation/util/PicturesUtils.java new file mode 100644 index 0000000..e4a173d --- /dev/null +++ b/health-consultation/health-consultation-biz/src/main/java/com/renkang/consultation/util/PicturesUtils.java @@ -0,0 +1,370 @@ +package com.renkang.consultation.util; + +import cn.hutool.json.JSONArray; +import cn.hutool.json.JSONObject; +import cn.hutool.json.XML; +import lombok.Data; +import lombok.extern.slf4j.Slf4j; +import org.apache.commons.collections4.map.HashedMap; +import org.apache.commons.io.IOUtils; +import org.apache.commons.lang3.StringUtils; +import org.apache.poi.openxml4j.opc.PackagePartName; +import org.apache.poi.ss.usermodel.Sheet; +import org.apache.poi.ss.usermodel.Workbook; +import org.apache.poi.ss.usermodel.WorkbookFactory; +import org.apache.poi.xssf.usermodel.*; + +import java.io.*; +import java.nio.charset.StandardCharsets; +import java.nio.file.Files; +import java.nio.file.Path; +import java.nio.file.Paths; +import java.util.HashMap; +import java.util.Iterator; +import java.util.List; +import java.util.Map; +import java.util.zip.ZipEntry; +import java.util.zip.ZipInputStream; + + +/** + * @author bianhl + * @version 1.0 + * @description 获取图片资源 + * @date 2024年4月28日08:44:42 + */ +@Slf4j +public class PicturesUtils { + + public static void main(String[] args) { + PicturesUtils picturesUtils = new PicturesUtils(); + byte[] fileData = picturesUtils.getFileStream(new File("你的文件路径")); + Map pictures = picturesUtils.getPictures(fileData); + pictures.forEach((id, xssfPictureData) -> { + System.out.println("id：" + id); + String fileName = xssfPictureData.getPackagePart().getPartName().getName(); + System.out.println("fileName：" + fileName); + File file = new File("C:\\Users\\XXX\\Desktop\\" + fileName); + File dir = file.getParentFile(); + dir.mkdirs(); + try { + Files.write(Paths.get(file.getPath()), xssfPictureData.getData()); + } catch (IOException e) { + e.printStackTrace(); + } + }); + } + + /** + * 获取浮动图片，以 map 形式返回，键为行列格式 x-y。 + * + * @param xssfSheet WPS 工作表 + * @return 浮动图片的 map + */ + public static Map getFloatingPictures(XSSFSheet xssfSheet) { + Map mapFloatingPictures = new HashMap<>(); + XSSFDrawing drawingPatriarch = xssfSheet.getDrawingPatriarch(); + if (drawingPatriarch != null) { + List shapes = drawingPatriarch.getShapes(); + for (XSSFShape shape : shapes) { + if (shape instanceof XSSFPicture) { + XSSFPicture picture = (XSSFPicture) shape; + XSSFClientAnchor anchor = (XSSFClientAnchor) picture.getAnchor(); + XSSFPictureData pictureData = picture.getPictureData(); + String key = anchor.getRow1() + "-" + anchor.getCol1(); + mapFloatingPictures.put(key, pictureData); + } + } + } + return mapFloatingPictures; + } + + + /** + * 处理 WPS 文件中的图片数据，返回图片信息 map。 + * + * @param stream 输入流 + * @param mapConfig 配置映射 + * @return 图片信息的 map + * @throws IOException + */ + private Map processPictures(ByteArrayInputStream stream, Map mapConfig) throws IOException { + Map mapPictures = new HashedMap<>(); + Workbook workbook = WorkbookFactory.create(stream); + List allPictures = (List) workbook.getAllPictures(); + for (XSSFPictureData pictureData : allPictures) { + PackagePartName partName = pictureData.getPackagePart().getPartName(); + String uri = partName.getURI().toString(); + if (mapConfig.containsKey(uri)) { + String strId = mapConfig.get(uri); + mapPictures.put(strId, pictureData); + } + } + return mapPictures; + } + + /** + * 获取 WPS 文档中的图片，包括嵌入式图片和浮动式图片。 + * + * @param data 二进制数据 + * @return 图片信息的 map + * @throws IOException + */ + public Map getPictures(byte[] data) { + try { + Map mapConfig = processZipEntries(new ByteArrayInputStream(data)); + Map mapPictures = processPictures(new ByteArrayInputStream(data), mapConfig); + Iterator sheetIterator = WorkbookFactory.create(new ByteArrayInputStream(data)).sheetIterator(); + while (sheetIterator.hasNext()) { + mapPictures.putAll(getFloatingPictures((XSSFSheet) sheetIterator.next())); + } + return mapPictures; + } catch (IOException e) { + return new HashedMap<>(); + } + } + + @Data + public static class PicturePosition { + private XSSFPictureData pictureData; + private int row; + private int col; + + public PicturePosition(XSSFPictureData pictureData, int row, int col) { + this.pictureData = pictureData; + this.row = row; + this.col = col; + } + } + + + public Map getPicturesWithRow(byte[] data) { + Map pictureMap = new HashMap<>(); + try { + Map mapConfig = processZipEntries(new ByteArrayInputStream(data)); + Map mapPictures = processPictures(new ByteArrayInputStream(data), mapConfig); + Iterator sheetIterator = WorkbookFactory.create(new ByteArrayInputStream(data)).sheetIterator(); + while (sheetIterator.hasNext()) { + XSSFSheet sheet = (XSSFSheet) sheetIterator.next(); + Map floatingPics = getFloatingPicturesWithRow(sheet); + pictureMap.putAll(floatingPics); + } + } catch (IOException e) { + // 处理异常 + } + return pictureMap; + } + + public static Map getFloatingPicturesWithRow(XSSFSheet xssfSheet) { + Map map = new HashMap<>(); + XSSFDrawing drawingPatriarch = xssfSheet.getDrawingPatriarch(); + if (drawingPatriarch != null) { + List shapes = drawingPatriarch.getShapes(); + for (XSSFShape shape : shapes) { + if (shape instanceof XSSFPicture) { + XSSFPicture picture = (XSSFPicture) shape; + XSSFClientAnchor anchor = (XSSFClientAnchor) picture.getAnchor(); + int row = anchor.getRow1(); + int col = anchor.getCol1(); + map.put(row, new PicturePosition(picture.getPictureData(), row, col)); + } + } + } + return map; + } + + /** + * 处理 Zip 文件中的条目，更新图片配置信息。 + * + * @param stream Zip 输入流 + * @return 配置信息的 map + * @throws IOException + */ + private Map processZipEntries(ByteArrayInputStream stream) throws IOException { + Map mapConfig = new HashedMap<>(); + ZipInputStream zipInputStream = new ZipInputStream(stream); + ZipEntry zipEntry; + while ((zipEntry = zipInputStream.getNextEntry()) != null) { + try { + final String fileName = zipEntry.getName(); + if ("xl/cellimages.xml".equals(fileName)) { + processCellImages(zipInputStream, mapConfig); + } else if ("xl/_rels/cellimages.xml.rels".equals(fileName)) { + return processCellImagesRels(zipInputStream, mapConfig); + } + } finally { + zipInputStream.closeEntry(); + } + } + return new HashedMap<>(); + } + + /** + * 处理 Zip 文件中的 cellimages.xml 文件，更新图片配置信息。 + * + * @param zipInputStream Zip 输入流 + * @param mapConfig 配置信息的 map + * @throws IOException + */ + private void processCellImages(ZipInputStream zipInputStream, Map mapConfig) throws IOException { + String content = IOUtils.toString(zipInputStream, StandardCharsets.UTF_8); + JSONObject jsonObject = XML.toJSONObject(content); + if (jsonObject != null) { + JSONObject cellImages = jsonObject.getJSONObject("etc:cellImages"); + if (cellImages != null) { + JSONArray cellImageArray = null; + Object cellImage = cellImages.get("etc:cellImage"); + if (cellImage != null && cellImage instanceof JSONArray) { + cellImageArray = (JSONArray) cellImage; + } else if (cellImage != null && cellImage instanceof JSONObject) { + JSONObject cellImageObj = (JSONObject) cellImage; // 显式类型转换 + if (cellImageObj != null) { + cellImageArray = new JSONArray(); + cellImageArray.add(cellImageObj); + } + } + if (cellImageArray != null) { + processImageItems(cellImageArray, mapConfig); + } + } + } + } + + /** + * 处理 cellImageArray 中的图片项，更新图片配置信息。 + * + * @param cellImageArray 图片项的 JSONArray + * @param mapConfig 配置信息的 map + */ + private void processImageItems(JSONArray cellImageArray, Map mapConfig) { + for (int i = 0; i < cellImageArray.size(); i++) { + JSONObject imageItem = cellImageArray.getJSONObject(i); + if (imageItem != null) { + JSONObject pic = imageItem.getJSONObject("xdr:pic"); + if (pic != null) { + processPic(pic, mapConfig); + } + } + } + } + + /** + * 处理 pic 中的图片信息，更新图片配置信息。 + * + * @param pic 图片的 JSONObject + * @param mapConfig 配置信息的 map + */ + private void processPic(JSONObject pic, Map mapConfig) { + JSONObject nvPicPr = pic.getJSONObject("xdr:nvPicPr"); + if (nvPicPr != null) { + JSONObject cNvPr = nvPicPr.getJSONObject("xdr:cNvPr"); + if (cNvPr != null) { + String name = cNvPr.getStr("name"); + if (StringUtils.isNotEmpty(name)) { + String strImageEmbed = updateImageEmbed(pic); + if (strImageEmbed != null) { + mapConfig.put(strImageEmbed, name); + } + } + } + } + } + + /** + * 获取嵌入式图片的 embed 信息。 + * + * @param pic 图片的 JSONObject + * @return embed 信息 + */ + private String updateImageEmbed(JSONObject pic) { + JSONObject blipFill = pic.getJSONObject("xdr:blipFill"); + if (blipFill != null) { + JSONObject blip = blipFill.getJSONObject("a:blip"); + if (blip != null) { + return blip.getStr("r:embed"); + } + } + return null; + } + + /** + * 处理 Zip 文件中的 relationship 条目，更新配置信息。 + * + * @param zipInputStream Zip 输入流 + * @param mapConfig 配置信息的 map + * @return 配置信息的 map + * @throws IOException + */ + private Map processCellImagesRels(ZipInputStream zipInputStream, Map mapConfig) throws IOException { + String content = IOUtils.toString(zipInputStream, StandardCharsets.UTF_8); + JSONObject jsonObject = XML.toJSONObject(content); + JSONObject relationships = jsonObject.getJSONObject("Relationships"); + if (relationships != null) { + JSONArray relationshipArray = null; + Object relationship = relationships.get("Relationship"); + + if (relationship != null && relationship instanceof JSONArray) { + relationshipArray = (JSONArray) relationship; + } else if (relationship != null && relationship instanceof JSONObject) { + JSONObject relationshipObj = (JSONObject) relationship; // Java 8 显式类型转换 + if (relationshipObj != null) { + relationshipArray = new JSONArray(); + relationshipArray.add(relationshipObj); + } + } + if (relationshipArray != null) { + return processRelationships(relationshipArray, mapConfig); + } + } + return null; + } + + /** + * 处理 relationshipArray 中的关系项，更新配置信息。 + * + * @param relationshipArray 关系项的 JSONArray + * @param mapConfig 配置信息的 map + * @return 配置信息的 map + */ + private Map processRelationships(JSONArray relationshipArray, Map mapConfig) { + Map mapRelationships = new HashedMap<>(); + for (int i = 0; i < relationshipArray.size(); i++) { + JSONObject relaItem = relationshipArray.getJSONObject(i); + if (relaItem != null) { + String id = relaItem.getStr("Id"); + String value = "/xl/" + relaItem.getStr("Target"); + if (mapConfig.containsKey(id)) { + String strImageId = mapConfig.get(id); + mapRelationships.put(value, strImageId); + } + } + } + return mapRelationships; + } + + /** + * @param file 数据文件 + * @return {@link byte[]} + * @description + * @author bianhl + * @date 2024/4/26 13:52 + */ + public byte[] getFileStream(File file) { + try (InputStream inputStream = new FileInputStream(file)) { + // 创建 ByteArrayOutputStream 来暂存流数据 + ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); + // 将 inputStream 读取到 byteArrayOutputStream 中 + byte[] buffer = new byte[1024]; + int length; + while ((length = inputStream.read(buffer)) != -1) { + byteArrayOutputStream.write(buffer, 0, length); + } + // 将 byteArrayOutputStream 的内容获取为字节数组 + return byteArrayOutputStream.toByteArray(); + } catch (IOException e) { + return null; + } + } + + +} diff --git a/jeecg-boot-base-core/src/main/java/org/jeecg/common/util/MyUploadUtil.java b/jeecg-boot-base-core/src/main/java/org/jeecg/common/util/MyUploadUtil.java index 260eb84..10dbdbd 100644 --- a/jeecg-boot-base-core/src/main/java/org/jeecg/common/util/MyUploadUtil.java +++ b/jeecg-boot-base-core/src/main/java/org/jeecg/common/util/MyUploadUtil.java @@ -41,6 +41,7 @@ import java.util.List; @Slf4j public class MyUploadUtil { public final static String IMPORT_BIZ = "import"; + public final static String TEMP_BIZ = "temp"; public final static String EXPORT_BIZ = "export"; private static String muUrl; private static String muToken;