tongtong/youlai-boot
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 下载代码zip

Browse Source 
下载zip模块。后端代码部分。
This commit is contained in:
Ky10
2024-08-01 17:49:07 +08:00
parent e5f46f63cc
commit 68a5725e06
3 changed files with 70 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/main/java/com/youlai/system/controller/GeneratorController.java
View File

@@ -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());
}
}