ExcelUtils.java 55 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360
  1. package com.zksy.common.utils.file;
  2. /**
  3. * @author Administrator
  4. * @version 1.0
  5. * @project credit-rating
  6. * @description Excel导入导出工具类
  7. * @date 2024/3/26 17:18:10
  8. */
  9. import com.alibaba.fastjson.JSONArray;
  10. import com.alibaba.fastjson.JSONObject;
  11. import com.zksy.common.exception.ServiceException;
  12. import jodd.io.FileUtil;
  13. import lombok.extern.slf4j.Slf4j;
  14. import org.apache.poi.hssf.usermodel.HSSFDataValidation;
  15. import org.apache.poi.hssf.usermodel.HSSFWorkbook;
  16. import org.apache.poi.poifs.filesystem.NotOLE2FileException;
  17. import org.apache.poi.ss.usermodel.*;
  18. import org.apache.poi.ss.usermodel.ClientAnchor.AnchorType;
  19. import org.apache.poi.ss.util.CellRangeAddress;
  20. import org.apache.poi.ss.util.CellRangeAddressList;
  21. import org.apache.poi.xssf.streaming.SXSSFWorkbook;
  22. import org.apache.poi.xssf.usermodel.*;
  23. import org.springframework.util.CollectionUtils;
  24. import org.springframework.web.multipart.MultipartFile;
  25. import javax.servlet.ServletOutputStream;
  26. import javax.servlet.http.HttpServletResponse;
  27. import java.io.*;
  28. import java.lang.reflect.Field;
  29. import java.math.BigDecimal;
  30. import java.math.RoundingMode;
  31. import java.net.URL;
  32. import java.text.DecimalFormat;
  33. import java.text.NumberFormat;
  34. import java.text.SimpleDateFormat;
  35. import java.time.LocalDate;
  36. import java.time.format.DateTimeFormatter;
  37. import java.time.format.DateTimeParseException;
  38. import java.util.*;
  39. import java.util.Map.Entry;
  40. import java.util.regex.Pattern;
  41. @SuppressWarnings("unused")
  42. @Slf4j
  43. public class ExcelUtils {
  44. private static final String XLSX = ".xlsx";
  45. private static final String XLS = ".xls";
  46. public static final String ROW_MERGE = "row_merge";
  47. public static final String COLUMN_MERGE = "column_merge";
  48. private static final String DATE_FORMAT = "yyyy-MM-dd HH:mm:ss";
  49. private static final String ROW_NUM = "rowNum";
  50. private static final String ROW_DATA = "rowData";
  51. private static final String ROW_TIPS = "rowTips";
  52. private static final int CELL_OTHER = 0;
  53. private static final int CELL_ROW_MERGE = 1;
  54. private static final int CELL_COLUMN_MERGE = 2;
  55. private static final int IMG_HEIGHT = 30;
  56. private static final int IMG_WIDTH = 30;
  57. private static final char LEAN_LINE = '/';
  58. private static final int BYTES_DEFAULT_LENGTH = 10240;
  59. private static final NumberFormat NUMBER_FORMAT = NumberFormat.getNumberInstance();
  60. public static <T> List<T> readFile(File file, Class<T> clazz,Integer cellNum) throws Exception {
  61. JSONArray array = readFile(file,cellNum);
  62. return getBeanList(array, clazz);
  63. }
  64. public static <T> List<T> readMultipartFile(MultipartFile mFile, Class<T> clazz,Integer cellNum) throws Exception {
  65. JSONArray array = readMultipartFile(mFile,cellNum);
  66. return getBeanList(array, clazz);
  67. }
  68. /**
  69. * TODO 商标信息导入(带图片)
  70. * @param mFile
  71. * @param clazz
  72. * @param cellNum
  73. * @return java.util.List<T>
  74. * @author Administrator
  75. * @date 2024/4/2 16:52:04
  76. */
  77. public static <T> List<T> readCrmEnterpriseTrademarkInformationMultipartFile(MultipartFile mFile, Class<T> clazz,Integer cellNum,Integer column,String localFileServerUploadDir,String localFileServerMappingPath) throws Exception {
  78. JSONArray array = readCrmEnterpriseTrademarkInformationMultipartFile(mFile,cellNum,column, localFileServerUploadDir, localFileServerMappingPath);
  79. return getBeanList(array, clazz);
  80. }
  81. public static JSONArray readFile(File file,Integer cellNum) throws Exception {
  82. return readExcel(null, file,cellNum);
  83. }
  84. public static JSONArray readMultipartFile(MultipartFile mFile,Integer cellNum) throws Exception {
  85. return readExcel(mFile, null,cellNum);
  86. }
  87. public static JSONArray readCrmEnterpriseTrademarkInformationMultipartFile(MultipartFile mFile,Integer cellNum,Integer column,String localFileServerUploadDir,String localFileServerMappingPath) throws Exception {
  88. return readCrmEnterpriseTrademarkInformationExcel(mFile, null,cellNum,column, localFileServerUploadDir, localFileServerMappingPath);
  89. }
  90. public static Map<String, JSONArray> readFileManySheet(File file,Integer cellNum) throws Exception {
  91. return readExcelManySheet(null, file,cellNum);
  92. }
  93. public static Map<String, JSONArray> readFileManySheet(MultipartFile file,Integer cellNum) throws Exception {
  94. return readExcelManySheet(file, null,cellNum);
  95. }
  96. private static <T> List<T> getBeanList(JSONArray array, Class<T> clazz) throws Exception {
  97. List<T> list = new ArrayList<>();
  98. Map<Integer, String> uniqueMap = new HashMap<>(50);
  99. for (int i = 0; i < array.size(); i++) {
  100. list.add(getBean(clazz, array.getJSONObject(i), uniqueMap));
  101. }
  102. return list;
  103. }
  104. /**
  105. * 获取每个对象的数据
  106. */
  107. private static <T> T getBean(Class<T> c, JSONObject obj, Map<Integer, String> uniqueMap) throws Exception {
  108. T t = c.newInstance();
  109. Field[] fields = c.getDeclaredFields();
  110. List<String> errMsgList = new ArrayList<>();
  111. boolean hasRowTipsField = false;
  112. StringBuilder uniqueBuilder = new StringBuilder();
  113. int rowNum = 0;
  114. for (Field field : fields) {
  115. // 行号
  116. if (field.getName().equals(ROW_NUM)) {
  117. rowNum = obj.getInteger(ROW_NUM);
  118. field.setAccessible(true);
  119. field.set(t, rowNum);
  120. continue;
  121. }
  122. // 是否需要设置异常信息
  123. if (field.getName().equals(ROW_TIPS)) {
  124. hasRowTipsField = true;
  125. continue;
  126. }
  127. // 原始数据
  128. if (field.getName().equals(ROW_DATA)) {
  129. field.setAccessible(true);
  130. field.set(t, obj.toString());
  131. continue;
  132. }
  133. // 设置对应属性值
  134. setFieldValue(t, field, obj, uniqueBuilder, errMsgList);
  135. }
  136. // 数据唯一性校验
  137. if (uniqueBuilder.length() > 0) {
  138. if (uniqueMap.containsValue(uniqueBuilder.toString())) {
  139. Set<Integer> rowNumKeys = uniqueMap.keySet();
  140. for (Integer num : rowNumKeys) {
  141. if (uniqueMap.get(num).equals(uniqueBuilder.toString())) {
  142. errMsgList.add(String.format("数据唯一性校验失败,(%s)与第%s行重复)", uniqueBuilder, num));
  143. }
  144. }
  145. } else {
  146. uniqueMap.put(rowNum, uniqueBuilder.toString());
  147. }
  148. }
  149. // 失败处理
  150. if (errMsgList.isEmpty() && !hasRowTipsField) {
  151. return t;
  152. }
  153. StringBuilder sb = new StringBuilder();
  154. int size = errMsgList.size();
  155. for (int i = 0; i < size; i++) {
  156. if (i == size - 1) {
  157. sb.append(errMsgList.get(i));
  158. } else {
  159. sb.append(errMsgList.get(i)).append(";");
  160. }
  161. }
  162. // 设置错误信息
  163. for (Field field : fields) {
  164. if (field.getName().equals(ROW_TIPS)) {
  165. field.setAccessible(true);
  166. field.set(t, sb.toString());
  167. }
  168. }
  169. return t;
  170. }
  171. private static <T> void setFieldValue(T t, Field field, JSONObject obj, StringBuilder uniqueBuilder, List<String> errMsgList) {
  172. // 获取 ExcelImport 注解属性
  173. ExcelImport annotation = field.getAnnotation(ExcelImport.class);
  174. if (annotation == null) {
  175. return;
  176. }
  177. String cname = annotation.value();
  178. if (cname.trim().length() == 0) {
  179. return;
  180. }
  181. // 获取具体值
  182. String val = null;
  183. //截取表头数据
  184. JSONObject jsonObject = new JSONObject();
  185. JSONObject resultJson = new JSONObject();
  186. StringBuffer sb = new StringBuffer();
  187. for(Entry<String,Object> entry : obj.entrySet()){
  188. String output = null;
  189. String result = null;
  190. if(entry.getKey() != null) {
  191. int newlineIndex = entry.getKey().indexOf('\n');
  192. if (newlineIndex != -1) {
  193. output = entry.getKey().substring(0, newlineIndex);
  194. } else {
  195. output = entry.getKey(); // 如果没有找到\n,则返回整个字符串
  196. }
  197. result = entry.getKey().replaceAll("\\s","");
  198. }
  199. jsonObject.put(output,entry.getValue());
  200. resultJson.put(result,entry.getValue());
  201. }
  202. if (jsonObject.containsKey(cname)) {
  203. //去掉换行后的表头
  204. val = getString(jsonObject.getString(cname));
  205. }else{
  206. val = getString(resultJson.getString(cname));
  207. }
  208. if (val == null) {
  209. return;
  210. }
  211. field.setAccessible(true);
  212. // 判断是否必填
  213. boolean require = annotation.required();
  214. if (require && val.isEmpty()) {
  215. errMsgList.add(String.format("[%s]不能为空", cname));
  216. return;
  217. }
  218. // 数据唯一性获取
  219. boolean unique = annotation.unique();
  220. if (unique) {
  221. if (uniqueBuilder.length() > 0) {
  222. uniqueBuilder.append("--").append(val);
  223. } else {
  224. uniqueBuilder.append(val);
  225. }
  226. }
  227. // 判断是否超过最大长度
  228. int maxLength = annotation.maxLength();
  229. if (maxLength > 0 && val.length() > maxLength) {
  230. errMsgList.add(String.format("[%s]长度不能超过%s个字符(当前%s个字符)", cname, maxLength, val.length()));
  231. }
  232. // 判断当前属性是否有映射关系
  233. LinkedHashMap<String, String> kvMap = getKvMap(annotation.kv());
  234. if (!kvMap.isEmpty()) {
  235. boolean isMatch = false;
  236. for (String key : kvMap.keySet()) {
  237. if (kvMap.get(key).equals(val)) {
  238. val = key;
  239. isMatch = true;
  240. break;
  241. }
  242. }
  243. if (!isMatch) {
  244. errMsgList.add(String.format("[%s]的值不正确(当前值为%s)", cname, val));
  245. return;
  246. }
  247. }
  248. // 其余情况根据类型赋值
  249. String fieldClassName = field.getType().getSimpleName();
  250. try {
  251. if ("String".equalsIgnoreCase(fieldClassName)) {
  252. field.set(t, val);
  253. } else if ("boolean".equalsIgnoreCase(fieldClassName)) {
  254. field.set(t, Boolean.valueOf(val));
  255. } else if ("int".equalsIgnoreCase(fieldClassName) || "Integer".equals(fieldClassName)) {
  256. try {
  257. field.set(t, Integer.valueOf(val));
  258. } catch (NumberFormatException e) {
  259. errMsgList.add(String.format("[%s]的值格式不正确(当前值为%s)", cname, val));
  260. }
  261. } else if ("double".equalsIgnoreCase(fieldClassName)) {
  262. field.set(t, Double.valueOf(val));
  263. } else if ("long".equalsIgnoreCase(fieldClassName)) {
  264. field.set(t, Long.valueOf(val));
  265. } else if ("BigDecimal".equalsIgnoreCase(fieldClassName)) {
  266. field.set(t, new BigDecimal(val));
  267. } else if ("Date".equalsIgnoreCase(fieldClassName)) {
  268. val = val.trim().replaceAll("\\u00A0", " ").trim();
  269. val = val.replace("/", "-");
  270. if (!val.contains("/") && !val.contains("-")) {
  271. String stringDate = convertExcelNumberToDate(Long.parseLong(val));
  272. SimpleDateFormat formatter = new SimpleDateFormat("yyyy/MM/dd");
  273. try {
  274. Date date = formatter.parse(stringDate);
  275. field.set(t, date); // 直接将 Date 对象赋值给 field
  276. } catch (IllegalAccessException e) {
  277. e.printStackTrace();
  278. }
  279. } else {
  280. // 如果 val 包含了分隔符,你可能需要解析这个字符串
  281. SimpleDateFormat sdf;
  282. if (val.contains(" ")) {
  283. sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  284. } else {
  285. sdf = new SimpleDateFormat("yyyy-MM-dd");
  286. }
  287. try {
  288. Date parsedDate = sdf.parse(val);
  289. field.set(t, parsedDate);
  290. } catch (Exception e) {
  291. e.printStackTrace();
  292. }
  293. }
  294. } else if ("LocalDate".equalsIgnoreCase(fieldClassName)) {
  295. val = val.trim().replaceAll("\\u00A0", " ").trim();
  296. val = val.replace("/", "-");
  297. if (!val.contains("/") && !val.contains("-")) {
  298. String stringDate = convertExcelNumberToDate(Long.parseLong(val));
  299. try {
  300. LocalDate date = LocalDate.parse(stringDate);
  301. field.set(t, date); // 直接将 Date 对象赋值给 field
  302. } catch (IllegalAccessException e) {
  303. e.printStackTrace();
  304. }
  305. } else {
  306. // 如果 val 包含了分隔符,你可能需要解析这个字符串
  307. DateTimeFormatter sdf;
  308. if (val.contains(" ")) {
  309. sdf = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
  310. } else {
  311. sdf = DateTimeFormatter.ofPattern("yyyy-MM-dd");
  312. }
  313. try {
  314. LocalDate parsedDate = LocalDate.parse(val,sdf);
  315. field.set(t, parsedDate);
  316. } catch (Exception e) {
  317. e.printStackTrace();
  318. }
  319. }
  320. }
  321. } catch (Exception e) {
  322. e.printStackTrace();
  323. }
  324. }
  325. public static String convertExcelNumberToDate(long excelDate) {
  326. Calendar cal = Calendar.getInstance();
  327. cal.setTime(DateUtil.getJavaDate(excelDate));
  328. return String.format("%d/%d/%d", cal.get(Calendar.YEAR), cal.get(Calendar.MONTH) + 1, cal.get(Calendar.DAY_OF_MONTH));
  329. }
  330. private static Map<String, JSONArray> readExcelManySheet(MultipartFile mFile, File file,Integer cellNum) throws IOException {
  331. Workbook book = getWorkbook(mFile, file);
  332. if (book == null) {
  333. return Collections.emptyMap();
  334. }
  335. Map<String, JSONArray> map = new LinkedHashMap<>();
  336. for (int i = 0; i < book.getNumberOfSheets(); i++) {
  337. Sheet sheet = book.getSheetAt(i);
  338. JSONArray arr = readSheet(sheet,cellNum);
  339. map.put(sheet.getSheetName(), arr);
  340. }
  341. book.close();
  342. return map;
  343. }
  344. private static JSONArray readExcel(MultipartFile mFile, File file,Integer cellNum) throws IOException {
  345. Workbook book = getWorkbook(mFile, file);
  346. if (book == null) {
  347. return new JSONArray();
  348. }
  349. JSONArray array = readSheet(book.getSheetAt(0),cellNum);
  350. book.close();
  351. return array;
  352. }
  353. private static JSONArray readCrmEnterpriseTrademarkInformationExcel(MultipartFile mFile, File file,Integer cellNum,Integer column,String localFileServerUploadDir,String localFileServerMappingPath) throws IOException {
  354. Workbook book = getWorkbook(mFile, file);
  355. if (book == null) {
  356. return new JSONArray();
  357. }
  358. JSONArray array = readCrmEnterpriseTrademarkInformationSheet(mFile,book.getSheetAt(0),cellNum,column, localFileServerUploadDir, localFileServerMappingPath);
  359. book.close();
  360. return array;
  361. }
  362. public static Map<String, PictureData> readPicture(MultipartFile mFile) throws IOException {
  363. if (mFile == null) {
  364. return null;
  365. }
  366. XSSFWorkbook workbook = new XSSFWorkbook(mFile.getInputStream());
  367. XSSFSheet sheet = workbook.getSheetAt(0);
  368. Map<String, PictureData> pictures = getPictures(sheet);
  369. return pictures;
  370. }
  371. private static String readPath(Map<String, PictureData> map,int row,Integer column,String localFileServerUploadDir,String localFileServerMappingPath) {
  372. String fileName = UUID.randomUUID()+".jpg";
  373. String path = "";
  374. String url = "";
  375. try {
  376. // 获取对应列
  377. String imageKey = row + ":" + column;
  378. if (map.containsKey(imageKey)) {
  379. PictureData pictureData = map.get(imageKey);
  380. byte[] image = pictureData.getData();
  381. // 通过hutool工具把图片保存到指定位置
  382. if (image != null) {
  383. //加个时间戳,尽量避免文件名称重复
  384. path = localFileServerUploadDir + fileName;
  385. //创建文件路径
  386. File dest = new File(path);
  387. //判断文件是否已经存在
  388. if (dest.exists()) {
  389. throw new Exception("文件夹已存在");
  390. }
  391. //判断文件父目录是否存在
  392. if (!dest.getParentFile().exists()) {
  393. dest.getParentFile().mkdir();
  394. }
  395. FileUtil.writeBytes(path, image);
  396. url = localFileServerMappingPath+"/"+fileName;
  397. }
  398. }
  399. }catch (Exception e){
  400. e.printStackTrace();
  401. }
  402. return url;
  403. }
  404. private static Workbook getWorkbook(MultipartFile mFile, File file) throws IOException {
  405. InputStream in = null;
  406. PushbackInputStream pushbackIn = null;
  407. try {
  408. if (mFile != null) {
  409. in = mFile.getInputStream();
  410. } else if (file != null && file.exists()) {
  411. in = new FileInputStream(file);
  412. } else {
  413. throw new IllegalArgumentException("Both MultipartFile and File are null or do not exist.");
  414. }
  415. // 使用PushbackInputStream包装原始输入流,以便在检测文件类型后回退到开始
  416. pushbackIn = new PushbackInputStream(in, 8);
  417. // 尝试读取前几个字节来判断文件类型,而不是直接创建POIFSFileSystem
  418. byte[] header = new byte[8];
  419. int bytesRead = pushbackIn.read(header);
  420. if (bytesRead == 8 && isOOXMLHeader(header)) {
  421. // 是.xlsx文件
  422. pushbackIn.unread(header); // 回退读取的字节
  423. return new XSSFWorkbook(pushbackIn);
  424. } else if (bytesRead > 0) {
  425. // 重置输入流到开始,可能是其他类型或旧版Excel文件
  426. pushbackIn.unread(header, 0, bytesRead);
  427. try {
  428. return new HSSFWorkbook(pushbackIn);
  429. } catch (NotOLE2FileException e) {
  430. throw new IllegalArgumentException("Unsupported file format. The file is neither OLE2 (.xls) nor OOXML (.xlsx).", e);
  431. }
  432. } else {
  433. throw new IOException("Failed to read file header.");
  434. }
  435. } catch (IOException e) {
  436. throw new ServiceException("无法读取文件或文件格式不支持,请确保它是有效的Excel文件(.xls或.xlsx)", e);
  437. } finally {
  438. if (pushbackIn != null) {
  439. pushbackIn.close(); // 关闭PushbackInputStream会自动关闭底层流
  440. } else if (in != null) {
  441. in.close();
  442. }
  443. }
  444. }
  445. private static boolean isOOXMLHeader(byte[] header) {
  446. // OOXML(.xlsx)文件的前几个字节应该是:50 4B 03 04
  447. return header[0] == 0x50 && header[1] == 0x4B && header[2] == 0x03 && header[3] == 0x04;
  448. }
  449. private static JSONArray readSheet(Sheet sheet,Integer cellNum) throws IOException {
  450. boolean isFirstRowMerged = false;
  451. for (int i = 0; i < sheet.getNumMergedRegions(); i++) {
  452. CellRangeAddress cellAddresses = sheet.getMergedRegion(i);
  453. if (cellAddresses.getFirstRow() == sheet.getFirstRowNum()) {
  454. isFirstRowMerged = true;
  455. break;
  456. }
  457. }
  458. int rowStart;
  459. if (isFirstRowMerged) {
  460. rowStart = sheet.getFirstRowNum() + 1;
  461. } else {
  462. rowStart = sheet.getFirstRowNum();
  463. }
  464. // 尾行下标
  465. int rowEnd = sheet.getLastRowNum();
  466. // 获取表头行
  467. Row headRow = sheet.getRow(rowStart);
  468. if (headRow == null) {
  469. return new JSONArray();
  470. }
  471. int cellStart = headRow.getFirstCellNum();
  472. int cellEnd = headRow.getLastCellNum();
  473. Map<Integer, String> keyMap = new HashMap<>();
  474. for (int j = cellStart; j < cellEnd; j++) {
  475. // 获取表头数据
  476. String val = getCellValue(headRow.getCell(j));
  477. if (val != null && val.trim().length() != 0 && !"序号".equals(val)) {
  478. keyMap.put(j, val);
  479. }
  480. }
  481. // 如果表头没有数据则不进行解析
  482. if (keyMap.isEmpty()) {
  483. return (JSONArray) Collections.emptyList();
  484. }
  485. //根据表头列数判断是否为正确表格
  486. if(keyMap.size() != cellNum && cellEnd != cellNum){
  487. throw new ServiceException("导入模板不正确");
  488. }
  489. // 获取每行JSON对象的值
  490. JSONArray array = new JSONArray();
  491. // 如果首行与尾行相同,表明只有一行,返回表头数据
  492. if (rowStart == rowEnd) {
  493. /*JSONObject obj = new JSONObject();
  494. // 添加行号
  495. obj.put(ROW_NUM, 1);
  496. for (int i : keyMap.keySet()) {
  497. obj.put(keyMap.get(i), "");
  498. }
  499. array.add(obj);*/
  500. return array;
  501. }
  502. // 获取工作表中所有合并单元格的信息
  503. List<CellRangeAddress> mergedRegions = sheet.getMergedRegions();
  504. for (int i = rowStart + 1; i <= rowEnd; i++) {
  505. Row eachRow = sheet.getRow(i);
  506. if (eachRow == null) {
  507. // 根据实际情况决定是否需要处理或记录这种情况,比如打印日志或者跳过
  508. continue; // 直接跳过这一行
  509. }
  510. JSONObject obj = new JSONObject();
  511. StringBuilder sb = new StringBuilder();
  512. // 遍历列前,先初始化一个变量来存储当前单元格是否属于合并单元格
  513. boolean isMergedCell = false;
  514. for (int k = cellStart; k < cellEnd; k++) {
  515. Cell cell = eachRow.getCell(k);
  516. if (cell != null) {
  517. String val = getCellValue(cell);
  518. // 检查当前单元格是否属于任何合并单元格
  519. for (CellRangeAddress region : mergedRegions) {
  520. if (region.isInRange(i, k)) { // 如果当前单元格在合并单元格内
  521. // 获取合并单元格的左上角单元格的实际值(假设数据只填写在左上角)
  522. val = getCellValue(sheet.getRow(region.getFirstRow()).getCell(region.getFirstColumn()));
  523. isMergedCell = true; // 标记此单元格已处理过合并情况
  524. break;
  525. }
  526. }
  527. // 所有数据添加到里面,用于判断该行是否为空
  528. sb.append(val);
  529. obj.put(keyMap.get(k), val);
  530. // 如果这是一个新的合并单元格的开始,重置isMergedCell为false,以便后续循环能正确处理下一个合并单元格或普通单元格
  531. if (isMergedCell && k == cellStart) {
  532. isMergedCell = false;
  533. }
  534. }
  535. }
  536. if (sb.length() > 0 || isMergedCell) { // 确保即使行数据仅由合并单元格组成也能被加入
  537. array.add(obj);
  538. }
  539. }
  540. return array;
  541. }
  542. private static JSONArray readCrmEnterpriseTrademarkInformationSheet(MultipartFile file,Sheet sheet,Integer cellNum,Integer column,String localFileServerUploadDir,String localFileServerMappingPath) throws IOException {
  543. Map<Integer ,CellRangeAddress> map = new HashMap<>();
  544. for(int i=0;i<sheet.getNumMergedRegions();i++){
  545. CellRangeAddress cellAddresses = sheet.getMergedRegion(i);
  546. map.put(cellAddresses.getFirstColumn(),cellAddresses);
  547. }
  548. int rowStart;
  549. if(sheet.getNumMergedRegions() > 0){
  550. rowStart = sheet.getFirstRowNum()+1;
  551. }else{
  552. // 首行下标
  553. rowStart = sheet.getFirstRowNum();
  554. }
  555. // 尾行下标
  556. int rowEnd = sheet.getLastRowNum();
  557. // 获取表头行
  558. Row headRow = sheet.getRow(rowStart);
  559. if (headRow == null) {
  560. return new JSONArray();
  561. }
  562. int cellStart = headRow.getFirstCellNum();
  563. int cellEnd = headRow.getLastCellNum();
  564. Map<Integer, String> keyMap = new HashMap<>();
  565. for (int j = cellStart; j < cellEnd; j++) {
  566. // 获取表头数据
  567. String val = getCellValue(headRow.getCell(j));
  568. if (val != null && val.trim().length() != 0 && !"序号".equals(val)) {
  569. keyMap.put(j, val);
  570. }
  571. }
  572. // 如果表头没有数据则不进行解析
  573. if (keyMap.isEmpty()) {
  574. return (JSONArray) Collections.emptyList();
  575. }
  576. //根据表头列数判断是否为正确表格
  577. if(keyMap.size() != cellNum){
  578. throw new ServiceException("导入模板不正确");
  579. }
  580. // 获取每行JSON对象的值
  581. JSONArray array = new JSONArray();
  582. // 如果首行与尾行相同,表明只有一行,返回表头数据
  583. if (rowStart == rowEnd) {
  584. /*JSONObject obj = new JSONObject();
  585. // 添加行号
  586. obj.put(ROW_NUM, 1);
  587. for (int i : keyMap.keySet()) {
  588. obj.put(keyMap.get(i), "");
  589. }
  590. array.add(obj);*/
  591. return array;
  592. }
  593. Map<String, PictureData> map1 = readPicture(file);
  594. for (int i = rowStart + 1; i <= rowEnd; i++) {
  595. Row eachRow = sheet.getRow(i);
  596. JSONObject obj = new JSONObject();
  597. // 添加行号
  598. obj.put(ROW_NUM, i + 1);
  599. StringBuilder sb = new StringBuilder();
  600. for (int k = cellStart; k < cellEnd; k++) {
  601. if (eachRow != null) {
  602. String val = getCellValue(eachRow.getCell(k));
  603. if(k==column){
  604. val = readPath(map1,eachRow.getCell(k).getRowIndex(),column, localFileServerUploadDir, localFileServerMappingPath);
  605. }
  606. // 所有数据添加到里面,用于判断该行是否为空
  607. sb.append(val);
  608. obj.put(keyMap.get(k), val);
  609. }
  610. }
  611. if (sb.length() > 0) {
  612. array.add(obj);
  613. }
  614. }
  615. return array;
  616. }
  617. private static Map<String, PictureData> getPictures(XSSFSheet sheet) {
  618. Map<String, PictureData> picMap = new HashMap<>();
  619. // 获取绘图父级
  620. Drawing<?> drawing = sheet.getDrawingPatriarch();
  621. // 检查绘图父级是否为 null
  622. if (drawing == null) {
  623. return picMap; // 如果没有绘图父级,直接返回空的图片映射
  624. }
  625. List<XSSFShape> list = sheet.getDrawingPatriarch().getShapes();
  626. if (CollectionUtils.isEmpty(list)) {
  627. return picMap;
  628. }
  629. for (XSSFShape hssfShape : list) {
  630. // 过滤出图片的数据
  631. if (hssfShape instanceof XSSFPicture) {
  632. XSSFPicture xssfPicture = (XSSFPicture) hssfShape;
  633. XSSFClientAnchor xssfClientAnchor = (XSSFClientAnchor) xssfPicture.getAnchor();
  634. PictureData pictureData = xssfPicture.getPictureData();
  635. picMap.put(xssfClientAnchor.getRow1() + ":" + xssfClientAnchor.getCol1(), pictureData);
  636. }
  637. }
  638. return picMap;
  639. }
  640. private static String getCellValue(Cell cell) {
  641. // 空白或空
  642. if (cell == null || cell.getCellType() == CellType.BLANK) {
  643. return "";
  644. }
  645. // String类型
  646. if (cell.getCellType() == CellType.STRING) {
  647. String val = cell.getStringCellValue();
  648. if (val == null || val.trim().length() == 0) {
  649. return "";
  650. }
  651. return val.trim();
  652. }
  653. // 数字类型
  654. if (cell.getCellType() == CellType.NUMERIC) {
  655. try {
  656. //时间格式
  657. DateTimeFormatter inputFormatter = DateTimeFormatter.ofPattern("d-LLL-yyyy", Locale.CHINESE);
  658. LocalDate date = LocalDate.parse(cell.toString(), inputFormatter);
  659. return date.toString();
  660. } catch (DateTimeParseException e) {
  661. if(String.valueOf(cell).contains("E")) {
  662. DataFormatter dataFormatter = new DataFormatter();
  663. String formattedValue = dataFormatter.formatCellValue(cell); // 使用 DataFormatter 获取格式化后的值
  664. return formattedValue;
  665. }else {
  666. String s = cell.getNumericCellValue() + "";
  667. // 去掉尾巴上的小数点0
  668. if (Pattern.matches(".*\\.0*", s)) {
  669. return s.split("\\.")[0];
  670. } else {
  671. return s;
  672. }
  673. }
  674. }
  675. }
  676. // 布尔值类型
  677. if (cell.getCellType() == CellType.BOOLEAN) {
  678. return cell.getBooleanCellValue() + "";
  679. }
  680. // 公式
  681. if (cell.getCellType() == CellType.FORMULA) {
  682. String s = cell.getNumericCellValue() + "";
  683. // 去掉尾巴上的小数点0
  684. if (Pattern.matches(".*\\.0*", s)) {
  685. return s.split("\\.")[0];
  686. } else if(s.contains("E")){
  687. DecimalFormat decimalFormat = new DecimalFormat("0");
  688. String decimalNumber = decimalFormat.format(Double.parseDouble(s));
  689. return decimalNumber;
  690. }else{
  691. return s;
  692. }
  693. }
  694. /*if (cell.getCellType() == CellType.NUMERIC) {
  695. return String.valueOf(cell.getNumericCellValue());
  696. }*/
  697. return "";
  698. }
  699. // 新增方法用于判断是否可能是日期
  700. private static boolean isLikelyDate(Cell cell) {
  701. // 这里可以根据实际情况调整判断逻辑,例如检查是否包含特定的分隔符、月份名称等
  702. // 下面的例子简单检查是否为负数,避免误将负数识别为日期
  703. return !cell.toString().startsWith("-") && cell.toString().matches("\\d{1,2}-\\p{L}{3}-\\d{4}"); // 或者直接尝试转换看是否成功,这需要根据实际情况调整
  704. }
  705. public static <T> void exportTemplate(HttpServletResponse response, String fileName, Class<T> clazz) {
  706. exportTemplate(response, fileName, fileName, clazz, false);
  707. }
  708. public static <T> void exportTemplate(HttpServletResponse response, String fileName, String sheetName,
  709. Class<T> clazz) {
  710. exportTemplate(response, fileName, sheetName, clazz, false);
  711. }
  712. public static <T> void exportTemplate(HttpServletResponse response, String fileName, Class<T> clazz,
  713. boolean isContainExample) {
  714. exportTemplate(response, fileName, fileName, clazz, isContainExample);
  715. }
  716. public static <T> void exportTemplate(HttpServletResponse response, String fileName, String sheetName,
  717. Class<T> clazz, boolean isContainExample) {
  718. // 获取表头字段
  719. List<ExcelClassField> headFieldList = getExcelClassFieldList(clazz);
  720. // 获取表头数据和示例数据
  721. List<List<Object>> sheetDataList = new ArrayList<>();
  722. List<Object> headList = new ArrayList<>();
  723. List<Object> exampleList = new ArrayList<>();
  724. Map<Integer, List<String>> selectMap = new LinkedHashMap<>();
  725. for (int i = 0; i < headFieldList.size(); i++) {
  726. ExcelClassField each = headFieldList.get(i);
  727. headList.add(each.getName());
  728. exampleList.add(each.getExample());
  729. LinkedHashMap<String, String> kvMap = each.getKvMap();
  730. if (kvMap != null && kvMap.size() > 0) {
  731. selectMap.put(i, new ArrayList<>(kvMap.values()));
  732. }
  733. }
  734. sheetDataList.add(headList);
  735. if (isContainExample) {
  736. sheetDataList.add(exampleList);
  737. }
  738. // 导出数据
  739. export(response, fileName, sheetName, sheetDataList, selectMap);
  740. }
  741. private static <T> List<ExcelClassField> getExcelClassFieldList(Class<T> clazz) {
  742. // 解析所有字段
  743. Field[] fields = clazz.getDeclaredFields();
  744. boolean hasExportAnnotation = false;
  745. Map<Integer, List<ExcelClassField>> map = new LinkedHashMap<>();
  746. List<Integer> sortList = new ArrayList<>();
  747. for (Field field : fields) {
  748. ExcelClassField cf = getExcelClassField(field);
  749. if (cf.getHasAnnotation() == 1) {
  750. hasExportAnnotation = true;
  751. }
  752. int sort = cf.getSort();
  753. if (map.containsKey(sort)) {
  754. map.get(sort).add(cf);
  755. } else {
  756. List<ExcelClassField> list = new ArrayList<>();
  757. list.add(cf);
  758. sortList.add(sort);
  759. map.put(sort, list);
  760. }
  761. }
  762. Collections.sort(sortList);
  763. // 获取表头
  764. List<ExcelClassField> headFieldList = new ArrayList<>();
  765. if (hasExportAnnotation) {
  766. for (Integer sort : sortList) {
  767. for (ExcelClassField cf : map.get(sort)) {
  768. if (cf.getHasAnnotation() == 1) {
  769. headFieldList.add(cf);
  770. }
  771. }
  772. }
  773. } else {
  774. headFieldList.addAll(map.get(0));
  775. }
  776. return headFieldList;
  777. }
  778. private static ExcelClassField getExcelClassField(Field field) {
  779. ExcelClassField cf = new ExcelClassField();
  780. String fieldName = field.getName();
  781. cf.setFieldName(fieldName);
  782. ExcelExport annotation = field.getAnnotation(ExcelExport.class);
  783. // 无 ExcelExport 注解情况
  784. if (annotation == null) {
  785. cf.setHasAnnotation(0);
  786. cf.setName(fieldName);
  787. cf.setSort(0);
  788. return cf;
  789. }
  790. // 有 ExcelExport 注解情况
  791. cf.setHasAnnotation(1);
  792. cf.setName(annotation.value());
  793. String example = getString(annotation.example());
  794. if (!example.isEmpty()) {
  795. if (isNumeric(example) && example.length() < 8) {
  796. cf.setExample(Double.valueOf(example));
  797. } else {
  798. cf.setExample(example);
  799. }
  800. } else {
  801. cf.setExample("");
  802. }
  803. cf.setSort(annotation.sort());
  804. // 解析映射
  805. String kv = getString(annotation.kv());
  806. cf.setKvMap(getKvMap(kv));
  807. return cf;
  808. }
  809. private static LinkedHashMap<String, String> getKvMap(String kv) {
  810. LinkedHashMap<String, String> kvMap = new LinkedHashMap<>();
  811. if (kv.isEmpty()) {
  812. return kvMap;
  813. }
  814. String[] kvs = kv.split(";");
  815. if (kvs.length == 0) {
  816. return kvMap;
  817. }
  818. for (String each : kvs) {
  819. String[] eachKv = getString(each).split("-");
  820. if (eachKv.length != 2) {
  821. continue;
  822. }
  823. String k = eachKv[0];
  824. String v = eachKv[1];
  825. if (k.isEmpty() || v.isEmpty()) {
  826. continue;
  827. }
  828. kvMap.put(k, v);
  829. }
  830. return kvMap;
  831. }
  832. /**
  833. * 导出表格到本地
  834. *
  835. * @param file 本地文件对象
  836. * @param sheetData 导出数据
  837. */
  838. public static void exportFile(File file, List<List<Object>> sheetData) {
  839. if (file == null) {
  840. System.out.println("文件创建失败");
  841. return;
  842. }
  843. if (sheetData == null) {
  844. sheetData = new ArrayList<>();
  845. }
  846. Map<String, List<List<Object>>> map = new HashMap<>();
  847. map.put(file.getName(), sheetData);
  848. export(null, file, file.getName(), map, null);
  849. }
  850. /**
  851. * 导出表格到本地
  852. *
  853. * @param <T> 导出数据类似,和K类型保持一致
  854. * @param filePath 文件父路径(如:D:/doc/excel/)
  855. * @param fileName 文件名称(不带尾缀,如:学生表)
  856. * @param list 导出数据
  857. * @throws IOException IO异常
  858. */
  859. public static <T> File exportFile(String filePath, String fileName, List<T> list) throws IOException {
  860. File file = getFile(filePath, fileName);
  861. List<List<Object>> sheetData = getSheetData(list);
  862. exportFile(file, sheetData);
  863. return file;
  864. }
  865. /**
  866. * 获取文件
  867. *
  868. * @param filePath filePath 文件父路径(如:D:/doc/excel/)
  869. * @param fileName 文件名称(不带尾缀,如:用户表)
  870. * @return 本地File文件对象
  871. */
  872. private static File getFile(String filePath, String fileName) throws IOException {
  873. String dirPath = getString(filePath);
  874. String fileFullPath;
  875. if (dirPath.isEmpty()) {
  876. fileFullPath = fileName;
  877. } else {
  878. // 判定文件夹是否存在,如果不存在,则级联创建
  879. File dirFile = new File(dirPath);
  880. if (!dirFile.exists()) {
  881. boolean mkdirs = dirFile.mkdirs();
  882. if (!mkdirs) {
  883. return null;
  884. }
  885. }
  886. // 获取文件夹全名
  887. if (dirPath.endsWith(String.valueOf(LEAN_LINE))) {
  888. fileFullPath = dirPath + fileName + XLSX;
  889. } else {
  890. fileFullPath = dirPath + LEAN_LINE + fileName + XLSX;
  891. }
  892. }
  893. System.out.println(fileFullPath);
  894. File file = new File(fileFullPath);
  895. if (!file.exists()) {
  896. boolean result = file.createNewFile();
  897. if (!result) {
  898. return null;
  899. }
  900. }
  901. return file;
  902. }
  903. private static <T> List<List<Object>> getSheetData(List<T> list) {
  904. // 获取表头字段
  905. List<ExcelClassField> excelClassFieldList = getExcelClassFieldList(list.get(0).getClass());
  906. List<String> headFieldList = new ArrayList<>();
  907. List<Object> headList = new ArrayList<>();
  908. Map<String, ExcelClassField> headFieldMap = new HashMap<>();
  909. for (ExcelClassField each : excelClassFieldList) {
  910. String fieldName = each.getFieldName();
  911. headFieldList.add(fieldName);
  912. headFieldMap.put(fieldName, each);
  913. headList.add(each.getName());
  914. }
  915. // 添加表头名称
  916. List<List<Object>> sheetDataList = new ArrayList<>();
  917. sheetDataList.add(headList);
  918. // 获取表数据
  919. for (T t : list) {
  920. Map<String, Object> fieldDataMap = getFieldDataMap(t);
  921. Set<String> fieldDataKeys = fieldDataMap.keySet();
  922. List<Object> rowList = new ArrayList<>();
  923. for (String headField : headFieldList) {
  924. if (!fieldDataKeys.contains(headField)) {
  925. continue;
  926. }
  927. Object data = fieldDataMap.get(headField);
  928. if (data == null) {
  929. rowList.add("");
  930. continue;
  931. }
  932. ExcelClassField cf = headFieldMap.get(headField);
  933. // 判断是否有映射关系
  934. LinkedHashMap<String, String> kvMap = cf.getKvMap();
  935. if (kvMap == null || kvMap.isEmpty()) {
  936. rowList.add(data);
  937. continue;
  938. }
  939. String val = kvMap.get(data.toString());
  940. if (isNumeric(val)) {
  941. rowList.add(Double.valueOf(val));
  942. } else {
  943. rowList.add(val);
  944. }
  945. }
  946. sheetDataList.add(rowList);
  947. }
  948. return sheetDataList;
  949. }
  950. private static <T> Map<String, Object> getFieldDataMap(T t) {
  951. Map<String, Object> map = new HashMap<>();
  952. Field[] fields = t.getClass().getDeclaredFields();
  953. try {
  954. for (Field field : fields) {
  955. String fieldName = field.getName();
  956. field.setAccessible(true);
  957. Object object = field.get(t);
  958. map.put(fieldName, object);
  959. }
  960. } catch (IllegalArgumentException | IllegalAccessException e) {
  961. e.printStackTrace();
  962. }
  963. return map;
  964. }
  965. public static void exportEmpty(HttpServletResponse response, String fileName) {
  966. List<List<Object>> sheetDataList = new ArrayList<>();
  967. List<Object> headList = new ArrayList<>();
  968. headList.add("导出无数据");
  969. sheetDataList.add(headList);
  970. export(response, fileName, sheetDataList);
  971. }
  972. public static void export(HttpServletResponse response, String fileName, List<List<Object>> sheetDataList) {
  973. export(response, fileName, fileName, sheetDataList, null);
  974. }
  975. public static void exportManySheet(HttpServletResponse response, String fileName, Map<String, List<List<Object>>> sheetMap) {
  976. export(response, null, fileName, sheetMap, null);
  977. }
  978. public static void export(HttpServletResponse response, String fileName, String sheetName,
  979. List<List<Object>> sheetDataList) {
  980. export(response, fileName, sheetName, sheetDataList, null);
  981. }
  982. public static void export(HttpServletResponse response, String fileName, String sheetName,
  983. List<List<Object>> sheetDataList, Map<Integer, List<String>> selectMap) {
  984. Map<String, List<List<Object>>> map = new HashMap<>();
  985. map.put(sheetName, sheetDataList);
  986. export(response, null, fileName, map, selectMap);
  987. }
  988. public static <T, K> void export(HttpServletResponse response, String fileName, List<T> list, Class<K> template) {
  989. // list 是否为空
  990. boolean lisIsEmpty = list == null || list.isEmpty();
  991. // 如果模板数据为空,且导入的数据为空,则导出空文件
  992. if (template == null && lisIsEmpty) {
  993. exportEmpty(response, fileName);
  994. return;
  995. }
  996. // 如果 list 数据,则导出模板数据
  997. if (lisIsEmpty) {
  998. exportTemplate(response, fileName, template);
  999. return;
  1000. }
  1001. // 导出数据
  1002. List<List<Object>> sheetDataList = getSheetData(list);
  1003. export(response, fileName, sheetDataList);
  1004. }
  1005. public static void export(HttpServletResponse response, String fileName, List<List<Object>> sheetDataList, Map<Integer, List<String>> selectMap) {
  1006. export(response, fileName, fileName, sheetDataList, selectMap);
  1007. }
  1008. private static void export(HttpServletResponse response, File file, String fileName,
  1009. Map<String, List<List<Object>>> sheetMap, Map<Integer, List<String>> selectMap) {
  1010. // 整个 Excel 表格 book 对象
  1011. SXSSFWorkbook book = new SXSSFWorkbook();
  1012. // 每个 Sheet 页
  1013. Set<Entry<String, List<List<Object>>>> entries = sheetMap.entrySet();
  1014. for (Entry<String, List<List<Object>>> entry : entries) {
  1015. List<List<Object>> sheetDataList = entry.getValue();
  1016. Sheet sheet = book.createSheet(entry.getKey());
  1017. Drawing<?> patriarch = sheet.createDrawingPatriarch();
  1018. // 设置表头背景色(灰色)
  1019. CellStyle headStyle = book.createCellStyle();
  1020. headStyle.setFillForegroundColor(IndexedColors.GREY_80_PERCENT.index);
  1021. headStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
  1022. headStyle.setAlignment(HorizontalAlignment.CENTER);
  1023. headStyle.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.index);
  1024. // 设置表身背景色(默认色)
  1025. CellStyle rowStyle = book.createCellStyle();
  1026. rowStyle.setAlignment(HorizontalAlignment.CENTER);
  1027. rowStyle.setVerticalAlignment(VerticalAlignment.CENTER);
  1028. // 设置表格列宽度(默认为15个字节)
  1029. sheet.setDefaultColumnWidth(15);
  1030. // 创建合并算法数组
  1031. int rowLength = sheetDataList.size();
  1032. int columnLength = sheetDataList.get(0).size();
  1033. int[][] mergeArray = new int[rowLength][columnLength];
  1034. for (int i = 0; i < sheetDataList.size(); i++) {
  1035. // 每个 Sheet 页中的行数据
  1036. Row row = sheet.createRow(i);
  1037. List<Object> rowList = sheetDataList.get(i);
  1038. for (int j = 0; j < rowList.size(); j++) {
  1039. // 每个行数据中的单元格数据
  1040. Object o = rowList.get(j);
  1041. int v = 0;
  1042. if (o instanceof URL) {
  1043. // 如果要导出图片的话, 链接需要传递 URL 对象
  1044. setCellPicture(book, row, patriarch, i, j, (URL) o);
  1045. } else {
  1046. Cell cell = row.createCell(j);
  1047. if (i == 0) {
  1048. // 第一行为表头行,采用灰色底背景
  1049. v = setCellValue(cell, o, headStyle);
  1050. } else {
  1051. // 其他行为数据行,默认白底色
  1052. v = setCellValue(cell, o, rowStyle);
  1053. }
  1054. }
  1055. mergeArray[i][j] = v;
  1056. }
  1057. }
  1058. // 合并单元格
  1059. mergeCells(sheet, mergeArray);
  1060. // 设置下拉列表
  1061. setSelect(sheet, selectMap);
  1062. }
  1063. // 写数据
  1064. if (response != null) {
  1065. // 前端导出
  1066. try {
  1067. write(response, book, fileName);
  1068. } catch (IOException e) {
  1069. e.printStackTrace();
  1070. }
  1071. } else {
  1072. // 本地导出
  1073. FileOutputStream fos;
  1074. try {
  1075. fos = new FileOutputStream(file);
  1076. ByteArrayOutputStream ops = new ByteArrayOutputStream();
  1077. book.write(ops);
  1078. fos.write(ops.toByteArray());
  1079. fos.close();
  1080. } catch (Exception e) {
  1081. e.printStackTrace();
  1082. }
  1083. }
  1084. }
  1085. /**
  1086. * 合并当前Sheet页的单元格
  1087. *
  1088. * @param sheet 当前 sheet 页
  1089. * @param mergeArray 合并单元格算法
  1090. */
  1091. private static void mergeCells(Sheet sheet, int[][] mergeArray) {
  1092. // 横向合并
  1093. for (int x = 0; x < mergeArray.length; x++) {
  1094. int[] arr = mergeArray[x];
  1095. boolean merge = false;
  1096. int y1 = 0;
  1097. int y2 = 0;
  1098. for (int y = 0; y < arr.length; y++) {
  1099. int value = arr[y];
  1100. if (value == CELL_COLUMN_MERGE) {
  1101. if (!merge) {
  1102. y1 = y;
  1103. }
  1104. y2 = y;
  1105. merge = true;
  1106. } else {
  1107. merge = false;
  1108. if (y1 > 0) {
  1109. sheet.addMergedRegion(new CellRangeAddress(x, x, (y1 - 1), y2));
  1110. }
  1111. y1 = 0;
  1112. y2 = 0;
  1113. }
  1114. }
  1115. if (y1 > 0) {
  1116. sheet.addMergedRegion(new CellRangeAddress(x, x, (y1 - 1), y2));
  1117. }
  1118. }
  1119. // 纵向合并
  1120. int xLen = mergeArray.length;
  1121. int yLen = mergeArray[0].length;
  1122. for (int y = 0; y < yLen; y++) {
  1123. boolean merge = false;
  1124. int x1 = 0;
  1125. int x2 = 0;
  1126. for (int x = 0; x < xLen; x++) {
  1127. int value = mergeArray[x][y];
  1128. if (value == CELL_ROW_MERGE) {
  1129. if (!merge) {
  1130. x1 = x;
  1131. }
  1132. x2 = x;
  1133. merge = true;
  1134. } else {
  1135. merge = false;
  1136. if (x1 > 0) {
  1137. sheet.addMergedRegion(new CellRangeAddress((x1 - 1), x2, y, y));
  1138. }
  1139. x1 = 0;
  1140. x2 = 0;
  1141. }
  1142. }
  1143. if (x1 > 0) {
  1144. sheet.addMergedRegion(new CellRangeAddress((x1 - 1), x2, y, y));
  1145. }
  1146. }
  1147. }
  1148. private static void write(HttpServletResponse response, SXSSFWorkbook book, String fileName) throws IOException {
  1149. response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
  1150. response.setCharacterEncoding("utf-8");
  1151. String name = new String(fileName.getBytes("GBK"), "ISO8859_1") + XLSX;
  1152. response.addHeader("Content-Disposition", "attachment;filename=" + name);
  1153. ServletOutputStream out = response.getOutputStream();
  1154. book.write(out);
  1155. out.flush();
  1156. out.close();
  1157. }
  1158. private static int setCellValue(Cell cell, Object o, CellStyle style) {
  1159. // 设置样式
  1160. cell.setCellStyle(style);
  1161. // 数据为空时
  1162. if (o == null) {
  1163. cell.setCellType(CellType.STRING);
  1164. cell.setCellValue("");
  1165. return CELL_OTHER;
  1166. }
  1167. // 是否为字符串
  1168. if (o instanceof String) {
  1169. String s = o.toString();
  1170. // 当数字类型长度超过8位时,改为字符串类型显示(Excel数字超过一定长度会显示为科学计数法)
  1171. if (isNumeric(s) && s.length() < 8) {
  1172. cell.setCellType(CellType.NUMERIC);
  1173. cell.setCellValue(Double.parseDouble(s));
  1174. return CELL_OTHER;
  1175. } else {
  1176. cell.setCellType(CellType.STRING);
  1177. cell.setCellValue(s);
  1178. }
  1179. if (s.equals(ROW_MERGE)) {
  1180. return CELL_ROW_MERGE;
  1181. } else if (s.equals(COLUMN_MERGE)) {
  1182. return CELL_COLUMN_MERGE;
  1183. } else {
  1184. return CELL_OTHER;
  1185. }
  1186. }
  1187. // 是否为字符串
  1188. if (o instanceof Integer || o instanceof Long || o instanceof Double || o instanceof Float) {
  1189. cell.setCellType(CellType.NUMERIC);
  1190. cell.setCellValue(Double.parseDouble(o.toString()));
  1191. return CELL_OTHER;
  1192. }
  1193. // 是否为Boolean
  1194. if (o instanceof Boolean) {
  1195. cell.setCellType(CellType.BOOLEAN);
  1196. cell.setCellValue((Boolean) o);
  1197. return CELL_OTHER;
  1198. }
  1199. // 如果是BigDecimal,则默认3位小数
  1200. if (o instanceof BigDecimal) {
  1201. cell.setCellType(CellType.NUMERIC);
  1202. cell.setCellValue(((BigDecimal) o).setScale(3, RoundingMode.HALF_UP).doubleValue());
  1203. return CELL_OTHER;
  1204. }
  1205. // 如果是Date数据,则显示格式化数据
  1206. if (o instanceof Date) {
  1207. cell.setCellType(CellType.STRING);
  1208. cell.setCellValue(formatDate((Date) o));
  1209. return CELL_OTHER;
  1210. }
  1211. // 如果是其他,则默认字符串类型
  1212. cell.setCellType(CellType.STRING);
  1213. cell.setCellValue(o.toString());
  1214. return CELL_OTHER;
  1215. }
  1216. private static void setCellPicture(SXSSFWorkbook wb, Row sr, Drawing<?> patriarch, int x, int y, URL url) {
  1217. // 设置图片宽高
  1218. sr.setHeight((short) (IMG_WIDTH * IMG_HEIGHT));
  1219. // (jdk1.7版本try中定义流可自动关闭)
  1220. try (InputStream is = url.openStream(); ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
  1221. byte[] buff = new byte[BYTES_DEFAULT_LENGTH];
  1222. int rc;
  1223. while ((rc = is.read(buff, 0, BYTES_DEFAULT_LENGTH)) > 0) {
  1224. outputStream.write(buff, 0, rc);
  1225. }
  1226. // 设置图片位置
  1227. XSSFClientAnchor anchor = new XSSFClientAnchor(0, 0, 0, 0, y, x, y + 1, x + 1);
  1228. // 设置这个,图片会自动填满单元格的长宽
  1229. anchor.setAnchorType(AnchorType.MOVE_AND_RESIZE);
  1230. patriarch.createPicture(anchor, wb.addPicture(outputStream.toByteArray(), HSSFWorkbook.PICTURE_TYPE_JPEG));
  1231. } catch (Exception e) {
  1232. e.printStackTrace();
  1233. }
  1234. }
  1235. private static String formatDate(Date date) {
  1236. if (date == null) {
  1237. return "";
  1238. }
  1239. SimpleDateFormat format = new SimpleDateFormat(DATE_FORMAT);
  1240. return format.format(date);
  1241. }
  1242. private static void setSelect(Sheet sheet, Map<Integer, List<String>> selectMap) {
  1243. if (selectMap == null || selectMap.isEmpty()) {
  1244. return;
  1245. }
  1246. Set<Entry<Integer, List<String>>> entrySet = selectMap.entrySet();
  1247. for (Entry<Integer, List<String>> entry : entrySet) {
  1248. int y = entry.getKey();
  1249. List<String> list = entry.getValue();
  1250. if (list == null || list.isEmpty()) {
  1251. continue;
  1252. }
  1253. String[] arr = new String[list.size()];
  1254. for (int i = 0; i < list.size(); i++) {
  1255. arr[i] = list.get(i);
  1256. }
  1257. DataValidationHelper helper = sheet.getDataValidationHelper();
  1258. CellRangeAddressList addressList = new CellRangeAddressList(1, 65000, y, y);
  1259. DataValidationConstraint dvc = helper.createExplicitListConstraint(arr);
  1260. DataValidation dv = helper.createValidation(dvc, addressList);
  1261. if (dv instanceof HSSFDataValidation) {
  1262. dv.setSuppressDropDownArrow(false);
  1263. } else {
  1264. dv.setSuppressDropDownArrow(true);
  1265. dv.setShowErrorBox(true);
  1266. }
  1267. sheet.addValidationData(dv);
  1268. }
  1269. }
  1270. private static boolean isNumeric(String str) {
  1271. if (Objects.nonNull(str) && "0.0".equals(str)) {
  1272. return true;
  1273. }
  1274. if ("".equals(str)){
  1275. return false;
  1276. }
  1277. for (int i = str.length(); --i >= 0; ) {
  1278. if (!Character.isDigit(str.charAt(i))) {
  1279. return false;
  1280. }
  1281. }
  1282. return true;
  1283. }
  1284. private static String getString(String s) {
  1285. if (s == null) {
  1286. return "";
  1287. }
  1288. if (s.isEmpty()) {
  1289. return s;
  1290. }
  1291. return s.trim();
  1292. }
  1293. }