From 2780ccc6fabaaf69ecdfdaace96a5d1ecdb493fe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=87=91=E6=88=9F?= <jinji.lf@alibaba-inc.com>
Date: Mon, 12 Jul 2021 20:37:13 +0800
Subject: [PATCH] for better performance

---
 .../testable/agent/transformer/TestableClassTransformer.java  | 4 ++--
 .../java/com/alibaba/testable/agent/util/DiagnoseUtil.java    | 4 +---
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/testable-agent/src/main/java/com/alibaba/testable/agent/transformer/TestableClassTransformer.java b/testable-agent/src/main/java/com/alibaba/testable/agent/transformer/TestableClassTransformer.java
index 558ca9e..ba3f986 100644
--- a/testable-agent/src/main/java/com/alibaba/testable/agent/transformer/TestableClassTransformer.java
+++ b/testable-agent/src/main/java/com/alibaba/testable/agent/transformer/TestableClassTransformer.java
@@ -253,7 +253,7 @@ public class TestableClassTransformer implements ClassFileTransformer {
     private String parseMockWithAnnotation(ClassNode cn, ClassType expectedType) {
         if (cn.visibleAnnotations != null) {
             for (AnnotationNode an : cn.visibleAnnotations) {
-                if (ClassUtil.toJavaStyleClassName(an.desc).equals(ConstPool.MOCK_WITH)) {
+                if ((ClassUtil.toByteCodeClassName(ConstPool.MOCK_WITH)).equals(an.desc)) {
                     ClassType type = AnnotationUtil.getAnnotationParameter(an, FIELD_TREAT_AS, ClassType.GuessByName,
                         ClassType.class);
                     if (isExpectedType(cn.name, type, expectedType)) {
@@ -271,7 +271,7 @@ public class TestableClassTransformer implements ClassFileTransformer {
     private String getDumpPathByAnnotation(ClassNode cn) {
         if (cn.visibleAnnotations != null) {
             for (AnnotationNode an : cn.visibleAnnotations) {
-                if (ClassUtil.toJavaStyleClassName(an.desc).equals(ConstPool.DUMP_TO)) {
+                if ((ClassUtil.toByteCodeClassName(ConstPool.DUMP_TO)).equals(an.desc)) {
                     String path = AnnotationUtil.getAnnotationParameter(an, FIELD_VALUE, null, String.class);
                     String fullPath = PathUtil.join(System.getProperty(PROPERTY_USER_DIR), path);
                     if (createFolder(fullPath)) {
diff --git a/testable-agent/src/main/java/com/alibaba/testable/agent/util/DiagnoseUtil.java b/testable-agent/src/main/java/com/alibaba/testable/agent/util/DiagnoseUtil.java
index 7e570e2..1507cdf 100644
--- a/testable-agent/src/main/java/com/alibaba/testable/agent/util/DiagnoseUtil.java
+++ b/testable-agent/src/main/java/com/alibaba/testable/agent/util/DiagnoseUtil.java
@@ -6,8 +6,6 @@ import com.alibaba.testable.core.util.LogUtil;
 import org.objectweb.asm.tree.AnnotationNode;
 import org.objectweb.asm.tree.ClassNode;
 
-import static com.alibaba.testable.agent.util.ClassUtil.toJavaStyleClassName;
-
 public class DiagnoseUtil {
 
     private static final String FIELD_VALUE = "value";
@@ -17,7 +15,7 @@ public class DiagnoseUtil {
             return;
         }
         for (AnnotationNode an : cn.visibleAnnotations) {
-            if (toJavaStyleClassName(an.desc).equals(ConstPool.MOCK_DIAGNOSE)) {
+            if (ClassUtil.toByteCodeClassName(ConstPool.MOCK_DIAGNOSE).equals(an.desc)) {
                 setupDiagnose(an, FIELD_VALUE);
             }
         }