Add support to access field of a record (finos#74)

Add support to access field of a record in the way .field
Snowflake-Labs · Dec 13, 2023 · 5c1bca0 · 5c1bca0
1 parent dd1a443
commit 5c1bca0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/Morphir/Snowpark/Backend.elm b/src/Morphir/Snowpark/Backend.elm
@@ -223,6 +223,8 @@ mapValue value ctx =
             mapIfThenElse condition thenExpr elseExpr ctx
         LetDefinition _ name definition body ->
             mapLetDefinition name definition body ctx
+        FieldFunction _ [name] ->
+            Constants.applySnowparkFunc "col" [(Scala.Literal (Scala.StringLit name))]
         _ ->
             Scala.Literal (Scala.StringLit ("Unsupported element"))
 

diff --git a/src/Morphir/Snowpark/MapFunctionsMapping.elm b/src/Morphir/Snowpark/MapFunctionsMapping.elm
@@ -247,8 +247,10 @@ processLambdaWithRecordBody functionExpr ctx mapValue =
                 Nothing
         ValueIR.Lambda (TypeIR.Function _ _  returnType) (ValueIR.AsPattern _ _ _) expr ->
              if isBasicType returnType then
-               Just [ Scala.ArgValue Nothing <| mapValue expr ctx ]
+                Just [ Scala.ArgValue Nothing <| mapValue expr ctx ]
              else  
                 Nothing
+        ValueIR.FieldFunction _ _ ->
+            Just [Scala.ArgValue Nothing (mapValue functionExpr ctx)]
         _ ->
             Nothing