public WriteFilesResult expand()

in src/main/java/com/google/cloud/solutions/autotokenize/common/CsvIO.java [543:563]

• 17 lines of code
• 2 McCabe index (conditional complexity)

    public WriteFilesResult<Void> expand(PCollection<T> input) {

      checkNotNull(outputFilePrefix(), "Provide outputFilePrefix for destination");

      var fileIO =
          FileIO.<T>write()
              .via(
                  CsvSink.<T>builder()
                      .csvFormat(csvFormat())
                      .transformFunction(transformFunction())
                      .writeHeaderRow(writeHeaderRow())
                      .build())
              .to(outputFilePrefix())
              .withSuffix(".csv");

      if (fileShards() != null) {
        fileIO = fileIO.withNumShards(fileShards());
      }

      return input.apply(fileIO);
    }