100 lines
2.9 KiB
Bash
Executable File
100 lines
2.9 KiB
Bash
Executable File
#!/usr/bin/env sh
|
|
|
|
#
|
|
# 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_path=$0
|
|
exporter_dir=$(cd $(dirname $0); pwd)
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--taskid=*)
|
|
task_id=${1#*=}
|
|
;;
|
|
--traceid=*)
|
|
trace_id=${1#*=}
|
|
;;
|
|
*)
|
|
export_path=$1
|
|
esac
|
|
shift
|
|
done
|
|
|
|
[[ ! ${task_id} || ! ${trace_id} || ! ${export_path} ]] \
|
|
&& echo 'Usage: sh tools/profile-exporter/profile_exporter.sh [--taskid] [--traceid] export_path' \
|
|
&& exit 1
|
|
|
|
[[ ! -d ${export_path} ]] \
|
|
&& echo "Cannot find export export_path path: ${export_path}" \
|
|
&& exit 1
|
|
|
|
# prepare paths
|
|
oap_libs_dir="${exporter_dir}/../../oap-libs"
|
|
exporter_log_file="${exporter_dir}/profile_exporter_log4j2.xml"
|
|
tool_application_config="${exporter_dir}/application.yml"
|
|
[[ ! -f ${tool_application_config} ]] \
|
|
&& echo "Cannot find oap application.yml" \
|
|
&& exit 1
|
|
[[ ! -d ${oap_libs_dir} ]] \
|
|
&& echo "Cannot find oap libs path" \
|
|
&& exit 1
|
|
|
|
# create current trace temporary path
|
|
work_dir="${export_path}/${trace_id}"
|
|
mkdir -p ${work_dir}
|
|
|
|
# prepare exporter files
|
|
mkdir -p "${work_dir}/config"
|
|
mkdir -p "${work_dir}/work"
|
|
cp ${exporter_log_file} ${work_dir}/config/log4j2.xml
|
|
# only persist core and storage module in application.yml config
|
|
cp ${tool_application_config} ${work_dir}/config/application.yml
|
|
|
|
# start export
|
|
echo "Exporting task: ${task_id}, trace: ${trace_id}, export_path: ${work_dir}"
|
|
JAVA_OPTS=" -Xms256M -Xmx512M"
|
|
_RUNJAVA=${JAVA_HOME}/bin/java
|
|
[ -z "$JAVA_HOME" ] && _RUNJAVA=java
|
|
|
|
CLASSPATH="${work_dir}/config:$CLASSPATH"
|
|
for i in "${oap_libs_dir}"/*.jar
|
|
do
|
|
CLASSPATH="$i:$CLASSPATH"
|
|
done
|
|
|
|
exec $_RUNJAVA ${JAVA_OPTS} -classpath $CLASSPATH org.apache.skywalking.oap.server.tool.profile.exporter.ProfileSnapshotExporter \
|
|
${task_id} ${trace_id} ${work_dir}/work &
|
|
wait
|
|
|
|
if [ `ls -l ${work_dir}/work | wc -l` -lt 2 ]; then
|
|
echo "Export failure!"
|
|
exit 1
|
|
fi
|
|
|
|
# compress files(only compress work data, no config)
|
|
echo "Compressing exported directory: ${work_dir}"
|
|
CURRENT_DIR="$(cd "$(dirname $0)"; pwd)"
|
|
cd ${work_dir}
|
|
tar zcvf "${trace_id}.tar.gz" "./work"
|
|
mv "${trace_id}.tar.gz" "../"
|
|
cd $CURRENT_DIR
|
|
|
|
# clear work files
|
|
rm -rf "${work_dir}"
|
|
|
|
echo "Profile export finished: ${work_dir}.tar.gz"
|