avoid running queries on the WriteOnly state table

Signed-off-by: Yang Keao <[email protected]>

Author: YangKeao

pkg/ddl/foreign_key_test.go

@@ -402,3 +402,41 @@ func TestAddForeignKey3(t *testing.T) {
 	tk.MustQuery("select * from t1 order by id").Check(testkit.Rows("1 1", "2 2", "3 3"))
 	tk.MustQuery("select * from t2 order by id").Check(testkit.Rows("1 1", "2 2", "3 3"))
 }
+
+func TestForeignKeyInWriteOnlyMode(t *testing.T) {
+	store := testkit.CreateMockStore(t)
+	tk := testkit.NewTestKit(t, store)
+	tk.MustExec("use test")
+
+	tkDDL := testkit.NewTestKit(t, store)
+	tkDDL.MustExec("use test")
+	tkDDL.MustExec("create table parent (id int key)")
+	tkDDL.MustExec("insert into parent values(1)")
+
+	var notExistErrs []error
+	testfailpoint.EnableCall(t, "github.com/pingcap/tidb/pkg/ddl/onJobRunBefore", func(job *model.Job) {
+		if job.Type == model.ActionCreateTable && job.TableName == "child" {
+			if job.SchemaState == model.StateWriteOnly {
+				// tk with the latest schema will insert data into child
+				_, err := tk.Exec("insert into child values (1, 1)")
+				notExistErrs = append(notExistErrs, err)
+				_, err = tk.Exec("update child set id = 2 where id = 1")
+				notExistErrs = append(notExistErrs, err)
+				_, err = tk.Exec("delete from child where id = 1")
+				notExistErrs = append(notExistErrs, err)
+				_, err = tk.Exec("delete child from child inner join parent where child.pid = parent.id")
+				notExistErrs = append(notExistErrs, err)
+				_, err = tk.Exec("delete parent from child inner join parent where child.pid = parent.id")
+				notExistErrs = append(notExistErrs, err)
+			}
+		}
+	})
+	tkDDL.MustExec("create table child (id int, pid int, index idx_pid(pid), foreign key (pid) references parent(id) on delete cascade);")
+
+	testfailpoint.Disable(t, "github.com/pingcap/tidb/pkg/ddl/onJobRunBefore")
+
+	for _, err := range notExistErrs {
+		require.Error(t, err)
+		require.Contains(t, err.Error(), "Table 'test.child' doesn't exist")
+	}
+}

pkg/planner/core/preprocess.go

@@ -1634,6 +1634,10 @@ func (p *preprocessor) handleTableName(tn *ast.TableName) {
 			return
 		}
 	}
+	if err := p.checkTableVisible(tableInfo); err != nil {
+		p.err = err
+		return
+	}
 	p.resolveCtx.AddTableName(&resolve.TableNameW{
 		TableName: tn,
 		DBInfo:    dbInfo,
@@ -1975,3 +1979,20 @@ func (p *preprocessor) skipLockMDL() bool {
 	// skip lock mdl for ANALYZE statement.
 	return p.flag&inImportInto > 0 || p.flag&inAnalyze > 0
 }
+
+// checkTableVisible returns an `ErrTableNotExists` error if the table is not visible (in None or WriteOnly state).
+func (p *preprocessor) checkTableVisible(tblInfo *model.TableInfo) error {
+	is := p.ensureInfoSchema()
+
+	if tblInfo.State != model.StatePublic {
+		schemaName := "unknown"
+		schema, ok := infoschema.SchemaByTable(is, tblInfo)
+		if ok {
+			// theoretically, this should always happen
+			schemaName = schema.Name.O
+		}
+		return infoschema.ErrTableNotExists.FastGenByArgs(schemaName, tblInfo.Name.O)
+	}
+
+	return nil
+}