91 lines
3.5 KiB
Bash
91 lines
3.5 KiB
Bash
#!/bin/sh
|
|
#
|
|
# Copyright 2020 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,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
set -eu
|
|
|
|
CUR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
# FIXME: auto-random is only stable on master currently.
|
|
check_cluster_version 4 0 0 AUTO_RANDOM || exit 0
|
|
|
|
for backend in tidb local; do
|
|
if [ "$backend" = 'local' ]; then
|
|
check_cluster_version 4 0 0 'local backend' || continue
|
|
fi
|
|
|
|
run_sql 'DROP DATABASE IF EXISTS auto_random;'
|
|
run_lightning --backend $backend
|
|
|
|
run_sql "SELECT count(*) from auto_random.t"
|
|
check_contains "count(*): 6"
|
|
|
|
run_sql "SELECT id & b'000001111111111111111111111111111111111111111111111111111111111' as inc FROM auto_random.t"
|
|
check_contains 'inc: 1'
|
|
check_contains 'inc: 2'
|
|
check_contains 'inc: 3'
|
|
if [ "$backend" = 'tidb' ]; then
|
|
check_contains 'inc: 4'
|
|
check_contains 'inc: 5'
|
|
check_contains 'inc: 6'
|
|
NEXT_AUTO_RAND_VAL=7
|
|
else
|
|
check_contains 'inc: 25'
|
|
check_contains 'inc: 26'
|
|
check_contains 'inc: 27'
|
|
NEXT_AUTO_RAND_VAL=28
|
|
fi
|
|
|
|
# tidb backend randomly generate the auto-random bit for each statement, so with 2 statements,
|
|
# the distinct auto_random prefix values can be 1 or 2, so we skip this check with tidb backend
|
|
if [ "$backend" != 'tidb' ]; then
|
|
run_sql "select count(distinct id >> 58) as count from auto_random.t"
|
|
check_contains "count: 2"
|
|
fi
|
|
|
|
# auto random base is 4
|
|
run_sql "SELECT max(id & b'000001111111111111111111111111111111111111111111111111111111111') >= $NEXT_AUTO_RAND_VAL as ge FROM auto_random.t"
|
|
check_contains 'ge: 0'
|
|
run_sql "INSERT INTO auto_random.t VALUES ();"
|
|
run_sql "SELECT max(id & b'000001111111111111111111111111111111111111111111111111111111111') >= $NEXT_AUTO_RAND_VAL as ge FROM auto_random.t"
|
|
check_contains 'ge: 1'
|
|
done
|
|
|
|
function run_for_auro_random_data2() {
|
|
create_table=$1
|
|
run_sql 'DROP DATABASE IF EXISTS auto_random;'
|
|
run_sql 'CREATE DATABASE IF NOT EXISTS auto_random;'
|
|
run_sql "$create_table"
|
|
run_lightning --backend $backend -d "$CUR/data2"
|
|
run_sql 'select count(*) as count from auto_random.t where c > 0'
|
|
check_contains "count: 2"
|
|
run_sql 'select count(*) as count from auto_random.t where a=1 and b=11'
|
|
check_contains "count: 1"
|
|
run_sql 'select count(*) as count from auto_random.t where a=2 and b=22'
|
|
check_contains "count: 1"
|
|
}
|
|
|
|
for backend in tidb local; do
|
|
if [ "$backend" = 'local' ]; then
|
|
check_cluster_version 4 0 0 'local backend' || continue
|
|
fi
|
|
|
|
run_for_auro_random_data2 'create table auto_random.t(c bigint auto_random primary key, a int, b int)'
|
|
run_for_auro_random_data2 'create table auto_random.t(a int, b int, c bigint auto_random primary key)'
|
|
# composite key and auto_random is the first column
|
|
run_for_auro_random_data2 'create table auto_random.t(c bigint auto_random, a int, b int, primary key(c, a))'
|
|
# composite key and auto_random is not the first column
|
|
run_for_auro_random_data2 'create table auto_random.t(a int, b int, c bigint auto_random, primary key(c, a))'
|
|
done
|