spring-boot配置Sharding分库分表时如果查询指定库表结构

        在使用sharding插件时,有时我们需要动态获取指定表的字段,类型等数据结构。但是在直接使用sql语句查询时sharding插件会将查询sql 转换错误。此时我们该怎么办呢。

      思路:

       第一步 :获取shardind数据源

        第二步:获取数据库链接信息

       第三补:重建数据源链接

       第四部: 关闭链接

不废话直接上代码:

/**
 * Sharding  分库分表需要这个来查询
 * @param SqlDbDataVo
 * @return
 */
private List<SqlColumnData> getColumns(SqlDbDataVo SqlDbDataVo){
   
    Connection con =null;
    ResultSet rs =null;
    Connection connection =null;
    List<SqlColumnData> data = new ArrayList<>();
    try {
        // 获取该服务链接的sharding数据源
        ShardingDataSource shardingDataSource = (ShardingDataSource) ApplicationContextConfig.getBean("dataSource");
        if(shardingDataSource == null)  return  null;
        connection =  shardingDataSource.getConnection();
        if(connection == null)  return  null;
       // 获取多数据源
        Map<String, DataSource> dataA = shardingDataSource.getDataSourceMap();
        // 轮询获取指定目标库表  这里测试直接全部轮询了
        for(String key : dataA.keySet()){
            DruidDataSource d = (DruidDataSource) dataA.get(key);
            Class.forName(d.getDriverClassName());
            con = DriverManager.getConnection(d.getUrl(), d.getUsername(), d.getPassword());
            if(!con.isClosed()) {//判断数据库是否链接成功
                //3、创建Statement对象
                Statement st = con.createStatement();
                //4、执行sql语句
                StringBuffer sql = new StringBuffer(" SELECT COLUMN_NAME, DATA_TYPE,COLUMN_COMMENT FROM   information_schema.Columns WHERE table_schema = '"+SqlDbDataVo.getTableSchema()+"'");
                sql.append(" AND TABLE_NAME ='"+SqlDbDataVo.getTableName()+"' ");
                if(SqlDbDataVo.getType() == 1 ){
                    sql.append(" AND DATA_TYPE !='datetime' AND COLUMN_NAME !='deleted'");
                }
                if(SqlDbDataVo.getType() == 2 ){
                    sql.append(" AND DATA_TYPE ='datetime' ");
                }
                sql.append(" AND COLUMN_NAME != '"+SqlDbDataVo.getFiled()+"'");
                try {
                    rs = st.executeQuery(sql.toString());//查询之后返回结果集
                    //5、打印出结果
                    while (rs.next()) {
                        data.add(new SqlColumnData(rs.getString("COLUMN_NAME"), rs.getString("DATA_TYPE"), rs.getString("COLUMN_COMMENT")));
                    }
                }catch (Exception e){
                    e.printStackTrace();
                }
            }
        }
    } catch (Exception e) {
        e.printStackTrace();
    }finally {
        try {
            if (rs != null) rs.close();//关闭资源
            if (con != null) con.close();//关闭数据库
            if (connection != null) connection.close();//关闭数据库
        }catch (Exception e){
             e.printStackTrace();
        }
    }
    return data;
}
@Data
public class SqlColumnData {
    @TableField("COLUMN_NAME")
    private String columnName;
    @TableField("DATA_TYPE")
    private String dataType;
    @TableField("COLUMN_COMMENT")
    private String columnComment;

    public SqlColumnData() {

    }

    public SqlColumnData(String columnName, String dataType, String columnComment) {
        this.columnName = columnName;
        this.dataType = dataType;
        this.columnComment = columnComment;
    }
}

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值