chore: support find

undb-io · Nov 1, 2024 · 9185187 · 9185187
1 parent ae474bd
commit 9185187
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/packages/formula/src/formula/formula.registry.ts b/packages/formula/src/formula/formula.registry.ts
@@ -313,3 +313,14 @@ globalFormulaRegistry.register(
     ["SWITCH({{field1}}, 1, 'one', 2, 'two', 3, 'three')", undefined],
   ],
 )
+
+globalFormulaRegistry.register(
+  "FIND",
+  [["string", "string"]],
+  "number",
+  "Returns the position of a substring within a string.",
+  [
+    ["FIND('Hello', 'e')", 1],
+    ["FIND({{field1}}, 'e')", undefined],
+  ],
+)
diff --git a/packages/persistence/src/underlying/underlying-formula.visitor.ts b/packages/persistence/src/underlying/underlying-formula.visitor.ts
@@ -120,6 +120,10 @@ export class UnderlyingFormulaVisitor extends FormulaParserVisitor<string> {
         sql += ` ELSE ${this.visit(defaultValue)} END`
         return `(${sql})`
       })
+      .with("FIND", () => {
+        const args = this.arguments(ctx)
+        return `(INSTR(LOWER(${args[1]}), LOWER(${args[0]})))`
+      })
       .with("ADD", "SUM", () => {
         const fn = this.arguments(ctx).join(" + ")
         return `(${fn})`