// Copyright 2018 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, // See the License for the specific language governing permissions and // limitations under the License. package executor_test import ( . "github.com/pingcap/check" "github.com/pingcap/tidb/util/testkit" ) func (s *testSuite1) TestExportRowID(c *C) { tk := testkit.NewTestKitWithInit(c, s.store) tk.Se.GetSessionVars().AllowWriteRowID = true defer func() { tk.Se.GetSessionVars().AllowWriteRowID = false }() tk.MustExec("drop table if exists t") tk.MustExec("create table t (a int, b int)") tk.MustExec("insert t values (1, 7), (1, 8), (1, 9)") tk.MustQuery("select *, _tidb_rowid from t"). Check(testkit.Rows("1 7 1", "1 8 2", "1 9 3")) tk.MustExec("update t set a = 2 where _tidb_rowid = 2") tk.MustQuery("select *, _tidb_rowid from t"). Check(testkit.Rows("1 7 1", "2 8 2", "1 9 3")) tk.MustExec("delete from t where _tidb_rowid = 2") tk.MustQuery("select *, _tidb_rowid from t"). Check(testkit.Rows("1 7 1", "1 9 3")) tk.MustExec("insert t (a, b, _tidb_rowid) values (2, 2, 2), (5, 5, 5)") tk.MustQuery("select *, _tidb_rowid from t"). Check(testkit.Rows("1 7 1", "2 2 2", "1 9 3", "5 5 5")) // If PK is handle, _tidb_rowid is unknown column. tk.MustExec("create table s (a int primary key)") tk.MustExec("insert s values (1)") _, err := tk.Exec("insert s (a, _tidb_rowid) values (1, 2)") c.Assert(err, NotNil) err = tk.ExecToErr("select _tidb_rowid from s") c.Assert(err, NotNil) _, err = tk.Exec("update s set a = 2 where _tidb_rowid = 1") c.Assert(err, NotNil) _, err = tk.Exec("delete from s where _tidb_rowid = 1") c.Assert(err, NotNil) // Make sure "AllowWriteRowID" is a session variable. tk1 := testkit.NewTestKit(c, s.store) tk1.MustExec("use test") _, err = tk1.Exec("insert into t (a, _tidb_rowid) values(10, 1);") c.Assert(err.Error(), Equals, "insert, update and replace statements for _tidb_rowid are not supported.") } func (s *testSuite1) TestNotAllowWriteRowID(c *C) { tk := testkit.NewTestKit(c, s.store) tk.MustExec("use test") tk.MustExec("create table tt(id binary(10), c int, primary key(id));") tk.MustExec("insert tt values (1, 10);") // select statement tk.MustQuery("select *, _tidb_rowid from tt"). Check(testkit.Rows("1\x00\x00\x00\x00\x00\x00\x00\x00\x00 10 1")) // insert statement _, err := tk.Exec("insert into tt (id, c, _tidb_rowid) values(30000,10,1);") c.Assert(err.Error(), Equals, "insert, update and replace statements for _tidb_rowid are not supported.") // replace statement _, err = tk.Exec("replace into tt (id, c, _tidb_rowid) values(30000,10,1);") c.Assert(err.Error(), Equals, "insert, update and replace statements for _tidb_rowid are not supported.") // update statement _, err = tk.Exec("update tt set id = 2, _tidb_rowid = 1 where _tidb_rowid = 1") c.Assert(err.Error(), Equals, "insert, update and replace statements for _tidb_rowid are not supported.") tk.MustExec("update tt set id = 2 where _tidb_rowid = 1") tk.MustExec("admin check table tt;") tk.MustExec("drop table tt") // There is currently no real support for inserting, updating, and replacing _tidb_rowid statements. // After we support it, the following operations must be passed. // tk.MustExec("insert into tt (id, c, _tidb_rowid) values(30000,10,1);") // tk.MustExec("admin check table tt;") }