ec517a53a8
upgrade clang-format version to 16 move thrift to fe-common fix core dump on pipeline engine when operator canceled and not prepared
119 lines
3.3 KiB
Bash
119 lines
3.3 KiB
Bash
#!/usr/bin/env bash
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you 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.
|
|
|
|
# print usage
|
|
usage() {
|
|
echo "
|
|
Description:
|
|
This script is used to restore the tablets from trash. It supports single mode
|
|
and batch mode.
|
|
In single mode, it will restore just one tablet.
|
|
In batch mode, it will restore all the tablets specified in file. The content
|
|
of the file is comma-split tablet id and schema hash, like the following:
|
|
12345,11111
|
|
12346,11111
|
|
12347,11111
|
|
|
|
Usage: $0 <options>
|
|
Optional options:
|
|
-h | --help print help info
|
|
-b | --backend backend http service, default: http://127.0.0.1/8040
|
|
-t | --tablet_id tablet id to restore
|
|
-s | --schema_hash tablet related schema hash
|
|
-f | --file file with lines containing comma-split tablet id and schema hash
|
|
|
|
Examples:
|
|
batch mode:
|
|
bash restore_tablet_tool.sh -b http://127.0.0.1:8040 -f tablets.txt
|
|
bash restore_tablet_tool.sh --backend http://127.0.0.1:8040 --file tablets.txt
|
|
|
|
single mode:
|
|
bash restore_tablet_tool.sh -b http://127.0.0.1:8040 -t 12345 -s 11111
|
|
bash restore_tablet_tool.sh --backend http://127.0.0.1:8040 --tablet_id 12345 --schema_hash 11111
|
|
"
|
|
exit 1
|
|
}
|
|
|
|
if ! OPTS=$(getopt \
|
|
-n "$0" \
|
|
-o 'b:t:s:f:' \
|
|
-l 'server:,tablet_id:,schema_hash:,file:,help' \
|
|
-- "$@"); then
|
|
usage
|
|
fi
|
|
|
|
eval set -- "${OPTS}"
|
|
|
|
SERVER='http://127.0.0.1/8040'
|
|
TABLET_ID=''
|
|
SCHEMA_HASH=''
|
|
FILENAME=''
|
|
BATCH_MODE=false
|
|
|
|
while true; do
|
|
case "$1" in
|
|
-b | --backend)
|
|
SERVER="$2"
|
|
shift 2
|
|
;;
|
|
-f | --file)
|
|
FILENAME="$2"
|
|
BATCH_MODE=true
|
|
shift 2
|
|
;;
|
|
-t | --tablet_id)
|
|
TABLET_ID="$2"
|
|
shift 2
|
|
;;
|
|
-s | --schema_hash)
|
|
SCHEMA_HASH="$2"
|
|
shift 2
|
|
;;
|
|
-h | --help)
|
|
usage
|
|
;;
|
|
--)
|
|
shift
|
|
break
|
|
;;
|
|
*)
|
|
echo "Internal error!"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
restore_tablet() {
|
|
echo "start to restore tablet id:$2, schema hash:$3"
|
|
curl -X POST "$1/api/restore_tablet?tablet_id=$2&schema_hash=$3"
|
|
echo -e "\n"
|
|
}
|
|
|
|
if ${BATCH_MODE}; then
|
|
while read -r line; do
|
|
# split the comma-split line
|
|
# format: tablet_id,schema_hash
|
|
read -r -a fields <<<"${line/,/ }"
|
|
TABLET_ID="${fields[0]}"
|
|
SCHEMA_HASH="${fields[1]}"
|
|
restore_tablet "${SERVER}" "${TABLET_ID}" "${SCHEMA_HASH}"
|
|
done <"${FILENAME}"
|
|
else
|
|
restore_tablet "${SERVER}" "${TABLET_ID}" "${SCHEMA_HASH}"
|
|
fi
|