mirror of
https://github.com/hanzoai/traces.git
synced 2026-10-30 18:20:41 +00:00
6c89137ee1
Bumps [github.com/expr-lang/expr](https://github.com/expr-lang/expr) from 1.16.9 to 1.17.0. - [Release notes](https://github.com/expr-lang/expr/releases) - [Commits](https://github.com/expr-lang/expr/compare/v1.16.9...v1.17.0) --- updated-dependencies: - dependency-name: github.com/expr-lang/expr dependency-type: indirect ... Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
38 lines
828 B
Go
38 lines
828 B
Go
package optimizer
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
. "github.com/expr-lang/expr/ast"
|
|
)
|
|
|
|
type sumArray struct{}
|
|
|
|
func (*sumArray) Visit(node *Node) {
|
|
if sumBuiltin, ok := (*node).(*BuiltinNode); ok &&
|
|
sumBuiltin.Name == "sum" &&
|
|
len(sumBuiltin.Arguments) == 1 {
|
|
if array, ok := sumBuiltin.Arguments[0].(*ArrayNode); ok &&
|
|
len(array.Nodes) >= 2 {
|
|
patchCopyType(node, sumArrayFold(array))
|
|
}
|
|
}
|
|
}
|
|
|
|
func sumArrayFold(array *ArrayNode) *BinaryNode {
|
|
if len(array.Nodes) > 2 {
|
|
return &BinaryNode{
|
|
Operator: "+",
|
|
Left: array.Nodes[0],
|
|
Right: sumArrayFold(&ArrayNode{Nodes: array.Nodes[1:]}),
|
|
}
|
|
} else if len(array.Nodes) == 2 {
|
|
return &BinaryNode{
|
|
Operator: "+",
|
|
Left: array.Nodes[0],
|
|
Right: array.Nodes[1],
|
|
}
|
|
}
|
|
panic(fmt.Errorf("sumArrayFold: invalid array length %d", len(array.Nodes)))
|
|
}
|