ir

Documentation

Overview

Package ir defines an intermediate representation (IR) for Rego.

The IR specifies an imperative execution model for Rego policies similar to a query plan in traditional databases.

Index

• func Pretty(w io.Writer, x interface{})
• func Walk(vis Visitor, x interface{}) error
• type ArrayAppendStmt
• type AssignBooleanStmt
• type AssignIntStmt
• type AssignVarOnceStmt
• type AssignVarStmt
• type Block
  • func (a *Block) String() string
• type BlockStmt
  • func (a *BlockStmt) String() string
• type BooleanConst
• type BreakStmt
• type BuiltinFunc
• type CallStmt
• type Const
• type DotStmt
• type EqualStmt
• type FloatConst
• type Func
  • func (a *Func) String() string
• type Funcs
  • func (a *Funcs) String() string
• type GreaterThanEqualStmt
• type GreaterThanStmt
• type IntConst
• type IsArrayStmt
• type IsDefinedStmt
• type IsObjectStmt
• type IsUndefinedStmt
• type LenStmt
• type LessThanEqualStmt
• type LessThanStmt
• type Local
• type MakeArrayStmt
• type MakeBooleanStmt
• type MakeNullStmt
• type MakeNumberFloatStmt
• type MakeNumberIntStmt
• type MakeNumberRefStmt
• type MakeObjectStmt
• type MakeSetStmt
• type MakeStringStmt
• type NotEqualStmt
• type NotStmt
• type NullConst
• type ObjectInsertOnceStmt
• type ObjectInsertStmt
• type ObjectMergeStmt
• type Plan
  • func (a *Plan) String() string
• type Policy
  • func (a *Policy) String() string
• type ResultSetAdd
• type ReturnLocalStmt
• type ScanStmt
• type SetAddStmt
• type Static
  • func (a *Static) String() string
• type Stmt
• type StringConst
• type Visitor
• type WithStmt

Functions

func Pretty

func Pretty(w io.Writer, x interface{})

Pretty writes a human-readable representation of an IR object to w.

func Walk

func Walk(vis Visitor, x interface{}) error

Walk invokes the visitor for nodes under x.

Types

type ArrayAppendStmt

type ArrayAppendStmt struct {
	Value Local
	Array Local
}

ArrayAppendStmt represents a dynamic append operation of a value onto an array.

type AssignBooleanStmt

type AssignBooleanStmt struct {
	Value  bool
	Target Local
}

AssignBooleanStmt represents an assignment of a boolean value to a local variable.

type AssignIntStmt

type AssignIntStmt struct {
	Value  int64
	Target Local
}

AssignIntStmt represents an assignment of an integer value to a local variable.

type AssignVarOnceStmt

type AssignVarOnceStmt struct {
	Target Local
	Source Local
}

AssignVarOnceStmt represents an assignment of one local variable to another. If the target is defined, execution aborts with a conflict error.

TODO(tsandall): is there a better name for this?

type AssignVarStmt

type AssignVarStmt struct {
	Source Local
	Target Local
}

AssignVarStmt represents an assignment of one local variable to another.

type Block

type Block struct {
	Stmts []Stmt
}

Block represents an ordered sequence of statements to execute. Blocks are executed until a return statement is encountered, a statement is undefined, or there are no more statements. If all statements are defined but no return statement is encountered, the block is undefined.

func (*Block) String

func (a *Block) String() string

type BlockStmt

type BlockStmt struct {
	Blocks []*Block
}

BlockStmt represents a nested block. Nested blocks and break statements can be used to short-circuit execution.

func (*BlockStmt) String

func (a *BlockStmt) String() string

type BooleanConst

type BooleanConst struct {
	Value bool
}

BooleanConst represents a boolean value.

type BreakStmt

type BreakStmt struct {
	Index uint32
}

BreakStmt represents a jump out of the current block. The index specifies how many blocks to jump starting from zero (the current block). Execution will continue from the end of the block that is jumped to.

type BuiltinFunc

type BuiltinFunc struct {
	Name string
}

BuiltinFunc represents a built-in function that may be required by the policy.

type CallStmt

type CallStmt struct {
	Func   string
	Args   []Local
	Result Local
}

CallStmt represents a named function call. The result should be stored in the result local.

type Const

type Const interface {
	// contains filtered or unexported methods
}

Const represents a constant value from the policy.

type DotStmt

type DotStmt struct {
	Source Local
	Key    Local
	Target Local
}

DotStmt represents a lookup operation on a value (e.g., array, object, etc.) The source of a DotStmt may be a scalar value in which case the statement will be undefined.

type EqualStmt

type EqualStmt struct {
	A Local
	B Local
}

EqualStmt represents an value-equality check of two local variables.

type FloatConst

type FloatConst struct {
	Value float64
}

FloatConst represents a floating-point constant.

type Func

type Func struct {
	Name   string
	Params []Local
	Return Local
	Blocks []*Block // TODO(tsandall): should this be a plan?
}

Func represents a named plan (function) that can be invoked. Functions accept one or more parameters and return a value. By convention, the input document and data documents are always passed as the first and second arguments (respectively).

func (*Func) String

func (a *Func) String() string

type Funcs

type Funcs struct {
	Funcs []*Func
}

Funcs represents a collection of planned functions to include in the policy.

func (*Funcs) String

func (a *Funcs) String() string

type GreaterThanEqualStmt

type GreaterThanEqualStmt struct {
	A Local
	B Local
}

GreaterThanEqualStmt represents a >= check of two local variables.

type GreaterThanStmt

type GreaterThanStmt struct {
	A Local
	B Local
}

GreaterThanStmt represents a > check of two local variables.

type IntConst

type IntConst struct {
	Value int64
}

IntConst represents an integer constant.

type IsArrayStmt

type IsArrayStmt struct {
	Source Local
}

IsArrayStmt represents a dynamic type check on a local variable.

type IsDefinedStmt

type IsDefinedStmt struct {
	Source Local
}

IsDefinedStmt represents a check of whether a local variable is defined.

type IsObjectStmt

type IsObjectStmt struct {
	Source Local
}

IsObjectStmt represents a dynamic type check on a local variable.

type IsUndefinedStmt

type IsUndefinedStmt struct {
	Source Local
}

IsUndefinedStmt represents a check of whether local variable is undefined.

type LenStmt

type LenStmt struct {
	Source Local
	Target Local
}

LenStmt represents a length() operation on a local variable. The result is stored in the target local variable.

type LessThanEqualStmt

type LessThanEqualStmt struct {
	A Local
	B Local
}

LessThanEqualStmt represents a <= check of two local variables.

type LessThanStmt

type LessThanStmt struct {
	A Local
	B Local
}

LessThanStmt represents a < check of two local variables.

type Local

type Local int

Local represents a plan-scoped variable.

TODO(tsandall): should this be int32 for safety?

const (
	// Input is the local variable that refers to the global input document.
	Input Local = iota

	// Data is the local variable that refers to the global data document.
	Data

	// Unused is the free local variable that can be allocated in a plan.
	Unused
)

type MakeArrayStmt

type MakeArrayStmt struct {
	Capacity int32
	Target   Local
}

MakeArrayStmt constructs a local variable that refers to an array value.

type MakeBooleanStmt

type MakeBooleanStmt struct {
	Value  bool
	Target Local
}

MakeBooleanStmt constructs a local variable that refers to a boolean value.

type MakeNullStmt

type MakeNullStmt struct {
	Target Local
}

MakeNullStmt constructs a local variable that refers to a null value.

type MakeNumberFloatStmt

type MakeNumberFloatStmt struct {
	Value  float64
	Target Local
}

MakeNumberFloatStmt constructs a local variable that refers to a floating-point number value.

type MakeNumberIntStmt

type MakeNumberIntStmt struct {
	Value  int64
	Target Local
}

MakeNumberIntStmt constructs a local variable that refers to an integer value.

type MakeNumberRefStmt

type MakeNumberRefStmt struct {
	Index  int
	Target Local
}

MakeNumberRefStmt constructs a local variable that refers to a number stored as a string.

type MakeObjectStmt

type MakeObjectStmt struct {
	Target Local
}

MakeObjectStmt constructs a local variable that refers to an object value.

type MakeSetStmt

type MakeSetStmt struct {
	Target Local
}

MakeSetStmt constructs a local variable that refers to a set value.

type MakeStringStmt

type MakeStringStmt struct {
	Index  int
	Target Local
}

MakeStringStmt constructs a local variable that refers to a string constant.

type NotEqualStmt

type NotEqualStmt struct {
	A Local
	B Local
}

NotEqualStmt represents a != check of two local variables.

type NotStmt

type NotStmt struct {
	Block *Block
}

NotStmt represents a negated statement.

type NullConst

type NullConst struct{}

NullConst represents a null value.

type ObjectInsertOnceStmt

type ObjectInsertOnceStmt struct {
	Key    Local
	Value  Local
	Object Local
}

ObjectInsertOnceStmt represents a dynamic insert operation of a key/value pair into an object. If the key already exists and the value differs, execution aborts with a conflict error.

type ObjectInsertStmt

type ObjectInsertStmt struct {
	Key    Local
	Value  Local
	Object Local
}

ObjectInsertStmt represents a dynamic insert operation of a key/value pair into an object.

type ObjectMergeStmt

type ObjectMergeStmt struct {
	A      Local
	B      Local
	Target Local
}

ObjectMergeStmt performs a recursive merge of two object values. If either of the locals refer to non-object values this operation will abort with a conflict error. Overlapping object keys are merged recursively.

type Plan

type Plan struct {
	Blocks []*Block
}

Plan represents an ordered series of blocks to execute. Plan execution stops when a return statement is reached. Blocks are executed in-order.

func (*Plan) String

func (a *Plan) String() string

type Policy

type Policy struct {
	Static *Static
	Plan   *Plan
	Funcs  *Funcs
}

Policy represents a planned policy query.

func (*Policy) String

func (a *Policy) String() string

type ResultSetAdd

type ResultSetAdd struct {
	Value Local
}

ResultSetAdd adds a value into the result set returned by the query plan.

type ReturnLocalStmt

type ReturnLocalStmt struct {
	Source Local
}

ReturnLocalStmt represents a return statement that yields a local value.

type ScanStmt

type ScanStmt struct {
	Source Local
	Key    Local
	Value  Local
	Block  *Block
}

ScanStmt represents a linear scan over a composite value. The source may be a scalar in which case the block will never execute.

type SetAddStmt

type SetAddStmt struct {
	Value Local
	Set   Local
}

SetAddStmt represents a dynamic add operation of an element into a set.

type Static

type Static struct {
	Strings      []*StringConst
	BuiltinFuncs []*BuiltinFunc
}

Static represents a static data segment that is indexed into by the policy.

func (*Static) String

func (a *Static) String() string

type Stmt

type Stmt interface {
}

Stmt represents an operation (e.g., comparison, loop, dot, etc.) to execute.

type StringConst

type StringConst struct {
	Value string
}

StringConst represents a string value.

type Visitor

type Visitor interface {
	Before(x interface{})
	Visit(x interface{}) (Visitor, error)
	After(x interface{})
}

Visitor defines the interface for visiting IR nodes.

type WithStmt

type WithStmt struct {
	Local Local
	Path  []int
	Value Local
	Block *Block
}

WithStmt replaces the Local or a portion of the document referred to by the Local with the Value and executes the contained block. If the Path is non-empty, the Value is upserted into the Local. If the intermediate nodes in the Local referred to by the Path do not exist, they will be created. When the WithStmt finishes the Local is reset to it's original value.