如果Java进程的用户 与 执行脚本的当前用户 不同，提示用户使用sudo再执行

这样可以避免在脚本做的sudo操作，需要用户输入，而Block脚本。
在服务器上脚本自动执行，不Block的特点很重要。

Author: oldratlee

README.md

@@ -21,7 +21,6 @@ source <(curl -fsSL https://raw.githubusercontent.com/oldratlee/useful-scripts/m
 
 1. [show-busy-java-threads.sh](docs/java.md#beer-show-busy-java-threadssh)  
     打印出在运行的`Java`进程中，消耗`CPU`最多的线程栈。用于快速排查`Java`的性能问题。
-    如果执行脚本的用户不是root,需要前面加上sudo
 1. [show-duplicate-java-classes](docs/java.md#beer-show-duplicate-java-classes)  
     找出`jar`文件和`class`目录中的重复类。用于排查`Java`类冲突问题。
 1. [find-in-jars.sh](docs/java.md#beer-find-in-jarssh)  

docs/java.md

@@ -17,6 +17,13 @@ show-busy-java-threads.sh -c <要显示的线程栈数>
 # 上面会从所有的Java进程中找出最消耗CPU的线程，这样用更方便。
 
 show-busy-java-threads.sh -c <要显示的线程栈数> -p <指定的Java Process>
+
+##############################
+# 注意：
+##############################
+# 如果Java进程的用户 与 执行脚本的当前用户 不同，则jstack不了这个Java进程。
+# 为了能切换到Java进程的用户，需要加sudo来执行，即可以解决：
+sudo show-busy-java-threads.sh
 ```
 
 ### 示例

show-busy-java-threads.sh

@@ -8,6 +8,7 @@
 # @author Jerry Lee
 
 readonly PROG=`basename $0`
+readonly -a COMMAND_LINE=("$0" "$@")
 
 usage() {
     cat <<EOF
@@ -58,6 +59,24 @@ redEcho() {
     } || echo "$@"
 }
 
+yellowEcho() {
+    [ -c /dev/stdout ] && {
+        # if stdout is console, turn on color output.
+        echo -ne "\033[1;33m"
+        echo -n "$@"
+        echo -e "\033[0m"
+    } || echo "$@"
+}
+
+blueEcho() {
+    [ -c /dev/stdout ] && {
+        # if stdout is console, turn on color output.
+        echo -ne "\033[1;36m"
+        echo -n "$@"
+        echo -e "\033[0m"
+    } || echo "$@"
+}
+
 # Check the existence of jstack command!
 if ! which jstack &> /dev/null; then
     [ -z "$JAVA_HOME" ] && {
@@ -96,18 +115,27 @@ printStackOfThread() {
 
         [ ! -f "${jstackFile}" ] && {
             {
-                if [ "${user}" == "${USER}" ];then
+                if [ "${user}" == "${USER}" ]; then
                     jstack ${pid} > ${jstackFile}
                 else
-                    sudo -u ${user} jstack ${pid} > ${jstackFile}
+                    if [ $UID == 0 ]; then
+                        sudo -u ${user} jstack ${pid} > ${jstackFile}
+                    else
+                        redEcho "[$((count++))] Fail to jstack Busy(${pcpu}%) thread(${threadId}/0x${threadId0x}) stack of java process(${pid}) under user(${user})."
+                        redEcho "User of java process($user) is not current user($USER), need sudo to run again:"
+                        yellowEcho "    sudo ${COMMAND_LINE[@]}"
+                        echo
+                        continue
+                    fi
                 fi
             } || {
-                redEcho "Fail to jstack java process ${pid}!"
+                redEcho "[$((count++))] Fail to jstack Busy(${pcpu}%) thread(${threadId}/0x${threadId0x}) stack of java process(${pid}) under user(${user})."
+                echo
                 rm ${jstackFile}
                 continue
             }
         }
-        redEcho "[$((count++))] Busy(${pcpu}%) thread(${threadId}/0x${threadId0x}) stack of java process(${pid}) under user(${user}):"
+        blueEcho "[$((count++))] Busy(${pcpu}%) thread(${threadId}/0x${threadId0x}) stack of java process(${pid}) under user(${user}):"
         sed "/nid=0x${threadId0x} /,/^$/p" -n ${jstackFile}
     done
 }