package com.kidgrow.generator.controller;
|
|
import com.kidgrow.common.model.PageResult;
|
import com.kidgrow.generator.service.SysGeneratorService;
|
import io.swagger.annotations.Api;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.io.IOUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
|
*
|
* @Description: 代码生成器<br>
|
* @Project: <br>
|
* @CreateDate: Created in 2020/2/24 09:40 <br>
|
* @Author: <a href="4345453@kidgrow.com">liuke</a>
|
*/
|
@RestController
|
@Api(tags = "代码生成器")
|
@RequestMapping("/generator")
|
@Slf4j
|
public class SysGeneratorController {
|
@Autowired
|
private SysGeneratorService sysGeneratorService;
|
|
/**
|
* 列表
|
*/
|
@ResponseBody
|
@GetMapping("/list")
|
public PageResult getTableList(@RequestParam Map<String, Object> params) {
|
return sysGeneratorService.queryList(params);
|
}
|
|
|
@GetMapping("/codeAll")
|
public void makeCodeAll(HttpServletResponse response) throws IOException {
|
|
Map<String, Object> map = new HashMap<String, Object>();
|
map.put("page",1);
|
map.put("limit",200);
|
|
PageResult pageResult = sysGeneratorService.queryList(map);
|
List<HashMap<String,String>> tableEntityList = pageResult.getData();
|
String[] tableNames = new String[tableEntityList.size()];
|
for(int i=0;i<tableEntityList.size();i++){
|
log.info(tableEntityList.get(i).get("tableName"));
|
tableNames[i]=tableEntityList.get(i).get("tableName");
|
}
|
|
byte[] data = sysGeneratorService.generatorCode(tableNames);
|
response.reset();
|
response.setHeader("Content-Disposition", "attachment; filename=\"generator.zip\"");
|
response.addHeader("Content-Length", "" + data.length);
|
response.setContentType("application/octet-stream; charset=UTF-8");
|
IOUtils.write(data, response.getOutputStream());
|
}
|
|
/**
|
* 生成代码FileUtil
|
*/
|
@GetMapping("/code")
|
public void makeCode(String tables, HttpServletResponse response) throws IOException {
|
byte[] data = sysGeneratorService.generatorCode(tables.split(","));
|
response.reset();
|
response.setHeader("Content-Disposition", "attachment; filename=\"generator.zip\"");
|
response.addHeader("Content-Length", "" + data.length);
|
response.setContentType("application/octet-stream; charset=UTF-8");
|
IOUtils.write(data, response.getOutputStream());
|
}
|
}
|
