1
0
mirror of https://github.com/MontFerret/ferret.git synced 2024-12-27 01:33:28 +02:00
ferret/pkg/runtime/expressions/operators/logical.go
2018-09-22 20:28:33 -04:00

86 lines
1.5 KiB
Go

package operators
import (
"context"
"github.com/MontFerret/ferret/pkg/runtime/core"
"github.com/MontFerret/ferret/pkg/runtime/values"
)
type (
LogicalOperatorType int
LogicalOperator struct {
*baseOperator
value LogicalOperatorType
}
)
var (
AndType LogicalOperatorType = 0
OrType LogicalOperatorType = 1
NotType LogicalOperatorType = 2
)
var logicalOperators = map[string]LogicalOperatorType{
"&&": AndType,
"AND": AndType,
"||": OrType,
"OR": OrType,
"NOT": NotType,
}
func NewLogicalOperator(
src core.SourceMap,
left core.Expression,
right core.Expression,
operator string,
) (*LogicalOperator, error) {
op, exists := logicalOperators[operator]
if !exists {
return nil, core.Error(core.ErrInvalidArgument, "value")
}
return &LogicalOperator{
&baseOperator{
src,
left,
right,
},
op,
}, nil
}
func (operator *LogicalOperator) Exec(ctx context.Context, scope *core.Scope) (core.Value, error) {
left, err := operator.left.Exec(ctx, scope)
if err != nil {
return nil, err
}
if operator.value == NotType {
return Not(left, values.None), nil
}
leftBool := values.ToBoolean(left)
if operator.value == AndType && leftBool == values.False {
if left.Type() == core.BooleanType {
return values.False, nil
}
return left, nil
}
if operator.value == OrType && leftBool == values.True {
return left, nil
}
right, err := operator.right.Exec(ctx, scope)
if err != nil {
return nil, err
}
return right, nil
}