netsting_variables_across_cells (byzer-org#1650)

ckeys · Dec 31, 2021 · 05a7b1c · 05a7b1c
1 parent 7e4ce80
commit 05a7b1c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/streamingpro-core/src/main/java/tech/mlsql/dsl/adaptor/SetAdaptor.scala b/streamingpro-core/src/main/java/tech/mlsql/dsl/adaptor/SetAdaptor.scala
@@ -159,7 +159,7 @@ class SetAdaptor(scriptSQLExecListener: ScriptSQLExecListener, stage: Stage.stag
         if (mergedValue != v) {
           scriptSQLExecListener.addEnv(k, mergedValue)
           if (scope == "session"){
-            scriptSQLExecListener.addSessionEnv(key, value,visibility.mkString(","))
+            scriptSQLExecListener.addSessionEnv(key, mergedValue,visibility.mkString(","))
           }
         }
     }

diff --git a/streamingpro-it/src/test/resources/sql/simple/20_nesting_variables_across_cells_test.mlsql b/streamingpro-it/src/test/resources/sql/simple/20_nesting_variables_across_cells_test.mlsql
@@ -0,0 +1,13 @@
+--%comparator=tech.mlsql.it.IgnoreResultComparator
+-- first cell
+
+set table_name="T_TableName" where scope="session";
+set ods_table_name="ODS_${table_name}" where scope="session";
+select "${ods_table_name}" as name as `${ods_table_name}`;
+
+-- second cell
+
+select name from `${ods_table_name}` as result;
+
+-- make sure The variable name should be ODS_T_TableName.
+!assert result ''':name=="ODS_T_TableName"'''  "The variable name should be ODS_T_TableName";