Packages

class ApplyInPandasWithStatePythonRunner extends BasePythonRunner[InType, OutType] with PythonArrowInput[InType] with PythonArrowOutput[OutType]

A variant implementation of ArrowPythonRunner to serve the operation applyInPandasWithState.

Unlike normal ArrowPythonRunner which both input and output (executor <-> python worker) are InternalRow, applyInPandasWithState has side data (state information) in both input and output along with data, which requires different struct on Arrow RecordBatch.

Linear Supertypes
PythonArrowOutput[OutType], PythonArrowInput[InType], BasePythonRunner[InType, OutType], Logging, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. ApplyInPandasWithStatePythonRunner
  2. PythonArrowOutput
  3. PythonArrowInput
  4. BasePythonRunner
  5. Logging
  6. AnyRef
  7. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new ApplyInPandasWithStatePythonRunner(funcs: Seq[ChainedPythonFunctions], evalType: Int, argOffsets: Array[Array[Int]], inputSchema: StructType, timeZoneId: String, initialWorkerConf: Map[String, String], stateEncoder: ExpressionEncoder[Row], keySchema: StructType, outputSchema: StructType, stateValueSchema: StructType, pythonMetrics: Map[String, SQLMetric])

Type Members

  1. class MonitorThread extends Thread
    Definition Classes
    BasePythonRunner
  2. abstract class ReaderIterator extends Iterator[OUT]
    Definition Classes
    BasePythonRunner
  3. class WriterMonitorThread extends Thread
    Definition Classes
    BasePythonRunner
  4. abstract class WriterThread extends Thread
    Definition Classes
    BasePythonRunner

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. val accumulator: PythonAccumulatorV2
    Attributes
    protected
    Definition Classes
    BasePythonRunner
  5. val argOffsets: Array[Array[Int]]
    Attributes
    protected
    Definition Classes
    BasePythonRunner
  6. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  7. val authSocketTimeout: Long
    Attributes
    protected
    Definition Classes
    BasePythonRunner
  8. val bufferSize: Int
    Definition Classes
    ApplyInPandasWithStatePythonRunner → BasePythonRunner
  9. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  10. def compute(inputIterator: Iterator[InType], partitionIndex: Int, context: TaskContext): Iterator[OutType]
    Definition Classes
    BasePythonRunner
  11. def deserializeColumnarBatch(batch: ColumnarBatch, schema: StructType): OutType

    Deserialize ColumnarBatch received from the Python worker to produce the output.

    Deserialize ColumnarBatch received from the Python worker to produce the output. Schema info for given ColumnarBatch is also provided as well.

    Attributes
    protected
    Definition Classes
    ApplyInPandasWithStatePythonRunner → PythonArrowOutput
  12. val envVars: Map[String, String]
    Attributes
    protected
    Definition Classes
    BasePythonRunner
  13. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  14. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  15. val evalType: Int
    Attributes
    protected
    Definition Classes
    BasePythonRunner
  16. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  17. val funcs: Seq[ChainedPythonFunctions]
    Attributes
    protected
    Definition Classes
    BasePythonRunner
  18. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  19. def handleMetadataAfterExec(stream: DataInputStream): Unit
    Attributes
    protected
    Definition Classes
    PythonArrowOutput
  20. def handleMetadataBeforeExec(stream: DataOutputStream): Unit

    This method sends out the additional metadata before sending out actual data.

    This method sends out the additional metadata before sending out actual data.

    Specifically, this class overrides this method to also write the schema for state value.

    Attributes
    protected
    Definition Classes
    ApplyInPandasWithStatePythonRunner → PythonArrowInput
  21. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  22. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  23. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  24. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  25. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  26. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  27. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  28. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  29. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  30. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  31. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  32. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  33. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  34. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  35. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  36. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  37. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  38. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  39. def newReaderIterator(stream: DataInputStream, writerThread: WriterThread, startTime: Long, env: SparkEnv, worker: Socket, pid: Option[Int], releasedOrClosed: AtomicBoolean, context: TaskContext): Iterator[OutType]
    Attributes
    protected
    Definition Classes
    PythonArrowOutput
  40. def newWriterThread(env: SparkEnv, worker: Socket, inputIterator: Iterator[InType], partitionIndex: Int, context: TaskContext): WriterThread
    Attributes
    protected
    Definition Classes
    PythonArrowInput
  41. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  42. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  43. val pythonExec: String
    Attributes
    protected
    Definition Classes
    BasePythonRunner
  44. val pythonMetrics: Map[String, SQLMetric]
    Definition Classes
    ApplyInPandasWithStatePythonRunner → PythonArrowOutput → PythonArrowInput
  45. val pythonVer: String
    Attributes
    protected
    Definition Classes
    BasePythonRunner
  46. val schema: StructType
    Attributes
    protected
    Definition Classes
    ApplyInPandasWithStatePythonRunner → PythonArrowInput
  47. val simplifiedTraceback: Boolean
    Definition Classes
    ApplyInPandasWithStatePythonRunner → BasePythonRunner
  48. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  49. val timeZoneId: String
    Attributes
    protected
    Definition Classes
    ApplyInPandasWithStatePythonRunner → PythonArrowInput
  50. def toString(): String
    Definition Classes
    AnyRef → Any
  51. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  52. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  53. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  54. val workerConf: Map[String, String]
    Attributes
    protected
    Definition Classes
    ApplyInPandasWithStatePythonRunner → PythonArrowInput
  55. def writeIteratorToArrowStream(root: VectorSchemaRoot, writer: ArrowStreamWriter, dataOut: DataOutputStream, inputIterator: Iterator[InType]): Unit

    Read the (key, state, values) from input iterator and construct Arrow RecordBatches, and write constructed RecordBatches to the writer.

    Read the (key, state, values) from input iterator and construct Arrow RecordBatches, and write constructed RecordBatches to the writer.

    See ApplyInPandasWithStateWriter for more details.

    Attributes
    protected
    Definition Classes
    ApplyInPandasWithStatePythonRunner → PythonArrowInput

Inherited from PythonArrowOutput[OutType]

Inherited from PythonArrowInput[InType]

Inherited from BasePythonRunner[InType, OutType]

Inherited from Logging

Inherited from AnyRef

Inherited from Any

Ungrouped