Flink中重写Watermark实践

2022-11-30 2025-02-26 约 1600 字预计阅读 4 分钟 - 次阅读

Apache Flink 是一个强大的流处理框架，有实时数据流处理的能力。在处理实时数据时，一个关键的东西是 Watermark。Watermarks 是一种特殊的时间戳，用于处理事件时间（event time）的流数据，以解决乱序事件和延迟数据的问题。然而，有时候我们可能需要根据特定的业务逻辑来自定义Watermark的生成。本文将探讨如何在Flink中重写Watermark，并提供一些实用的技巧和示例。

1 什么是Watermark

在Flink中，Watermark 是一个事件时间的标记，它表示在这个时间戳之前的数据已经被处理。Flink使用Watermark来确定何时可以触发基于事件时间的窗口操作。如果一个事件的时间戳早于当前的Watermark，那么这个事件被认为是“迟到”的，可能会被丢弃或者放入一个特殊的侧输出流。

2 为什么需要重写

在实际应用中，原始数据流中的Watermark可能不符合我们的需求。例如，我们可能需要根据业务逻辑来调整Watermark的生成策略，或者处理一些特殊情况，如数据延迟、系统故障等。这时，我们就需要自定义Watermark的生成逻辑。

3 如何在Flink中重写Watermark

在Flink中，我们可以通过实现 WatermarkStrategy 接口来自定义Watermark的生成。一般要做到以下4点：

定义Watermark策略：创建一个继承自 WatermarkStrategy 的类，并实现 createTimestampAssigner 和 createWatermarkGenerator 方法。
实现TimestampAssigner：在 createTimestampAssigner 方法中，返回一个 TimestampAssigner 实例，负责为每个事件分配时间戳。
实现WatermarkGenerator：在 createWatermarkGenerator 方法中，返回一个 WatermarkGenerator 实例，负责生成Watermark。
应用Watermark策略：在创建数据流（DataStream）时，使用 WatermarkStrategy 来设置Watermark的生成策略。

4 示例：自定义Watermark生成策略

假设我们有一个日志数据流，其中包含了用户活动的时间戳。我们希望根据用户活动的频率来动态调整Watermark，以更好地处理迟到的数据。以下是一个简单的示例：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33


public class CustomWatermarkStrategy implements WatermarkStrategy<LogEvent> {
    @Override
    public TimestampAssigner<LogEvent> createTimestampAssigner(WatermarkGenerator<?> generator) {
        return new TimestampAssigner<LogEvent>() {
            @Override
            public long extractTimestamp(LogEvent event) {
                // 从事件中提取时间戳
                return event.getTimestamp();
            }
        };
    }

    @Override
    public WatermarkGenerator<LogEvent> createWatermarkGenerator(TimestampAssigner<LogEvent> timestampAssigner) {
        return new WatermarkGenerator<LogEvent>() {
            private long lastTimestamp = Long.MIN_VALUE;

            @Override
            public void onEvent(LogEvent event, long eventTimestamp, WatermarkOutput output) {
                // 根据事件时间戳动态调整Watermark
                long watermark = Math.max(lastTimestamp, eventTimestamp) - 10000; // 延迟10秒
                output.emitWatermark(new Watermark(watermark));
                lastTimestamp = eventTimestamp;
            }

            @Override
            public void onPeriodicEmit(WatermarkOutput output) {
                // 定期发射Watermark
                output.emitWatermark(new Watermark(lastTimestamp - 10000));
            }
        };
    }
}

这个示例中创建了一个自定义的Watermark策略，它根据事件的时间戳动态调整Watermark。这样，我们就可以更好地处理迟到的数据，而不会因为过于严格的Watermark策略而丢失太多信息。

4.1 针对时间格式改造

时间信息是形如"2022/10/22 10:34"的字符串，怎样针对这种时间信息进行重写呢？

如果这个字符串已经被解析为一个Java时间对象，例如java.util.Date或java.time.Instant，可以直接在assignTimestampsAndWatermarks函数中将它作为参数传入。例如：

1
2
3
4
5
6
7


Copy codeDataStream<YourType> stream = ...;
stream.assignTimestampsAndWatermarks(new BoundedOutOfOrdernessTimestampExtractor<YourType>(Time.seconds(10)) {
    @Override
    public long extractTimestamp(YourType element) {
        return element.getEventTime().getTime();
    }
});

如果字符串未被解析为Java时间对象，则需要先将其解析为一个时间对象，然后再在assignTimestampsAndWatermarks函数中使用它。可以使用SimpleDateFormat类来解析时间字符串：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10


Copy codeDataStream<YourType> stream = ...;
SimpleDateFormat format = new SimpleDateFormat("yyyy/MM/dd HH:mm");
final Date eventTime = format.parse(timeString);

stream.assignTimestampsAndWatermarks(new BoundedOutOfOrdernessTimestampExtractor<YourType>(Time.seconds(10)) {
    @Override
    public long extractTimestamp(YourType element) {
        return eventTime.getTime();
    }
});

5 总结

在Flink中重写Watermark是一个高级功能，能根据具体的业务需求来调整流处理的行为。通过自定义Watermark策略，可以更灵活地处理实时数据流，提高数据处理的准确性和效率。在实际应用中，合理地使用这一操作可以提升对大数据的处理能力。

Buy me a coffee~

赞赏

支付宝

贝宝

微信