前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >使用druid的工具包改sql的库名,实现单实例分库

使用druid的工具包改sql的库名,实现单实例分库

作者头像
王念博客
发布2023-01-10 13:40:24
4130
发布2023-01-10 13:40:24
举报
文章被收录于专栏:王念博客王念博客

前言,没时间解释了,先贴代码

1.新增自定义Visitor                                                                                                                                                                                                                           

代码语言:javascript
复制
 /**
     * 自定义Visitor ,这里是MySQL,其他数据库请自行更换
     **/
    public static class MyReplaceTableVisitor extends MySqlASTVisitorAdapter {

        private final String databaseName;

        public MyReplaceTableVisitor(String databaseName) {
            this.databaseName = databaseName;
        }

        @Override
        public boolean visit(SQLExprTableSource x) {
            String originTableName = x.getExpr().toString();
            if (databaseName != null) {
                if (originTableName.contains(".")) {
                    return true;
                }
                //修改表名
                x.setExpr(databaseName + "." + originTableName);
                logger.info(" replace table:" + originTableName + " -> " + databaseName + "." + originTableName);
            }
            return true;
        }
    }

   2.替换方法                                                                                                                                                                                                                                         

代码语言:javascript
复制
    public static String replaceTableName(String sql, String tableName) {
        if (tableName == null) {
            return sql;
        }
        List<SQLStatement> statements = SQLUtils.parseStatements(sql, JdbcConstants.MYSQL);
        MyReplaceTableVisitor visitor = new MyReplaceTableVisitor(tableName);
        for (SQLStatement stmt : statements) {
            stmt.accept(visitor);
        }
        return SQLUtils.toSQLString(statements, JdbcConstants.MYSQL);
    }

调用测试方法                                                                                                                                                                                                                                         

代码语言:javascript
复制
    public static void main(String[] args) {
        String sql = "select c.*, (select a.name as province_name from portal_area a where a.id = c.province_id) as province_name, (select a.name as city_name from portal_area a where a.id = c.city_id) as city_name, (CASE WHEN c.area_id IS NULL THEN NULL ELSE (select a.name as area_name from portal_area a where a.id = c.area_id)  END )as area_name  from portal.portal_company c where no = ?";
        String s = replaceTableName(sql, "abc");
        System.out.println(s);
    }
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2023-01-10,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
云数据库 MySQL
腾讯云数据库 MySQL(TencentDB for MySQL)为用户提供安全可靠,性能卓越、易于维护的企业级云数据库服务。其具备6大企业级特性,包括企业级定制内核、企业级高可用、企业级高可靠、企业级安全、企业级扩展以及企业级智能运维。通过使用腾讯云数据库 MySQL,可实现分钟级别的数据库部署、弹性扩展以及全自动化的运维管理,不仅经济实惠,而且稳定可靠,易于运维。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档