mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 08:32:02 +00:00
40 lines
1.4 KiB
Bash
Executable File
40 lines
1.4 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
CURDIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
# shellcheck source=../shell_config.sh
|
|
. "$CURDIR"/../shell_config.sh
|
|
|
|
|
|
function wait_query_started()
|
|
{
|
|
local query_id="$1"
|
|
$CLICKHOUSE_CLIENT --query "SYSTEM FLUSH LOGS"
|
|
while [[ $($CLICKHOUSE_CLIENT --query="SELECT count() FROM system.query_log WHERE query_id='$query_id' AND current_database = currentDatabase()") == 0 ]]; do
|
|
sleep 0.1;
|
|
$CLICKHOUSE_CLIENT --query "SYSTEM FLUSH LOGS;"
|
|
done
|
|
}
|
|
|
|
function kill_query()
|
|
{
|
|
local query_id="$1"
|
|
$CLICKHOUSE_CLIENT --query "KILL QUERY WHERE query_id='$query_id'" >/dev/null
|
|
while [[ $($CLICKHOUSE_CLIENT --query="SELECT count() FROM system.processes WHERE query_id='$query_id'") != 0 ]]; do sleep 0.1; done
|
|
}
|
|
|
|
|
|
sleep_query_id="sleep_query_id_02932_kill_query_sleep_${CLICKHOUSE_DATABASE}_$RANDOM"
|
|
|
|
# This sleep query wants to sleep for 1000 seconds (which is too long).
|
|
# We're going to cancel this query later.
|
|
sleep_query="SELECT sleep(1000)"
|
|
|
|
$CLICKHOUSE_CLIENT --query_id="$sleep_query_id" --function_sleep_max_microseconds_per_block="1000000000" --query "$sleep_query" >/dev/null 2>&1 &
|
|
wait_query_started "$sleep_query_id"
|
|
|
|
echo "Cancelling query"
|
|
kill_query "$sleep_query_id"
|
|
|
|
$CLICKHOUSE_CLIENT --query "SYSTEM FLUSH LOGS;"
|
|
$CLICKHOUSE_CLIENT --query "SELECT exception FROM system.query_log WHERE query_id='$sleep_query_id' AND current_database = currentDatabase()" | grep -oF "QUERY_WAS_CANCELLED"
|