Add SQL function for creating an array
This commit is contained in:
parent
8076afd414
commit
cacdf6dd49
4 changed files with 69 additions and 0 deletions
|
@ -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
27
util/fncs/array.go
Normal 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
39
util/fncs/array_test.go
Normal 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})
|
||||
})
|
||||
|
||||
}
|
|
@ -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":
|
||||
|
|
Loading…
Reference in a new issue