如何在spark streaming中更改已完成的批量计数？

import org.apache.spark.streaming.{Seconds, StreamingContext}
import java.util.concurrent.atomic.AtomicLong

object SparkStreamingExample {
  def main(args: Array[String]): Unit = {
    val ssc = new StreamingContext("local[*]", "SparkStreamingExample", Seconds(1))

    // 创建全局计数器
    val completedBatchCount = new AtomicLong(0)

    // 创建DStream并处理每个批次
    val lines = ssc.socketTextStream("localhost", 9999)
    lines.foreachRDD { rdd =>
      // 处理每个批次的逻辑

      // 获取当前批次的计数值
      val batchCount = rdd.count()

      // 将当前批次的计数值累加到全局计数器中
      completedBatchCount.addAndGet(batchCount)

      // 如果需要更改已完成的批量计数，可以直接修改全局计数器的值
      completedBatchCount.set(100) // 修改已完成的批量计数为100
    }

    ssc.start()
    ssc.awaitTermination()
  }
}