infoschema: fix select temporary table with view will panic problem (p…

…ingcap#42566) (pingcap#42702)

close pingcap#42563

infoschema/infoschema.go

@@ -700,15 +700,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,
 	}
 }

table/temptable/BUILD.bazel

@@ -34,6 +34,7 @@ go_test(
     srcs = [
         "ddl_test.go",
         "interceptor_test.go",
+        "intergration_test.go",
         "main_test.go",
     ],
     embed = [":temptable"],
@@ -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",

table/temptable/infoschema.go

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

table/temptable/intergration_test.go

@@ -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"))
+}