| # |
| # 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. |
| # |
| |
| #!/bin/bash |
| |
| echo --------------------- |
| echo Start collecting info |
| echo --------------------- |
| |
| if [ -z "${IOTDB_HOME}" ]; then |
| export IOTDB_HOME="`dirname "$0"`/../.." |
| fi |
| |
| COLLECTION_FILE="collection.txt" |
| COLLECTION_DIR="iotdb-info" |
| |
| HELP="Usage: $0 [-h <ip>] [-p <port>] [-u <username>] [-pw <password>] [-jp <jdk_path>] [-dd <data_dir>]" |
| |
| user_param="root" |
| passwd_param="root" |
| host_param="127.0.0.1" |
| port_param="6667" |
| jdk_path_param="" |
| data_dir_param="data/datanode/data" |
| |
| get_property_value() { |
| local file="$1" # Properties path |
| local key="$2" # key name |
| local value="" |
| |
| if [ -f "$file" ]; then |
| local line=$(grep "^$key=" "$file") |
| value="${line#*=}" |
| fi |
| |
| echo "$value" |
| } |
| |
| convert_unit() { |
| local size=$1 |
| local target_unit=$2 |
| local converted_size=$size |
| |
| if [ "$target_unit" == "M" ]; then |
| converted_size=$(awk "BEGIN {printf \"%.2f\", $size / 1024}") |
| elif [ "$target_unit" == "G" ]; then |
| converted_size=$(awk "BEGIN {printf \"%.2f\", $size / (1024 * 1024)}") |
| elif [ "$target_unit" == "T" ]; then |
| converted_size=$(awk "BEGIN {printf \"%.2f\", $size / (1024 * 1024 * 1024)}") |
| fi |
| |
| echo "$converted_size" |
| } |
| |
| choose_unit() { |
| local size=$1 |
| local unit="" |
| |
| if [ "$size" -lt 1024 ]; then |
| unit="K" |
| elif [ "$size" -lt 1048576 ]; then |
| unit="M" |
| elif [ "$size" -lt 1073741824 ]; then |
| unit="G" |
| else |
| unit="T" |
| fi |
| |
| echo "$unit" |
| } |
| |
| if [ -f "$IOTDB_HOME/conf/iotdb-system.properties" ]; then |
| properties_file="$IOTDB_HOME/conf/iotdb-system.properties" |
| else |
| properties_file="$IOTDB_HOME/conf/iotdb-datanode.properties" |
| fi |
| data_dir_key="dn_data_dirs" |
| value=$(get_property_value "$properties_file" "$data_dir_key") |
| if [ -n "$value" ]; then |
| data_dir_param=$value |
| fi |
| |
| while true; do |
| case "$1" in |
| -u) |
| user_param="$2" |
| shift 2 |
| ;; |
| -pw) |
| passwd_param="$2" |
| shift 2 |
| ;; |
| -h) |
| host_param="$2" |
| shift 2 |
| ;; |
| -p) |
| port_param="$2" |
| shift 2 |
| ;; |
| -jp) |
| jdk_path_param="$2" |
| shift 2 |
| ;; |
| -dd) |
| data_dir_param="$2" |
| shift 2 |
| ;; |
| "") |
| break |
| ;; |
| *) |
| echo "Unrecognized options:$1" |
| echo "${HELP}" |
| exit 1 |
| ;; |
| esac |
| done |
| |
| timestamp=$(date +"%Y%m%d%H%M%S") |
| |
| zip_name="collection-$timestamp.zip" |
| |
| zip_directory="$IOTDB_HOME/" |
| |
| files_to_zip="${COLLECTION_FILE} $IOTDB_HOME/conf" |
| |
| rm -rf $IOTDB_HOME/$COLLECTION_DIR |
| mkdir -p $IOTDB_HOME/$COLLECTION_DIR/logs |
| |
| { |
| echo '===================== System Info =====================' |
| case "$(uname)" in |
| Linux) |
| read -r system_memory unused_memory <<< "$(free | awk '/Mem:/{print $2, $4}')" |
| system_cpu_cores=$(grep -c 'processor' /proc/cpuinfo) |
| system_cpu_name=$(grep 'model name' /proc/cpuinfo | head -n 1 | awk -F ':' '{$1=$1}1') |
| system_os_info=$(get_property_value /etc/os-release PRETTY_NAME) |
| ;; |
| FreeBSD) |
| read -r system_memory_in_bytes unused_memory <<< "$(sysctl -n hw.realmem vm.stats.vm.v_inactive_count)" |
| system_memory=$((system_memory_in_bytes / 1024)) |
| unused_memory=$(sysctl -n vm.stats.vm.v_inactive_count) |
| system_cpu_cores=$(sysctl -n hw.ncpu) |
| system_cpu_name=$(sysctl -n hw.model) |
| system_os_info=$(uname -r) |
| ;; |
| SunOS) |
| read -r system_memory unused_memory <<< "$(prtconf | awk '/Memory size:/ {print $3}') $(kstat -p unix:0:system_pages:pagestotal -s unix:0:system_pages:pagesfree | awk '{print $2}')" |
| system_cpu_cores=$(psrinfo | wc -l) |
| system_cpu_name=$(psrinfo -pv | grep "The" | awk -F':' '{print $2}' | awk '{$1=$1}1' | head -n 1) |
| system_os_info=$(uname -v) |
| ;; |
| Darwin) |
| read -r system_memory_in_bytes unused_memory <<< "$(sysctl -n hw.memsize) $(vm_stat | awk '/Pages free/ {print $3 * 4}')" |
| system_memory=$((system_memory_in_bytes / 1024)) |
| system_cpu_cores=$(sysctl -n hw.ncpu) |
| system_cpu_name=$(sysctl -n machdep.cpu.brand_string) |
| system_os_info=$(sw_vers -productName) |
| ;; |
| *) |
| system_memory="Unknown" |
| unused_memory="Unknown" |
| system_cpu_cores="Unknown" |
| system_cpu_name="Unknown" |
| system_os_info="Unknown" |
| ;; |
| esac |
| echo "Operating System: $system_os_info" |
| echo "CPU Name: $system_cpu_name" |
| echo "CPU Cores: $system_cpu_cores" |
| total_unit=$(choose_unit "$system_memory") |
| echo "System Memory Total: $(convert_unit "$system_memory" "$total_unit") $total_unit" |
| unuse_unit=$(choose_unit "$unused_memory") |
| echo "Unused Memory: $(convert_unit "$unused_memory" "$unuse_unit") $unuse_unit" |
| |
| } >> "$COLLECTION_FILE" |
| |
| { |
| echo '===================== JDK Version=====================' |
| if [ -n "$jdk_path_param" ]; then |
| if [ -d "$jdk_path_param" ]; then |
| $jdk_path_param/bin/java -version 2>&1 |
| else |
| echo "Invalid JDK path: $jdk_path_param" |
| exit 1 |
| fi |
| else |
| if [ -n "$JAVA_HOME" ]; then |
| for java in "$JAVA_HOME"/bin/amd64/java "$JAVA_HOME"/bin/java; do |
| if [ -x "$java" ]; then |
| JAVA="$java" |
| break |
| fi |
| done |
| else |
| JAVA=java |
| fi |
| |
| if [ -z $JAVA ] ; then |
| echo Unable to find java executable. Check JAVA_HOME and PATH environment variables. > /dev/stderr |
| exit 1; |
| fi |
| java -version 2>&1 |
| fi |
| } >> "$COLLECTION_FILE" |
| |
| { |
| echo '=================== Activation Info====================' |
| if [ -d "$(dirname "$0")/$IOTDB_HOME/activation" ]; then |
| if [ -f "$(dirname "$0")/$IOTDB_HOME/activation/license" ]; then |
| echo "Active" |
| else |
| echo "Not active" |
| fi |
| else |
| echo "Open source version" |
| fi |
| } >> "$COLLECTION_FILE" |
| |
| { |
| if [[ -d "$IOTDB_HOME/logs/" ]]; then |
| for file in $IOTDB_HOME/logs/*.log; do |
| if [[ $file =~ \.log$ ]]; then |
| cp "$file" "$IOTDB_HOME/$COLLECTION_DIR/logs" |
| fi |
| done |
| else |
| echo "Directory $IOTDB_HOME/logs/ does not exist." |
| fi |
| } |
| |
| calculate_directory_size() { |
| local file_type="$1" |
| local total_size=0 |
| IFS=',' read -ra dirs <<< "$data_dir_param" |
| for dir in "${dirs[@]}"; do |
| if [[ $dir == /* ]]; then |
| iotdb_data_dir="$dir/$file_type" |
| else |
| iotdb_data_dir="$IOTDB_HOME/$dir/$file_type" |
| fi |
| if [ -d "$iotdb_data_dir" ]; then |
| local size=$(du -s "$iotdb_data_dir" | awk '{print $1}') |
| total_size=$((total_size + size)) |
| fi |
| done |
| echo "$total_size" |
| } |
| |
| calculate_file_num() { |
| local file_type="$1" |
| local total_num=0 |
| IFS=',' read -ra dirs <<< "$data_dir_param" |
| for dir in "${dirs[@]}"; do |
| if [[ $dir == /* ]]; then |
| iotdb_data_dir="$dir/$file_type" |
| else |
| iotdb_data_dir="$IOTDB_HOME/$dir/$file_type" |
| fi |
| if [ -d "$iotdb_data_dir" ]; then |
| local num=$(find "$iotdb_data_dir" -type f ! -name "*.tsfile.resource" | wc -l) |
| total_num=$((total_num + num)) |
| fi |
| done |
| echo "$total_num" |
| } |
| |
| { |
| echo '===================== TsFile Info=====================' |
| sequence="sequence" |
| unsequence="unsequence" |
| echo "sequence(tsfile number): $(calculate_file_num "$sequence")" |
| echo "unsequence(tsfile number): $(calculate_file_num "$unsequence")" |
| total_size=$(calculate_directory_size "$sequence") |
| unit=$(choose_unit "$total_size") |
| total_size_in_unit=$(convert_unit "$total_size" "$unit") |
| echo "sequence(tsfile size): $total_size_in_unit $unit" |
| total_size=$(calculate_directory_size "$unsequence") |
| unit=$(choose_unit "$total_size") |
| total_size_in_unit=$(convert_unit "$total_size" "$unit") |
| echo "unsequence(tsfile size): $total_size_in_unit $unit" |
| } >> "$COLLECTION_FILE" |
| |
| execute_command_and_append_to_file() { |
| local command=$1 |
| { |
| echo "=================== $command ====================" |
| if [ -n "$jdk_path_param" ]; then |
| export JAVA_HOME="$jdk_path_param";"$IOTDB_HOME"/sbin/start-cli.sh -h "$host_param" -p "$port_param" -u "$user_param" -pw "$passwd_param" -e "$command" |
| else |
| "$IOTDB_HOME"/sbin/start-cli.sh -h "$host_param" -p "$port_param" -u "$user_param" -pw "$passwd_param" -e "$command" |
| fi |
| } >> "$COLLECTION_FILE" |
| } |
| |
| execute_command_and_append_to_file 'show version' |
| execute_command_and_append_to_file 'show cluster details' |
| execute_command_and_append_to_file 'show regions' |
| execute_command_and_append_to_file 'show databases' |
| execute_command_and_append_to_file 'count devices' |
| execute_command_and_append_to_file 'count timeseries' |
| |
| |
| mv $COLLECTION_FILE $IOTDB_HOME/$COLLECTION_DIR |
| cp -r $IOTDB_HOME/conf $IOTDB_HOME/$COLLECTION_DIR |
| zip -r "$IOTDB_HOME/$zip_name" $IOTDB_HOME/$COLLECTION_DIR |
| echo "Program execution completed, file name is $zip_name" |
