java原生代码实现行转列

最新推荐文章于 2024-03-24 09:16:51 发布

会降低房价是否

最新推荐文章于 2024-03-24 09:16:51 发布

阅读量1.4k

点赞数 1

分类专栏：行转列文章标签： spring java

本文链接：https://blog.csdn.net/bwl258wd/article/details/124423733

版权

行转列专栏收录该内容

1 篇文章 0 订阅

订阅专栏

背景

大家应该都用过数据库自带的内置函数PIVOT和UNPIVOT来进行行列的互相转化，但是当我们在做数据转储的时候，如果这个动作是集成在自动化服务当中，想用java代码直接实现从源数据读出数据，经过转储以后直接插入目标表中，能不能实现呢？今天我们就来尝试一下！

方案

源表：

a	b	qty_2022_04_26	qty_2022_04_27	qty_2022_04_28
aa	bb	11	22	33

目标表：

a	b	qty_date	qty
aa	bb	2022_04_26	11
aa	bb	2022_04_27	22
aa	bb	2022_04_28	33

我们所要实现的就是把源表经过行转列处理后变成目标表的格式，话不多说，直接上才艺。

code

public void test(Connection conn, String table, String Condition) throws SQLException, ClassNotFoundException {
        ResultSet input_rs;
        Statement st = conn.createStatement();

        if (ObjectUtils.isEmpty(Condition)) {
            Condition = "1=1";
        }
        String sql = "SELECT * FROM tableName where " + Condition;
        input_rs = st.executeQuery(sql);

        saveTable(input_rs);

        input_rs.close();
        conn.close();
    }
public void saveTable(ResultSet input_rs) throws SQLException, ClassNotFoundException {
        ResultSetMetaData rsmdl = input_rs.getMetaData();
        Connection con = JDBCUtil.getConnection();
        con.setAutoCommit(false);
        String insert = "insert into dbo.targetTable values(?,?,?,?,?,?,?,?,?) ";
        PreparedStatement ps = con.prepareStatement(insert);
        Map<Integer, String> qtyColumnMap = new HashMap<>();
        ResultSetMetaData rsmd = input_rs.getMetaData();
        for (int i = 1; i <= rsmd.getColumnCount(); i++) {
            String columnName = rsmd.getColumnName(i);
            if (columnName.startsWith("QTY_")) {
                qtyColumnMap.put(i, columnName);
            }
        }


        int count = rsmdl.getColumnCount();
        int i = 0;
        while (input_rs.next()) {
            List<String> newColumn = new ArrayList<>();
            for (int j = 1; j <= count; j++) {
                if (Objects.nonNull(qtyColumnMap.get(j))) {
                    for (int k = 1; k <= newColumn.size(); k++) {
                        ps.setString(k, newColumn.get(k - 1));
                    }
                    ps.setString(newColumn.size() + 1, qtyColumnMap.get(j).substring("QTY_".length())); // qty_date
                    ps.setFloat(newColumn.size() + 2, input_rs.getFloat(j)); // qty
                    ps.addBatch();
                    i++;
                } else {
                    newColumn.add(input_rs.getString(j));
                }
            }

            if (i >= 2000) {
                ps.executeBatch();
                con.commit();
                ps.clearBatch();
                i = 0;
            }

        }

        ps.executeBatch();
        con.commit();
        ps.clearBatch();
        JDBCUtil.close(ps, con);

    }