增加smartboxserver

bingbing

2020-07-14 e0bb13f823169282241f5797f7a0cbaabc1d3365

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
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());
    }
}