手写可视化逆向工程(一键生成mapper-＞controller-＞vue)

﹏ヽ暗。殇╰゛Y 2022-12-25 12:56 149阅读 0赞

[https://github.com/wangchao550586585/codegen][https_github.com_wangchao550586585_codegen]

设计目的  
公司目前从0开发CRM,主要开发文章,视频等内容,发现非常多通用的逻辑可直接代码生成,于是花费3天写了一个复合公司业务场景需要的**可视化**逆向工程脚手架.

功能:

*  一键生成mapper->service->controller->vue(尚未公开,公司内部使用)增删改查通用模板,即拿即用,无需任何修改,集成**swagger接口注释文档**

使用方法

*  启动项目

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70][]

*  访问[http://localhost:8080/][http_localhost_8080]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 1][]

*  配置数据源

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 2][]

*  点击任意一个数据源,即可生成数据源下所有表,以ZIP形式下载

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 3][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 4][]

*  ZIP自带包名,copy项目即可.无需任何修改.

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 5][]

涉及技术

*  spring-boot(动态数据源)
 *  freemarker

# ftl模板技术核心代码 #

![20210308211624549.png][]

@SneakyThrows
        @Override
        public byte[] code(GeneratorConf generatorConf) {
            //解析生成root
            TableEntity root = getRoot(generatorConf);
    
            //获取Configuration
            Configuration cfg = getConfiguration();
    
            //生成文件
            ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
            ZipOutputStream zip = new ZipOutputStream(outputStream);
            genFile(new File(path), root, cfg, zip);
            IoUtil.close(zip);
            return outputStream.toByteArray();
        }

获取ftl配置

/**
         * 获取模板配置
         * @return
         * @throws IOException
         */
        private Configuration getConfiguration() throws IOException {
            Configuration cfg = new Configuration(Configuration.VERSION_2_3_22);
            cfg.setDirectoryForTemplateLoading(new File(path));
            cfg.setDefaultEncoding("UTF-8");
            cfg.setTemplateExceptionHandler(TemplateExceptionHandler.RETHROW_HANDLER);
            return cfg;
        }

生成文件

private void genFile(File file, TableEntity root, Configuration cfg, ZipOutputStream zip) throws IOException, TemplateException {
            if (file.isDirectory()) {
                File[] files = file.listFiles();
                for (int i = 0; i < files.length; i++) {
                    genFile(files[i], root, cfg, zip);
                }
            } else {
                String fileName = file.getName();
                String s = fileName.split("\\.")[0];
                String genFileName = fileName.equals("Entity.java.ftl") ? ".java" : fileName.replaceAll(".ftl", "");
    
                //构造文件路径
                String filePath = root.getPackageName().concat("." + s.toLowerCase());
                filePath = filePath.replace(".", File.separator).concat(File.separator + root.getClassName() + genFileName);
                filePath = filePath.replace("serviceimpl", "service" + File.separator + "impl");
    
                //获取模板
                Template temp = cfg.getTemplate(fileName);
    
                //生成zip
                genZip(root, zip, filePath, temp);
            }
        }

生成ZIP

private void genZip(TableEntity root, ZipOutputStream zip, String path, Template temp) throws TemplateException, IOException {
            StringWriter sw = new StringWriter();
            temp.process(root, sw);
            zip.putNextEntry(new ZipEntry(Objects.requireNonNull(path)));
            IoUtil.write(zip, StandardCharsets.UTF_8, false, sw.toString());
    
            IoUtil.close(sw);
            zip.closeEntry();
        }

# 动态数据源 #

首先注册动态数据源对象

@Bean
        public DataSource dataSource() {
            HikariDataSource dataSource = new HikariDataSource();
            dataSource.setPassword(globalConfig.getUsername());
            dataSource.setUsername(globalConfig.getPassword());
            dataSource.setDriverClassName(globalConfig.getDriverClassName());
            dataSource.setJdbcUrl(globalConfig.getJdbcUrl());
    
            //自定义动态数据源类
            DynamicDataSource dynamicDataSource = new DynamicDataSource();
            HashMap<Object, Object> objectObjectHashMap = new HashMap<>();
            objectObjectHashMap.put("defaultDataSource", dataSource);
            dynamicDataSource.setTargetDataSources(objectObjectHashMap);
            dynamicDataSource.setDefaultTargetDataSource(dataSource);
    
            return dynamicDataSource;
        }

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 6][]

public static class DataSourceContext {
    
            private static ThreadLocal<DataSourceBean> threadLocal = new InheritableThreadLocal<DataSourceBean>();
    
            /**
             * 获取数据源
             */
            public static DataSourceBean getDataSource() {
                return threadLocal.get();
            }
    
            /**
             * 设置数据源
             */
            public static void setDataSource(DataSourceBean dataSourceBean) {
                threadLocal.set(dataSourceBean);
            }
    
            /**
             * 清除数据源
             * 清除后,数据源为默认时间
             */
            public static void toDefault() {
                threadLocal.remove();
            }
        }

/**
         * 连接数据源前,调用该方法
         */
        @Override
        protected Object determineCurrentLookupKey() {
            DataSourceBean dataSourceBean = DataSourceContext.getDataSource();
            if (dataSourceBean == null) {
                return null;
            }
            try {
                Map<Object, Object> targetSourceMap = getTargetSource();
                synchronized (this) {
                    if (!targetSourceMap.keySet().contains(dataSourceBean.getBeanName())) {
                        Object dataSource = createDataSource(dataSourceBean);
                        targetSourceMap.put(dataSourceBean.getBeanName(), dataSource);
                        super.afterPropertiesSet();
                    }
                }
                return dataSourceBean.getBeanName();
            } catch (Exception e) {
                e.printStackTrace();
            }
            return null;
        }
    
        public Object createDataSource(DataSourceBean dataSourceBean) throws IllegalAccessException {
            ConfigurableApplicationContext context = (ConfigurableApplicationContext) applicationContext;
            DefaultListableBeanFactory beanFactory = (DefaultListableBeanFactory) context.getBeanFactory();
            BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(HikariDataSource.class);
    
            Map<String, Object> propertyKeyValues = getPropertyKeyValues(DataSourceBean.class, dataSourceBean);
            for (Map.Entry<String, Object> entry : propertyKeyValues.entrySet()) {
                beanDefinitionBuilder.addPropertyValue(entry.getKey(), entry.getValue());
            }
            beanFactory.registerBeanDefinition(dataSourceBean.getBeanName(), beanDefinitionBuilder.getBeanDefinition());
            return context.getBean(dataSourceBean.getBeanName());
        }

[https_github.com_wangchao550586585_codegen]: https://github.com/wangchao550586585/codegen
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70]: /images/20221120/7ed947e8bffc43ad87c7dfcd93f22ebe.png
[http_localhost_8080]: http://localhost:8080/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 1]: /images/20221120/0197d21123d841e08a699d2b3d3fb089.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20201202090416831.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 3]: https://img-blog.csdnimg.cn/20201202090538382.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 4]: https://img-blog.csdnimg.cn/202012020906024.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 5]: https://img-blog.csdnimg.cn/2020120209075496.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw==,size_16,color_FFFFFF,t_70
[20210308211624549.png]: https://img-blog.csdnimg.cn/20210308211624549.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw_size_16_color_FFFFFF_t_70 6]: https://img-blog.csdnimg.cn/20210308211944415.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L051YW5fRmVuZw==,size_16,color_FFFFFF,t_70