Merge pull request #496 from mcarmonaa/fix/uast-extract-returned-type

internal/function: fix uast_extract returned type

Author: ajnavarro

.travis.yml

@@ -23,6 +23,7 @@ before_script:
   - docker run -d --name bblfshd --privileged -p 9432:9432 -v /var/lib/bblfshd:/var/lib/bblfshd bblfsh/bblfshd
   - docker exec -it bblfshd bblfshctl driver install python bblfsh/python-driver
   - docker exec -it bblfshd bblfshctl driver install php bblfsh/php-driver
+  - docker exec -it bblfshd bblfshctl driver install go bblfsh/go-driver
 
 script:
   - make test-coverage codecov

integration_test.go

@@ -230,6 +230,18 @@ func TestIntegration(t *testing.T) {
 				{"e8d3ffab552895c19b9fcf7aa264d277cde33881", "some code in a branch\n", "php", "dbd3641b371024f44d0e469a9c8f5457b0660de1"},
 			},
 		},
+		{
+			`SELECT
+				file_path, array_length(uast_extract(uast(blob_content, language(file_path)), "@type"))
+			FROM
+				files
+			WHERE
+				language(file_path)="Go"
+			LIMIT 1`,
+			[]sql.Row{
+				{"go/example.go", int32(1)},
+			},
+		},
 	}
 
 	runTests := func(t *testing.T) {

internal/function/uast.go

@@ -458,7 +458,7 @@ func (u *UASTExtract) Eval(ctx *sql.Context, row sql.Row) (out interface{}, err
 		return nil, nil
 	}
 
-	extracted := make([][]string, len(nodes))
+	extracted := make([]interface{}, len(nodes))
 	for i, n := range nodes {
 		extracted[i] = extractInfo(n, key)
 	}