From 93655d1915250c4d623eaf033efc4cd35ca32dad Mon Sep 17 00:00:00 2001
From: "Ray.Hao" <1490493387@qq.com>
Date: Fri, 13 Sep 2024 18:31:42 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BB=A3=E7=A0=81=E4=B8=8B=E8=BD=BD?=
 =?UTF-8?q?=E8=A7=A3=E5=8E=8B=E6=8F=90=E7=A4=BA=E6=95=B0=E6=8D=AE=E4=B8=8D?=
 =?UTF-8?q?=E5=AE=8C=E6=95=B4=E9=97=AE=E9=A2=98=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../boot/module/codegen/service/impl/CodegenServiceImpl.java   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/youlai/boot/module/codegen/service/impl/CodegenServiceImpl.java b/src/main/java/com/youlai/boot/module/codegen/service/impl/CodegenServiceImpl.java
index 89aa25a0..7e54dc91 100644
--- a/src/main/java/com/youlai/boot/module/codegen/service/impl/CodegenServiceImpl.java
+++ b/src/main/java/com/youlai/boot/module/codegen/service/impl/CodegenServiceImpl.java
@@ -273,7 +273,8 @@ public class CodegenServiceImpl implements CodegenService {
             for (String tableName : tableNames) {
                 generateAndZipCode(tableName, zip);
             }
-
+            // 确保所有压缩数据写入输出流，避免数据残留在内存缓冲区引发的数据不完整
+            zip.finish();
             return outputStream.toByteArray();
 
         } catch (IOException e) {