How to check dataframe size in Scala?

import org.apache.spark.sql.{DataFrame, SparkSession}

object DataFrameSizeCheck {
  def main(args: Array[String]): Unit = {
    // Create SparkSession
    val spark = SparkSession.builder()
      .appName("DataFrameSizeCheck")
      .master("local[*]")
      .getOrCreate()

    // Sample DataFrame (replace this with your actual DataFrame)
    val dataframe: DataFrame = spark.emptyDataFrame

    // Get the size of DataFrame
    val size = dataframe.count()

    // Print the size
    println(s"DataFrame size: $size")

    // Stop SparkSession
    spark.stop()
  }
}

import org.apache.spark.sql.{DataFrame, SparkSession}
import org.apache.spark.sql.types.{IntegerType, StructField, StructType}

object DataFrameSizeCheck {
  def main(args: Array[String]): Unit = {
    // Create SparkSession
    val spark = SparkSession.builder()
      .appName("DataFrameSizeCheck")
      .master("local[*]")
      .getOrCreate()

    // Sample data for DataFrame
    val data = Seq(
      (1, "John"),
      (2, "Alice"),
      (3, "Bob")
    )

    // Define the schema
    val schema = StructType(
      Seq(
        StructField("ID", IntegerType, nullable = false),
        StructField("Name", StringType, nullable = false)
      )
    )

    // Create DataFrame
    val dataframe: DataFrame = spark.createDataFrame(spark.sparkContext.parallelize(data), schema)

    // Get the size of DataFrame
    val size = dataframe.count()

    // Print the size
    println(s"DataFrame size: $size")

    // Stop SparkSession
    spark.stop()
  }
}