go-pkg/expr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

operator-length.go: now returns index+1 when used with iterators; that is the number of iterations

Browse Source
This commit is contained in:
Celestino Amoroso 2024-05-04 01:23:32 +02:00
parent a736bba2c7
commit e6174aca82
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
operator-length.go
View File

@ -17,22 +17,22 @@ func newLengthTerm(tk *Token) (inst *term) {
}
func evalLength(ctx ExprContext, self *term) (v any, err error) {
var rightValue any
var childValue any
if rightValue, err = self.evalPrefix(ctx); err != nil {
if childValue, err = self.evalPrefix(ctx); err != nil {
return
}
if isList(rightValue) {
list, _ := rightValue.([]any)
if isList(childValue) {
list, _ := childValue.([]any)
v = int64(len(list))
} else if isString(rightValue) {
s, _ := rightValue.(string)
} else if isString(childValue) {
s, _ := childValue.(string)
v = int64(len(s))
} else if it, ok := rightValue.(Iterator); ok {
v = int64(it.Index())
} else if it, ok := childValue.(Iterator); ok {
v = int64(it.Index() + 1)
} else {
err = self.errIncompatibleType(rightValue)
err = self.errIncompatibleType(childValue)
}
return
}