forked from pingcap/tidb
-
Notifications
You must be signed in to change notification settings - Fork 0
/
func_first_row_test.go
109 lines (99 loc) · 4.66 KB
/
func_first_row_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
// Copyright 2019 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package aggfuncs_test
import (
"testing"
"time"
"github.com/pingcap/tidb/executor/aggfuncs"
"github.com/pingcap/tidb/parser/ast"
"github.com/pingcap/tidb/parser/mysql"
"github.com/pingcap/tidb/types"
"github.com/pingcap/tidb/types/json"
"github.com/pingcap/tidb/util/chunk"
)
func TestMergePartialResult4FirstRow(t *testing.T) {
t.Parallel()
elems := []string{"e", "d", "c", "b", "a"}
enumC, _ := types.ParseEnumName(elems, "c", mysql.DefaultCollationName)
enumE, _ := types.ParseEnumName(elems, "e", mysql.DefaultCollationName)
setED, _ := types.ParseSetName(elems, "e,d", mysql.DefaultCollationName)
setE, _ := types.ParseSetName(elems, "e", mysql.DefaultCollationName)
tests := []aggTest{
buildAggTester(ast.AggFuncFirstRow, mysql.TypeLonglong, 5, 0, 2, 0),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeFloat, 5, 0.0, 2.0, 0.0),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeDouble, 5, 0.0, 2.0, 0.0),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeNewDecimal, 5, types.NewDecFromInt(0), types.NewDecFromInt(2), types.NewDecFromInt(0)),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeString, 5, "0", "2", "0"),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeDate, 5, types.TimeFromDays(365), types.TimeFromDays(367), types.TimeFromDays(365)),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeDuration, 5, types.Duration{Duration: time.Duration(0)}, types.Duration{Duration: time.Duration(2)}, types.Duration{Duration: time.Duration(0)}),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeJSON, 5, json.CreateBinary(int64(0)), json.CreateBinary(int64(2)), json.CreateBinary(int64(0))),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeEnum, 5, enumE, enumC, enumE),
buildAggTester(ast.AggFuncFirstRow, mysql.TypeSet, 5, setE, setED, setE),
}
for _, test := range tests {
testMergePartialResult(t, test)
}
}
func TestMemFirstRow(t *testing.T) {
t.Parallel()
tests := []aggMemTest{
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeLonglong, 5,
aggfuncs.DefPartialResult4FirstRowIntSize, defaultUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeFloat, 5,
aggfuncs.DefPartialResult4FirstRowFloat32Size, defaultUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeDouble, 5,
aggfuncs.DefPartialResult4FirstRowFloat64Size, defaultUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeNewDecimal, 5,
aggfuncs.DefPartialResult4FirstRowDecimalSize, defaultUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeString, 5,
aggfuncs.DefPartialResult4FirstRowStringSize, firstRowUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeDate, 5,
aggfuncs.DefPartialResult4FirstRowTimeSize, defaultUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeDuration, 5,
aggfuncs.DefPartialResult4FirstRowDurationSize, defaultUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeJSON, 5,
aggfuncs.DefPartialResult4FirstRowJSONSize, firstRowUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeEnum, 5,
aggfuncs.DefPartialResult4FirstRowEnumSize, firstRowUpdateMemDeltaGens, false),
buildAggMemTester(ast.AggFuncFirstRow, mysql.TypeSet, 5,
aggfuncs.DefPartialResult4FirstRowSetSize, firstRowUpdateMemDeltaGens, false),
}
for _, test := range tests {
testAggMemFunc(t, test)
}
}
func firstRowUpdateMemDeltaGens(srcChk *chunk.Chunk, dataType *types.FieldType) (memDeltas []int64, err error) {
for i := 0; i < srcChk.NumRows(); i++ {
row := srcChk.GetRow(i)
if i > 0 {
memDeltas = append(memDeltas, int64(0))
continue
}
switch dataType.Tp {
case mysql.TypeString:
val := row.GetString(0)
memDeltas = append(memDeltas, int64(len(val)))
case mysql.TypeJSON:
jsonVal := row.GetJSON(0)
memDeltas = append(memDeltas, int64(len(string(jsonVal.Value))))
case mysql.TypeEnum:
enum := row.GetEnum(0)
memDeltas = append(memDeltas, int64(len(enum.Name)))
case mysql.TypeSet:
typeSet := row.GetSet(0)
memDeltas = append(memDeltas, int64(len(typeSet.Name)))
}
}
return memDeltas, nil
}