mirror of
https://github.com/johnkerl/miller.git
synced 2026-01-23 02:14:13 +00:00
a977617797
* address some staticcheck issues * address some staticcheck issues * address some staticcheck issues * address some staticcheck issues
90 lines
2.6 KiB
Go
90 lines
2.6 KiB
Go
// ================================================================
|
|
// This is for things that get us out of statement blocks: break, continue,
|
|
// return.
|
|
// ================================================================
|
|
|
|
package cst
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/johnkerl/miller/internal/pkg/dsl"
|
|
"github.com/johnkerl/miller/internal/pkg/lib"
|
|
"github.com/johnkerl/miller/internal/pkg/runtime"
|
|
)
|
|
|
|
// ----------------------------------------------------------------
|
|
type BreakNode struct {
|
|
}
|
|
|
|
func (root *RootNode) BuildBreakNode(astNode *dsl.ASTNode) (*BreakNode, error) {
|
|
lib.InternalCodingErrorIf(astNode.Type != dsl.NodeTypeBreak)
|
|
lib.InternalCodingErrorIf(astNode.Children == nil)
|
|
lib.InternalCodingErrorIf(len(astNode.Children) != 0)
|
|
|
|
return &BreakNode{}, nil
|
|
}
|
|
|
|
func (node *BreakNode) Execute(state *runtime.State) (*BlockExitPayload, error) {
|
|
return &BlockExitPayload{
|
|
BLOCK_EXIT_BREAK,
|
|
nil,
|
|
}, nil
|
|
}
|
|
|
|
// ----------------------------------------------------------------
|
|
type ContinueNode struct {
|
|
}
|
|
|
|
func (root *RootNode) BuildContinueNode(astNode *dsl.ASTNode) (*ContinueNode, error) {
|
|
lib.InternalCodingErrorIf(astNode.Type != dsl.NodeTypeContinue)
|
|
lib.InternalCodingErrorIf(astNode.Children == nil)
|
|
lib.InternalCodingErrorIf(len(astNode.Children) != 0)
|
|
|
|
return &ContinueNode{}, nil
|
|
}
|
|
|
|
func (node *ContinueNode) Execute(state *runtime.State) (*BlockExitPayload, error) {
|
|
return &BlockExitPayload{
|
|
BLOCK_EXIT_CONTINUE,
|
|
nil,
|
|
}, nil
|
|
}
|
|
|
|
// ----------------------------------------------------------------
|
|
type ReturnNode struct {
|
|
returnValueExpression IEvaluable
|
|
}
|
|
|
|
func (root *RootNode) BuildReturnNode(astNode *dsl.ASTNode) (*ReturnNode, error) {
|
|
lib.InternalCodingErrorIf(astNode.Type != dsl.NodeTypeReturn)
|
|
lib.InternalCodingErrorIf(astNode.Children == nil)
|
|
if len(astNode.Children) == 0 {
|
|
return &ReturnNode{returnValueExpression: nil}, nil
|
|
} else if len(astNode.Children) == 1 {
|
|
returnValueExpression, err := root.BuildEvaluableNode(astNode.Children[0])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &ReturnNode{returnValueExpression: returnValueExpression}, nil
|
|
} else {
|
|
lib.InternalCodingErrorIf(true)
|
|
}
|
|
return nil, fmt.Errorf("internal coding error: statement should not be reached.")
|
|
}
|
|
|
|
func (node *ReturnNode) Execute(state *runtime.State) (*BlockExitPayload, error) {
|
|
if node.returnValueExpression == nil {
|
|
return &BlockExitPayload{
|
|
BLOCK_EXIT_RETURN_VOID,
|
|
nil,
|
|
}, nil
|
|
} else {
|
|
// The return value can be of type MT_ERROR but we do not use Go-level error return here.
|
|
returnValue := node.returnValueExpression.Evaluate(state)
|
|
return &BlockExitPayload{
|
|
BLOCK_EXIT_RETURN_VALUE,
|
|
returnValue.Copy(),
|
|
}, nil
|
|
}
|
|
}
|