fix test err

fix
risingwavelabs · Oct 23, 2023 · ed5498a · ed5498a
1 parent a3fcb92
commit ed5498a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 12 deletions.
diff --git a/proto/plan_common.proto b/proto/plan_common.proto
@@ -106,6 +106,7 @@ enum EncodeType {
   ENCODE_TYPE_PROTOBUF = 4;
   ENCODE_TYPE_JSON = 5;
   ENCODE_TYPE_BYTES = 6;
+  ENCODE_TYPE_TEMPLATE = 7;
 }
 
 enum RowFormatType {

diff --git a/src/connector/src/sink/formatter/mod.rs b/src/connector/src/sink/formatter/mod.rs
@@ -116,11 +116,9 @@ impl SinkFormatterImpl {
                         Ok(SinkFormatterImpl::AppendOnlyProto(formatter))
                     }
                     SinkEncode::Avro => err_unsupported(),
-                    SinkEncode::Template => {
-                        return Err(SinkError::Config(anyhow!(
-                            "Template only support with redis sink"
-                        )))
-                    }
+                    SinkEncode::Template => Err(SinkError::Config(anyhow!(
+                        "Template only support with redis sink"
+                    ))),
                 }
             }
             SinkFormat::Debezium => {
@@ -227,11 +225,9 @@ impl SinkFormatterImpl {
                     ))),
                 }
             }
-            _ => {
-                return Err(SinkError::Config(anyhow!(
-                    "Redis sink only support Json and Template"
-                )))
-            }
+            _ => Err(SinkError::Config(anyhow!(
+                "Redis sink only support Json and Template"
+            ))),
         }
     }
 }

diff --git a/src/connector/src/sink/redis.rs b/src/connector/src/sink/redis.rs
@@ -71,7 +71,7 @@ pub struct RedisSink {
     format_desc: SinkFormatDesc,
 }
 
-fn check_string_format(format: &String, set: &HashSet<String>) -> Result<()> {
+fn check_string_format(format: &str, set: &HashSet<String>) -> Result<()> {
     // We will check if the string inside {} corresponds to a column name in rw.
     // In other words, the content within {} should exclusively consist of column names from rw,
     // which means '{{column_name}}' or '{{column_name1},{column_name2}}' would be incorrect.
@@ -380,7 +380,7 @@ mod test {
         let format_desc = SinkFormatDesc {
             format: SinkFormat::AppendOnly,
             encode: SinkEncode::Template,
-            options: BTreeMap::default(),
+            options: btree_map,
         };
 
         let mut redis_sink_writer = RedisSinkWriter::mock(schema, vec![0], &format_desc).unwrap();