如何在flink数据流中获取拼图文件的文件名

import org.apache.flink.api.common.functions.FlatMapFunction;
import org.apache.flink.api.java.tuple.Tuple2;
import org.apache.flink.core.fs.FileInputSplit;
import org.apache.flink.util.Collector;

public class FileProcessingJob {

  public static void main(String[] args) throws Exception {
    // 创建流处理环境
    StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();

    // 读取文件，并将其转换为数据流
    DataStream<String> inputStream = env.readFile(new TextInputFormat(new Path("path/to/puzzle/files")), "path/to/puzzle/files");

    // 在数据流中获取文件名
    DataStream<String> fileNames = inputStream.flatMap(new FileNameExtractor());

    // 打印文件名
    fileNames.print();

    // 执行作业
    env.execute("File Processing Job");
  }

  public static class FileNameExtractor implements FlatMapFunction<String, String> {
    @Override
    public void flatMap(String input, Collector<String> out) {
      // 获取InputSplit对象
      InputSplit split = getRuntimeContext().getInputSplit();

      // 如果是FileInputSplit，获取文件名
      if (split instanceof FileInputSplit) {
        FileInputSplit fileSplit = (FileInputSplit) split;
        String fileName = fileSplit.getPath().getName();

        // 发出文件名
        out.collect(fileName);
      }
    }
  }
}