mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-27 01:33:28 +02:00
86 lines
1.5 KiB
Go
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
|
|
}
|