Merge branch 'master' of gitee.com:youlaiorg/youlai-boot

src/main/java/com/youlai/system/controller/GeneratorController.java

@@ -11,9 +11,12 @@ import com.youlai.system.service.GeneratorService;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.Parameter;
 import io.swagger.v3.oas.annotations.tags.Tag;
+import jakarta.servlet.http.HttpServletResponse;
 import lombok.RequiredArgsConstructor;
+import org.apache.commons.io.IOUtils;
 import org.springframework.web.bind.annotation.*;
 
+import java.io.IOException;
 import java.util.List;
 
 /**
@@ -71,4 +74,22 @@ public class GeneratorController {
         return Result.success(list);
     }
 
+
+
+    @Operation(summary = "下载代码zip")
+    @GetMapping("/{tableName}/downloadZip")
+    public void downloadZip(HttpServletResponse response, @PathVariable String tableName) throws IOException {
+        String[] tableNames = tableName.split(",");
+        byte[] data = generatorService.downloadCode(tableNames);
+        response.reset();
+        response.setHeader("Content-Disposition", "attachment; filename=\"youlai-admin-code.zip\"");
+        response.addHeader("Content-Length", "" + data.length);
+        response.addHeader("Access-Control-Allow-Origin", "*");
+        response.setHeader("Pragma", "no-cache");
+        response.setHeader("Cache-Control", "no-cache");
+        response.addHeader("Access-Control-Expose-Headers", "Content-Disposition");
+        response.setContentType("application/octet-stream; charset=UTF-8");
+        response.setDateHeader("Expires", 0);
+        IOUtils.write(data, response.getOutputStream());
+    }
 }

src/main/java/com/youlai/system/service/GeneratorService.java

@@ -55,4 +55,11 @@ public interface GeneratorService {
      * @return
      */
     void deleteGenConfig(String tableName);
+
+    /**
+     * 下载代码
+     * @param tableNames 表名
+     * @return
+     */
+    byte[] downloadCode(String[] tableNames);
 }

src/main/java/com/youlai/system/service/impl/GeneratorServiceImpl.java

@@ -32,11 +32,15 @@ import com.youlai.system.service.GenConfigService;
 import com.youlai.system.service.GenFieldConfigService;
 import com.youlai.system.service.SysMenuService;
 import lombok.RequiredArgsConstructor;
+import org.apache.commons.io.IOUtils;
 import org.springframework.stereotype.Service;
 
+import java.io.ByteArrayOutputStream;
 import java.io.File;
 import java.util.*;
 import java.util.stream.Collectors;
+import java.util.zip.ZipOutputStream;
+
 
 /**
  * 数据库服务实现类
@@ -383,5 +387,43 @@ public class GeneratorServiceImpl implements GeneratorService {
         return content;
     }
 
+    /**
+     * 下载代码
+     * @param tableNames 表名，可以支持多张表。
+     * @return 压缩文件字节数组
+     */
+    @Override
+    public byte[] downloadCode(String[] tableNames) {
+        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
+        ZipOutputStream zip = new ZipOutputStream(outputStream);
+        for (String tableName : tableNames)
+        {
+            generatorCode(tableName, zip);
+        }
+        IOUtils.closeQuietly(zip);
+        return outputStream.toByteArray();
+    }
+
+    /**
+     * 根据表名生成代码并且压缩到zip文件中
+     *
+     * @param tableName 单个表名
+     * @param zip 压缩文件
+     */
+    private void generatorCode(String tableName, ZipOutputStream zip) {
+        List<GeneratorPreviewVO> previewVOList = getTablePreviewData(tableName);
+        for (GeneratorPreviewVO previewVO : previewVOList) {
+            String fileName = previewVO.getFileName();
+            String content = previewVO.getContent();
+            String path = previewVO.getPath();
+            try {
+                zip.putNextEntry(new java.util.zip.ZipEntry(path + File.separator + fileName));
+                zip.write(content.getBytes("UTF-8"));
+                zip.closeEntry();
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        }
+    }
 
 }