rocksdb/t/unique_check.test

--source include/have_rocksdb.inc
--source include/have_debug_sync.inc
--source include/count_sessions.inc

# For GitHub issue#167 -- Unique key check doesn't work

connect (con1, localhost, root,,);
connect (con2, localhost, root,,);
connect (con3, localhost, root,,);
connection default;

create table t1 (id int, value int, primary key (id)) engine=rocksdb;
create table t2 (id int, id2 int, value int, primary key (id), unique key (id2)) engine=rocksdb;

# 1) second insert should be blocked at GetForUpdate(), then returning duplicate key error
# after con1's commit
connection con1;
begin;
insert into t1 values (1,1);

connection con2;
let $ID= `select connection_id()`;
set session rocksdb_lock_wait_timeout=50;
begin;
send insert into t1 values (1,2);

connection con1;
let $wait_condition= select 1 from INFORMATION_SCHEMA.PROCESSLIST
                     where ID = $ID and STATE = "Waiting for row lock";
--source include/wait_condition.inc
commit;

connection con2;
--error ER_DUP_ENTRY
reap;
commit;
select * from t1;
truncate table t1;

# 2) same as 1) but using secondary unique key constraint
connection con1;
begin;
insert into t2 values (1,1,1);

connection con2;
begin;
send insert into t2 values (2,1,2);

connection con1;
--source include/wait_condition.inc
commit;

connection con2;
--error ER_DUP_ENTRY
reap;
commit;
select * from t2;
truncate table t2;

# 3) similar to 1),2) but rolled back
connection con1;
begin;
insert into t1 values (1,1);

connection con2;
begin;
send insert into t1 values (1,2);

connection con1;
--source include/wait_condition.inc
rollback;

connection con2;
reap;
commit;
select * from t1;
truncate table t1;

connection con1;
begin;
insert into t2 values (1,1,1);

connection con2;
begin;
send insert into t2 values (2,1,2);

connection con1;
--source include/wait_condition.inc
rollback;

connection con2;
reap;
commit;
select * from t2;
truncate table t2;


# 4) simulating T1 GetForUpdate() -> T2 GetForUpdate(). T2 should fail with lock wait timeout.
connection con1;
set debug_sync='rocksdb.update_write_row_after_unique_check SIGNAL parked1 WAIT_FOR go1';
send insert into t1 values (1,1);

connection con2;
set debug_sync='rocksdb.update_write_row_after_unique_check SIGNAL parked2 WAIT_FOR go2';
send insert into t2 values (1,1,1);

connection default;
set debug_sync='now WAIT_FOR parked1';
set debug_sync='now WAIT_FOR parked2';

connection con3;
set session rocksdb_lock_wait_timeout=1;
--error ER_LOCK_WAIT_TIMEOUT
insert into t1 values (1,2);
--error ER_LOCK_WAIT_TIMEOUT
insert into t2 values (2,1,2);

connection default;
set debug_sync='now SIGNAL go1';
set debug_sync='now SIGNAL go2';

connection con1;
reap;

connection con2;
reap;

connection default;
--error ER_DUP_ENTRY
insert into t1 values (1,2);
--error ER_DUP_ENTRY
insert into t2 values (2,1,2);
select * from t1;
select * from t2;
# Cleanup
connection default;
set debug_sync='RESET';
disconnect con1;
disconnect con2;
disconnect con3;
drop table t1, t2;

# skip_unique checks should skip checks only for tables that don't have
# secondary indexes

# table with PK only
create table t1 (id int, value int, primary key (id)) engine=rocksdb;
# table with PK and SK
create table t2 (id int, id2 int, value int, primary key (id), unique key (id2)) engine=rocksdb;
# table with hidden PK
create table t3 (id int, value int) engine=rocksdb;

SET @old_val = @@session.unique_checks;
set @@session.unique_checks = FALSE;

insert into t1 values (1, 1), (1, 2);
--error ER_DUP_ENTRY
insert into t2 values (1, 1, 1), (1, 2, 1);
insert into t3 values (1, 1), (1, 1);

set @@session.unique_checks = @old_val;
# cleanup
drop table t1, t2, t3;

--echo #
--echo #  Issue#1026: MyRocks may return "Can't find record" which is unexpected
--echo #

connect (con1, localhost, root,,);
connection default;

CREATE TABLE t1 (
  id bigint(20) unsigned NOT NULL AUTO_INCREMENT,
  a varchar(36) NOT NULL ,
  b varchar(128) NOT NULL ,
  c varchar(10240) NOT NULL ,
  PRIMARY KEY (id),
  UNIQUE KEY uniq_idx (a,b)
) ENGINE=ROCKSDB;

insert into t1 values (1,1,1,1), (2,2,2,2);

## TRX A:
begin;
set debug_sync='rocksdb.after_unique_pk_check SIGNAL trx_a_sleep WAIT_FOR trx_a_cont';
send
insert into t1(a,b,c) values (10,'file_type','trx-a') on duplicate key update c=values(c);

connection con1;
# TRX B:
set debug_sync='now WAIT_FOR trx_a_sleep';
begin;
insert into t1(a,b,c) values (10,'file_type','trx-b') on duplicate key update c=values(c);
commit;
set debug_sync='now SIGNAL trx_a_cont';


connection default;
--error ER_LOCK_DEADLOCK
reap;

disconnect con1;
rollback;
drop table t1;

set debug_sync='RESET';

--echo #
--echo # Issue#1026, testcase #2 (with READ-COMMITTED).
--echo #

CREATE table t1 (
  pk0 int primary key auto_increment,
  sk int,
  val int default 0,
  unique(sk)
) engine=rocksdb;

insert into t1 (sk) values (20), (30);

connect (con1, localhost, root,,);

connection con1;
set transaction_isolation='READ-COMMITTED';
begin;
set debug_sync='rocksdb.after_unique_sk_check SIGNAL here WAIT_FOR go';
send
insert into t1 (sk) values (1), (2) on duplicate key update val = val + 1;

connection default;
set debug_sync='now WAIT_FOR here';
begin;
insert into t1 (sk) values (2);
commit;
set debug_sync='now SIGNAL go';

connection con1;
reap;
commit;

connection default;
disconnect con1;

select * from t1;

drop table t1;

set debug_sync='RESET';

--echo #
--echo # Issue#1026, testcase #3 (with READ-COMMITTED and PK).
--echo #

CREATE table t1 (
  pk int primary key,
  val int default 0
) engine=rocksdb;

insert into t1 (pk) values (20), (30);

connect (con1, localhost, root,,);

connection con1;
set transaction_isolation='READ-COMMITTED';
begin;
set debug_sync='rocksdb.after_unique_pk_check SIGNAL here WAIT_FOR go';
send
insert into t1 (pk) values (1), (2) on duplicate key update val = val + 1;

connection default;
set debug_sync='now WAIT_FOR here';
begin;
insert into t1 (pk) values (2);
commit;
set debug_sync='now SIGNAL go';

connection con1;
reap;
commit;

connection default;
disconnect con1;

select * from t1;

drop table t1;

set debug_sync='RESET';

--source include/wait_until_count_sessions.inc