[FLINK-6298] Local execution is not setting RuntimeContext for RichOutputFormat

• Type: Bug
• Status: Closed
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 1.1.0, 1.2.0
• Fix Version/s: 1.2.2, 1.3.0
• Component/s: API / DataStream
• Assignee: Wenlong Lyu
• Reporter: Mateusz Zakarczemny
• Created: 11/Apr/17 18:01
• Updated: 14/Apr/17 01:35
• Resolved: 14/Apr/17 01:35

Description

RuntimeContext is never set in RichOutputFormat. I tested it in local execution. RichMapFunction is setup correctly.

Following code will never print "//////Context set in RichOutputFormat"

import org.apache.flink.api.common.functions.{RichMapFunction, RuntimeContext}
import org.apache.flink.api.common.io.RichOutputFormat
import org.apache.flink.api.scala._
import org.apache.flink.configuration.Configuration
import org.apache.flink.streaming.api.scala.StreamExecutionEnvironment

object Startup {
  def main(args: Array[String]): Unit = {

    val mapFunction = new RichMapFunction[String, String] {
      def open(taskNumber: Int, numTasks: Int) { getRuntimeContext }
      def map(event: String) = { event }
      override def setRuntimeContext(t: RuntimeContext) = {
        println("//////Context set in RichMapFunction")
        super.setRuntimeContext(t)
      }
    }

    val outputFormat = new RichOutputFormat[String] {
      override def setRuntimeContext(t: RuntimeContext) = {
        println("//////Context set in RichOutputFormat")
        super.setRuntimeContext(t)
      }
      def open(taskNumber: Int, numTasks: Int) {}
      def writeRecord(event: String) {
        println(event)
      }
      def configure(parameters: Configuration): Unit = {}
      def close(): Unit = {}
    }

    val see = StreamExecutionEnvironment.getExecutionEnvironment
    val eventsStream = see.fromElements[String]("A", "B", "C").map(mapFunction)
    eventsStream.writeUsingOutputFormat(outputFormat)
    see.execute("test-job")
  }
}

Issue Links

• links to: GitHub Pull Request #3716