function: correctly transform up explode nodes

Signed-off-by: Miguel Molina <miguel@erizocosmi.co>
src-d · Jun 20, 2019 · fc71411 · fc71411
1 parent eaab179
commit fc71411
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/engine_test.go b/engine_test.go
@@ -2481,6 +2481,16 @@ var generatorQueries = []struct {
 			{int64(3), "f", "third"},
 		},
 	},
+	{
+		`SELECT EXPLODE(SPLIT(c, "")) FROM t LIMIT 5`,
+		[]sql.Row{
+			{"f"},
+			{"i"},
+			{"r"},
+			{"s"},
+			{"t"},
+		},
+	},
 	{
 		`SELECT a, EXPLODE(b) AS x, c FROM t WHERE x = 'e'`,
 		[]sql.Row{

diff --git a/sql/expression/function/explode.go b/sql/expression/function/explode.go
@@ -86,7 +86,7 @@ func (e *Generate) String() string {
 
 // TransformUp implements the sql.Expression interface.
 func (e *Generate) TransformUp(f sql.TransformExprFunc) (sql.Expression, error) {
-	c, err := f(e.Child)
+	c, err := e.Child.TransformUp(f)
 	if err != nil {
 		return nil, err
 	}