60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
// Copyright (c) 2024-2026 Celestino Amoroso (celestino.amoroso@gmail.com).
|
|
// All rights reserved.
|
|
|
|
// operator-length.go
|
|
package expr
|
|
|
|
import (
|
|
"git.portale-stac.it/go-pkg/expr/kern"
|
|
"git.portale-stac.it/go-pkg/expr/scan"
|
|
)
|
|
|
|
//-------- length term
|
|
|
|
func newLengthTerm(tk *scan.Token) (inst *scan.Term) {
|
|
return &scan.Term{
|
|
Tk: *tk,
|
|
Children: make([]*scan.Term, 0, 1),
|
|
Position: scan.PosPrefix,
|
|
Priority: scan.PriSign,
|
|
EvalFunc: evalLength,
|
|
}
|
|
}
|
|
|
|
func evalLength(ctx kern.ExprContext, opTerm *scan.Term) (v any, err error) {
|
|
var childValue any
|
|
|
|
if childValue, err = opTerm.EvalPrefix(ctx); err != nil {
|
|
return
|
|
}
|
|
|
|
if kern.IsList(childValue) {
|
|
ls, _ := childValue.(*kern.ListType)
|
|
v = int64(len(*ls))
|
|
} else if kern.IsString(childValue) {
|
|
s, _ := childValue.(string)
|
|
v = int64(len(s))
|
|
} else if kern.IsDict(childValue) {
|
|
m, _ := childValue.(*kern.DictType)
|
|
v = int64(len(*m))
|
|
} else if lls, ok := childValue.(*kern.LinkedList); ok {
|
|
v = int64(lls.Len())
|
|
} else if it, ok := childValue.(kern.Iterator); ok {
|
|
v = int64(it.Count())
|
|
// if extIt, ok := childValue.(ExtIterator); ok && extIt.HasOperation(CountName) {
|
|
// count, _ := extIt.CallOperation(CountName, nil)
|
|
// v, _ = ToGoInt(count, "")
|
|
// } else {
|
|
// v = int64(it.Index() + 1)
|
|
// }
|
|
} else {
|
|
err = opTerm.ErrIncompatiblePrefixPostfixType(childValue)
|
|
}
|
|
return
|
|
}
|
|
|
|
// init
|
|
func init() {
|
|
scan.RegisterTermConstructor(scan.SymHash, newLengthTerm)
|
|
}
|