feat: record dubbo request param type (#167)

lucas-myx · web-flow · commit 0626b50878c9 · 2023-05-18T15:05:09.000+08:00
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/DubboAdapter.java b/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/DubboAdapter.java
@@ -21,6 +21,7 @@
 import java.util.HashSet;
 import java.util.Map;
 import java.util.Set;
+import java.util.function.Function;
 
 import static io.arex.inst.runtime.model.ArexConstants.DUBBO_STREAM_NAME;
 import static io.arex.inst.runtime.model.ArexConstants.DUBBO_STREAM_PROTOCOL;
@@ -59,11 +60,24 @@ public String getServiceOperation() {
         return getPath() + "." + getOperationName();
     }
     public String getRequest() {
-        Object request = null;
-        if (invocation.getArguments() != null && invocation.getArguments().length > 0) {
-            request = invocation.getArguments()[0];
+        return parseRequest(invocation.getArguments(), Serializer::serialize);
+    }
+    public String getRequestParamType() {
+        Class<?>[] parameterTypes = invocation.getParameterTypes();
+        if (parameterTypes != null && parameterTypes.length > 0) {
+            return parameterTypes[0].getName();
+        }
+        return parseRequest(invocation.getArguments(), TypeUtil::getName);
+    }
+    public static String parseRequest(Object request, Function<Object, String> parser) {
+        if (request instanceof Object[]) {
+            Object[] requests = (Object[]) request;
+            if (requests.length > 0) {
+                // take only the first request parameter, and in most cases will be packed as one object
+                return parser.apply(requests[0]);
+            }
         }
-        return Serializer.serialize(request);
+        return parser.apply(request);
     }
     public String getReturnType() {
         Type returnType = null;
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/DubboConsumerExtractor.java b/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/DubboConsumerExtractor.java
@@ -21,6 +21,7 @@ public Result record(Result result) {
     private Mocker makeMocker() {
         Mocker mocker = MockUtils.createDubboConsumer(adapter.getServiceOperation());
         mocker.getTargetRequest().setBody(adapter.getRequest());
+        mocker.getTargetRequest().setType(adapter.getRequestParamType());
         mocker.getTargetResponse().setType(adapter.getReturnType());
         return mocker;
     }
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/DubboProviderExtractor.java b/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/DubboProviderExtractor.java
@@ -70,6 +70,7 @@ private static Mocker makeMocker(DubboAdapter adapter) {
         headerMap.put("protocol", adapter.getProtocol());
         mocker.getTargetRequest().setAttributes(Collections.singletonMap("Headers", Serializer.serialize(headerMap)));
         mocker.getTargetRequest().setBody(adapter.getRequest());
+        mocker.getTargetRequest().setType(adapter.getRequestParamType());
         String responseHeader = Serializer.serialize(RpcContext.getServerAttachment().getObjectAttachments());
         mocker.getTargetResponse().setAttributes(Collections.singletonMap("Headers", responseHeader));
         return mocker;
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamAdapter.java b/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamAdapter.java
@@ -3,13 +3,17 @@
 import io.arex.agent.bootstrap.model.MockResult;
 import io.arex.agent.bootstrap.model.MockStrategyEnum;
 import io.arex.agent.bootstrap.model.Mocker;
+import io.arex.inst.dubbo.DubboAdapter;
 import io.arex.inst.runtime.config.Config;
 import io.arex.inst.runtime.context.ContextManager;
 import io.arex.inst.runtime.serializer.Serializer;
 import io.arex.inst.runtime.util.MockUtils;
+import io.arex.inst.runtime.util.TypeUtil;
 import org.apache.dubbo.rpc.model.MethodDescriptor;
 import org.apache.dubbo.rpc.protocol.tri.stream.Stream;
 
+import java.lang.reflect.ParameterizedType;
+import java.lang.reflect.Type;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -18,16 +22,18 @@
  */
 public class DubboStreamAdapter {
     private String streamId;
-    private DubboStreamAdapter(String streamId) {
+    private MethodDescriptor methodDescriptor;
+    private DubboStreamAdapter(String streamId, MethodDescriptor methodDescriptor) {
         this.streamId = streamId;
+        this.methodDescriptor = methodDescriptor;
     }
 
-    public static DubboStreamAdapter of(String streamId) {
-        return new DubboStreamAdapter(streamId);
+    public static DubboStreamAdapter of(String streamId, MethodDescriptor methodDescriptor) {
+        return new DubboStreamAdapter(streamId, methodDescriptor);
     }
 
-    public static DubboStreamAdapter of(Stream stream) {
-        return of(generateStreamId(stream));
+    public static DubboStreamAdapter of(Stream stream, MethodDescriptor methodDescriptor) {
+        return of(generateStreamId(stream), methodDescriptor);
     }
 
     public void saveRequest(byte[] message) {
@@ -78,4 +84,26 @@ public void clearRequest() {
     public static String generateStreamId(Stream stream) {
         return Integer.toHexString(System.identityHashCode(stream));
     }
+
+    public String getRequest(Object request) {
+        return DubboAdapter.parseRequest(request, Serializer::serialize);
+    }
+
+    public String getRequestParamType(Object request) {
+        if (methodDescriptor.getMethod() == null) {
+            return DubboAdapter.parseRequest(request, TypeUtil::getName);
+        }
+        Type[] genericParameters = methodDescriptor.getMethod().getGenericParameterTypes();
+        if (genericParameters.length <= 0) {
+            return DubboAdapter.parseRequest(request, TypeUtil::getName);
+        }
+        Type genericType = genericParameters[0];
+        if (genericType instanceof ParameterizedType) {
+            Type[] actualTypes = ((ParameterizedType) genericType).getActualTypeArguments();
+            if (actualTypes != null && actualTypes.length > 0) {
+                return actualTypes[0].getTypeName();
+            }
+        }
+        return DubboAdapter.parseRequest(request, TypeUtil::getName);
+    }
 }
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamConsumerExtractor.java b/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamConsumerExtractor.java
@@ -67,7 +67,8 @@ public void record(RequestMetadata requestMetadata, byte[] message, Throwable th
             StreamModel.DataModel requestModel = getUnRecordRequest(dataModels);
             if (requestModel.getData() != null) {
                 Object request = requestMetadata.packableMethod.parseRequest(requestModel.getData());
-                mocker.getTargetRequest().setBody(Serializer.serialize(request));
+                mocker.getTargetRequest().setBody(adapter.getRequest(request));
+                mocker.getTargetRequest().setType(adapter.getRequestParamType(request));
             }
             if (throwable != null) {
                 mocker.getTargetResponse().setBody(Serializer.serialize(throwable));
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamConsumerInstrumentation.java b/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamConsumerInstrumentation.java
@@ -70,7 +70,7 @@ public static boolean onEnter(@Advice.Argument(0) Object message,
                                       @Advice.Local("mockResult") List<MockResult> mockResults) {
             if (ContextManager.needRecordOrReplay()) {
                 RepeatedCollectManager.enter();
-                extractor = new DubboStreamConsumerExtractor(DubboStreamAdapter.of(stream));
+                extractor = new DubboStreamConsumerExtractor(DubboStreamAdapter.of(stream, requestMetadata.method));
                 if (ContextManager.needReplay()) {
                     mockResults = extractor.replay(message, requestMetadata);
                 }
@@ -113,7 +113,8 @@ public static void onExit(@Advice.Argument(0) byte[] message,
                                   @Advice.FieldValue("stream") ClientStream stream,
                                   @Advice.FieldValue("requestMetadata") RequestMetadata requestMetadata) {
             if (ContextManager.needRecord()) {
-                DubboStreamConsumerExtractor extractor = new DubboStreamConsumerExtractor(DubboStreamAdapter.of(stream));
+                DubboStreamConsumerExtractor extractor = new DubboStreamConsumerExtractor(
+                        DubboStreamAdapter.of(stream, requestMetadata.method));
                 extractor.record(requestMetadata, message, null);
             }
         }
@@ -125,7 +126,8 @@ public static void onEnter(@Advice.Argument(0) TriRpcStatus status,
                                    @Advice.FieldValue("stream") ClientStream stream,
                                    @Advice.FieldValue("requestMetadata") RequestMetadata requestMetadata) {
             if (ContextManager.needRecord()) {
-                DubboStreamConsumerExtractor extractor = new DubboStreamConsumerExtractor(DubboStreamAdapter.of(stream));
+                DubboStreamConsumerExtractor extractor = new DubboStreamConsumerExtractor(
+                        DubboStreamAdapter.of(stream, requestMetadata.method));
                 extractor.complete(status, requestMetadata);
             }
         }
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamProviderExtractor.java b/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamProviderExtractor.java
@@ -126,7 +126,8 @@ public void record(Map<String, Object> requestHeader, Object response, String se
                     StreamModel.DataModel dataModel = requestsList.get(i);
                     if (dataModel.getData() != null) {
                         Object request = packableMethod.parseRequest(dataModel.getData());
-                        mocker.getTargetRequest().setBody(Serializer.serialize(request));
+                        mocker.getTargetRequest().setBody(adapter.getRequest(request));
+                        mocker.getTargetRequest().setType(adapter.getRequestParamType(request));
                     }
                     // The result is recorded for the last time, the previous requests are all compensate record
                     if (i == (requestTimes - 1)) {
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamProviderInstrumentation.java b/arex-instrumentation/dubbo/arex-dubbo/src/main/java/io/arex/inst/dubbo/stream/DubboStreamProviderInstrumentation.java
@@ -76,7 +76,8 @@ public static void onExit(@Advice.Argument(0) Object message,
                                   @Advice.FieldValue("stream") ServerStream stream,
                                   @Advice.FieldValue("packableMethod") PackableMethod packableMethod) {
             if (ContextManager.needRecord()) {
-                DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(DubboStreamAdapter.of(stream));
+                DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(
+                        DubboStreamAdapter.of(stream, methodDescriptor));
                 extractor.record(requestMetadata, message, serviceName, methodDescriptor, packableMethod);
             }
         }
@@ -90,7 +91,8 @@ public static void onEnter(@Advice.Argument(0) byte[] message,
                                    @Advice.FieldValue("methodDescriptor") MethodDescriptor methodDescriptor,
                                    @Advice.FieldValue("packableMethod") PackableMethod packableMethod) {
             if (ContextManager.needRecordOrReplay()) {
-                DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(DubboStreamAdapter.of(stream));
+                DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(
+                        DubboStreamAdapter.of(stream, methodDescriptor));
                 if (ContextManager.needReplay()) {
                     extractor.replay(message, serviceName, methodDescriptor, packableMethod);
                 }
@@ -110,7 +112,8 @@ public static void onEnter(@Advice.Argument(0) TriRpcStatus status,
                                    @Advice.FieldValue("stream") ServerStream stream,
                                    @Advice.FieldValue("packableMethod") PackableMethod packableMethod) {
             if (ContextManager.needRecord()) {
-                DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(DubboStreamAdapter.of(stream));
+                DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(
+                        DubboStreamAdapter.of(stream, methodDescriptor));
                 extractor.complete(status, requestMetadata, serviceName, methodDescriptor, packableMethod);
             }
         }
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/test/java/io/arex/inst/dubbo/DubboAdapterTest.java b/arex-instrumentation/dubbo/arex-dubbo/src/test/java/io/arex/inst/dubbo/DubboAdapterTest.java
@@ -83,6 +83,15 @@ void getRequest() {
         assertNull(adapter.getRequest());
     }
 
+    @Test
+    void getRequestParamType() {
+        Mockito.when(invocation.getParameterTypes()).thenReturn(new Class<?>[]{String.class});
+        assertNotNull(adapter.getRequestParamType());
+        Mockito.when(invocation.getParameterTypes()).thenReturn(new Class<?>[]{});
+        Mockito.when(invocation.getArguments()).thenReturn(new Object[]{});
+        assertNotNull(adapter.getRequestParamType());
+    }
+
     @Test
     void getReturnType() {
         Type type1 = Mockito.mock(Type.class);
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/test/java/io/arex/inst/dubbo/stream/DubboStreamAdapterTest.java b/arex-instrumentation/dubbo/arex-dubbo/src/test/java/io/arex/inst/dubbo/stream/DubboStreamAdapterTest.java
@@ -20,6 +20,7 @@
 import org.mockito.Mockito;
 import org.mockito.junit.jupiter.MockitoExtension;
 
+import java.lang.reflect.Method;
 import java.util.List;
 import java.util.Objects;
 import java.util.function.Predicate;
@@ -32,10 +33,12 @@
 class DubboStreamAdapterTest {
     static DubboStreamAdapter adapter;
     static MockedStatic<DubboStreamCache> dubboStreamCacheMocker;
+    static MethodDescriptor methodDescriptor;
 
     @BeforeAll
-    static void setUp() {
-        adapter = DubboStreamAdapter.of(Mockito.mock(Stream.class));
+    static void setUp() throws NoSuchMethodException {
+        methodDescriptor = Mockito.mock(MethodDescriptor.class);
+        adapter = DubboStreamAdapter.of(Mockito.mock(Stream.class), methodDescriptor);
         dubboStreamCacheMocker = Mockito.mockStatic(DubboStreamCache.class);
         Mockito.mockStatic(ContextManager.class);
         Mockito.mockStatic(MockUtils.class);
@@ -93,4 +96,62 @@ void clearRequest() {
         adapter.clearRequest();
         dubboStreamCacheMocker.verify(() -> DubboStreamCache.remove(any()));
     }
+
+    @Test
+    void getRequest() {
+        assertNull(adapter.getRequest(new Object[]{"mock"}));
+    }
+
+    @ParameterizedTest
+    @MethodSource("getRequestParamTypeCase")
+    void getRequestParamType(Runnable mocker, Predicate<String> predicate) {
+        mocker.run();
+        String result = adapter.getRequestParamType(new Object[]{"mock"});
+        assertTrue(predicate.test(result));
+    }
+
+    static java.util.stream.Stream<Arguments> getRequestParamTypeCase() {
+        Runnable emptyMocker = () -> {};
+        Runnable mocker1 = () -> {
+            Method method = null;
+            try {
+                method = DubboStreamAdapterTest.class.getDeclaredMethod(
+                        "testWithoutParameter");
+            } catch (NoSuchMethodException e) {}
+            Mockito.when(methodDescriptor.getMethod()).thenReturn(method);
+        };
+        Runnable mocker2 = () -> {
+            Method method = null;
+            try {
+                method = DubboStreamAdapterTest.class.getDeclaredMethod(
+                        "testWithGenericParameter", List.class);
+            } catch (NoSuchMethodException e) {}
+            Mockito.when(methodDescriptor.getMethod()).thenReturn(method);
+        };
+        Runnable mocker3 = () -> {
+            Method method = null;
+            try {
+                method = DubboStreamAdapterTest.class.getDeclaredMethod(
+                        "testWithGenericParameter", String.class, List.class);
+            } catch (NoSuchMethodException e) {}
+            Mockito.when(methodDescriptor.getMethod()).thenReturn(method);
+        };
+        Predicate<String> predicate1 = Objects::nonNull;
+        return java.util.stream.Stream.of(
+                arguments(emptyMocker, predicate1),
+                arguments(mocker1, predicate1),
+                arguments(mocker2, predicate1),
+                arguments(mocker3, predicate1)
+        );
+    }
+
+    String testWithoutParameter() {
+        return "mock";
+    }
+    String testWithGenericParameter(List<String> list) {
+        return "mock";
+    }
+    String testWithGenericParameter(String param, List<String> list) {
+        return "mock";
+    }
 }
diff --git a/arex-instrumentation/dubbo/arex-dubbo/src/test/java/io/arex/inst/dubbo/stream/DubboStreamConsumerInstrumentationTest.java b/arex-instrumentation/dubbo/arex-dubbo/src/test/java/io/arex/inst/dubbo/stream/DubboStreamConsumerInstrumentationTest.java
@@ -31,6 +31,7 @@
 class DubboStreamConsumerInstrumentationTest {
     static DubboStreamConsumerInstrumentation target;
     static DubboStreamConsumerExtractor extractor;
+    static RequestMetadata requestMetadata;
 
     @BeforeAll
     static void setUp() {
@@ -41,6 +42,7 @@ static void setUp() {
         Mockito.when(ContextManager.needRecord()).thenReturn(true);
         Mockito.mockStatic(RepeatedCollectManager.class);
         extractor = Mockito.mock(DubboStreamConsumerExtractor.class);
+        requestMetadata = Mockito.mock(RequestMetadata.class);
     }
 
     @AfterAll
@@ -65,7 +67,7 @@ void sendMessageOnEnter() {
         try (MockedConstruction<DubboStreamConsumerExtractor> mocked = Mockito.mockConstruction(DubboStreamConsumerExtractor.class, (mock, context) -> {
             Mockito.when(mock.replay(any(), any())).thenReturn(Collections.singletonList(MockResult.success("mock")));
         })) {
-            assertTrue(DubboStreamConsumerInstrumentation.SendMessageAdvice.onEnter(null, null, null, null, null));
+            assertTrue(DubboStreamConsumerInstrumentation.SendMessageAdvice.onEnter(null, null, requestMetadata, null, null));
         }
     }
 
@@ -114,7 +116,7 @@ void onStartOnEnter() {
     void onMessageOnExit() {
         try (MockedConstruction<DubboStreamConsumerExtractor> mocked = Mockito.mockConstruction(DubboStreamConsumerExtractor.class, (mock, context) -> {
         })) {
-            DubboStreamConsumerInstrumentation.OnMessageAdvice.onExit(null, null, null);
+            DubboStreamConsumerInstrumentation.OnMessageAdvice.onExit(null, null, requestMetadata);
             DubboStreamConsumerExtractor consumerExtractor = mocked.constructed().get(0);
             verify(consumerExtractor).record(any(), any(), any());
         }
@@ -124,7 +126,7 @@ void onMessageOnExit() {
     void onCompleteOnEnter() {
         try (MockedConstruction<DubboStreamConsumerExtractor> mocked = Mockito.mockConstruction(DubboStreamConsumerExtractor.class, (mock, context) -> {
         })) {
-            DubboStreamConsumerInstrumentation.OnCompleteAdvice.onEnter(null, null, null);
+            DubboStreamConsumerInstrumentation.OnCompleteAdvice.onEnter(null, null, requestMetadata);
             DubboStreamConsumerExtractor consumerExtractor = mocked.constructed().get(0);
             verify(consumerExtractor).complete(any(), any());
         }

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,7 @@ public Result record(Result result) {`
`21`	`21`	`private Mocker makeMocker() {`
`22`	`22`	`Mocker mocker = MockUtils.createDubboConsumer(adapter.getServiceOperation());`
`23`	`23`	`mocker.getTargetRequest().setBody(adapter.getRequest());`
	`24`	`+ mocker.getTargetRequest().setType(adapter.getRequestParamType());`
`24`	`25`	`mocker.getTargetResponse().setType(adapter.getReturnType());`
`25`	`26`	`return mocker;`
`26`	`27`	`}`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ public static boolean onEnter(@Advice.Argument(0) Object message,`
`70`	`70`	`@Advice.Local("mockResult") List<MockResult> mockResults) {`
`71`	`71`	`if (ContextManager.needRecordOrReplay()) {`
`72`	`72`	`RepeatedCollectManager.enter();`
`73`		`- extractor = new DubboStreamConsumerExtractor(DubboStreamAdapter.of(stream));`
	`73`	`+ extractor = new DubboStreamConsumerExtractor(DubboStreamAdapter.of(stream, requestMetadata.method));`
`74`	`74`	`if (ContextManager.needReplay()) {`
`75`	`75`	`mockResults = extractor.replay(message, requestMetadata);`
`76`	`76`	`}`
`@@ -113,7 +113,8 @@ public static void onExit(@Advice.Argument(0) byte[] message,`
`113`	`113`	`@Advice.FieldValue("stream") ClientStream stream,`
`114`	`114`	`@Advice.FieldValue("requestMetadata") RequestMetadata requestMetadata) {`
`115`	`115`	`if (ContextManager.needRecord()) {`
`116`		`- DubboStreamConsumerExtractor extractor = new DubboStreamConsumerExtractor(DubboStreamAdapter.of(stream));`
	`116`	`+ DubboStreamConsumerExtractor extractor = new DubboStreamConsumerExtractor(`
	`117`	`+ DubboStreamAdapter.of(stream, requestMetadata.method));`
`117`	`118`	`extractor.record(requestMetadata, message, null);`
`118`	`119`	`}`
`119`	`120`	`}`
`@@ -125,7 +126,8 @@ public static void onEnter(@Advice.Argument(0) TriRpcStatus status,`
`125`	`126`	`@Advice.FieldValue("stream") ClientStream stream,`
`126`	`127`	`@Advice.FieldValue("requestMetadata") RequestMetadata requestMetadata) {`
`127`	`128`	`if (ContextManager.needRecord()) {`
`128`		`- DubboStreamConsumerExtractor extractor = new DubboStreamConsumerExtractor(DubboStreamAdapter.of(stream));`
	`129`	`+ DubboStreamConsumerExtractor extractor = new DubboStreamConsumerExtractor(`
	`130`	`+ DubboStreamAdapter.of(stream, requestMetadata.method));`
`129`	`131`	`extractor.complete(status, requestMetadata);`
`130`	`132`	`}`
`131`	`133`	`}`
Original file line number	Diff line number	Diff line change
`@@ -76,7 +76,8 @@ public static void onExit(@Advice.Argument(0) Object message,`
`76`	`76`	`@Advice.FieldValue("stream") ServerStream stream,`
`77`	`77`	`@Advice.FieldValue("packableMethod") PackableMethod packableMethod) {`
`78`	`78`	`if (ContextManager.needRecord()) {`
`79`		`- DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(DubboStreamAdapter.of(stream));`
	`79`	`+ DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(`
	`80`	`+ DubboStreamAdapter.of(stream, methodDescriptor));`
`80`	`81`	`extractor.record(requestMetadata, message, serviceName, methodDescriptor, packableMethod);`
`81`	`82`	`}`
`82`	`83`	`}`
`@@ -90,7 +91,8 @@ public static void onEnter(@Advice.Argument(0) byte[] message,`
`90`	`91`	`@Advice.FieldValue("methodDescriptor") MethodDescriptor methodDescriptor,`
`91`	`92`	`@Advice.FieldValue("packableMethod") PackableMethod packableMethod) {`
`92`	`93`	`if (ContextManager.needRecordOrReplay()) {`
`93`		`- DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(DubboStreamAdapter.of(stream));`
	`94`	`+ DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(`
	`95`	`+ DubboStreamAdapter.of(stream, methodDescriptor));`
`94`	`96`	`if (ContextManager.needReplay()) {`
`95`	`97`	`extractor.replay(message, serviceName, methodDescriptor, packableMethod);`
`96`	`98`	`}`
`@@ -110,7 +112,8 @@ public static void onEnter(@Advice.Argument(0) TriRpcStatus status,`
`110`	`112`	`@Advice.FieldValue("stream") ServerStream stream,`
`111`	`113`	`@Advice.FieldValue("packableMethod") PackableMethod packableMethod) {`
`112`	`114`	`if (ContextManager.needRecord()) {`
`113`		`- DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(DubboStreamAdapter.of(stream));`
	`115`	`+ DubboStreamProviderExtractor extractor = new DubboStreamProviderExtractor(`
	`116`	`+ DubboStreamAdapter.of(stream, methodDescriptor));`
`114`	`117`	`extractor.complete(status, requestMetadata, serviceName, methodDescriptor, packableMethod);`
`115`	`118`	`}`
`116`	`119`	`}`