From 8a6de6383143a056926ce0930efe8fc1cfd0a55e Mon Sep 17 00:00:00 2001 From: xxchan Date: Fri, 10 Jan 2025 21:25:01 +0800 Subject: [PATCH] fix Signed-off-by: xxchan --- src/bench/sink_bench/main.rs | 6 +++--- src/meta/service/src/cloud_service.rs | 6 ++---- src/meta/src/stream/source_manager.rs | 2 +- 3 files changed, 6 insertions(+), 8 deletions(-) diff --git a/src/bench/sink_bench/main.rs b/src/bench/sink_bench/main.rs index 445a9ae628ea..1f104cbd2946 100644 --- a/src/bench/sink_bench/main.rs +++ b/src/bench/sink_bench/main.rs @@ -56,10 +56,10 @@ use risingwave_connector::sink::{ SINK_TYPE_APPEND_ONLY, SINK_TYPE_UPSERT, }; use risingwave_connector::source::datagen::{ - DatagenProperties, DatagenSplitEnumerator, DatagenSplitReader, + DatagenProperties, DatagenSplit, DatagenSplitEnumerator, DatagenSplitReader, }; use risingwave_connector::source::{ - Column, DataType, SourceContext, SourceEnumeratorContext, SplitReader, + Column, DataType, SourceContext, SourceEnumeratorContext, SplitEnumerator, SplitReader, }; use risingwave_stream::executor::test_utils::prelude::ColumnDesc; use risingwave_stream::executor::{Barrier, Message, MessageStreamItem, StreamExecutorError}; @@ -303,7 +303,7 @@ impl MockDatagenSource { datagen_split_readers.push( DatagenSplitReader::new( properties.clone(), - vec![splits], + vec![DatagenSplit::try_from(splits).unwrap()], parser_config.clone(), SourceContext::dummy().into(), Some(source_schema.clone()), diff --git a/src/meta/service/src/cloud_service.rs b/src/meta/service/src/cloud_service.rs index 9ed72f245a46..e1853db5650f 100644 --- a/src/meta/service/src/cloud_service.rs +++ b/src/meta/service/src/cloud_service.rs @@ -18,10 +18,8 @@ use std::sync::LazyLock; use async_trait::async_trait; use regex::Regex; use risingwave_connector::error::ConnectorResult; -use risingwave_connector::source::{ - ConnectorProperties, SourceEnumeratorContext, SourceProperties, SplitEnumerator, -}; -use risingwave_connector::{dispatch_source_prop, WithOptionsSecResolved}; +use risingwave_connector::source::{ConnectorProperties, SourceEnumeratorContext, SplitEnumerator}; +use risingwave_connector::WithOptionsSecResolved; use risingwave_pb::cloud_service::cloud_service_server::CloudService; use risingwave_pb::cloud_service::rw_cloud_validate_source_response::{Error, ErrorType}; use risingwave_pb::cloud_service::{ diff --git a/src/meta/src/stream/source_manager.rs b/src/meta/src/stream/source_manager.rs index 3c163f6c68ec..761d6b768b05 100644 --- a/src/meta/src/stream/source_manager.rs +++ b/src/meta/src/stream/source_manager.rs @@ -27,7 +27,7 @@ use risingwave_common::metrics::LabelGuardedIntGauge; use risingwave_connector::error::ConnectorResult; use risingwave_connector::source::{ fill_adaptive_split, ConnectorProperties, SourceEnumeratorContext, SourceEnumeratorInfo, - SourceProperties, SplitEnumerator, SplitId, SplitImpl, SplitMetaData, + SplitEnumerator, SplitId, SplitImpl, SplitMetaData, }; use risingwave_connector::WithOptionsSecResolved; use risingwave_meta_model::SourceId;