我有一个令人尴尬的并行任务,我使用Spark来分配计算。这些计算是在Python中进行的,我使用PySpark来读取和预处理数据。我的任务的输入数据存储在HBase中。不幸的是,我还没有找到一种令人满意的(即易于使用和可扩展的)方法来使用Python从/向Spark读取/写入HBase数据。
我之前探讨的内容:
使用happybase
在我的Python流程中进行连接。该软件包允许使用HBase的Thrift API从Python连接到HBase。这样,我基本上跳过Spark进行数据读/写,并且错过了潜在的HBase-Spark优化。读取速度似乎相当快,但写入速度很慢。这是目前我最好的解决方案。
使用SparkContext的newAPIHadoopRDD
和saveAsNewAPIHadoopDataset
来利用HBase的MapReduce接口。这方面的示例曾经包含在Spark代码库(see here)中。但是,这些现在被认为是过时的,有利于HBase的Spark绑定(see here)。我也发现这个方法很慢而且很麻烦(读取,编写效果很好),例如,从newAPIHadoopRDD
返回的字符串必须以各种方式进行解析和转换,最终得到Python对象通缉。它一次只支持一列。
我知道的替代方案:
我目前正在使用Cloudera的CDH,版本5.7.0提供hbase-spark
(CDH release notes和a detailed blog post)。该模块(以前称为SparkOnHBase
)将正式成为HBase 2.0的一部分。不幸的是,这个奇妙的解决方案似乎只适用于Scala / Java。
华为的Spark-SQL-on-HBase / Astro(我看不出两者之间存在差异......)。它看起来不像我想要的解决方案那样强大且得到良好支持。
答案 0 :(得分:18)
我找到hbase-spark
的一位制作人this comment,这似乎表明有一种方法可以使用PySpark使用Spark SQL查询HBase。
事实上,the pattern described here可以应用于使用PySpark使用Spark SQL查询HBase,如下例所示:
from pyspark import SparkContext
from pyspark.sql import SQLContext
sc = SparkContext()
sqlc = SQLContext(sc)
data_source_format = 'org.apache.hadoop.hbase.spark'
df = sc.parallelize([('a', '1.0'), ('b', '2.0')]).toDF(schema=['col0', 'col1'])
# ''.join(string.split()) in order to write a multi-line JSON string here.
catalog = ''.join("""{
"table":{"namespace":"default", "name":"testtable"},
"rowkey":"key",
"columns":{
"col0":{"cf":"rowkey", "col":"key", "type":"string"},
"col1":{"cf":"cf", "col":"col1", "type":"string"}
}
}""".split())
# Writing
df.write\
.options(catalog=catalog)\ # alternatively: .option('catalog', catalog)
.format(data_source_format)\
.save()
# Reading
df = sqlc.read\
.options(catalog=catalog)\
.format(data_source_format)\
.load()
我已尝试hbase-spark-1.2.0-cdh5.7.0.jar
(由Cloudera分发),但遇到了麻烦(写org.apache.hadoop.hbase.spark.DefaultSource does not allow create table as select
时,java.util.NoSuchElementException: None.get
阅读时遇到麻烦。事实证明,当前版本的CDH不包括允许Spark SQL-HBase集成的hbase-spark
更改。
为我工作的是shc
Spark包,找到了here。我必须对上述脚本进行的唯一更改是更改:
data_source_format = 'org.apache.spark.sql.execution.datasources.hbase'
以下是我在CDH群集中提交上述脚本的方法,遵循shc
自述文件中的示例:
spark-submit --packages com.hortonworks:shc:1.0.0-1.6-s_2.10 --repositories http://repo.hortonworks.com/content/groups/public/ --files /opt/cloudera/parcels/CDH/lib/hbase/conf/hbase-site.xml example.py
shc
上的大多数工作似乎已经合并到HBase的hbase-spark
模块中,以便在2.0版中发布。这样,使用上述模式可以使用Spark SQL查询HBase(有关详细信息,请参阅:https://hbase.apache.org/book.html#_sparksql_dataframes)。上面的例子显示了PySpark用户的样子。
最后,一个警告:我上面的示例数据只有字符串。 shc
不支持Python数据转换,因此我遇到的问题是整数和浮点数没有出现在HBase中或者出现奇怪的值。