Method Call

runtime/vam/expr/function/regexp.go:20–59

(args ...vector.Any)

}

func (r *Regexp) Call(args ...vector.Any) vector.Any {
	if vec, ok := expr.CheckForNullThenError(args); ok {
		return vec
	}
	args = underAll(args)
	regVec, inputVec := args[0], args[1]
	if regVec.Type().ID() != super.IDString {
		return vector.NewWrappedError(r.sctx, "regexp: string required for first arg", args[0])
	}
	if inputVec.Type().ID() != super.IDString {
		return vector.NewWrappedError(r.sctx, "regexp: string required for second arg", args[1])
	}
	errMsg := vector.NewStringEmpty(0)
	var errs []uint32
	inner := vector.NewStringEmpty(0)
	out := vector.NewArray(r.sctx.LookupTypeArray(super.TypeString), []uint32{0}, inner)
	for i := range regVec.Len() {
		re := vector.StringValue(regVec, i)
		if r.restr != re {
			r.restr = re
			r.re, r.err = regexp.Compile(r.restr)
		}
		if r.err != nil {
			errMsg.Append(regexpErrMsg("regexp", r.err))
			errs = append(errs, i)
			continue
		}
		s := vector.StringValue(inputVec, i)
		match := r.re.FindStringSubmatch(s)
		for _, b := range match {
			inner.Append(b)
		}
		out.Offsets = append(out.Offsets, inner.Len())
	}
	c := vector.NewCombiner(out)
	if len(errs) > 0 {
		c.Add(errs, vector.NewVecWrappedError(r.sctx, errMsg, vector.Pick(regVec, errs)))
	}
	return c.Result()
}

type RegexpReplace struct {
	sctx  *super.Context