From faee94d77e4c40411242b13a5cf573496b940f1d Mon Sep 17 00:00:00 2001
From: ReaJason <reajason1225@gmail.com>
Date: Sat, 31 May 2025 13:02:19 +0800
Subject: [PATCH] refactor: use TypeDescription

---
 .../memshell/generator/ListenerGenerator.java  | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/generator/ListenerGenerator.java b/generator/src/main/java/com/reajason/javaweb/memshell/generator/ListenerGenerator.java
index 193700ba..cd9fb9f3 100644
--- a/generator/src/main/java/com/reajason/javaweb/memshell/generator/ListenerGenerator.java
+++ b/generator/src/main/java/com/reajason/javaweb/memshell/generator/ListenerGenerator.java
@@ -6,12 +6,13 @@ import net.bytebuddy.asm.Advice;
 import net.bytebuddy.asm.AsmVisitorWrapper;
 import net.bytebuddy.description.modifier.Ownership;
 import net.bytebuddy.description.modifier.Visibility;
+import net.bytebuddy.description.type.TypeDescription;
 import net.bytebuddy.dynamic.DynamicType;
 import net.bytebuddy.implementation.FixedValue;
 
 import java.util.Collections;
 
-import static net.bytebuddy.matcher.ElementMatchers.named;
+import static net.bytebuddy.matcher.ElementMatchers.*;
 
 /**
  * @author ReaJason
@@ -20,13 +21,6 @@ import static net.bytebuddy.matcher.ElementMatchers.named;
 public class ListenerGenerator {
 
     public static DynamicType.Builder<?> build(DynamicType.Builder<?> builder, Class<?> implInterceptor, Class<?> targetClass, String newClassName) {
-        boolean needAddGetFieldValue = false;
-        try {
-            targetClass.getMethod("getFieldValue", Object.class, String.class);
-        } catch (NoSuchMethodException e) {
-            needAddGetFieldValue = true;
-        }
-
         builder = builder.visit(new AsmVisitorWrapper.ForDeclaredMethods()
                         .method(named("getResponseFromRequest"),
                                 new MethodCallReplaceVisitorWrapper(
@@ -36,7 +30,13 @@ public class ListenerGenerator {
                 )
                 .visit(Advice.to(implInterceptor).on(named("getResponseFromRequest")));
 
-        if (needAddGetFieldValue) {
+        boolean methodNotFound = TypeDescription.ForLoadedType.of(targetClass)
+                .getDeclaredMethods()
+                .filter(named("getFieldValue")
+                        .and(takesArguments(Object.class, String.class)))
+                .isEmpty();
+
+        if (methodNotFound) {
             builder = builder.defineMethod("getFieldValue", Object.class, Visibility.PUBLIC, Ownership.STATIC)
                     .withParameters(Object.class, String.class)
                     .intercept(FixedValue.nullValue())