diff --git a/connector/kafka-0-10-sql/src/test/scala/org/apache/spark/sql/kafka010/KafkaRealTimeModeSuite.scala b/connector/kafka-0-10-sql/src/test/scala/org/apache/spark/sql/kafka010/KafkaRealTimeModeSuite.scala index 468d1da7f467f..ae23b53fc35ad 100644 --- a/connector/kafka-0-10-sql/src/test/scala/org/apache/spark/sql/kafka010/KafkaRealTimeModeSuite.scala +++ b/connector/kafka-0-10-sql/src/test/scala/org/apache/spark/sql/kafka010/KafkaRealTimeModeSuite.scala @@ -25,7 +25,7 @@ import org.scalatest.time.SpanSugar._ import org.apache.spark.{SparkConf, SparkContext, SparkIllegalStateException} import org.apache.spark.sql.execution.datasources.v2.LowLatencyClock import org.apache.spark.sql.execution.streaming._ -import org.apache.spark.sql.execution.streaming.sources.ContinuousMemorySink +import org.apache.spark.sql.execution.streaming.sources.{ContinuousMemorySink, LowLatencyMemoryStream} import org.apache.spark.sql.execution.streaming.state.RocksDBStateStoreProvider import org.apache.spark.sql.internal.SQLConf import org.apache.spark.sql.kafka010.consumer.KafkaDataConsumer diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/sources/LowLatencyMemoryStream.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/sources/LowLatencyMemoryStream.scala index 5b16a2a3be80c..666adba1e33dd 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/sources/LowLatencyMemoryStream.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/sources/LowLatencyMemoryStream.scala @@ -15,7 +15,7 @@ * limitations under the License. */ -package org.apache.spark.sql.execution.streaming +package org.apache.spark.sql.execution.streaming.sources import java.util.concurrent.atomic.AtomicInteger import javax.annotation.concurrent.GuardedBy @@ -47,6 +47,7 @@ import org.apache.spark.sql.connector.read.streaming.{ } import org.apache.spark.sql.connector.read.streaming.SupportsRealTimeRead.RecordStatus import org.apache.spark.sql.execution.datasources.v2.LowLatencyClock +import org.apache.spark.sql.execution.streaming.Offset import org.apache.spark.sql.execution.streaming.runtime._ import org.apache.spark.util.{Clock, RpcUtils} diff --git a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeAllowlistSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeAllowlistSuite.scala index 4306e5a860120..a9b8d54a6c61b 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeAllowlistSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeAllowlistSuite.scala @@ -20,7 +20,7 @@ package org.apache.spark.sql.streaming import scala.concurrent.duration._ import org.apache.spark.SparkIllegalArgumentException -import org.apache.spark.sql.execution.streaming.LowLatencyMemoryStream +import org.apache.spark.sql.execution.streaming.sources.LowLatencyMemoryStream import org.apache.spark.sql.functions._ import org.apache.spark.sql.internal.SQLConf diff --git a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeE2ESuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeE2ESuite.scala index 3615edc75cb2c..8387f07b640cf 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeE2ESuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeE2ESuite.scala @@ -26,8 +26,8 @@ import org.scalatest.time.SpanSugar._ import org.apache.spark.SparkContext import org.apache.spark.sql.{ForeachWriter, Row} import org.apache.spark.sql.execution.datasources.v2.LowLatencyClock -import org.apache.spark.sql.execution.streaming.LowLatencyMemoryStream import org.apache.spark.sql.execution.streaming.runtime.StreamingQueryWrapper +import org.apache.spark.sql.execution.streaming.sources.LowLatencyMemoryStream import org.apache.spark.sql.functions._ import org.apache.spark.sql.streaming.util.GlobalSingletonManualClock import org.apache.spark.sql.test.TestSparkSession diff --git a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeSuite.scala index 2ff8fd90bb9c9..cea70676bb7bc 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeSuite.scala @@ -25,9 +25,9 @@ import scala.concurrent.duration.Duration import org.scalatest.concurrent.PatienceConfiguration.Timeout import org.apache.spark.{SparkIllegalArgumentException, SparkIllegalStateException, TaskContext} -import org.apache.spark.sql.execution.streaming.{LowLatencyMemoryStream, RealTimeTrigger} +import org.apache.spark.sql.execution.streaming.RealTimeTrigger import org.apache.spark.sql.execution.streaming.runtime.MemoryStream -import org.apache.spark.sql.execution.streaming.sources.ContinuousMemorySink +import org.apache.spark.sql.execution.streaming.sources.{ContinuousMemorySink, LowLatencyMemoryStream} import org.apache.spark.sql.functions.udf import org.apache.spark.sql.internal.SQLConf diff --git a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeSuiteBase.scala b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeSuiteBase.scala index 7ec5d8e51f098..7a8a1957b4e11 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeSuiteBase.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamRealTimeModeSuiteBase.scala @@ -27,8 +27,9 @@ import org.scalatest.time.SpanSugar._ import org.apache.spark.{SparkConf, SparkContext} import org.apache.spark.sql.ForeachWriter import org.apache.spark.sql.execution.datasources.v2.LowLatencyClock -import org.apache.spark.sql.execution.streaming.{LowLatencyMemoryStream, RealTimeTrigger} +import org.apache.spark.sql.execution.streaming.RealTimeTrigger import org.apache.spark.sql.execution.streaming.runtime.StreamingQueryWrapper +import org.apache.spark.sql.execution.streaming.sources.LowLatencyMemoryStream import org.apache.spark.sql.internal.SQLConf import org.apache.spark.sql.streaming.util.GlobalSingletonManualClock import org.apache.spark.sql.test.TestSparkSession