Signed-off-by: hongming <talonwan@yunify.com>
This commit is contained in:
hongming
2020-03-19 22:44:05 +08:00
parent 23f6be88c6
commit 9769357005
332 changed files with 69808 additions and 4129 deletions

View File

@@ -0,0 +1,75 @@
// Copyright 2018 The OPA Authors. All rights reserved.
// Use of this source code is governed by an Apache2
// license that can be found in the LICENSE file.
package topdown
import (
"github.com/open-policy-agent/opa/ast"
"github.com/open-policy-agent/opa/topdown/builtins"
)
func builtinArrayConcat(a, b ast.Value) (ast.Value, error) {
arrA, err := builtins.ArrayOperand(a, 1)
if err != nil {
return nil, err
}
arrB, err := builtins.ArrayOperand(b, 2)
if err != nil {
return nil, err
}
arrC := make(ast.Array, 0, len(arrA)+len(arrB))
for _, elemA := range arrA {
arrC = append(arrC, elemA)
}
for _, elemB := range arrB {
arrC = append(arrC, elemB)
}
return arrC, nil
}
func builtinArraySlice(a, i, j ast.Value) (ast.Value, error) {
arr, err := builtins.ArrayOperand(a, 1)
if err != nil {
return nil, err
}
startIndex, err := builtins.IntOperand(i, 2)
if err != nil {
return nil, err
}
stopIndex, err := builtins.IntOperand(j, 3)
if err != nil {
return nil, err
}
// Return empty array if bounds cannot be clamped sensibly.
if (startIndex >= stopIndex) || (startIndex <= 0 && stopIndex <= 0) {
return arr[0:0], nil
}
// Clamp bounds to avoid out-of-range errors.
if startIndex < 0 {
startIndex = 0
}
if stopIndex > len(arr) {
stopIndex = len(arr)
}
arrb := arr[startIndex:stopIndex]
return arrb, nil
}
func init() {
RegisterFunctionalBuiltin2(ast.ArrayConcat.Name, builtinArrayConcat)
RegisterFunctionalBuiltin3(ast.ArraySlice.Name, builtinArraySlice)
}