Java Spark 简单示例(二)累加器 广播变量

广播变量在最后补充。

今天的示例主要介绍一下累加器的使用。Spark官方文档的Action介绍中有提到 foreach

foreach(func):Run a function func on each element of the dataset. This is usually done for side effects such as updating an Accumulator or interacting with external storage systems.
Note: modifying variables other than Accumulators outside of the foreach() may result in undefined behavior. See Understanding closures for more details.

意思是说,调用foreach函数可以遍历rdd中的每一个元素,但是在foreach外部,不可以更新累加器以外的变量,否则会出现与期望值不符的结果。

例如,你想做一个简单求和计数,首先声明一个变量,然后在foreach内部增值。请注意,这样是行不通的。

官方文档中这样介绍:By default, when Spark runs a function in parallel as a set of tasks on different nodes, it ships a copy of each variable used in the function to each task. 默认情况下,当Spark以不同节点上的一组任务并行运行一个函数时,它会将该函数中使用的每个变量的副本发送给每个任务。

为了解决这个问题,除了之前示例中用过的reduce函数外,还可以引入累加器。

package com.yzy.spark;

import org.apache.spark.SparkConf;
import org.apache.spark.api.java.JavaRDD;
import org.apache.spark.api.java.JavaSparkContext;
import org.apache.spark.api.java.function.VoidFunction;
import org.apache.spark.util.LongAccumulator;

import java.util.Arrays;
import java.util.List;

public class demo3 {
    private static String appName = "spark.demo";
    private static String master = "local[*]";

    public static void main(String[] args) {
        JavaSparkContext sc = null;
        try {
            //初始化 JavaSparkContext
            SparkConf conf = new SparkConf().setAppName(appName).setMaster(master);
            sc = new JavaSparkContext(conf);

            //构造数据源
            List<Integer> data = Arrays.asList(1, 2, 3, 4, 5);

            //并行化创建rdd
            JavaRDD<Integer> rdd = sc.parallelize(data);

            //累加器
            final LongAccumulator accumulator = sc.sc().longAccumulator();

            rdd.foreach(new VoidFunction<Integer>() {
                public void call(Integer integer) throws Exception {
                    accumulator.add(integer);
                }
            });

            System.out.println("执行结果:" + accumulator.value());
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (sc != null) {
                sc.close();
            }
        }
    }
}

执行结果

//省略若干行
18/06/25 14:16:24 INFO TaskSetManager: Finished task 3.0 in stage 0.0 (TID 3) in 89 ms on localhost (executor driver) (1/4)
18/06/25 14:16:24 INFO TaskSetManager: Finished task 2.0 in stage 0.0 (TID 2) in 92 ms on localhost (executor driver) (2/4)
18/06/25 14:16:24 INFO TaskSetManager: Finished task 1.0 in stage 0.0 (TID 1) in 94 ms on localhost (executor driver) (3/4)
18/06/25 14:16:24 INFO TaskSetManager: Finished task 0.0 in stage 0.0 (TID 0) in 113 ms on localhost (executor driver) (4/4)
18/06/25 14:16:24 INFO TaskSchedulerImpl: Removed TaskSet 0.0, whose tasks have all completed, from pool 
18/06/25 14:16:24 INFO DAGScheduler: ResultStage 0 (foreach at demo3.java:35) finished in 0.298 s
执行结果:15

累加器的使用比较简单,但是需要注意的是,以下两种情况会造成累加器计数出错

第一种:计数为0

//累加器
final LongAccumulator accumulator = sc.sc().longAccumulator();

//map && reduce
JavaRDD<Integer> javaRDD = rdd.map(new Function<Integer, Integer>() {
    public Integer call(Integer integer) throws Exception {
        accumulator.add(integer);
        return integer;
    }
});

System.out.println("执行结果:" + accumulator.value());
-----------------------
执行结果:0

可以看到,我们把累加器放到map中执行的时候,结果输出为0。这个很容易想到,因为map是转换操作,是惰性的。rdd没有启动Actionmap不会执行,因此输出为0

第二种:计数翻倍

//累加器
final LongAccumulator accumulator = sc.sc().longAccumulator();

//map && reduce
JavaRDD<Integer> javaRDD = rdd.map(new Function<Integer, Integer>() {
    public Integer call(Integer integer) throws Exception {
        accumulator.add(integer);
        return integer;
    }
});

//reduce
javaRDD.reduce(new Function2<Integer, Integer, Integer>() {
    public Integer call(Integer integer, Integer integer2) throws Exception {
        return integer + integer2;
    }
});

//count
javaRDD.count();

System.out.println("执行结果:" + accumulator.value());
-----------------------
执行结果:30

当执行reduce操作时,启动作业,map开始执行,累加器的值应为15。但是rdd又执行了一次count操作,count也是Actionmap操作再次执行,由于累加器此时不为0,所以最终结果翻倍,输出30

解决这个问题的方法是:开启rdd 缓存

//reduce,rdd缓存
javaRDD.cache().reduce(new Function2<Integer, Integer, Integer>() {
    public Integer call(Integer integer, Integer integer2) throws Exception {
        return integer + integer2;
    }
});

rdd 第一次执行action算子的时候开启cache缓存,后面的action算子不会触发map执行。

实际上rdd缓存函数是persist函数,它支持不同类型的缓存策略,cache()等同于persist(StorageLevel.MEMORY_ONLY())。其他缓存级别可参考官网。

补充:缓存删除Spark会自动监视每个节点上的缓存使用情况,并以最近最少使用(LRU)的方式删除旧数据分区。如果您想手动删除RDD而不是等待它退出缓存,请使用该RDD.unpersist()方法。

广播变量

官方介绍:Broadcast variables allow the programmer to keep a read-only variable cached on each machine rather than shipping a copy of it with tasks. They can be used, for example, to give every node a copy of a large input dataset in an efficient manner. Spark also attempts to distribute broadcast variables using efficient broadcast algorithms to reduce communication cost.广播变量是Spark中另一种共享变量,允许程序将一个只读的变量发送到Executor,一个Executor只需要在第一个Task启动时,获得一份Broadcast数据,之后的Task都从本节点的BlockManager中获取相关数据。

使用方法非常简单

 //定义一个String类型的变量
final Broadcast<String> broadcast = sc.broadcast("broadcast");
...
//rdd中使用
broadcast.value()
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,772评论 6 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,458评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,610评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,640评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,657评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,590评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,962评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,631评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,870评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,611评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,704评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,386评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,969评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,944评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,179评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,742评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,440评论 2 342

推荐阅读更多精彩内容