Skip to content

Commit

Permalink
This is an automated cherry-pick of #50928
Browse files Browse the repository at this point in the history
Signed-off-by: ti-chi-bot <ti-community-prow-bot@tidb.io>
  • Loading branch information
wjhuang2016 authored and ti-chi-bot committed Apr 12, 2024
1 parent 425cda9 commit 3c5557d
Show file tree
Hide file tree
Showing 3 changed files with 106 additions and 2 deletions.
2 changes: 1 addition & 1 deletion ddl/metadatalocktest/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ go_test(
"mdl_test.go",
],
flaky = True,
shard_count = 32,
shard_count = 34,
deps = [
"//config",
"//ddl",
Expand Down
84 changes: 84 additions & 0 deletions ddl/metadatalocktest/mdl_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
22 changes: 21 additions & 1 deletion planner/core/preprocess.go
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand All @@ -157,6 +157,13 @@ const (
inSequenceFunction
// initTxnContextProvider is set when we should init txn context in preprocess
initTxnContextProvider
<<<<<<< HEAD:planner/core/preprocess.go
=======
// inImportInto is set when visiting an import into statement.
inImportInto
// inAnalyze is set when visiting an analyze statement.
inAnalyze
>>>>>>> fce2805f2a5 (planner: skip MDL when analyzing table (#50928)):pkg/planner/core/preprocess.go
)

// Make linter happy.
Expand Down Expand Up @@ -393,6 +400,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
}
Expand Down Expand Up @@ -1907,3 +1916,14 @@ func tryLockMDLAndUpdateSchemaIfNecessary(sctx sessionctx.Context, dbName model.
}
return tbl, nil
}
<<<<<<< HEAD:planner/core/preprocess.go
=======

// skipLockMDL returns true if the preprocessor should skip the lock of MDL.
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.
// skip lock mdl for ANALYZE statement.
return p.flag&inImportInto > 0 || p.flag&inAnalyze > 0
}
>>>>>>> fce2805f2a5 (planner: skip MDL when analyzing table (#50928)):pkg/planner/core/preprocess.go

0 comments on commit 3c5557d

Please sign in to comment.