#!/bin/bash
#
# author: dijia478
# date: 2020-8-20 18:14:38
# desc: 本腳本需要放到arthas的目錄中，連同整個目錄一起復制到docker容器中去。主要用途為在容器中切換目標服務的用戶，并啟動arthas

echo "開始查詢目標服務的進程id和用戶..."
PID=`ps -eo pid,user=aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa -o args | grep java | grep -v grep | awk '{print $1}'`
echo "目標服務的進程id為${PID}"
USER=`ps -eo pid,user=aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa -o args | grep java | grep -v grep | awk '{print $2}'`
echo "目標服務的用戶為${USER}"

if [[ ! -d "/home/${USER}" ]]
then
  mkdir -p /home/${USER}
  echo "創(chuàng)建目錄/home/${USER}"
fi
chmod 777 /home/${USER}

echo "開始切換用戶并啟動arthas..."
# 下面的arthas路徑需要修改，并且要和startArthas.sh腳本中保持一致
ARTHAS_PATH="/opt/arthas"
su ${USER} -c "java -jar ${ARTHAS_PATH}/arthas-client.jar 127.0.0.1 3658 -c 'stop'"
su ${USER} -c "java -jar ${ARTHAS_PATH}/arthas-boot.jar ${PID}"

#! /bin/bash
#
# author: dijia478
# date: 2020-9-18 10:36:27
# desc: 本腳本主要用途為啟動arthas診斷工具來診斷某個docker中java服務

if [[ ${1} == '' ]]
then
  echo "請選擇一個服務："
  sudo docker ps | awk 'NR>1 {print $2}'
  exit 0
fi

echo "開始尋找服務${1}的容器..."
DOCKER_LIST=`sudo docker ps | awk 'NR>1 {print $2}'`
FLAG=0
for i in ${DOCKER_LIST[@]}
do
  if [[ ${i} == ${1} ]]
  then
    FLAG=1
    break
  fi
done

if [[ ${FLAG} == 0 ]]
then
  DOCKER_NAME=`sudo docker ps | awk 'NR>1 {print $2}' | grep ${1}`
  if [[ ${DOCKER_NAME} == '' ]]
  then
    echo "未找到該服務的容器，請重新選擇服務："
    sudo docker ps | awk 'NR>1 {print $2}'
  else
    echo "請輸入服務的完整名稱："
    sudo docker ps | awk 'NR>1 {print $2}' | grep ${1}
  fi

else
  ID=`sudo docker ps --filter ancestor=${1} | awk '{print $1}' | sed -n '2p'`
  echo "找到容器${ID}"

  echo "開始復制arthas到容器中..."
  # 下面的arthas路徑需要修改，并且要和arthasDocker.sh腳本中保持一致
  ARTHAS_PATH="/opt/arthas"
  sudo docker exec -it ${ID} /bin/bash -c "rm -rf ${ARTHAS_PATH}"
  sudo docker cp ${ARTHAS_PATH} ${ID}:${ARTHAS_PATH}
  echo "復制完成"

  echo "即將進入容器中..."
  sudo docker exec -it ${ID} /bin/bash -c "bash ${ARTHAS_PATH}/arthasDocker.sh"
fi