expression: implement vectorized evaluation for builtinCastDurationAsTimeSig (#13244)
* expression: implement vectorized evaluation for builtinCastDurationAsTimeSig * expression: use GoDurations instead of GetDuration
This commit is contained in:
committed by
goroutine
parent
be128c6090
commit
c7932476ef
@ -478,11 +478,49 @@ func (b *builtinCastDecimalAsDecimalSig) vecEvalDecimal(input *chunk.Chunk, resu
|
||||
}
|
||||
|
||||
func (b *builtinCastDurationAsTimeSig) vectorized() bool {
|
||||
return false
|
||||
return true
|
||||
}
|
||||
|
||||
func (b *builtinCastDurationAsTimeSig) vecEvalTime(input *chunk.Chunk, result *chunk.Column) error {
|
||||
return errors.Errorf("not implemented")
|
||||
n := input.NumRows()
|
||||
buf, err := b.bufAllocator.get(types.ETDuration, n)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer b.bufAllocator.put(buf)
|
||||
if err := b.args[0].VecEvalDuration(b.ctx, input, buf); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
result.ResizeTime(n, false)
|
||||
result.MergeNulls(buf)
|
||||
var duration types.Duration
|
||||
ds := buf.GoDurations()
|
||||
times := result.Times()
|
||||
stmtCtx := b.ctx.GetSessionVars().StmtCtx
|
||||
fsp := int8(b.tp.Decimal)
|
||||
for i := 0; i < n; i++ {
|
||||
if result.IsNull(i) {
|
||||
continue
|
||||
}
|
||||
|
||||
duration.Duration = ds[i]
|
||||
duration.Fsp = fsp
|
||||
tm, err := duration.ConvertToTime(stmtCtx, b.tp.Tp)
|
||||
if err != nil {
|
||||
if err = handleInvalidTimeError(b.ctx, err); err != nil {
|
||||
return err
|
||||
}
|
||||
result.SetNull(i, true)
|
||||
continue
|
||||
}
|
||||
tm, err = tm.RoundFrac(stmtCtx, fsp)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
times[i] = tm
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b *builtinCastIntAsStringSig) vectorized() bool {
|
||||
|
||||
@ -70,6 +70,7 @@ var vecBuiltinCastCases = map[string][]vecExprBenchCase{
|
||||
&timeStrGener{},
|
||||
&dataStrGener{},
|
||||
}},
|
||||
{retEvalType: types.ETDatetime, childrenTypes: []types.EvalType{types.ETDuration}},
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user