Spark使用Java读Hive写入HBase

最新推荐文章于 2024-05-24 10:54:11 发布

原创最新推荐文章于 2024-05-24 10:54:11 发布 · 886 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#spark #hbase

Spark 同时被 3 个专栏收录

16 篇文章

订阅专栏

Java

10 篇文章

订阅专栏

Hbase

3 篇文章

订阅专栏

该博客介绍了如何使用Spark的Java API读取Hive表的数据，并将这些数据写入到HBase中。通过创建`Put`对象，设置列族和列，然后使用HBaseWriter方法批量写入HBase。示例代码展示了如何处理行数据，包括处理空值情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Spark使用Java读Hive写入HBase

Spark读取Hive数据得方法在下面之前得一篇博客中有提及，在此就不重复提了，如果感兴趣得可以查阅如下连接中得博客

Spark使用Java读Hive得博客： https://blog.youkuaiyun.com/Aaron_ch/article/details/113151157

rowDataset.foreachPartition((ForeachPartitionFunction<Row>) t -> {
            String columnFamily="test";
            int i = 0;
            List<Put> putList = new ArrayList<>();
            Put put = null;  //rowkey
            Row row;
            while (t.hasNext()) {
                row = t.next();
                if (put == null) {
                    put = new Put(row.getString(0).getBytes());
                }
                put.addColumn(Bytes.toBytes(columnFamily), Bytes.toBytes("colName1"), Bytes.toBytes(row.getString(0)));
                put.addColumn(Bytes.toBytes(columnFamily), Bytes.toBytes("colName2"), Bytes.toBytes(row.getString(1)));
                
                if (row.getString(2) == null) {
                    put.addColumn(Bytes.toBytes(columnFamily), Bytes.toBytes("colName3"), null);
                } else {
                    put.addColumn(Bytes.toBytes(columnFamily), Bytes.toBytes("colName3"), Bytes.toBytes(row.getString(2)));
                }               
                putList.add(put);
                hBaseWriter(putList, zk);
                put = null;
                putList.clear();             
            }
        });

如下是写入得方法：

private static void hBaseWriter(List<Put> putList, String zk) {

        for (int y = 0; y < putList.size(); y++) {
            System.out.println(putList.get(y).toString());
        }

        System.out.println("start to write into hbase~");
        String tableName = "xxx:tableName";
        Configuration conf = HBaseConfiguration.create();
        conf.set("hbase.zookeeper.quorum", zk);
        conf.set("hbase.zookeeper.property.clientPort", "2181");
        Connection connection = null;
        Table table = null;
        try {
            connection = ConnectionFactory.createConnection(conf);
            table = connection.getTable(TableName.valueOf(tableName));

            if (putList.size() > 0) {
                table.put(putList);
                System.out.println("writer into hbase success!!!");
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (table != null) {
                try {
                    table.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (connection != null) {
                try {
                    connection.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }