first commit

懒羊羊

2023-08-30 1ac2bc1590406d9babec036e154d8d08f34a6aa1

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
package cn.stylefeng.guns.gen.core.generator.restful.controller;
 
import cn.stylefeng.guns.gen.core.generator.base.AbstractCustomGenerator;
import org.beetl.core.Template;
 
import java.io.File;
import java.util.Map;
 
/**
 * 带restful接口控制器生成器
 *
 * @author fengshuonan
 * @date 2018-12-13-2:20 PM
 */
public class RestfulControllerGenerator extends AbstractCustomGenerator {
 
    public RestfulControllerGenerator(Map<String, Object> tableContext) {
        super(tableContext);
    }
 
    @Override
    public void bindingOthers(Template template) {
        template.binding("controllerPackage", contextParam.getProPackage() + ".controller");
    }
 
    @Override
    public String getTemplateResourcePath() {
        if (contextParam.getSwagger()) {
            return "/feignTemplates/RestfulControllerSwagger.btl";
        } else {
            return "/feignTemplates/RestfulController.btl";
        }
    }
 
    @Override
    public String getGenerateFileTempPath() {
        String proPackage = this.contextParam.getProPackage();
        String proPath = proPackage.replaceAll("\\.", "/");
        File file = new File(contextParam.getOutputPath() + "/" + proPath + "/controller/" + tableContext.get("entity") + "Controller.java");
        return file.getAbsolutePath();
    }
 
    @Override
    public String getGenerateFileDirectPath() {
        return  this.getGenerateFileTempPath();
    }
}