From baf41aa59f2910be92c25da938e8f3f5ce39abd0 Mon Sep 17 00:00:00 2001
From: portman xu <iportman@qq.com>
Date: Wed, 28 Aug 2019 14:12:14 +0800
Subject: [PATCH] change prompt to arthas@pid (#803)

---
 .../taobao/arthas/core/shell/impl/ShellImpl.java    | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/core/src/main/java/com/taobao/arthas/core/shell/impl/ShellImpl.java b/core/src/main/java/com/taobao/arthas/core/shell/impl/ShellImpl.java
index fb8c3f91f..883dd9584 100644
--- a/core/src/main/java/com/taobao/arthas/core/shell/impl/ShellImpl.java
+++ b/core/src/main/java/com/taobao/arthas/core/shell/impl/ShellImpl.java
@@ -46,6 +46,7 @@ public class ShellImpl implements Shell {
     private Term term;
     private String welcome;
     private Job currentForegroundJob;
+    private String prompt;
 
     public ShellImpl(ShellServer server, Term term, InternalCommandManager commandManager,
             Instrumentation instrumentation, int pid, JobControllerImpl jobController) {
@@ -64,6 +65,8 @@ public class ShellImpl implements Shell {
         if (term != null) {
             term.setSession(session);
         }
+
+        this.setPrompt();
     }
 
     public JobController jobController() {
@@ -106,6 +109,14 @@ public class ShellImpl implements Shell {
         this.welcome = welcome;
     }
 
+    private void setPrompt(){
+        StringBuilder stringBuilder = new StringBuilder();
+        stringBuilder.append("[arthas@");
+        stringBuilder.append(session.getPid());
+        stringBuilder.append("]$ ");
+        this.prompt = stringBuilder.toString();
+    }
+
     public ShellImpl init() {
         term.interruptHandler(new InterruptHandler(this));
         term.suspendHandler(new SuspendHandler(this));
@@ -142,7 +153,7 @@ public class ShellImpl implements Shell {
     }
 
     public void readline() {
-        term.readline(Constants.DEFAULT_PROMPT, new ShellLineHandler(this),
+        term.readline(prompt, new ShellLineHandler(this),
                 new CommandManagerCompletionHandler(commandManager));
     }