如何在Kotlin中扩展DoFn？

import org.apache.beam.sdk.transforms.DoFn
import org.apache.beam.sdk.transforms.DoFn.ProcessElement

class DoFnExtensions {
    companion object {
        @ProcessElement
        fun <T> DoFn<T, T>.process(context: ProcessContext) {
            val input = context.element() // 获取输入数据
            val output = processInput(input) // 调用自定义的处理函数
            context.output(output) // 输出处理结果
        }

        private fun <T> processInput(input: T): T {
            // 自定义的数据处理逻辑
            // 可以使用Kotlin的语法和标准库函数来处理数据
            return input
        }
    }
}

import org.apache.beam.sdk.Pipeline
import org.apache.beam.sdk.io.TextIO
import org.apache.beam.sdk.transforms.ParDo

fun main() {
    val pipeline = Pipeline.create()

    pipeline
        .apply(TextIO.read().from("input.txt"))
        .apply(ParDo.of(DoFnExtensions.process()))

    pipeline.run().waitUntilFinish()
}