In spark, how does broadcast work?

2020-05-21 04:28发布

This is a very simple question: in spark, broadcast can be used to send variables to executors efficiently. How does this work ?

More precisely:

  • when are values sent : as soon as I call broadcast, or when the values are used ?
  • Where exactly is the data sent : to all executors, or only to the ones that will need it ?
  • where is the data stored ? In memory, or on disk ?
  • Is there a difference in how simple variables and broadcast variables are accessed ? What happens under the hood when I call the .value method ?

2条回答
Evening l夕情丶
2楼-- · 2020-05-21 05:08

Short answer

  • Values are sent the first time they are needed in an executor. Nothing is sent when sc.broadcast(variable) is called.
  • The data is sent only to the nodes that contain an executor that needs it.
  • The data is stored in memory. If not enough memory is available, the disk is used.
  • Yes, there is a big difference between accessing a local variable and a broadcast variable. Broadcast variables have to be downloaded the first time they are accessed.

Long answer

The answer is in Spark's source, in TorrentBroadcast.scala.

  1. When sc.broadcast is called, a new TorrentBroadcast object is instantiated from BroadcastFactory.scala. The following happens in writeBlocks(), which is called when the TorrentBroadcast object is initialized:

    1. The object is cached unserialized locally using the MEMORY_AND_DISK policy.
    2. It is serialized.
    3. The serialized version is split into 4Mb blocks, that are compressed[0], and saved locally[1].
  2. When new executors are created, they only have the lightweight TorrentBroadcast object, that only contains the broadcast object's identifier, and its number of blocks.

  3. The TorrentBroadcast object has a lazy[2] property that contains its value. When the value method is called, this lazy property is returned. So the first time this value function is called on a task, the following happens:

    1. In a random order, blocks are fetched from the local block manager and uncompressed.
    2. If they are not present in the local block manager, getRemoteBytes is called on the block manager to fetch them. Network traffic happens only at that time.
    3. If the block wasn't present locally, it is cached using MEMORY_AND_DISK_SER.

[0] Compressed with lz4 by default. This can be tuned.

[1] The blocks are stored in the local block manager, using MEMORY_AND_DISK_SER, which means that it spills partitions that don't fit in memory to disk. Each block has an unique identifier, computed from the identifier of the broadcast variable, and its offset. The size of blocks can be configured; it is 4Mb by default.

[2] A lazy val in scala is a variable whose value is evaluated the first time it is accessed, and then cached. See the documentation.

查看更多
beautiful°
3楼-- · 2020-05-21 05:22
  • as soon as it is broadcasted
  • it is send to all executors using torrent protocol but loaded only when needed
  • once loaded variables are stored deserialized in memory
  • it:

    • validates that broadcast hasn't been destroyed
    • lazily loads variable from blockManager
查看更多
登录 后发表回答