expression: implement vectorized evaluation for builtinCastDurationAsTimeSig (#13244)

* expression: implement vectorized evaluation for builtinCastDurationAsTimeSig

* expression: use GoDurations instead of GetDuration
This commit is contained in:
Eugene Kalinin
2019-11-08 11:23:22 +03:00
committed by goroutine
parent be128c6090
commit c7932476ef
2 changed files with 41 additions and 2 deletions

View File

@ -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 {

View File

@ -70,6 +70,7 @@ var vecBuiltinCastCases = map[string][]vecExprBenchCase{
&timeStrGener{},
&dataStrGener{},
}},
{retEvalType: types.ETDatetime, childrenTypes: []types.EvalType{types.ETDuration}},
},
}