Spark Drive 端的 List 无法获取 Executor 中的数据吗?

你好,我是 shengjk1,多年大厂经验,努力构建 通俗易懂的、好玩的编程语言教程。 欢迎关注!你会有如下收益:

  1. 了解大厂经验
  2. 拥有和大厂相匹配的技术等

希望看什么,评论或者私信告诉我!

一、背景

好久没有写 Spark 相关的代码了,结果写了一个如下类似的代码,在 drive 端 new 了一个 arrayList,然后再 Executor 端进行 add 操作,最后再在 drive 端打印 这个 list .size。结果发现这个 size 为0,顿时好奇心就上来了。

bash 复制代码
 	    List<SuggestFeedback> slows = new ArrayList<>();

        rowJavaRDD.foreachPartition(rowIterator -> {
            while (rowIterator.hasNext()) {
                Row row = rowIterator.next();
                SuggestFeedback suggestFeedback = new SuggestFeedback();
                suggestFeedback.setDeviceId( Objects.isNull(row.getAs("device_id")) ? "" : row.getAs("device_id").toString());
                String suggest = suggestFeedback.getLevel1_mass_category();
                if (suggest.equals("卡顿问题")) {
                    slows.add(suggestFeedback);
                } 
       }
       System.out.print(slow.size())

二、spark 中解决方案( 其他分布式计算引擎也类似 )

在 Apache Spark 中,Driver 端和 Executor 端(TaskManager)运行在不同的 JVM 进程中,因此它们之间的内存是不共享的。如果你在 Driver 端创建了一个 List,并尝试在 Executor 端(TaskManager)通过 list.add 添加数据,这是不会生效的,因为 Executor 端操作的是它自己的内存空间,而不是 Driver 端的内存空间。

解决方案

如果你需要在 Driver 端收集 Executor 端的数据,可以使用以下几种方式:

1. 使用 collect 方法

collect 方法会将 Executor 端的数据收集到 Driver 端。你可以在 Executor 端处理数据,然后使用 collect 将结果返回到 Driver 端。

scala 复制代码
val rdd = sc.parallelize(Seq(1, 2, 3, 4, 5))
val result = rdd.map(x => x * 2).collect()  // 将数据收集到 Driver 端
val list = result.toList  // 转换为 List

2. 使用 aggregatereduce 方法

如果你需要在 Executor 端进行聚合操作,并将结果返回到 Driver 端,可以使用 aggregatereduce 方法。

scala 复制代码
val rdd = sc.parallelize(Seq(1, 2, 3, 4, 5))
val sum = rdd.reduce(_ + _)  // 在 Executor 端进行聚合操作,结果返回到 Driver 端

3. 使用累加器(Accumulator)

累加器是一种特殊的变量,可以在 Executor 端进行累加操作,最终结果会在 Driver 端可见。

scala 复制代码
val accum = sc.longAccumulator("My Accumulator")
rdd.foreach(x => accum.add(x))
println(accum.value)  // 在 Driver 端获取累加器的值

4. 使用广播变量(Broadcast Variable)

如果你需要在 Executor 端访问 Driver 端的数据,可以使用广播变量。广播变量会将 Driver 端的数据发送到每个 Executor 端,但 Executor 端不能修改广播变量的值。

scala 复制代码
val list = List(1, 2, 3, 4, 5)
val broadcastVar = sc.broadcast(list)
val rdd = sc.parallelize(Seq(1, 2, 3))
val result = rdd.map(x => x + broadcastVar.value.sum).collect()

总结

在 Spark 中,Driver 端和 Executor 端的内存是隔离的,因此无法直接在 Executor 端修改 Driver 端的数据结构(如 List)。你需要通过 collectaggregate、累加器或广播变量等方式在 Driver 端和 Executor 端之间传递数据。

三、为什么不会生效和报错

在 Apache Spark 或 Apache Flink 中,如果你在 Executor 端(TaskManager) 尝试修改 Driver 端(JobManager) 创建的 List 或其他数据结构,程序不会直接报错,但修改不会生效。这是因为 Executor 端和 Driver 端运行在不同的 JVM 进程中,内存是隔离的。以下详细解释为什么不会报错,以及背后的原理。


1. 为什么不会报错?

  • SparkFlink 都是分布式计算框架,它们的任务执行模型是基于 Driver-ExecutorJobManager-TaskManager 的架构。
  • Driver 端(JobManager) 负责协调任务,而 Executor 端(TaskManager) 负责实际执行任务。
  • Executor 端Driver 端 运行在不同的 JVM 进程中,它们之间的内存是隔离的。

Java 对象的序列化和传递

  • 当你将一个 List 或其他对象传递给 Executor 端 时,Spark 或 Flink 会将该对象序列化并发送到 Executor 端
  • Executor 端 ,这个对象会被反序列化,生成一个 新的对象 ,这个对象与 Driver 端 的对象是完全独立的。
  • 因此,在 Executor 端 对这个对象的任何修改(如 list.add)只会影响 Executor 端 的副本,而不会影响 Driver 端 的原始对象。

为什么不会报错?

  • Executor 端 的操作是在自己的 JVM 进程中执行的,修改的是自己的副本,因此不会触发任何异常。
  • Driver 端Executor 端 之间的内存隔离是设计上的特性,而不是错误。框架本身不会对这种行为进行检查或报错。

2. 为什么修改不会生效?

内存隔离

  • Driver 端Executor 端 运行在不同的 JVM 进程中,它们的内存是完全隔离的。
  • 即使你在 Executor 端 修改了 List,这些修改只会影响 Executor 端 的副本,而不会同步回 Driver 端

对象的序列化和反序列化

  • 当你将一个对象(如 List)传递给 Executor 端 时,Spark 或 Flink 会将该对象序列化并发送到 Executor 端
  • Executor 端 ,这个对象会被反序列化,生成一个 新的对象 。这个新对象与 Driver 端 的对象是完全独立的。
  • 因此,在 Executor 端 对这个对象的任何修改(如 list.add)只会影响 Executor 端 的副本,而不会影响 Driver 端 的原始对象。

3. 如何避免这种问题?

为了避免在 Executor 端 修改 Driver 端 的数据结构而不生效的问题,可以采用以下方法:

在 Spark 中

  1. 使用 collect 方法

    • Executor 端 的数据收集到 Driver 端 ,然后在 Driver 端 进行修改。
    scala 复制代码
    val rdd = sc.parallelize(Seq(1, 2, 3, 4, 5))
    val result = rdd.map(x => x * 2).collect()  // 将数据收集到 Driver 端
    val list = result.toList  // 转换为 List
  2. 使用累加器(Accumulator)

    • 累加器可以在 Executor 端 进行累加操作,最终结果会在 Driver 端 可见。
    scala 复制代码
    val accum = sc.longAccumulator("My Accumulator")
    rdd.foreach(x => accum.add(x))
    println(accum.value)  // 在 Driver 端获取累加器的值
  3. 使用广播变量(Broadcast Variable)

    • 广播变量可以将 Driver 端 的数据分发到所有 Executor 端 ,但 Executor 端 只能读取广播变量,不能修改。
    scala 复制代码
    val list = List(1, 2, 3, 4, 5)
    val broadcastVar = sc.broadcast(list)
    val rdd = sc.parallelize(Seq(1, 2, 3))
    val result = rdd.map(x => x + broadcastVar.value.sum).collect()
  1. 使用 DataSetDataStream API

    • TaskManager 端 的数据返回到 JobManager 端
    java 复制代码
    ExecutionEnvironment env = ExecutionEnvironment.getExecutionEnvironment();
    DataSet<Integer> data = env.fromElements(1, 2, 3, 4, 5);
    DataSet<Integer> result = data.map(x -> x * 2);
    List<Integer> resultList = result.collect(); // 将数据收集到 JobManager 端
  2. 使用累加器(Accumulator)

    • 累加器可以在 TaskManager 端 进行累加操作,最终结果会在 JobManager 端 可见。
    java 复制代码
    StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
    LongCounter counter = new LongCounter();
    env.fromElements(1, 2, 3, 4, 5)
       .map(x -> {
           counter.add(1); // 在 TaskManager 端累加
           return x;
       })
       .print(); // 触发任务执行
    env.execute();
    System.out.println("Counter value: " + counter.getLocalValue()); // 在 JobManager 端获取累加器的值
  3. 使用广播变量(Broadcast Variable)

    • 广播变量可以将 JobManager 端 的数据分发到所有 TaskManager 端 ,但 TaskManager 端 只能读取广播变量,不能修改。
    java 复制代码
    StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
    DataStream<Integer> data = env.fromElements(1, 2, 3, 4, 5);
    DataStream<Integer> broadcastData = env.fromElements(10, 20, 30);
    
    data.connect(broadcastData.broadcast())
        .process(new BroadcastProcessFunction<Integer, Integer, Integer>() {
            @Override
            public void processElement(Integer value, ReadOnlyContext ctx, Collector<Integer> out) {
                // 访问广播变量
                for (Integer broadcastValue : ctx.getBroadcastState().get("broadcastState")) {
                    out.collect(value + broadcastValue);
                }
            }
    
            @Override
            public void processBroadcastElement(Integer value, Context ctx, Collector<Integer> out) {
                // 更新广播状态
                ctx.getBroadcastState().put("broadcastState", value);
            }
        })
        .print();
    env.execute();

4. 总结

  • Executor 端Driver 端 的内存是隔离的,因此在 Executor 端 修改 Driver 端 的数据结构不会生效。
  • 这种行为不会报错,因为 Executor 端 操作的是自己的副本,而不是 Driver 端 的原始对象。
  • 为了避免这种问题,可以使用框架提供的机制(如 collect、累加器、广播变量等)来实现 Driver 端Executor 端 之间的数据传递和共享。

四、总结

Driver 端 和 Executor 端 运行在不同的 JVM 进程中,内存隔离导致 Executor 端 对 Driver 端 数据结构的修改不会生效。

数据从 Driver 端 传递到 Executor 端 是通过序列化和反序列化实现的,这是一个单向过程,修改不会同步回 Driver 端。

解决方案包括使用 collect 方法将数据收集到 Driver 端、使用累加器进行分布式累加、使用广播变量共享数据等。

这种行为不会报错,因为 Executor 端 操作的是自己的副本,而不是 Driver 端 的原始对象。

相关推荐
boring_1113 分钟前
Apache Pulsar 消息、流、存储的融合
分布式·后端
墨绿色的摆渡人4 分钟前
pytorch小记(二十二):全面解读 PyTorch 的 `torch.cumprod`——累积乘积详解与实战示例
人工智能·pytorch·python
moonsims11 分钟前
低空态势感知:基于AI的DAA技术是低空飞行的重要安全保障-机载端&地面端
人工智能·安全
若叶时代22 分钟前
数据分析_Python
人工智能·python·数据分析
虾球xz26 分钟前
游戏引擎学习第286天:开始解耦实体行为
c++·人工智能·学习·游戏引擎
武子康28 分钟前
大语言模型 11 - 从0开始训练GPT 0.25B参数量 MiniMind2 准备数据与训练模型 DPO直接偏好优化
人工智能·gpt·ai·语言模型·自然语言处理
AI+程序员在路上41 分钟前
REST架构风格介绍
物联网·架构·restful·web
羽凌寒2 小时前
图像对比度调整(局域拉普拉斯滤波)
人工智能·计算机视觉
大模型铲屎官2 小时前
【Python-Day 14】玩转Python字典(上篇):从零开始学习创建、访问与操作
开发语言·人工智能·pytorch·python·深度学习·大模型·字典
一点.点2 小时前
计算机视觉的简单介绍
人工智能·深度学习·计算机视觉