init commit

duhuizhe

2024-04-19 64c4fd3c7067c7626dc960a70b4bc2c3662bc653

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
package com.yqzx.generator.action.config;
 
import cn.hutool.core.util.StrUtil;
import com.yqzx.generator.action.model.GenQo;
import com.yqzx.generator.util.SpringUtil;
 
/**
 * @description: 默认的代码生成web配置
 * @author: chaoyapeng
 * @time: 2020/8/18 15:53
 */
public class WebGeneratorConfig extends AbstractGeneratorConfig {
 
    private GenQo genQo;
 
    public WebGeneratorConfig(GenQo genQo) {
        this.genQo = genQo;
    }
 
    protected void dataSourceConfig() {
        dataSourceConfig.setDriverName(StrUtil.isNotEmpty(genQo.getDriverName()) ? genQo.getDriverName() : getProperty().getDriverName());
        dataSourceConfig.setUsername(StrUtil.isNotEmpty(genQo.getUserName()) ? genQo.getUrl() : getProperty().getUserName());
        dataSourceConfig.setPassword(StrUtil.isNotEmpty(genQo.getPassword()) ? genQo.getUrl() : getProperty().getPassword());
        // 处理配置文件里的固定ip
        if (StrUtil.isEmpty(genQo.getIp())) {
            genQo.setIp(getProperty().getIp());
        }
        if (StrUtil.isEmpty(genQo.getPort())) {
            genQo.setPort(getProperty().getPort());
        }
        if (StrUtil.isEmpty(genQo.getDataBaseName())) {
            genQo.setDataBaseName(getProperty().getDataBaseName());
        }
        dataSourceConfig.setUrl(genQo.getUrl());
    }
 
    protected void contextConfig() {
        contextConfig.setAuthor(genQo.getAuthor());
        contextConfig.setProjectPath(genQo.getProjectPath());
        contextConfig.setPackageName(genQo.getProjectPackage());
        contextConfig.setTableName(genQo.getTableName());
        contextConfig.setBizChName(genQo.getBizName());
        // 生成器开关
        contextConfig.setControllerSwitch(genQo.getControllerSwitch() || getProperty().getControllerSwitch());
        contextConfig.setServiceSwitch(genQo.getServiceSwitch() || getProperty().getServiceSwitch());
        contextConfig.setMapperSwitch(genQo.getMapperSwitch() || getProperty().getMapperSwitch());
        contextConfig.setModelSwitch(genQo.getModelSwitch() || getProperty().getMapperSwitch());
        contextConfig.setQuerySwitch(genQo.getQuerySwitch() || getProperty().getQuerySwitch());
 
        // 查询参数对象
        super.setQueryTableInfo(genQo.getQueryTableInfo());
    }
 
    @Override
    protected void config() {
        dataSourceConfig();
        contextConfig();
    }
 
    private GenQo getProperty() {
        return SpringUtil.getBean(GenQo.class);
    }
}