switch to javascript udf

Signed-off-by: Runji Wang <[email protected]>
risingwavelabs · Feb 8, 2024 · 0eec285 · 0eec285
1 parent d8c2c38
commit 0eec285
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 12 deletions.
diff --git a/ci/scripts/sql/nexmark/q14.sql b/ci/scripts/sql/nexmark/q14.sql
@@ -1,10 +1,14 @@
 -- noinspection SqlNoDataSourceInspectionForFile
 -- noinspection SqlResolveForFile
 
-CREATE FUNCTION count_char(s varchar, c varchar) RETURNS int LANGUAGE rust AS $$
-  fn count_char(s: &str, c: &str) -> i32 {
-      s.matches(c).count() as i32
-  }
+CREATE FUNCTION count_char(s varchar, c varchar) RETURNS int LANGUAGE javascript AS $$
+    var count = 0;
+    for (var cc of s) {
+        if (cc === c) {
+            count++;
+        }
+    }
+    return count;
 $$;
 
 CREATE SINK nexmark_q14 AS

diff --git a/e2e_test/streaming/nexmark/sinks/q14.slt.part b/e2e_test/streaming/nexmark/sinks/q14.slt.part
@@ -1,8 +1,12 @@
 statement ok
-CREATE FUNCTION count_char(s varchar, c varchar) RETURNS int LANGUAGE rust AS $$
-  fn count_char(s: &str, c: &str) -> i32 {
-      s.matches(c).count() as i32
-  }
+CREATE FUNCTION count_char(s varchar, c varchar) RETURNS int LANGUAGE javascript AS $$
+    var count = 0;
+    for (var cc of s) {
+        if (cc === c) {
+            count++;
+        }
+    }
+    return count;
 $$;
 
 statement ok

diff --git a/e2e_test/streaming/nexmark/views/q14.slt.part b/e2e_test/streaming/nexmark/views/q14.slt.part
@@ -1,8 +1,12 @@
 statement ok
-CREATE FUNCTION count_char(s varchar, c varchar) RETURNS int LANGUAGE rust AS $$
-  fn count_char(s: &str, c: &str) -> i32 {
-      s.matches(c).count() as i32
-  }
+CREATE FUNCTION count_char(s varchar, c varchar) RETURNS int LANGUAGE javascript AS $$
+    var count = 0;
+    for (var cc of s) {
+        if (cc === c) {
+            count++;
+        }
+    }
+    return count;
 $$;
 
 statement ok