ddl: fix CREATE GLOBAL TEMPORARY TABLE ... LIKE creates a normal table

ddl/ddl_api.go

@@ -1638,12 +1638,15 @@ func checkTableInfoValid(tblInfo *model.TableInfo) error {
  return checkInvisibleIndexOnPK(tblInfo)
 }
 
-func buildTableInfoWithLike(ident ast.Ident, referTblInfo *model.TableInfo, s *ast.CreateTableStmt) (*model.TableInfo, error) {
+func buildTableInfoWithLike(ctx sessionctx.Context, ident ast.Ident, referTblInfo *model.TableInfo, s *ast.CreateTableStmt) (*model.TableInfo, error) {
  // Check the referred table is a real table object.
  if referTblInfo.IsSequence() || referTblInfo.IsView() {
  return nil, ErrWrongObject.GenWithStackByArgs(ident.Schema, referTblInfo.Name, "BASE TABLE")
  }
  tblInfo := *referTblInfo
+ if err := setTemporaryType(ctx, &tblInfo, s); err != nil {
+ return nil, errors.Trace(err)
+ }
  // Check non-public column and adjust column offset.
  newColumns := referTblInfo.Cols()
  newIndices := make([]*model.IndexInfo, 0, len(tblInfo.Indices))
@@ -1731,22 +1734,8 @@ func buildTableInfoWithStmt(ctx sessionctx.Context, s *ast.CreateTableStmt, dbCh
  if err != nil {
  return nil, errors.Trace(err)
  }
- switch s.TemporaryKeyword {
- case ast.TemporaryGlobal:
- tbInfo.TempTableType = model.TempTableGlobal
- if !ctx.GetSessionVars().EnableGlobalTemporaryTable {
- return nil, errors.New("global temporary table is experimental and it is switched off by tidb_enable_global_temporary_table")
- }
- // "create global temporary table ... on commit preserve rows"
- if !s.OnCommitDelete {
- return nil, errors.Trace(errUnsupportedOnCommitPreserve)
- }
- case ast.TemporaryLocal:
- // TODO: set "tbInfo.TempTableType = model.TempTableLocal" after local temporary table is supported.
- tbInfo.TempTableType = model.TempTableNone
- ctx.GetSessionVars().StmtCtx.AppendWarning(errors.New("local TEMPORARY TABLE is not supported yet, TEMPORARY will be parsed but ignored"))
- case ast.TemporaryNone:
- tbInfo.TempTableType = model.TempTableNone
+ if err = setTemporaryType(ctx, tbInfo, s); err != nil {
+ return nil, errors.Trace(err)
  }
 
  if err = setTableAutoRandomBits(ctx, tbInfo, colDefs); err != nil {
@@ -1808,7 +1797,7 @@ func (d *ddl) CreateTable(ctx sessionctx.Context, s *ast.CreateTableStmt) (err e
  // build tableInfo
  var tbInfo *model.TableInfo
  if s.ReferTable != nil {
- tbInfo, err = buildTableInfoWithLike(ident, referTbl.Meta(), s)
+ tbInfo, err = buildTableInfoWithLike(ctx, ident, referTbl.Meta(), s)
  } else {
  tbInfo, err = buildTableInfoWithStmt(ctx, s, schema.Charset, schema.Collate)
  }
@@ -1828,6 +1817,27 @@ func (d *ddl) CreateTable(ctx sessionctx.Context, s *ast.CreateTableStmt) (err e
  return d.CreateTableWithInfo(ctx, schema.Name, tbInfo, onExist, false /*tryRetainID*/)
 }
 
+func setTemporaryType(ctx sessionctx.Context, tbInfo *model.TableInfo, s *ast.CreateTableStmt) error {
+ switch s.TemporaryKeyword {
+ case ast.TemporaryGlobal:
+ tbInfo.TempTableType = model.TempTableGlobal
+ if !ctx.GetSessionVars().EnableGlobalTemporaryTable {
+ return errors.New("global temporary table is experimental and it is switched off by tidb_enable_global_temporary_table")
+ }
+ // "create global temporary table ... on commit preserve rows"
+ if !s.OnCommitDelete {
+ return errors.Trace(errUnsupportedOnCommitPreserve)
+ }
+ case ast.TemporaryLocal:
+ // TODO: set "tbInfo.TempTableType = model.TempTableLocal" after local temporary table is supported.
+ tbInfo.TempTableType = model.TempTableNone
+ ctx.GetSessionVars().StmtCtx.AppendWarning(errors.New("local TEMPORARY TABLE is not supported yet, TEMPORARY will be parsed but ignored"))
+ default:
+ tbInfo.TempTableType = model.TempTableNone
+ }
+ return nil
+}
+
 func (d *ddl) CreateTableWithInfo(
  ctx sessionctx.Context,
  dbName model.CIStr,

ddl/serial_test.go

@@ -613,6 +613,23 @@ func (s *testSerialSuite) TestCreateTableWithLikeAtTemporaryMode(c *C) {
  c.Assert(err, IsNil)
  tableInfo := table.Meta()
  c.Assert(len(tableInfo.ForeignKeys), Equals, 0)
+
+ // Issue 25613.
+ tk.MustExec("drop table if exists tb2, tb3")
+ tk.MustExec("create table tb2(id int);")
+ tk.MustExec("create global temporary table tb3 like tb2 on commit delete rows;")
+ defer tk.MustExec("drop table if exists tb2, tb3")
+ tk.MustQuery("show create table tb3;").Check(testkit.Rows("tb3 CREATE GLOBAL TEMPORARY TABLE `tb3` (\n" +
+ " `id` int(11) DEFAULT NULL\n" +
+ ") ENGINE=memory DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin ON COMMIT DELETE ROWS"))
+
+ tk.MustExec("drop table if exists tb4, tb5")
+ tk.MustExec("create table tb4(id int);")
+ tk.MustExec("create table tb5 like tb4")
+ defer tk.MustExec("drop table if exists tb4, tb5")
+ tk.MustQuery("show create table tb5;").Check(testkit.Rows("tb5 CREATE TABLE `tb5` (\n" +
+ " `id` int(11) DEFAULT NULL\n" +
+ ") ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin"))
 }
 
 // TestCancelAddIndex1 tests canceling ddl job when the add index worker is not started.