mirror of
https://github.com/open-policy-agent/opa.git
synced 2026-08-12 19:32:48 -06:00
Anders Eknert
20fe70e321
This PR adds interning of strings representing common integer values, which greatly speeds up "to string" operations on numbers, and updates some built-ins commonly used for this to make use of interned values where possible. This is "light" version of a previous PR that did this more aggressively, but also came with more caveats. Importantly, interning of new strings is now never done at "runtime", but only allowed at init time. The API for interning is marked experimental and should not relied upon by anyone who expects a stable API. Signed-off-by: Anders Eknert <anders@styra.com>
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)
|
|
}
|