完成匹配、跟价、权限部分

2026-04-17 12:52:16 +08:00
parent ecc8b6ce6c
commit d5764f3b50
63 changed files with 8858 additions and 1473 deletions
--- a/backend-java/src/main/java/com/nanri/aiimage/common/util/ExcelStreamReader.java
+++ b/backend-java/src/main/java/com/nanri/aiimage/common/util/ExcelStreamReader.java
@@ -0,0 +1,126 @@
+package com.nanri.aiimage.common.util;
+
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.context.AnalysisContext;
+import com.alibaba.excel.event.AnalysisEventListener;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+public final class ExcelStreamReader {
+
+    private ExcelStreamReader() {
+    }
+
+    public static void readFirstSheet(File file, SheetRowHandler handler) throws IOException {
+        try (InputStream inputStream = new FileInputStream(file)) {
+            readFirstSheet(inputStream, handler);
+        }
+    }
+
+    public static void readFirstSheet(InputStream inputStream, SheetRowHandler handler) throws IOException {
+        runRead(inputStream, false, handler);
+    }
+
+    public static void readAllSheets(File file, SheetRowHandler handler) throws IOException {
+        try (InputStream inputStream = new FileInputStream(file)) {
+            readAllSheets(inputStream, handler);
+        }
+    }
+
+    public static void readAllSheets(InputStream inputStream, SheetRowHandler handler) throws IOException {
+        runRead(inputStream, true, handler);
+    }
+
+    private static void runRead(InputStream inputStream, boolean readAllSheets, SheetRowHandler handler) throws IOException {
+        StreamingListener listener = new StreamingListener(handler);
+        try {
+            if (readAllSheets) {
+                EasyExcel.read(inputStream, listener).headRowNumber(1).doReadAll();
+            } else {
+                EasyExcel.read(inputStream, listener).sheet(0).headRowNumber(1).doRead();
+            }
+        } catch (WrappedRuntimeException ex) {
+            if (ex.getCause() instanceof IOException ioException) {
+                throw ioException;
+            }
+            throw ex;
+        }
+    }
+
+    public interface SheetRowHandler {
+        default void onHeader(String sheetName, Integer sheetNo, Map<Integer, String> headerMap) throws Exception {
+        }
+
+        void onRow(String sheetName, Integer sheetNo, int rowIndex, Map<Integer, String> headerMap, Map<Integer, String> rowMap) throws Exception;
+    }
+
+    private static final class StreamingListener extends AnalysisEventListener<Map<Integer, String>> {
+
+        private final SheetRowHandler handler;
+        private Map<Integer, String> currentHeaderMap = Map.of();
+
+        private StreamingListener(SheetRowHandler handler) {
+            this.handler = handler;
+        }
+
+        @Override
+        public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {
+            Map<Integer, String> normalizedHeadMap = new LinkedHashMap<>();
+            for (Map.Entry<Integer, String> entry : headMap.entrySet()) {
+                normalizedHeadMap.put(entry.getKey(), entry.getValue());
+            }
+            currentHeaderMap = normalizedHeadMap;
+            try {
+                handler.onHeader(sheetName(context), sheetNo(context), currentHeaderMap);
+            } catch (RuntimeException ex) {
+                throw ex;
+            } catch (Exception ex) {
+                throw new WrappedRuntimeException(ex);
+            }
+        }
+
+        @Override
+        public void invoke(Map<Integer, String> data, AnalysisContext context) {
+            Map<Integer, String> normalizedRow = new LinkedHashMap<>();
+            for (Map.Entry<Integer, String> entry : data.entrySet()) {
+                normalizedRow.put(entry.getKey(), entry.getValue());
+            }
+            try {
+                handler.onRow(
+                        sheetName(context),
+                        sheetNo(context),
+                        context.readRowHolder().getRowIndex(),
+                        currentHeaderMap,
+                        normalizedRow
+                );
+            } catch (RuntimeException ex) {
+                throw ex;
+            } catch (Exception ex) {
+                throw new WrappedRuntimeException(ex);
+            }
+        }
+
+        @Override
+        public void doAfterAllAnalysed(AnalysisContext context) {
+        }
+
+        private String sheetName(AnalysisContext context) {
+            return context.readSheetHolder() == null ? "" : context.readSheetHolder().getSheetName();
+        }
+
+        private Integer sheetNo(AnalysisContext context) {
+            return context.readSheetHolder() == null ? null : context.readSheetHolder().getSheetNo();
+        }
+    }
+
+    private static final class WrappedRuntimeException extends RuntimeException {
+        private WrappedRuntimeException(Throwable cause) {
+            super(cause);
+        }
+    }
+}