使用Apache Beam將Kafka Stream輸出寫入多個目錄

Question

我想使用數據流將Kafka主題的數據保存到谷歌存儲。

我在本地寫了一個示例代碼，它運行良好。

public static void main(String[] args) {
    PipelineOptions options = PipelineOptionsFactory.create();
    Pipeline p = Pipeline.create(options);
    p.apply(KafkaIO.<Long, String>read().withBootstrapServers("localhost:9092").withTopic("my-topic")
            .withKeyDeserializer(LongDeserializer.class).withValueDeserializer(StringDeserializer.class))
            .apply(Window
                    .<KafkaRecord<Long, String>>
                    into(FixedWindows.of(Duration.standardMinutes(1)))
            )
            .apply(FlatMapElements.into(TypeDescriptors.strings())
                    .via((KafkaRecord<Long, String> line) -> TextUtil.splitLine(line.getKV().getValue())))
            .apply(Filter.by((String word) -> StringUtils.isNotEmpty(word))).apply(Count.perElement())
            .apply(MapElements.into(TypeDescriptors.strings())
                    .via((KV<String, Long> lineCount) -> lineCount.getKey() + ": " + lineCount.getValue()))
            .apply(TextIO.write().withWindowedWrites().withNumShards(1)
                    .to("resources/temp/wc-kafka-op/wc"));

    p.run().waitUntilFinish();
}

上面的代碼完美無缺。 但我想將每個窗口的輸出保存在單獨的目錄中。

例如{BasePath} / {Window} / {prefix} {Suffice}

我無法讓它發揮作用。

Answer 1

當您可以指定名稱的派生方式時，TextIO支持windowedWrites。 請參閱JavaDoc 。