ClickHouse/tests/queries/0_stateless/transactions.lib
Alexander Tokmakov 0fc8c259c7 fix tests
2021-12-24 16:14:17 +03:00

79 lines
2.6 KiB
Bash
Executable File

#!/usr/bin/env bash
# shellcheck disable=SC2015
# Useful to run queries in parallel sessions
function tx()
{
tx_num=$1
query=$2
session="${CLICKHOUSE_TEST_ZOOKEEPER_PREFIX}_tx$tx_num"
query_id="${session}_${RANDOM}"
url_without_session="https://${CLICKHOUSE_HOST}:${CLICKHOUSE_PORT_HTTPS}/?"
url="${url_without_session}session_id=$session&query_id=$query_id&database=$CLICKHOUSE_DATABASE"
${CLICKHOUSE_CURL} -m 60 -sSk "$url" --data "$query" | sed "s/^/tx$tx_num\t/"
}
# Waits for the last query in session to finish
function tx_wait() {
tx_num=$1
session="${CLICKHOUSE_TEST_ZOOKEEPER_PREFIX}_tx$tx_num"
# try get pid of previous query
query_pid=""
tmp_file_name="${CLICKHOUSE_TMP}/tmp_tx_${CLICKHOUSE_TEST_ZOOKEEPER_PREFIX}"
query_id_and_pid=$(grep -F "$session" "$tmp_file_name" 2>/dev/null | tail -1) ||:
read -r query_id query_pid <<< "$query_id_and_pid" ||:
# wait for previous query in transaction
if [ -n "$query_pid" ]; then
timeout 5 tail --pid=$query_pid -f /dev/null && return ||:
fi
# there is no pid (or maybe we got wrong one), so wait using system.processes (it's less reliable)
count=0
while [[ $($CLICKHOUSE_CLIENT -q "SELECT count() FROM system.processes WHERE query_id LIKE '$session%'") -gt 0 ]]; do
sleep 0.5
count=$((count+1))
if [ "$count" -gt 120 ]; then
echo "timeout while waiting for $tx_num"
break
fi
done;
}
# Wait for previous query in session to finish, starts new one asynchronously
function tx_async()
{
tx_num=$1
query=$2
tx_wait "$tx_num"
session="${CLICKHOUSE_TEST_ZOOKEEPER_PREFIX}_tx$tx_num"
query_id="${session}_${RANDOM}"
url_without_session="https://${CLICKHOUSE_HOST}:${CLICKHOUSE_PORT_HTTPS}/?"
url="${url_without_session}session_id=$session&query_id=$query_id&database=$CLICKHOUSE_DATABASE"
# We cannot be sure that query will actually start execution and appear in system.processes before the next call to tx_wait
# Also we cannot use global map in bash to store last query_id for each tx_num, so we use tmp file...
tmp_file_name="${CLICKHOUSE_TMP}/tmp_tx_${CLICKHOUSE_TEST_ZOOKEEPER_PREFIX}"
# run query asynchronously
${CLICKHOUSE_CURL} -m 60 -sSk "$url" --data "$query" | sed "s/^/tx$tx_num\t/" &
query_pid=$!
echo -e "$query_id\t$query_pid" >> "$tmp_file_name"
}
# Wait for previous query in session to finish, execute the next one synchronously
function tx_sync()
{
tx_num=$1
query=$2
tx_wait "$tx_num"
tx "$tx_num" "$query"
}