[#12434] Refactor BasicSpan

Author: emeroad

collector/src/main/java/com/navercorp/pinpoint/collector/heatmap/service/HeatmapService.java

@@ -43,7 +43,7 @@ public void insertSpanChunk(SpanChunkBo spanChunkBo) {
 
     @Override
     public void insertSpan(SpanBo spanBo) {
-        HeatmapStat heatmapStat = new HeatmapStat(spanBo.getApplicationId(), spanBo.getAgentId(), spanBo.getCollectorAcceptTime(), spanBo.getElapsed(), spanBo.getErrCode());
+        HeatmapStat heatmapStat = new HeatmapStat(spanBo.getApplicationName(), spanBo.getAgentId(), spanBo.getCollectorAcceptTime(), spanBo.getElapsed(), spanBo.getErrCode());
         heatmapDao.insert(heatmapStat);
     }
 }

collector/src/main/java/com/navercorp/pinpoint/collector/service/HbaseTraceService.java

@@ -22,6 +22,7 @@
 import com.navercorp.pinpoint.collector.dao.TraceDao;
 import com.navercorp.pinpoint.collector.event.SpanStorePublisher;
 import com.navercorp.pinpoint.common.profiler.logging.ThrottledLogger;
+import com.navercorp.pinpoint.common.server.bo.BasicSpan;
 import com.navercorp.pinpoint.common.server.bo.SpanBo;
 import com.navercorp.pinpoint.common.server.bo.SpanChunkBo;
 import com.navercorp.pinpoint.common.server.bo.SpanEventBo;
@@ -96,8 +97,8 @@ public void insertSpanChunk(@Valid final SpanChunkBo spanChunkBo) {
         publisher.publishEvent(event, true);
     }
 
-    private ServiceType getApplicationServiceType(SpanChunkBo spanChunk) {
-        final int applicationServiceTypeCode = spanChunk.getApplicationServiceType();
+    private ServiceType getApplicationServiceType(BasicSpan basicSpan) {
+        final int applicationServiceTypeCode = basicSpan.getApplicationServiceType();
         return registry.findServiceType(applicationServiceTypeCode);
     }
 
@@ -158,12 +159,6 @@ private void insertAcceptorHost(SpanBo span) {
         }
     }
 
-    private ServiceType getApplicationServiceType(SpanBo span) {
-        // Check if applicationServiceType is set. If not, use span's service type.
-        final int applicationServiceTypeCode = span.getApplicationServiceType();
-        return registry.findServiceType(applicationServiceTypeCode);
-    }
-
     private void insertSpanStat(SpanBo span) {
         final ServiceType applicationServiceType = getApplicationServiceType(span);
         final ServiceType spanServiceType = registry.findServiceType(span.getServiceType());

commons-server/src/main/java/com/navercorp/pinpoint/common/server/bo/BasicSpan.java

@@ -17,6 +17,7 @@
 package com.navercorp.pinpoint.common.server.bo;
 
 import com.navercorp.pinpoint.common.profiler.util.TransactionId;
+import com.navercorp.pinpoint.common.trace.ServiceType;
 
 /**
  * @author Woonduk Kang(emeroad)
@@ -55,6 +56,15 @@ public interface BasicSpan {
     TransactionId getTransactionId();
 //    void setTransactionId(TransactionId transactionId);
 
+    int getApplicationServiceType();
+    void setApplicationServiceType(int applicationServiceType);
+    default boolean hasApplicationServiceType() {
+        final int type = getApplicationServiceType();
+        return type != 0 && type != ServiceType.UNDEFINED.getCode();
+    }
+
+    long getCollectorAcceptTime();
+    void setCollectorAcceptTime(long collectorAcceptTime);
 
 //    List<SpanEventBo> getSpanEventBoList();
 }

commons-server/src/main/java/com/navercorp/pinpoint/common/server/bo/SpanBo.java

@@ -19,7 +19,6 @@
 import com.navercorp.pinpoint.common.profiler.util.TransactionId;
 import com.navercorp.pinpoint.common.server.util.ByteUtils;
 import com.navercorp.pinpoint.common.server.util.StringPrecondition;
-import com.navercorp.pinpoint.common.trace.ServiceType;
 import jakarta.validation.constraints.NotBlank;
 import jakarta.validation.constraints.PositiveOrZero;
 
@@ -309,10 +308,12 @@ public void setRemoteAddr(String remoteAddr) {
         this.remoteAddr = remoteAddr;
     }
 
+    @Override
     public long getCollectorAcceptTime() {
         return collectorAcceptTime;
     }
 
+    @Override
     public void setCollectorAcceptTime(long collectorAcceptTime) {
         this.collectorAcceptTime = collectorAcceptTime;
     }
@@ -348,14 +349,12 @@ public void setExceptionClass(String exceptionClass) {
         this.exceptionClass = exceptionClass;
     }
 
+    @Override
     public void setApplicationServiceType(int applicationServiceType) {
         this.applicationServiceType = applicationServiceType;
     }
 
-    public boolean hasApplicationServiceType() {
-        return applicationServiceType != 0 && applicationServiceType != ServiceType.UNDEFINED.getCode();
-    }
-
+    @Override
     public int getApplicationServiceType() {
         if (hasApplicationServiceType()) {
             return this.applicationServiceType;

commons-server/src/main/java/com/navercorp/pinpoint/common/server/bo/SpanChunkBo.java

@@ -147,18 +147,22 @@ public void setEndPoint(String endPoint) {
         this.endPoint = endPoint;
     }
 
+    @Override
     public long getCollectorAcceptTime() {
         return collectorAcceptTime;
     }
 
+    @Override
     public void setCollectorAcceptTime(long collectorAcceptTime) {
         this.collectorAcceptTime = collectorAcceptTime;
     }
 
+    @Override
     public void setApplicationServiceType(int applicationServiceType) {
         this.applicationServiceType  = applicationServiceType;
     }
 
+    @Override
     public int getApplicationServiceType() {
         return this.applicationServiceType;
     }