From faa2a5d8d3cba12ff98c261713bff4d9cea99b79 Mon Sep 17 00:00:00 2001
From: skyouc
Date: 星期三, 09 四月 2025 17:23:40 +0800
Subject: [PATCH] Merge branch 'devlop' of http://47.97.1.152:5880/r/wms-master into devlop

---
 rsf-server/src/main/java/com/vincent/rsf/server/common/utils/ExcelUtil.java |  197 +++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 164 insertions(+), 33 deletions(-)

diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/common/utils/ExcelUtil.java b/rsf-server/src/main/java/com/vincent/rsf/server/common/utils/ExcelUtil.java
index ebffe54..43131e0 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/common/utils/ExcelUtil.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/common/utils/ExcelUtil.java
@@ -1,11 +1,19 @@
 package com.vincent.rsf.server.common.utils;
 
+import cn.afterturn.easypoi.excel.annotation.Excel;
+import cn.afterturn.easypoi.excel.entity.ImportParams;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.vincent.rsf.framework.common.Cools;
+import com.vincent.rsf.framework.common.SpringUtils;
+import com.vincent.rsf.server.manager.entity.excel.MatnrsTemplate;
+import com.vincent.rsf.server.manager.entity.excel.annotation.ExcelComment;
+import com.vincent.rsf.server.system.entity.Fields;
+import com.vincent.rsf.server.system.service.FieldsService;
 import io.swagger.annotations.ApiModelProperty;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
 import org.apache.poi.hssf.usermodel.HSSFWorkbook;
-import org.apache.poi.ss.usermodel.Row;
-import org.apache.poi.ss.usermodel.Sheet;
-import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.ss.usermodel.*;
 
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
@@ -13,14 +21,13 @@
 import java.lang.reflect.Modifier;
 import java.net.URLEncoder;
 import java.text.SimpleDateFormat;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
 
 /**
  * Created by vincent on 2/17/2024
  */
+@Slf4j
 public class ExcelUtil {
-
     public static void build(Workbook workbook, HttpServletResponse response) {
         response.reset();
         Http.cors(response);
@@ -28,16 +35,27 @@
         try {
             response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode("export", "UTF-8"));
             workbook.write(response.getOutputStream());
-        } catch (IOException ignore) {}
+        } catch (IOException ignore) {
+        }
     }
 
     public static <T> Workbook create(List<T> list, Class<T> clz) {
+        return create(list, clz, false);
+    }
+
+    public static <T> Workbook create(List<T> list, Class<T> clz, boolean flagTemplate) {
         HSSFWorkbook workbook = new HSSFWorkbook();
         Sheet sheet = workbook.createSheet(clz.getSimpleName());
 
         Row header = sheet.createRow(0);
         Field[] fields = Cools.getAllFields(clz);
         int headerIdx = 0;
+        FieldsService itemService = SpringUtils.getBean(FieldsService.class);
+        List<Fields> sysFields = itemService.list(new LambdaQueryWrapper<Fields>()
+                .eq(Fields::getStatus, 1)
+                .eq(Fields::getFlagEnable, 1));
+
+
         for (Field field : fields) {
             if (Modifier.isFinal(field.getModifiers())
                     || Modifier.isStatic(field.getModifiers())
@@ -45,44 +63,66 @@
                 continue;
             }
             String memo = "Undefined";
+            if (flagTemplate) {
+                memo = field.getName();
+            } else {
+                if (field.isAnnotationPresent(Excel.class)) {
+                    memo = field.getAnnotation(Excel.class).name();
+                }
+            }
             if (field.isAnnotationPresent(ApiModelProperty.class)) {
                 memo = field.getAnnotation(ApiModelProperty.class).value();
             }
             header.createCell(headerIdx).setCellValue(memo);
             headerIdx++;
         }
-
-        int rowIndex = 1;
-        for (T t : list) {
-            Row row = sheet.createRow(rowIndex++);
-
-            int cellIndex = 0;
-            for (Field field : fields) {
-                if (Modifier.isFinal(field.getModifiers())
-                        || Modifier.isStatic(field.getModifiers())
-                        || Modifier.isTransient(field.getModifiers())) {
-                    continue;
-                }
-
-                field.setAccessible(true);  // 姝よ寰堥噸瑕侊紝鐗瑰埆鏄瓧娈典负private鏃�
-                Object value = null;
-                try {
-                    value = field.get(t);
-                } catch (IllegalAccessException e) {
-                    e.printStackTrace();
-                }
-                if (value != null) {
-                    if (value instanceof Date) {
-                        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-                        row.createCell(cellIndex).setCellValue(sdf.format((Date) value));
+        if (clz.equals(MatnrsTemplate.class)) {
+            //娣诲姞鎵╁睍瀛楁鍒悕
+            if (!sysFields.isEmpty()) {
+                for (Fields field : sysFields) {
+                    if (flagTemplate) {
+                        header.createCell(headerIdx).setCellValue(field.getFields());
                     } else {
-                        row.createCell(cellIndex).setCellValue(value.toString());
+                        header.createCell(headerIdx).setCellValue(field.getFieldsAlise());
                     }
+                    headerIdx++;
                 }
-                cellIndex++;
             }
         }
 
+        int rowIndex = 1;
+        if (!Objects.isNull(list)) {
+            for (T t : list) {
+                Row row = sheet.createRow(rowIndex++);
+
+                int cellIndex = 0;
+                for (Field field : fields) {
+                    if (Modifier.isFinal(field.getModifiers())
+                            || Modifier.isStatic(field.getModifiers())
+                            || Modifier.isTransient(field.getModifiers())) {
+                        continue;
+                    }
+
+                    // 姝よ寰堥噸瑕侊紝鐗瑰埆鏄瓧娈典负private鏃�
+                    field.setAccessible(true);
+                    Object value = null;
+                    try {
+                        value = field.get(t);
+                    } catch (IllegalAccessException e) {
+                        e.printStackTrace();
+                    }
+                    if (value != null) {
+                        if (value instanceof Date) {
+                            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+                            row.createCell(cellIndex).setCellValue(sdf.format((Date) value));
+                        } else {
+                            row.createCell(cellIndex).setCellValue(value.toString());
+                        }
+                    }
+                    cellIndex++;
+                }
+            }
+        }
         for (int i = 0; i <= fields.length; i++) {
             sheet.autoSizeColumn(i);
         }
@@ -90,6 +130,97 @@
         return workbook;
     }
 
+    /**
+     * 娣诲姞瀵煎叆excel閰嶇疆鍙傛暟
+     * 娉細榛樿閰嶇疆鍙弧瓒冲綋鍓嶉渶姹�
+     *
+     * @return
+     */
+    public static ImportParams getDefaultImportParams() {
+        ImportParams importParams = new ImportParams();
+        importParams.setTitleRows(0);
+//        importParams.setNeedVerify(true);
+        importParams.setHeadRows(1);
+        importParams.setSheetNum(1);
+        return importParams;
+    }
+
+    /**
+     * 鏍规嵁 {@code tClass} 鐩稿叧鎴愬憳鍙橀噺鐨� {@link ExcelComment#example()} 瀛楁鍒涘缓妯℃嫙鏁版嵁锛屾殏涓嶆敮鎸� 澶嶆潅绫诲瀷
+     *
+     * @param tClass
+     * @return
+     */
+    public static <T> T mockData(Class<T> tClass) {
+        if (tClass == null) {
+            return null;
+        }
+        T instance = null;
+
+        try {
+            instance = tClass.newInstance();
+            Field[] declaredFields = tClass.getDeclaredFields();
+            for (Field declaredField : declaredFields) {
+                ExcelComment comment = declaredField.getAnnotation(ExcelComment.class);
+                if (comment == null) {
+                    continue;
+                }
+                declaredField.setAccessible(true);
+                Class<?> fieldType = declaredField.getType();
+
+                String exampleValue = comment.example();
+                Object value = null;
+
+                if (fieldType == int.class || fieldType == Integer.class) {
+                    value = StringUtils.isBlank(exampleValue) ? 0 : Integer.parseInt(exampleValue);
+                } else if (fieldType == short.class || fieldType == Short.class) {
+                    value = StringUtils.isBlank(exampleValue) ? 0 : Short.parseShort(exampleValue);
+                } else if (fieldType == long.class || fieldType == Long.class) {
+                    value = StringUtils.isBlank(exampleValue) ? 0 : Long.parseLong(exampleValue);
+                } else if (fieldType == double.class || fieldType == Double.class) {
+                    value = StringUtils.isBlank(exampleValue) ? 0 : Double.parseDouble(exampleValue);
+                } else if (fieldType == boolean.class || fieldType == Boolean.class) {
+                    value = StringUtils.isNotBlank(exampleValue) && Boolean.parseBoolean(exampleValue);
+                } else if (fieldType == String.class) {
+                    value = exampleValue;
+                } else if (fieldType == Date.class) {
+                    value = DateUtils.parse(exampleValue);
+                }
+
+                if (value == null && !isBaseType(fieldType)) {
+                    declaredField.set(instance, null);
+                } else {
+                    declaredField.set(instance, value);
+                }
+            }
+        } catch (Exception e) {
+            log.error("鏁版嵁鏋勯�犲け璐ワ紝璇锋煡璇㈣缁嗕俊鎭�", e);
+            return instance;
+        }
+
+        return instance;
+    }
+
+
+    /**
+     * 鏄惁鏄熀纭�鏁版嵁绫诲瀷
+     *
+     * @param className
+     * @return
+     */
+    private static boolean isBaseType(Class<?> className) {
+        if (className.equals(java.lang.Integer.class) ||
+                className.equals(java.lang.Byte.class) ||
+                className.equals(java.lang.Long.class) ||
+                className.equals(java.lang.Double.class) ||
+                className.equals(java.lang.Float.class) ||
+                className.equals(java.lang.Character.class) ||
+                className.equals(java.lang.Short.class) ||
+                className.equals(java.lang.Boolean.class)) {
+            return true;
+        }
+        return false;
+    }
 
 
 }

--
Gitblit v1.9.1