github
/
RuoYi
mirror of https://gitee.com/y_project/RuoYi.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

!90 抽取重复代码到一个方法 

Merge pull request !90 from lywjlong/master
pull/90/MERGE
若依 2019-05-25 09:28:12 +08:00 committed by Gitee
parent 89c4659911 30a7806328
commit 661db11765
1 changed files with 94 additions and 89 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
ruoyi-generator/src/main/java/com/ruoyi/generator/controller/GenController.java
View File

@ -60,12 +60,7 @@ public class GenController extends BaseController
public void genCode(HttpServletResponse response, @PathVariable("tableName") String tableName) throws IOException
{
byte[] data = genService.generatorCode(tableName);
response.reset();
response.setHeader("Content-Disposition", "attachment; filename=\"ruoyi.zip\"");
response.addHeader("Content-Length", "" + data.length);
response.setContentType("application/octet-stream; charset=UTF-8");
IOUtils.write(data, response.getOutputStream());
genCode(response, data);
}
/**
@ -79,6 +74,16 @@ public class GenController extends BaseController
{
String[] tableNames = Convert.toStrArray(tables);
byte[] data = genService.generatorCode(tableNames);
genCode(response, data);
}
/**
* 生成zip文件
* @param response
* @param data
* @throws IOException
*/
private void genCode(HttpServletResponse response, byte[] data) throws IOException {
response.reset();
response.setHeader("Content-Disposition", "attachment; filename=\"ruoyi.zip\"");
response.addHeader("Content-Length", "" + data.length);