From a88bc20a97b6c5768771bda7070c5d4498b7363e Mon Sep 17 00:00:00 2001
From: ga-ram <galam.choi@navercorp.com>
Date: Thu, 15 Jun 2023 15:18:11 +0900
Subject: [PATCH] [#10041] Add auto option to profiler.span.collected.uri

---
 agent/src/main/resources/pinpoint-root.config |  7 +++---
 .../grpc/GrpcSpanMessageConverter.java        | 23 +++++++++++++------
 .../GrpcSpanMessageConverterProvider.java     |  2 +-
 3 files changed, 21 insertions(+), 11 deletions(-)

diff --git a/agent/src/main/resources/pinpoint-root.config b/agent/src/main/resources/pinpoint-root.config
index 21d798f07a9c7..70f7d39807cd9 100644
--- a/agent/src/main/resources/pinpoint-root.config
+++ b/agent/src/main/resources/pinpoint-root.config
@@ -408,9 +408,10 @@ pinpoint.banner.configs=pinpoint.profiler.profiles.active,\
 ###########################################################
 # SPAN COLLECTED URL TYPE                                 #
 ###########################################################
-# Collected URI Type for profiler : TEMPLATE (default), RAW
+# Collected URI Type for profiler : TEMPLATE, RAW, AUTO (default)
 # e.g.)
 # Collected URI for each type for a request `/randomResponseTime/50` mapped to `@RequestMapping("/randomResponseTime/**")`
-# TEMPLATE (default) : `/randomResponseTime/**`
+# TEMPLATE : `/randomResponseTime/**`
 # RAW : `/randomResponseTime/50`
-profiler.span.collected.uri.type=TEMPLATE
+# AUTO (default) : `/randomResponseTime/**` for supported frameworks, `/randomResponseTime/50` for unsupported frameworks
+profiler.span.collected.uri.type=AUTO
\ No newline at end of file
diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/grpc/GrpcSpanMessageConverter.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/grpc/GrpcSpanMessageConverter.java
index 2813e4b5bcde7..76748561e0233 100644
--- a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/grpc/GrpcSpanMessageConverter.java
+++ b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/grpc/GrpcSpanMessageConverter.java
@@ -81,7 +81,7 @@ public class GrpcSpanMessageConverter implements MessageConverter<SpanType, Gene
     private final PAnnotation.Builder pAnnotationBuilder = PAnnotation.newBuilder();
 
     public enum SpanUriType {
-        TEMPLATE, RAW
+        TEMPLATE, RAW, AUTO
     }
 
     private final SpanUriType spanCollectedUriType;
@@ -196,12 +196,7 @@ private PAcceptEvent newAcceptEvent(Span span) {
         }
 
         final Shared shared = span.getTraceRoot().getShared();
-        final String rpc;
-        if (spanCollectedUriType == SpanUriType.RAW) {
-            rpc = shared.getRpcName();
-        } else {
-            rpc = shared.getUriTemplate();
-        }
+        final String rpc = getCollectedUri(shared, spanCollectedUriType);
 
         if (StringUtils.isEmpty(rpc)) {
             hasEmptyValue = true;
@@ -228,6 +223,20 @@ private PAcceptEvent newAcceptEvent(Span span) {
         return builder.build();
     }
 
+    private String getCollectedUri(Shared shared, SpanUriType spanCollectedUriType) {
+        if (spanCollectedUriType == SpanUriType.RAW) {
+            return shared.getRpcName();
+        } else if (spanCollectedUriType == SpanUriType.TEMPLATE){
+            return shared.getUriTemplate();
+        } else {
+            String uriTemplate = shared.getUriTemplate();
+            if (StringUtils.isEmpty(uriTemplate)) {
+                return shared.getRpcName();
+            }
+            return uriTemplate;
+        }
+    }
+
     private PParentInfo newParentInfo(Span span) {
         final PParentInfo.Builder builder = PParentInfo.newBuilder();
         // For the Queue service type, the acceptorHost value can be stored even without the parentApplicationName value.
diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/grpc/GrpcSpanMessageConverterProvider.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/grpc/GrpcSpanMessageConverterProvider.java
index b8dc365ded685..f6d70fc1f5a39 100644
--- a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/grpc/GrpcSpanMessageConverterProvider.java
+++ b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/grpc/GrpcSpanMessageConverterProvider.java
@@ -52,7 +52,7 @@ public GrpcSpanMessageConverterProvider(@AgentId String agentId, @ApplicationSer
         this.applicationServiceTypeCode = applicationServiceType.getCode();
         this.spanPostProcessor = Objects.requireNonNull(spanPostProcessor, "spanPostProcessor");
         Objects.requireNonNull(profilerConfig, "profilerConfig");
-        this.spanCollectedUriType = profilerConfig.readString(SPAN_COLLECTED_URI_CONFIG, "TEMPLATE");
+        this.spanCollectedUriType = profilerConfig.readString(SPAN_COLLECTED_URI_CONFIG, "AUTO");
     }
 
     @Override