package com.ots.project.tool.gen.controller;
|
import com.ots.common.utils.text.Convert;
|
import com.ots.framework.aspectj.lang.annotation.Log;
|
import com.ots.framework.aspectj.lang.enums.BusinessType;
|
import com.ots.framework.web.controller.BaseController;
|
import com.ots.framework.web.domain.AjaxResult;
|
import com.ots.framework.web.page.TableDataInfo;
|
import com.ots.project.tool.gen.domain.GenTable;
|
import com.ots.project.tool.gen.domain.GenTableColumn;
|
import com.ots.project.tool.gen.service.IGenTableColumnService;
|
import com.ots.project.tool.gen.service.IGenTableService;
|
import org.apache.commons.io.IOUtils;
|
import org.apache.shiro.authz.annotation.RequiresPermissions;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.ui.ModelMap;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.*;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.util.List;
|
import java.util.Map;
|
|
@Controller
|
@RequestMapping("/tool/gen")
|
public class GenController extends BaseController {
|
private String prefix = "tool/gen";
|
@Autowired
|
private IGenTableService genTableService;
|
@Autowired
|
private IGenTableColumnService genTableColumnService;
|
@RequiresPermissions("tool:gen:view")
|
@GetMapping()
|
public String gen() {
|
return prefix + "/gen";
|
}
|
|
@RequiresPermissions("tool:gen:list")
|
@PostMapping("/list")
|
@ResponseBody
|
public TableDataInfo genList(GenTable genTable) {
|
startPage();
|
List<GenTable> list = genTableService.selectGenTableList(genTable);
|
return getDataTable(list);
|
}
|
|
@RequiresPermissions("tool:gen:list")
|
@PostMapping("/db/list")
|
@ResponseBody
|
public TableDataInfo dataList(GenTable genTable) {
|
startPage();
|
List<GenTable> list = genTableService.selectDbTableList(genTable);
|
return getDataTable(list);
|
}
|
|
@RequiresPermissions("tool:gen:list")
|
@PostMapping("/column/list")
|
@ResponseBody
|
public TableDataInfo columnList(GenTableColumn genTableColumn) {
|
TableDataInfo dataInfo = new TableDataInfo();
|
List<GenTableColumn> list = genTableColumnService.selectGenTableColumnListByTableId(genTableColumn);
|
dataInfo.setRows(list);
|
dataInfo.setTotal(list.size());
|
return dataInfo;
|
}
|
|
@RequiresPermissions("tool:gen:list")
|
@GetMapping("/importTable")
|
public String importTable() {
|
return prefix + "/importTable";
|
}
|
|
@RequiresPermissions("tool:gen:list")
|
@Log(title = "代码生成", businessType = BusinessType.IMPORT)
|
@PostMapping("/importTable")
|
@ResponseBody
|
public AjaxResult importTableSave(String tables) {
|
String[] tableNames = Convert.toStrArray(tables);
|
|
List<GenTable> tableList = genTableService.selectDbTableListByNames(tableNames);
|
genTableService.importGenTable(tableList);
|
return AjaxResult.success();
|
}
|
|
@GetMapping("/edit/{tableId}")
|
public String edit(@PathVariable("tableId") Long tableId, ModelMap mmap) {
|
GenTable table = genTableService.selectGenTableById(tableId);
|
mmap.put("table", table);
|
return prefix + "/edit";
|
}
|
|
@RequiresPermissions("tool:gen:edit")
|
@Log(title = "代码生成", businessType = BusinessType.UPDATE)
|
@PostMapping("/edit")
|
@ResponseBody
|
public AjaxResult editSave(@Validated GenTable genTable) {
|
genTableService.validateEdit(genTable);
|
genTableService.updateGenTable(genTable);
|
return AjaxResult.success();
|
}
|
@RequiresPermissions("tool:gen:remove")
|
@Log(title = "代码生成", businessType = BusinessType.DELETE)
|
@PostMapping("/remove")
|
@ResponseBody
|
public AjaxResult remove(String ids) {
|
genTableService.deleteGenTableByIds(ids);
|
return AjaxResult.success();
|
}
|
|
@RequiresPermissions("tool:gen:preview")
|
@GetMapping("/preview/{tableId}")
|
@ResponseBody
|
public AjaxResult preview(@PathVariable("tableId") Long tableId) throws IOException {
|
Map<String, String> dataMap = genTableService.previewCode(tableId);
|
return AjaxResult.success(dataMap);
|
}
|
|
@RequiresPermissions("tool:gen:code")
|
@Log(title = "代码生成", businessType = BusinessType.GENCODE)
|
@GetMapping("/genCode/{tableName}")
|
public void genCode(HttpServletResponse response, @PathVariable("tableName") String tableName) throws IOException {
|
byte[] data = genTableService.generatorCode(tableName);
|
genCode(response, data);
|
}
|
|
@RequiresPermissions("tool:gen:code")
|
@Log(title = "代码生成", businessType = BusinessType.GENCODE)
|
@GetMapping("/batchGenCode")
|
@ResponseBody
|
public void batchGenCode(HttpServletResponse response, String tables) throws IOException {
|
String[] tableNames = Convert.toStrArray(tables);
|
byte[] data = genTableService.generatorCode(tableNames);
|
genCode(response, data);
|
}
|
|
private void genCode(HttpServletResponse response, byte[] data) throws IOException {
|
response.reset();
|
response.setHeader("Content-Disposition", "attachment; filename=\"ots.zip\"");
|
response.addHeader("Content-Length", "" + data.length);
|
response.setContentType("application/octet-stream; charset=UTF-8");
|
IOUtils.write(data, response.getOutputStream());
|
}
|
}
|
