From 34aac1ecc5d8769bb36b650be5fad4022e37b32d Mon Sep 17 00:00:00 2001
From: gongdewei <kylixs@qq.com>
Date: Mon, 8 Jun 2020 17:17:28 +0800
Subject: [PATCH] change resultViewMap's key to model class

---
 .../core/command/view/ResultViewResolver.java       | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/core/src/main/java/com/taobao/arthas/core/command/view/ResultViewResolver.java b/core/src/main/java/com/taobao/arthas/core/command/view/ResultViewResolver.java
index 0ef7d8d30..6e8c3fce4 100644
--- a/core/src/main/java/com/taobao/arthas/core/command/view/ResultViewResolver.java
+++ b/core/src/main/java/com/taobao/arthas/core/command/view/ResultViewResolver.java
@@ -18,7 +18,7 @@ public class ResultViewResolver {
     private static final Logger logger = LoggerFactory.getLogger(ResultViewResolver.class);
 
     // modelClass -> view
-    private Map<String, ResultView> resultViewMap = new ConcurrentHashMap<String, ResultView>();
+    private Map<Class, ResultView> resultViewMap = new ConcurrentHashMap<Class, ResultView>();
 
     private static ResultViewResolver viewResolver;
 
@@ -52,9 +52,13 @@ public class ResultViewResolver {
     private ResultViewResolver() {
     }
 
+    public ResultView getResultView(ResultModel model) {
+        return resultViewMap.get(model.getClass());
+    }
+
     public void registerView(Class modelClass, ResultView view) {
         //TODO 检查model的type是否重复，减少复制代码带来的bug
-        this.resultViewMap.put(modelClass.getName(), view);
+        this.resultViewMap.put(modelClass, view);
     }
 
     public void registerView(ResultView view) {
@@ -75,11 +79,6 @@ public class ResultViewResolver {
         this.registerView(view);
     }
 
-    public ResultView getResultView(ResultModel model) {
-        return resultViewMap.get(model.getClass().getName());
-    }
-
-
     /**
      * Get model class of result view
      *