fix(sink): license SQL Server sink (#17910)

risingwavelabs · Aug 5, 2024 · faf4ecd · faf4ecd
1 parent bf1c1db
commit faf4ecd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/connector/src/sink/sqlserver.rs b/src/connector/src/sink/sqlserver.rs
@@ -144,6 +144,10 @@ impl Sink for SqlServerSink {
     const SINK_NAME: &'static str = SQLSERVER_SINK;
 
     async fn validate(&self) -> Result<()> {
+        risingwave_common::license::Feature::SqlServerSink
+            .check_available()
+            .map_err(|e| anyhow::anyhow!(e))?;
+
         if !self.is_append_only && self.pk_indices.is_empty() {
             return Err(SinkError::Config(anyhow!(
                 "Primary key not defined for upsert SQL Server sink (please define in `primary_key` field)")));

diff --git a/src/license/src/feature.rs b/src/license/src/feature.rs
@@ -47,6 +47,7 @@ macro_rules! for_all_features {
             { TimeTravel,           Paid,       "Query historical data within the retention period."},
             { GlueSchemaRegistry,   Paid,       "Use Schema Registry from AWS Glue rather than Confluent." },
             { SecretManagement,     Paid,       "Secret management." },
+            { SqlServerSink,        Paid,       "Sink data from RisingWave to SQL Server." },
         }
     };
 }