Skip to content

Notebooks — streaming_ingest-events-spark-iceberg

Auto-extracted from jupyter/notebook.ipynb and zeppelin/notebook.zpln. Both notebooks implement identical logic in PySpark and Scala.

1. Section map

Subsection Scala (Zeppelin) PySpark (Jupyter)
2.1 Setup
2.2 Read
2.3 Transform
2.4 Write
2.5 Verify

2. Walkthrough

2.1 Setup

Scala (Zeppelin):

import spark.implicits._
import org.apache.spark.sql.functions._
import org.apache.spark.sql.types._

PySpark (Jupyter):

from pyspark.sql import SparkSession
from pyspark.sql import functions as F
from pyspark.sql.types import StringType, StructType, TimestampType

spark = SparkSession.builder.remote("sc://spark-connect:15002").getOrCreate()

2.2 Read

Scala (Zeppelin):

val raw = spark.readStream
  .format("kafka")
  .option("kafka.bootstrap.servers", "redpanda:9092")
  .option("subscribe", "events")
  .option("startingOffsets", "earliest")
  .load()

PySpark (Jupyter):

raw = (
    spark.readStream
    .format("kafka")
    .option("kafka.bootstrap.servers", "redpanda:9092")
    .option("subscribe", "events")
    .option("startingOffsets", "earliest")
    .load()
)

2.3 Transform

Scala (Zeppelin):

val schema = new StructType()
  .add("user_id", StringType)
  .add("event", StringType)
  .add("ts", TimestampType)

val events = raw
  .select(from_json($"value".cast("string"), schema).as("e"))
  .select("e.*")

PySpark (Jupyter):

schema = (
    StructType()
    .add("user_id", StringType())
    .add("event", StringType())
    .add("ts", TimestampType())
)

events = raw.select(
    F.from_json(F.col("value").cast("string"), schema).alias("e")
).select("e.*")

2.4 Write

Scala (Zeppelin):

val query = events.writeStream
  .format("iceberg")
  .outputMode("append")
  .option("checkpointLocation", "s3a://checkpoints/events")
  .toTable("lakehouse.bronze.events")

// Run query.awaitTermination() to keep the stream alive for a live Redpanda topic

PySpark (Jupyter):

query = (
    events.writeStream
    .format("iceberg")
    .outputMode("append")
    .option("checkpointLocation", "s3a://checkpoints/events")
    .toTable("lakehouse.bronze.events")
)

# Run query.awaitTermination() to keep the stream alive for a live Redpanda topic

2.5 Verify

Scala (Zeppelin):

spark.table("lakehouse.bronze.events").count()

PySpark (Jupyter):

spark.table("lakehouse.bronze.events").count()

3. Scala / PySpark parity

Both notebooks share the same numbered sections and produce identical Iceberg tables; only the language and interpreter differ.

4. How to run

Open the scenario's zeppelin/notebook.zpln on the Atlas Zeppelin UI or jupyter/notebook.ipynb on JupyterHub, then run all paragraphs/cells top to bottom.