RisingWave x 特征工程:解锁实时特征新范式

RisingWave x 特征工程:解锁实时特征新范式
2024年12月25日 13:01 DataFunTalk

导读本文将介绍 RisingWave 在实时特征工程中的应用。RisingWave 是一款开源的流式数据库,具有易用、健壮、上下游生态系统开放、性价比高等特点,支持 SQL 和 UDF 扩展,其架构包含接入层、计算层和存储引擎,支持多种数据源和下游系统,通过物化视图等实现增量实时计算。在实时特征工程中,它能够助力数据摄入、数据清洗、特征构建、样本拼接和特征查询等环节,提供高效的状态管理和 UDF 支持。此外,RisingWave 2.0 带来了如 Premium 版本、云版本增强、对流批统一的改进等新特性。通过阅读本文,读者可深入了解 RisingWave 在实时数据处理领域的优势与应用。

今天的介绍会围绕下面四点展开:

1. RisingWave 介绍

2. RisingWave 在实时特征工程中的应用

3.RisingWave 其他使用场景

4. RisingWave 2.0 更新内容

分享嘉宾|Patrick HuangRisingWave Labs    VP of Engineering

编辑整理|李硕

内容校对|李瑶

出品社区|DataFun

01

RisingWave 介绍

1. 项目背景与基本信息

RisingWave 是一款具有创新性的开源流处理系统,在实时数据处理领域展现出独特优势。其开源项目背景源于对革新流处理和数据库管理的追求,于 2021 年初创立,并在 2022 年 4 月以 Apache2.0 协议在 GitHub 开源。经过三年打磨,已在全球多领域落地应用。

RisingWave是基于Rust的自研项目,采用存算分离架构,交互接口与 PostgreSQL 协议兼容,并可通过 UDF 拓展。其包含接入层、计算层与存储层三层架构,由 meta 节点协调,计算节点执行流作业并带有多级缓存,状态持久化至基于对象存储的存储引擎。产品使命为解决易用性问题,降低实时应用开发、运维与运行成本,无论对实时计算新手还是资深从业者,都致力于提供便捷、稳定且高效的流处理方案。目前,应用领域涵盖互联网、金融、能源、供应链等多个行业,在实时监控告警、流表实时打宽、规则引擎、实时数据市场等场景均有应用。截至当前,全球日活集群已超 1700 个。

2. RisingWave 特点

(1)易用性

RisingWave 通过 SQL 作为交互接口,兼容 PostgreSQL 协议,用户通过简单的 SQL 即可实现复杂的实时需求,同时支持通过不同语言的 UDF 进行拓展。另外,RisingWave 不仅仅是流式计算引擎,而且带有自研的存储引擎,除了支持有状态的复杂流计算外,实时分析的结果可以以物化视图的方式通过 SQL 在 RisingWave 中查询,我们称其为 Serving。同时 RisingWave 流算子的内部状态都抽象成了关系型表,也可以通过 SQL 查询,大大提升了流计算的可观测性。

(2)健壮性

RisingWave 定位为数据库,所以健壮稳定是首要要求。实时性方面,可以达到亚秒级新鲜度,并实现了 Exactly Once。支持强一致持久化 checkpoint,当出现故障时可以立即从上一 checkpoint 恢复。基于存算分离的架构,可以实现 zero downtime 的弹性伸缩和快速恢复。同时,RisingWave 支持 20+ 路多流 join 和复杂流式变换,并且支持长时间窗口大状态的流处理。

(3)开放的上下游生态系统

RisingWave 作为流处理系统,具备开放且多元的上下游生态系统。在上游 Source 方面,它支持多种常见的消息队列(如 Kafka 等)、各类数据库的变更数据捕获(CDC),涵盖 MySQL、PostgreSQL、Oracle 等关系型数据库以及 MongoDB 等非关系型数据库,并且支持如 Debezium 等多种 CDC 格式,同时也接纳如数据湖、文件系统内文件等批式数据源。而在下游 Sink,不仅支持消息队列,还支持 ClickHouse、StarRocks 等分析型数据库以及 Elasticsearch、Redis 等非关系型数据库,此外还实现了实时入湖功能。这种开放的生态系统,极大地拓展了 RisingWave 在不同数据场景下的应用范围,使其能更好地融入多样化的数据处理链路中。

(4)高性价比

实时计算相比于离线计算通常成本更高,而 RisingWave 通过多种优化,实现了高性价比。首先,使用低成本的对象存储作为存储后端,我们自研了基于 LSM 的存储引擎降低存储成本。RisingWave 支持多种对象存储,比如 S3、Azure Blob 等,也可以自己部署 MinIO、HDFS、DFS。采用存算分离架构,计算和存储可以独立扩缩容。计算节点采用多级缓存,可以根据需求调整,并且支持 serverless compaction。

3. RisingWave 架构

RisingWave 的架构主要分为三层。最上层是接入层(Frontend),它负责解析和优化用户请求,并生成执行计划,这些计划会被分布式调度到第二层 —— 计算层(Compute)执行。在流作业中,有状态的算子其状态会持久化到基于对象存储(ObjectStore)的存储引擎中。在这些组件之上,有一个 Meta 节点负责协调,起到控制器的作用。整体架构体现了 RisingWave 在流处理方面的高效设计,同时兼顾了存储和协调功能。

02

RisingWave 在实时特征工程中的应用

1. 特征工程步骤与链路

实时特征工程包含 Training 链路和 Inference 链路。Training 链路包括从上游数据源摄入数据、清洗选择、特征构建、样本拼接和实时模型训练。Inference 链路包括摄入数据构建行为特征、查询 Feature Store 特征拼接和向 Model 喂入特征完成 Inference。

实时特征工程在架构上存在挑战。引入的组件越多,运维越困难,工程师需熟悉多个系统。同时,组件增多会使稳定性难以保障,一个组件故障就可能影响整体。此外,影响实时性的因素变多,且上线周期变长,工程师需学习不同接口与组件交互,数据分散也导致回测困难。

2. RisingWave 的助力

RisingWave 在实时特征工程方面有诸多助力。它能用 SQL + UDF 构建 Streaming Pipeline,提供统一的数据源存储,支持 Serving 查询,并具备实时流式 Sink 功能,能够有效简化和优化实时特征工程的流程,提升效率。

接下来具体看一下链路中的每个步骤。

(1)数据摄入

在数据摄入环节,RisingWave 中可以使用 source connector 轻松接入多种数据源。

Source 相关助力

  • 多样化数据源支持

    消息队列(MQ):支持 Kafka、Pulsar、MQTT 等。

    变更数据捕获(CDC):支持 MySQL、PostgreSQL、TiDB、MongoDB 等数据库的 CDC。

    批处理数据源:支持 File System、Object Store、Iceberg 等。

  • 消息编码支持

    支持 AVRO、JSON、PROTOBUF、CSV、BYTES 等编码格式。

  • 消息队列支持指定消费位置指定

  • 支持从 Schema Registry 自动获取上游 Schema

Table 相关助力

  • 数据源支持广泛

    Table 可以消费所有 Source 支持的数据源,能够将各种来源的数据进行整合。

  • 物化数据支持

    将 Source 的数据物化到表,支持主键,便于数据的管理和查询。

  • 上游 CDC 支持

    支持常见的 OLTP 数据库(如 MySQL、PostgreSQL、Oracle、TiDB 等)和 NoSQL 数据库(如 MongoDB)的 CDC。

  • DML 支持

    支持增删改查(DML)操作,方便对数据进行处理和维护。

  • 消息格式支持

    支持多种消息格式,如 PLAIN、DEBEZIUM、CANAL、MAXWELL、UPSERT 等,便于与不同系统进行数据交互。

通过这些功能,RisingWave 在数据摄入环节能够灵活、高效地处理各种数据源的数据,并提供方便的数据管理和操作功能。

(2)数据选择和清洗

在 RisingWave 中,丰富的 SQL 函数可以帮助用户轻松定义数据选择和清洗的逻辑,同时通过物化视图(MaterializedView)构建特征工程的 Streaming Pipeline。

基于 SQL 进行数据选择和清洗

  • 离散化(Categorization)

    可以使用 SQL 语句将数据离散化到多个桶中。例如,根据一定的条件将数据划分到不同的类别。

  • 异常值处理(Filtering)

    通过 WHERE 条件来处理异常值。例如,筛选出符合特定范围的数据,排除异常数据。

  • 去重(Distinct On)

    使用 DISTINCT ON 语句可以对指定列的数据进行去重操作,只保留一条记录。

  • 缺失值处理(Coalescing)

    利用 SQL 函数(如 LAG)来填补缺失值,使缺失值变为上一个有效值。

基于物化视图构建 Pipeline

物化视图是一个增量实时维护流处理结果的抽象。当上游数据到来时,物化视图会自动、实时、同步地增量维护流处理的结果。

  • 支持 MV - on - MV 构建层级化的流处理管道,可以堆叠物化视图来构建多层级的流处理流程。

  • 物化视图支持丰富的 SQL 语法,包括 JOIN、窗口函数、子查询、分组等,还支持高级的流处理特性如 watermark,以及半结构化数据的处理函数。

  • 物化视图的结果是实时可查询的,用户可以通过 SQL 查询来获取物化视图的结果,方便进行数据验证和调试。

SQL 即流处理

RisingWave 中的 SQL 即流处理具有诸多优势。它基于 SQL 构建流作业,具备丰富的查询优化功能,如列裁剪、Filter 下推等。还支持子查询解关联、Join 重排序等操作,能够将用户编写的 SQL 优化成高效的分布式流作业,方便用户操作。

(3)特征构建

特征构建是实时特征工程的关键环节,下面我们从一些常用特征出发,看一下如何通过 RisingWave 进行特征构建

聚合特征和 Over 窗口计算

  • 通过 CREATE MATERIALIZEDVIEW 语句实现,例如计算用户最近 30 天行为聚合统计,从清洗后的数据表(如 cleaned_events)中筛选出特定时间范围内(NOW() - INTERVAL'30 DAYS'到NOW())的数据,按用户 ID(user_id)和事件类型(event_type)进行分组,计算访问次数(COUNT())和最后访问时间(MAX(event_timestamp))。还可进一步计算如用户过去 30 天最常浏览的 Top2 商品类别,先按用户 ID 分区并按访问次数降序排序,然后选择排名前 2 的类别。

窗口特征

  • Hop Window 和 Tumble Window:如创建 2 分钟 hop 窗口聚合特征,从数据源(如 taxi_trips)中,以 completed_at 为时间字段,按 2 分钟间隔进行窗口聚合,计算行程数量(count(trip_id))和总距离(sum(distance))。同样,对于 2 分钟 tumble 窗口聚合特征,使用 TUMBLE 函数并设置相应参数实现。这些窗口计算为时间序列数据的分析提供了灵活的方式。

  • Session Window 与 Watermark:在源头表(如 user_views)上定义 5 分钟间隔的 watermark,用于处理乱序数据。然后创建 5 分钟 session 窗口聚合特征,按用户 ID 分区,以 viewed_at 为时间字段,计算每个会话的起始时间(first_value(viewed_at))和结束时间(last_value(viewed_at))。session 窗口能有效捕捉用户在一段时间内的连续行为,对于分析用户行为模式非常有用。

实时多流 Join

  • Inner Join 示例

    计算用户过去一天内浏览的商品种类分布,通过 CREATEMATERIALIZED VIEW 将 user_clicks 表与 product_metadata 表进行 JOIN 操作,连接条件为 user_clicks.product_id =product_metadata.product_id,筛选出过去一天内的数据(user_clicks.event_time>= NOW() - INTERVAL '1 DAY'),按用户 ID 和商品类别分组,统计各类别浏览次数(COUNT())。

  • Outer Join 应用

    可用于维度特征关联,如将 user_events 表分别与 product_info、store_info 和 user_info 表进行左外连接(LEFT OUTER JOIN),获取更丰富的用户行为相关信息,包括产品、店铺和用户自身的详细信息,为后续分析提供多维度数据。

  • Window Join 功能

    实现窗口特征拼接,例如将两个以 completed_at 为时间字段、2 分钟间隔的窗口(TUMBLE (taxi_trips, completed_at,INTERVAL '2 MINUTES')和 TUMBLE (taxi_fare, completed_at,INTERVAL '2 MINUTES'))进行连接,连接条件为行程 ID(trip_id)和窗口起始时间(window_start)相等,按窗口起始时间排序,从而整合行程和费用相关的窗口特征,为分析出租车业务数据提供了全面的视角。

实时多流 Join 是 RisingWave 的一个高亮特性,除了上面介绍的 Regular Join 和 Interval Join,还支持 Temporal Join,以及基于 Watermark 的 Windows Join。多流 Join 是流处理中的一个难点,而 RisingWave 凭借其架构优势和丰富的优化,让用户在不感知调度和实现细节的情况下,可以轻松通过 SQL 构建包含多流 Join 的实时特征。

高效状态管理

  • 状态过期清理

    基于 DynamicFilter 算子实现,能够生成正确强一致的流变更和存储 delete tombstone,确保状态存储和 SQL 语义完全一致。在处理如用户最近 30 天行为聚合统计等特征构建时,自动管理状态的过期,避免无效数据占用存储空间,保证数据的时效性和准确性。

  • 长周期大状态处理

    算子状态持久化在对象存储,无单机状态上限。

    基于存算分离架构可实现秒级扩缩容。

    自研云原生 LSM 存储引擎。

在 RisingWave 中做了大量工作去优化状态远端存储带来的延迟。通过多级缓存机制,用户可以根据实际场景在性能与成本间做出权衡。

  • 内部状态 SQL 可查

    流算子内部状态抽象成关系型 StateTable。

    可以通过 SHOW INTERNALTABLES 查看算子内部状态表,也可以通过 SQL 查询。

    适用于排查线上数据问题、优化流作业 SQL、学习流算子的状态管理制等场景。

  • 状态复用

特征工程中,Source 数据清洗后的原始数据可以会物化成 MV,基于这些 MV 又可以创建不同的下游 MV,MV 之间还可以 join,这样分层构建流作业,天然支持状态复用。Source Table 支持 DML 进行数据订正,订正引起的变更会自动地同步到各个下游。

UDF

支持通过 CREATE FUNCTION 和 CREATE AGGREGATE 方式定义 UDF。

(4)Feature Serving

在 RisingWave中,Feature Serving 是实时特征工程的重要组成部分,提供了强大的功能用于特征查询、数据分发和服务优化。

查询与结果一致性

  • 可查询性

    Materialized View 和 Table 均可查询,支持 Batch Query 和 Streaming Query。用户可以通过 SELECT 语句直接查询物化视图(如 user_feature)获取特征数据,例如查询特定用户 ID(user_id = 15213)的特征。这种查询方式方便快捷,能够满足不同场景下对特征数据的获取需求。

  • 结果一致性与调试回溯

    Streaming 和 Batch Query 结果一致,这一特性使得用户在开发和调试过程中更加便捷。用户在创建物化视图前可以先运行 Batch Query 来查看结果是否符合预期,进行数据验证和逻辑调试。如果发现问题,可以方便地回溯和排查,因为两种查询方式的结果具有一致性,保证了数据的可靠性和可追溯性。

  • 支持创建索引加速 Serving 查询

索引加速查询

  • 索引创建与应用

    支持在 Materialized View 和 Table 上创建索引来加速 Serving 查询。用户可以在 timestamp 列创建索引(如 CREATE INDEX idx_timestampon user_feature(timestamp)),然后在查询时利用该索引加速对 timestamp 列的范围查询(如 SELECT  FROM user_featureWHERE timestamp

  • 索引特性支持

    支持指定 Include 列、Distributed 列,还支持表达式索引。例如,在 customers 表上创建索引加速点查(CREATE INDEX idx_c_phone on customer(c_phone)),在 orders 表上创建索引加速 JOIN 操作(CREATE INDEX idx_o_custkey ON orders(o_custkey)),以及在包含 JSONB 类型列的表上创建表达式索引。这些丰富的索引特性为优化查询提供了多种选择,适应不同的数据结构和查询需求。

隔离与伸缩性

  • 隔离 Streaming 与 Serving

    支持隔离 Streaming 与 Serving,允许独立伸缩。这意味着用户可以根据实际需求分别调整 Streaming 和 Serving 的资源配置,优化系统性能。例如,在高并发查询场景下,可以为 Serving 分配更多的计算资源以满足查询需求,而不会影响 Streaming 的实时数据处理能力。

  • 资源优化与灵活性

    通过独立伸缩,用户可以更好地平衡系统资源的利用,提高系统的整体效率和稳定性。无论是处理大规模实时数据的摄入和处理(Streaming),还是应对高并发的特征查询(Serving),都能够灵活配置资源,确保系统在不同负载下的良好性能表现。

数据分发到下游系统

支持将数据变更 Sink 到下游系统。

  • Sink 功能与支持的系统

    通过 Sink 可以实时将数据发送到多种下游系统,支持的 Connector 包括 Redis、Kafka、JDBC、Clickhouse、StarRocks、Doris、ElasticSearch、Cassandra、File、Iceberg 等。用户可以根据实际业务需求选择合适的下游系统进行数据分发,实现数据的进一步处理和分析。

  • 数据格式与输入源

    支持多种数据格式,如 APPEND_ONLY、UPSERT、DEBEZIUM 等。Sink 的输入可以是 Table/Materialized View,也可以是 SQL query。

支持 Subscription 订阅变更

支持 Python-SDK 执行 SQL 和订阅变更

让我们再来整体回顾一下 RisingWave 在特征工程各环节起到的助力作用。首先是数据摄入,利用 RisingWave 可以便捷地导入不同数据源;接下来是数据选择和清洗,基于 SQL 和 UDF,利用物化视图分层构建流处理 pipeline;特征构建完成后,可以用 SQL 或 Python 进行特征查询;最后,可以采用 push-based 也就是 sink 的方式将变更输出到下游,也可以采用 pull-based subscribe 的方式获取变更。

03

RisingWave 其他使用场景

1. 实时监控告警

用户借助 RisingWave 实时处理数据,一旦监测到如设备故障等异常情况,便能迅速发出告警,实现自动修复或及时通知相关人员处理。

2. 流表实时打宽

当上游存在多个不同数据源的数据表时,RisingWave 可将这些表整合打宽成一张大宽表,以便在数据库中生成报表或进行深入分析,为决策提供全面的数据支持。

3. 规则引擎

用户通过 SQL 定义规则,利用其与 PostgreSQL 协议 的兼容性,结合如 Superset 等 BI 工具,可直观展示和分析数据,依据规则对数据进行处理和判断,如在金融交易中检测异常交易行为。

4. 实时数据市场

不同部门利用 RisingWave 构建物化视图,维护数据的可见性与权限。借助 dbt 工具,清晰管理数据血缘,保障数据质量与可追溯性,促进部门间高效的数据协作与共享。

04

RisingWave 2.0 更新内容

RisingWave 2.0 作为最新发布的版本,带来了诸多重要更新。

首先,新增 Premium 版本,专为自部署集群打造,提供企业级支持,有力保障自部署时的稳定性与性能表现。同时,RisingWave 的 Cloud 版本在应用性方面持续增强,尤其在 2.0 版本中,针对 Streaming 和 Batch 的统一支持进行了显著改进。例如,对 Batch Source、Batch Sink 以及 Batch Query 均进行了优化,提升了批量数据处理的效率与性能。

此外,该版本实现了自动的 SchemaChange 和自动的 Schema Mapping 功能。这意味着当上游数据存在 Schema 时,用户导入数据无需手动编写 Schema,并且上游数据列的增减操作能够自动同步至 RisingWave 中,极大地简化了数据管理流程。同时针对创建 MV 时回填历史数据这一资源消耗大且一次性的操作提供了进一步地的优化,优化了数据处理的完整性和效率。

RisingWave 2.0 通过这些更新,致力于为用户提供更优质、高效、便捷的服务,期待用户深入了解并反馈使用体验,共同推动产品的持续优化。

以上就是本次分享的内容,谢谢大家。

分享嘉宾

INTRODUCTION

Patrick Huang

RisingWave Labs

VP of Engineering

RisingWave Labs -> VP of Engineering (DB Kernel & Storage)

Wechat Infra -> Staff Engineer (Large-scale Online Storage)

LinkedIn Infra -> Senior Engineer (Kafka & Next-gen Streaming Platform)

CMU DB Labs -> Research Assistant (Peloton DBMS)

财经自媒体联盟更多自媒体作者

新浪首页 语音播报 相关新闻 返回顶部