Files
tidb/executor/simpletest/chunk_reuse_test.go

139 lines
11 KiB
Go

// Copyright 2023 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 simpletest
import (
"testing"
"github.com/pingcap/tidb/testkit"
"github.com/stretchr/testify/require"
)
func TestLongBlobReuse(t *testing.T) {
store := testkit.CreateMockStore(t)
tk := testkit.NewTestKit(t, store)
tk.MustExec("use test")
tk.MustExec("create table t1 (id1 int ,id2 char(10) ,id3 text,id4 blob,id5 json,id6 varchar(1000),id7 varchar(1001), PRIMARY KEY (`id1`) clustered,key id2(id2))")
tk.MustExec("insert into t1 (id1,id2)values(1,1);")
tk.MustExec("insert into t1 (id1,id2)values(2,2),(3,3);")
tk.MustExec("create table t2 (id1 int ,id2 char(10) ,id3 text,id4 blob,id5 json,id6 varchar(1000),PRIMARY KEY (`id1`) clustered,key id2(id2))")
tk.MustExec("insert into t2 (id1,id2)values(1,1);")
tk.MustExec("insert into t2 (id1,id2)values(2,2),(3,3);")
//IndexRangeScan
res := tk.MustQuery("explain select t1.id1 from t1,t2 where t1.id2 > '1' and t2.id2 > '1'")
require.Regexp(t, ".*IndexRangeScan*", res.Rows()[4][0])
tk.MustQuery("select t1.id1 from t1,t2 where t1.id2 > '1' and t2.id2 > '1'").Check(testkit.Rows("2", "2", "3", "3"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1,t1.id2 from t1,t2 where t1.id2 > '1' and t2.id2 > '1' ").Check(testkit.Rows("2 2", "2 2", "3 3", "3 3"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1,t1.id3 from t1,t2 where t1.id2 > '1' and t2.id2 > '1' ").Check(testkit.Rows("2 <nil>", "2 <nil>", "3 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id4 from t1,t2 where t1.id2 > '1' and t2.id2 > '1' ").Check(testkit.Rows("2 <nil>", "2 <nil>", "3 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id5 from t1,t2 where t1.id2 > '1' and t2.id2 > '1' ").Check(testkit.Rows("2 <nil>", "2 <nil>", "3 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id6 from t1,t2 where t1.id2 > '1' and t2.id2 > '1' ").Check(testkit.Rows("2 <nil>", "2 <nil>", "3 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1,t1.id7 from t1,t2 where t1.id2 > '1' and t2.id2 > '1' ").Check(testkit.Rows("2 <nil>", "2 <nil>", "3 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
//TableFullScan
res = tk.MustQuery("explain select t1.id1 from t1,t2 where t1.id2 > '1'and t1.id1 = t2.id1")
require.Regexp(t, ".*TableFullScan*", res.Rows()[2][0])
tk.MustQuery("select t1.id1 from t1,t2 where t1.id2 > '1' and t1.id1 = t2.id1").Check(testkit.Rows("2", "3"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1 ,t1.id3 from t1,t2 where t1.id2 > '1' and t1.id1 = t2.id1").Check(testkit.Rows("2 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1 ,t1.id4 from t1,t2 where t1.id2 > '1' and t1.id1 = t2.id1").Check(testkit.Rows("2 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1 ,t1.id5 from t1,t2 where t1.id2 > '1' and t1.id1 = t2.id1").Check(testkit.Rows("2 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1 ,t1.id6 from t1,t2 where t1.id2 > '1' and t1.id1 = t2.id1").Check(testkit.Rows("2 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1 ,t1.id7 from t1,t2 where t1.id2 > '1' and t1.id1 = t2.id1").Check(testkit.Rows("2 <nil>", "3 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
//Point_Get
res = tk.MustQuery("explain select t1.id1 from t1,t2 where t1.id1 = 1 and t2.id1 = 1")
require.Regexp(t, ".*Point_Get*", res.Rows()[1][0])
tk.MustQuery("select t1.id1 from t1,t2 where t1.id1 = 1 and t2.id1 = 1").Check(testkit.Rows("1"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1,t1.id2 from t1,t2 where t1.id1 = 1 and t2.id1 = 1 ").Check(testkit.Rows("1 1"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1,t1.id3 from t1,t2 where t1.id1 = 1 and t2.id1 = 1 ").Check(testkit.Rows("1 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id4 from t1,t2 where t1.id1 = 1 and t2.id1 = 1 ").Check(testkit.Rows("1 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id5 from t1,t2 where t1.id1 = 1 and t2.id1 = 1 ").Check(testkit.Rows("1 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id6 from t1,t2 where t1.id1 = 1 and t2.id1 = 1 ").Check(testkit.Rows("1 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1,t1.id7 from t1,t2 where t1.id1 = 1 and t2.id1 = 1 ").Check(testkit.Rows("1 <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
//IndexLookUp
res = tk.MustQuery("explain select t1.id1,t1.id6 ,t2.id6 from t1 join t2 on t1.id2 = '1' and t2.id2 = '2' ;")
require.Regexp(t, ".*IndexLookUp*", res.Rows()[1][0])
tk.MustQuery("select t1.id1,t1.id6 ,t2.id6 from t1 join t2 on t1.id2 = '1' and t2.id2 = '2'").Check(testkit.Rows("1 <nil> <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select t1.id1,t1.id3 ,t2.id6 from t1 join t2 on t1.id2 = '1' and t2.id2 = '2'").Check(testkit.Rows("1 <nil> <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id4 ,t2.id6 from t1 join t2 on t1.id2 = '1' and t2.id2 = '2'").Check(testkit.Rows("1 <nil> <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id5 ,t2.id6 from t1 join t2 on t1.id2 = '1' and t2.id2 = '2'").Check(testkit.Rows("1 <nil> <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id7 ,t2.id6 from t1 join t2 on t1.id2 = '1' and t2.id2 = '2'").Check(testkit.Rows("1 <nil> <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select t1.id1,t1.id6 ,t2.id3 from t1 join t2 on t1.id2 = '1' and t2.id2 = '2'").Check(testkit.Rows("1 <nil> <nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
// IndexMerge
tk.MustExec("create table t3 (id1 int ,id2 char(10),id8 int ,id3 text,id4 blob,id5 json,id6 varchar(1000),id7 varchar(1001), PRIMARY KEY (`id1`) clustered,key id2(id2),key id8(id8))")
tk.MustExec("insert into t3 (id1,id2,id8)values(1,1,1),(2,2,2),(3,3,3);")
res = tk.MustQuery("explain select id1 from t3 where id2 > '3' or id8 < 10 union (select id1 from t3 where id2 > '4' or id8 < 7);")
require.Regexp(t, ".*IndexMerge*", res.Rows()[3][0])
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select id1 from t3 where id2 > '4' or id8 < 7);").Sort().Check(testkit.Rows("1", "2", "3"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select id3 from t3 where id2 > '4' or id8 < 7);").Sort().Check(testkit.Rows("1", "2", "3", "<nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select id4 from t3 where id2 > '4' or id8 < 7);").Sort().Check(testkit.Rows("1", "2", "3", "<nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select id5 from t3 where id2 > '4' or id8 < 7);").Sort().Check(testkit.Rows("1", "2", "3", "<nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select id6 from t3 where id2 > '4' or id8 < 7);").Sort().Check(testkit.Rows("1", "2", "3", "<nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select id7 from t3 where id2 > '4' or id8 < 7);").Sort().Check(testkit.Rows("1", "2", "3", "<nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
//IndexReader
tk.MustExec("set tidb_enable_clustered_index = OFF")
tk.MustExec("create table t4 (id1 int ,id2 char(10),id8 int ,id3 text,id4 blob,id5 json,id6 varchar(1000),id7 varchar(1001), PRIMARY KEY (`id1`),key id2(id2),key id8(id8,id2))")
tk.MustExec("insert into t4 (id1,id2,id8)values(1,1,1),(2,2,2),(3,3,3);")
res = tk.MustQuery("explain select id2 from t4 where id2 > '3' union (select id2 from t4 where id2 > '4');")
require.Regexp(t, ".*IndexReader*", res.Rows()[2][0])
tk.MustQuery("select id2 from t4 where id2 > '3' union (select id2 from t4 where id2 > '4');").Sort().Check(testkit.Rows())
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
//function
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select CHAR_LENGTH(id3) from t3 where id2 > '4' or id8 < 7);").Sort().Check(testkit.Rows("1", "2", "3", "<nil>"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select CHAR_LENGTH(id2) from t3 where id2 > '4' or id8 < 7);").Sort().Check(testkit.Rows("1", "2", "3"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("1"))
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select id2 from t3 where id2 > '4' or id8 < 7 and id3 is null);").Sort().Check(testkit.Rows("1", "2", "3"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
tk.MustQuery("select id1 from t3 where id2 > '3' or id8 < 10 union (select id2 from t3 where id2 > '4' or id8 < 7 and char_length(id3) > 0);").Sort().Check(testkit.Rows("1", "2", "3"))
tk.MustQuery("select @@last_sql_use_alloc").Check(testkit.Rows("0"))
}