diff --git a/pkg/ddl/tests/metadatalock/BUILD.bazel b/pkg/ddl/tests/metadatalock/BUILD.bazel index a73297ca95700..1aa96403e0207 100644 --- a/pkg/ddl/tests/metadatalock/BUILD.bazel +++ b/pkg/ddl/tests/metadatalock/BUILD.bazel @@ -8,7 +8,7 @@ go_test( "mdl_test.go", ], flaky = True, - shard_count = 32, + shard_count = 34, deps = [ "//pkg/config", "//pkg/ddl", diff --git a/pkg/ddl/tests/metadatalock/mdl_test.go b/pkg/ddl/tests/metadatalock/mdl_test.go index 8ad85021fa2bf..667e8b055284c 100644 --- a/pkg/ddl/tests/metadatalock/mdl_test.go +++ b/pkg/ddl/tests/metadatalock/mdl_test.go @@ -432,6 +432,90 @@ func TestMDLAutoCommitReadOnly(t *testing.T) { require.Greater(t, ts1, ts2) } +func TestMDLAnalyze(t *testing.T) { + store, dom := testkit.CreateMockStoreAndDomain(t) + sv := server.CreateMockServer(t, store) + + sv.SetDomain(dom) + dom.InfoSyncer().SetSessionManager(sv) + defer sv.Close() + + conn1 := server.CreateMockConn(t, sv) + tk := testkit.NewTestKitWithSession(t, store, conn1.Context().Session) + conn2 := server.CreateMockConn(t, sv) + tkDDL := testkit.NewTestKitWithSession(t, store, conn2.Context().Session) + tk.MustExec("use test") + tk.MustExec("set global tidb_enable_metadata_lock=1") + tk.MustExec("create table t(a int);") + tk.MustExec("insert into t values(1);") + + var wg sync.WaitGroup + wg.Add(2) + var ts2 time.Time + var ts1 time.Time + + go func() { + tk.MustExec("begin") + tk.MustExec("analyze table t;") + tk.MustQuery("select sleep(2);") + tk.MustExec("commit") + ts1 = time.Now() + wg.Done() + }() + + go func() { + tkDDL.MustExec("alter table test.t add column b int;") + ts2 = time.Now() + wg.Done() + }() + + wg.Wait() + require.Greater(t, ts1, ts2) +} + +func TestMDLAnalyzePartition(t *testing.T) { + store, dom := testkit.CreateMockStoreAndDomain(t) + sv := server.CreateMockServer(t, store) + + sv.SetDomain(dom) + dom.InfoSyncer().SetSessionManager(sv) + defer sv.Close() + + conn1 := server.CreateMockConn(t, sv) + tk := testkit.NewTestKitWithSession(t, store, conn1.Context().Session) + conn2 := server.CreateMockConn(t, sv) + tkDDL := testkit.NewTestKitWithSession(t, store, conn2.Context().Session) + tk.MustExec("use test") + tk.MustExec("set @@tidb_partition_prune_mode='dynamic'") + tk.MustExec("set global tidb_enable_metadata_lock=1") + tk.MustExec("create table t(a int) partition by range(a) ( PARTITION p0 VALUES LESS THAN (0), PARTITION p1 VALUES LESS THAN (100), PARTITION p2 VALUES LESS THAN MAXVALUE );") + tk.MustExec("insert into t values(1), (2), (3), (4);") + + var wg sync.WaitGroup + wg.Add(2) + var ts2 time.Time + var ts1 time.Time + + go func() { + tk.MustExec("begin") + tk.MustExec("analyze table t;") + tk.MustExec("analyze table t partition p1;") + tk.MustQuery("select sleep(2);") + tk.MustExec("commit") + ts1 = time.Now() + wg.Done() + }() + + go func() { + tkDDL.MustExec("alter table test.t drop partition p2;") + ts2 = time.Now() + wg.Done() + }() + + wg.Wait() + require.Greater(t, ts1, ts2) +} + func TestMDLAutoCommitNonReadOnly(t *testing.T) { store, dom := testkit.CreateMockStoreAndDomain(t) sv := server.CreateMockServer(t, store) diff --git a/pkg/planner/core/preprocess.go b/pkg/planner/core/preprocess.go index 92879e2ecd923..a5849a1af24fa 100644 --- a/pkg/planner/core/preprocess.go +++ b/pkg/planner/core/preprocess.go @@ -139,7 +139,7 @@ func Preprocess(ctx context.Context, sctx sessionctx.Context, node ast.Node, pre return errors.Trace(v.err) } -type preprocessorFlag uint8 +type preprocessorFlag uint64 const ( // inPrepare is set when visiting in prepare statement. @@ -159,6 +159,8 @@ const ( initTxnContextProvider // inImportInto is set when visiting an import into statement. inImportInto + // inAnalyze is set when visiting an analyze statement. + inAnalyze ) // Make linter happy. @@ -401,6 +403,8 @@ func (p *preprocessor) Enter(in ast.Node) (out ast.Node, skipChildren bool) { p.sctx.GetSessionVars().StmtCtx.IsStaleness = true p.IsStaleness = true } + case *ast.AnalyzeTableStmt: + p.flag |= inAnalyze default: p.flag &= ^parentIsJoin } @@ -1911,5 +1915,6 @@ func tryLockMDLAndUpdateSchemaIfNecessary(sctx sessionctx.Context, dbName model. func (p *preprocessor) skipLockMDL() bool { // skip lock mdl for IMPORT INTO statement, // because it's a batch process and will do both DML and DDL. - return p.flag&inImportInto > 0 + // skip lock mdl for ANALYZE statement. + return p.flag&inImportInto > 0 || p.flag&inAnalyze > 0 }