Курнат Андрей
57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
// Copyright 2020 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/internal/uuid"
|
|
"github.com/open-policy-agent/opa/v1/ast"
|
|
"github.com/open-policy-agent/opa/v1/topdown/builtins"
|
|
)
|
|
|
|
type uuidCachingKey string
|
|
|
|
func builtinUUIDRFC4122(bctx BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
|
|
|
|
var key = uuidCachingKey(operands[0].Value.String())
|
|
|
|
val, ok := bctx.Cache.Get(key)
|
|
if ok {
|
|
return iter(val.(*ast.Term))
|
|
}
|
|
|
|
s, err := uuid.New(bctx.Seed)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
result := ast.StringTerm(s)
|
|
bctx.Cache.Put(key, result)
|
|
|
|
return iter(result)
|
|
}
|
|
|
|
func builtinUUIDParse(_ BuiltinContext, operands []*ast.Term, iter func(term *ast.Term) error) error {
|
|
str, err := builtins.StringOperand(operands[0].Value, 1)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
parsed, err := uuid.Parse(string(str))
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
val, err := ast.InterfaceToValue(parsed)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return iter(ast.NewTerm(val))
|
|
}
|
|
|
|
func init() {
|
|
RegisterBuiltinFunc(ast.UUIDRFC4122.Name, builtinUUIDRFC4122)
|
|
RegisterBuiltinFunc(ast.UUIDParse.Name, builtinUUIDParse)
|
|
}
|