如何将JavaPairDStream写入Redis?
How to write JavaPairDStream to Redis?
我正在使用 spark 1.5.0 和 java 7.
输入来自具有 type
字段的不同 json 对象形式的 kafka。例如:
{'type': 'alpha', ...}
{'type': 'beta', ...}
...
我正在根据与每种事件类型的计数相对应的输入数据创建 JavaPairDStream<String, Integer>
。
我想把这个数据存储到redis。我该怎么做呢?
使用foreachRDD
和forEach
函数实现如下:
wordCounts.foreachRDD(
new Function<JavaPairRDD<String, Integer>, Void>() {
public Void call(JavaPairRDD<String, Integer> rdd) {
rdd.foreach(
new VoidFunction<Tuple2<String,Integer>>() {
public void call(Tuple2<String,Integer> wordCount) {
System.out.println(wordCount._1() + ":" + wordCount._2());
JedisPool pool = new JedisPool(new JedisPoolConfig(), "localhost");
Jedis jedis = pool.getResource();
jedis.select(0);
jedis.set(wordCount._1(), wordCount._2().toString());
}
}
);
return null;
}
}
);
为每个 RDD 创建一个新的连接池是非常低效的。我建议为每个分区创建一个连接:
wordCount.mapPartitions(p->{
Jedis jd = new Jedis(getJedisConfig());
while (p->hasNext()) {
Tuple2<String,Integer> data = p.next();
String word = data._1();
Integer cnt = data._2();
jd.set(word,count); // or any other format of save to Redis
}
}
)
我正在使用 spark 1.5.0 和 java 7.
输入来自具有 type
字段的不同 json 对象形式的 kafka。例如:
{'type': 'alpha', ...}
{'type': 'beta', ...}
...
我正在根据与每种事件类型的计数相对应的输入数据创建 JavaPairDStream<String, Integer>
。
我想把这个数据存储到redis。我该怎么做呢?
使用foreachRDD
和forEach
函数实现如下:
wordCounts.foreachRDD(
new Function<JavaPairRDD<String, Integer>, Void>() {
public Void call(JavaPairRDD<String, Integer> rdd) {
rdd.foreach(
new VoidFunction<Tuple2<String,Integer>>() {
public void call(Tuple2<String,Integer> wordCount) {
System.out.println(wordCount._1() + ":" + wordCount._2());
JedisPool pool = new JedisPool(new JedisPoolConfig(), "localhost");
Jedis jedis = pool.getResource();
jedis.select(0);
jedis.set(wordCount._1(), wordCount._2().toString());
}
}
);
return null;
}
}
);
为每个 RDD 创建一个新的连接池是非常低效的。我建议为每个分区创建一个连接:
wordCount.mapPartitions(p->{
Jedis jd = new Jedis(getJedisConfig());
while (p->hasNext()) {
Tuple2<String,Integer> data = p.next();
String word = data._1();
Integer cnt = data._2();
jd.set(word,count); // or any other format of save to Redis
}
}
)