tzy-admin/py-base/src/main/java/com/poyee/util/TableGenerationTool.java

package com.poyee.util;

import com.poyee.datasource.common.TableInitUtil;
import com.poyee.datasource.modal.TableConfigModal;
import lombok.extern.slf4j.Slf4j;

import java.util.ArrayList;
import java.util.List;

/**
 * 	表生成 工具类
 */
@Slf4j
public class TableGenerationTool {
    public static void main(String[] args) throws Exception {

        TableInitUtil.generation(initTableConfig("card_group_info"));

    }

    /**
     * @param tableName
     * @return
     */
    public static TableConfigModal initTableConfig(String tableName){
        TableConfigModal config = new TableConfigModal();
        config.setDbUrl("jdbc:postgresql://m2-dev.hobbystocks.cn:5432/tzy_system");
        config.setDbUsername("postgres");
        config.setDbPwd("123456");
        config.setDbType("postgresql");
        config.setAuthor("lsz");
        config.setJavaProjectPath("C:\\poyi\\project\\poyee-dashboard\\py-data\\src\\main\\");
        config.setOpenSwagger(true);
        config.setJavaPackage("com.poyee");
        config.setTemplates(templates());
        config.setTableName(tableName);
        return config;
    }
    public static List<String> templates(){
        return new ArrayList<String>(){{
            add("entity");
            add("mapper");
            add("mapperXml");
            add("service");
            add("serviceImpl");
            add("controller");
            add("Dto");
        }};
    }

}