Skip to content

Commit

Permalink
infoschema: fix select temporary table with view will panic
Browse files Browse the repository at this point in the history
  • Loading branch information
lcwangchao committed Mar 25, 2023
1 parent f1df935 commit 785ab33
Show file tree
Hide file tree
Showing 4 changed files with 47 additions and 12 deletions.
13 changes: 4 additions & 9 deletions infoschema/infoschema.go
Original file line number Diff line number Diff line change
Expand Up @@ -751,15 +751,10 @@ func (ts *SessionExtendedInfoSchema) HasTemporaryTable() bool {
return ts.LocalTemporaryTables != nil && ts.LocalTemporaryTables.Count() > 0 || ts.InfoSchema.HasTemporaryTable()
}

// AttachMDLTableInfoSchema attach MDL related table information schema to is
func AttachMDLTableInfoSchema(is InfoSchema) InfoSchema {
mdlTables := NewSessionTables()
if iss, ok := is.(*SessionExtendedInfoSchema); ok {
iss.MdlTables = mdlTables
return iss
}
// DetachTemporaryTableInfoSchema returns a new SessionExtendedInfoSchema without temporary tables
func (ts *SessionExtendedInfoSchema) DetachTemporaryTableInfoSchema() *SessionExtendedInfoSchema {
return &SessionExtendedInfoSchema{
InfoSchema: is,
MdlTables: mdlTables,
InfoSchema: ts.InfoSchema,
MdlTables: ts.MdlTables,
}
}
3 changes: 3 additions & 0 deletions table/temptable/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ go_test(
srcs = [
"ddl_test.go",
"interceptor_test.go",
"intergration_test.go",
"main_test.go",
],
embed = [":temptable"],
Expand All @@ -42,13 +43,15 @@ go_test(
"//infoschema",
"//kv",
"//meta/autoid",
"//parser/auth",
"//parser/model",
"//parser/mysql",
"//sessionctx",
"//store/driver/txn",
"//store/mockstore",
"//table",
"//tablecodec",
"//testkit",
"//testkit/testsetup",
"//types",
"//util/codec",
Expand Down
4 changes: 1 addition & 3 deletions table/temptable/infoschema.go
Original file line number Diff line number Diff line change
Expand Up @@ -41,9 +41,7 @@ func AttachLocalTemporaryTableInfoSchema(sctx sessionctx.Context, is infoschema.
// DetachLocalTemporaryTableInfoSchema detach local temporary table information schema from is
func DetachLocalTemporaryTableInfoSchema(is infoschema.InfoSchema) infoschema.InfoSchema {
if attachedInfoSchema, ok := is.(*infoschema.SessionExtendedInfoSchema); ok {
newIs := attachedInfoSchema
newIs.LocalTemporaryTables = nil
return newIs
return attachedInfoSchema.DetachTemporaryTableInfoSchema()
}

return is
Expand Down
39 changes: 39 additions & 0 deletions table/temptable/intergration_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
// 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 temptable_test

import (
"testing"

"github.com/pingcap/tidb/parser/auth"
"github.com/pingcap/tidb/testkit"
"github.com/stretchr/testify/require"
)

func TestSelectTemporaryTableUnionView(t *testing.T) {
// see the issue: https://github.com/pingcap/tidb/issues/42563
store := testkit.CreateMockStore(t)
tk := testkit.NewTestKit(t, store)
require.NoError(t, tk.Session().Auth(&auth.UserIdentity{Username: "root", Hostname: "%"}, nil, nil))
tk.MustExec("use test")
tk.MustExec("create table t(a int)")
tk.MustExec("insert into t values(1)")
tk.MustExec("create view tv as select a from t")
tk.MustExec("create temporary table t(a int)")
tk.MustExec("insert into t values(2)")
tk.MustQuery("select * from tv").Check(testkit.Rows("1"))
tk.MustQuery("select * from t").Check(testkit.Rows("2"))
tk.MustQuery("select * from (select a from t union all select a from tv) t1 order by a").Check(testkit.Rows("1", "2"))
}

0 comments on commit 785ab33

Please sign in to comment.