fix: add validation when json format has unstructured flag enabled (A…

…rroyoSystems#735)
jrcribb · Sep 5, 2024 · 4741624 · 4741624
1 parent abbde01
commit 4741624
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/crates/arroyo-rpc/src/api_types/connections.rs b/crates/arroyo-rpc/src/api_types/connections.rs
@@ -283,6 +283,16 @@ impl ConnectionSchema {
                     bail!("raw_string format requires a schema with a single field called `value` of type TEXT");
                 }
             }
+            Some(Format::Json(json_format)) => {
+                if json_format.unstructured
+                    && (self.fields.len() != 1
+                        || self.fields.first().unwrap().field_type.r#type
+                            != FieldType::Primitive(PrimitiveType::Json)
+                        || self.fields.first().unwrap().field_name != "value")
+                {
+                    bail!("json format with unstructured flag enabled requires a schema with a single field called `value` of type JSON");
+                }
+            }
             _ => {
                 // Right now only RawString has checks, but we may add checks for other formats in the future
             }