Initial QSfera import
This commit is contained in:
+48
@@ -0,0 +1,48 @@
|
||||
// Copyright 2016 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/v1/ast"
|
||||
|
||||
type compareFunc func(a, b ast.Value) bool
|
||||
|
||||
func compareGreaterThan(a, b ast.Value) bool {
|
||||
return ast.Compare(a, b) > 0
|
||||
}
|
||||
|
||||
func compareGreaterThanEq(a, b ast.Value) bool {
|
||||
return ast.Compare(a, b) >= 0
|
||||
}
|
||||
|
||||
func compareLessThan(a, b ast.Value) bool {
|
||||
return ast.Compare(a, b) < 0
|
||||
}
|
||||
|
||||
func compareLessThanEq(a, b ast.Value) bool {
|
||||
return ast.Compare(a, b) <= 0
|
||||
}
|
||||
|
||||
func compareNotEq(a, b ast.Value) bool {
|
||||
return ast.Compare(a, b) != 0
|
||||
}
|
||||
|
||||
func compareEq(a, b ast.Value) bool {
|
||||
return ast.Compare(a, b) == 0
|
||||
}
|
||||
|
||||
func builtinCompare(cmp compareFunc) BuiltinFunc {
|
||||
return func(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
||||
return iter(ast.InternedTerm(cmp(operands[0].Value, operands[1].Value)))
|
||||
}
|
||||
}
|
||||
|
||||
func init() {
|
||||
RegisterBuiltinFunc(ast.GreaterThan.Name, builtinCompare(compareGreaterThan))
|
||||
RegisterBuiltinFunc(ast.GreaterThanEq.Name, builtinCompare(compareGreaterThanEq))
|
||||
RegisterBuiltinFunc(ast.LessThan.Name, builtinCompare(compareLessThan))
|
||||
RegisterBuiltinFunc(ast.LessThanEq.Name, builtinCompare(compareLessThanEq))
|
||||
RegisterBuiltinFunc(ast.NotEqual.Name, builtinCompare(compareNotEq))
|
||||
RegisterBuiltinFunc(ast.Equal.Name, builtinCompare(compareEq))
|
||||
}
|
||||
Reference in New Issue
Block a user