Add SQL function for creating an array

This commit is contained in:
Tobie Morgan Hitchcock 2017-12-12 00:54:18 +00:00
parent 8076afd414
commit cacdf6dd49
4 changed files with 69 additions and 0 deletions

View file

@ -98,6 +98,7 @@ var aggrs = map[string]bool{
var funcs = map[string]map[int]interface{}{
"array": {-1: nil},
"batch": {2: nil},
"binary": {1: nil},
"difference": {-1: nil},

27
util/fncs/array.go Normal file
View file

@ -0,0 +1,27 @@
// Copyright © 2016 Abcum Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package fncs
import (
"context"
)
func array(ctx context.Context, args ...interface{}) ([]interface{}, error) {
out := make([]interface{}, len(args))
for i, arg := range args {
out[i] = arg
}
return out, nil
}

39
util/fncs/array_test.go Normal file
View file

@ -0,0 +1,39 @@
// Copyright © 2016 Abcum Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package fncs
import (
"time"
"context"
"testing"
. "github.com/smartystreets/goconvey/convey"
)
func TestArray(t *testing.T) {
var res interface{}
timer := time.Now()
Convey("array() works properly", t, func() {
res, _ = Run(context.Background(), "array", 1, 2, 3)
So(res, ShouldResemble, []interface{}{1, 2, 3})
res, _ = Run(context.Background(), "array", nil, true, "test", timer)
So(res, ShouldResemble, []interface{}{nil, true, "test", timer})
})
}

View file

@ -20,6 +20,8 @@ func Run(ctx context.Context, name string, args ...interface{}) (interface{}, er
switch name {
case "array":
return array(ctx, args...)
case "batch":
return batch(ctx, args...)
case "difference":