From 0ade922855816e7d8608355db5cc60c1c7ad4178 Mon Sep 17 00:00:00 2001
From: hengyunabc <hengyunabc@gmail.com>
Date: Wed, 2 Sep 2020 20:25:07 +0800
Subject: [PATCH] fix do not unReg listener error. #1468

---
 .../taobao/arthas/core/shell/system/impl/ProcessImpl.java  | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/core/src/main/java/com/taobao/arthas/core/shell/system/impl/ProcessImpl.java b/core/src/main/java/com/taobao/arthas/core/shell/system/impl/ProcessImpl.java
index 5180f4179..74adfd1bd 100644
--- a/core/src/main/java/com/taobao/arthas/core/shell/system/impl/ProcessImpl.java
+++ b/core/src/main/java/com/taobao/arthas/core/shell/system/impl/ProcessImpl.java
@@ -546,14 +546,15 @@ public class ProcessImpl implements Process {
         }
 
         @Override
-        public void register(AdviceListener listener, ClassFileTransformer transformer) {
-            if (listener instanceof ProcessAware) {
-                ProcessAware processAware = (ProcessAware) listener;
+        public void register(AdviceListener adviceListener, ClassFileTransformer transformer) {
+            if (adviceListener instanceof ProcessAware) {
+                ProcessAware processAware = (ProcessAware) adviceListener;
                 // listener 有可能是其它 command 创建的
                 if(processAware.getProcess() == null) {
                     processAware.setProcess(this.process);
                 }
             }
+            this.listener = adviceListener;
             AdviceWeaver.reg(listener);
             
             this.transformer = transformer;