[#229] Support TraceRoot

*  Refactoring Trace, AsyncTrace, DisableTrace and AsyncDisableTrace With TraceRoot
* continueTrace parent Information header parsing
* implements TraceId

Author: feelform

lib/config.js

@@ -10,8 +10,6 @@ const path = require('path')
 const fs = require('fs')
 const defaultConfig = require('./pinpoint-config-default')
 const log = require('./utils/logger')
-const { setLog } = require('./supports')
-const { makeLogLevelLog } = require('./utils/log/log-level-logger')
 const ServiceConfigBuilder = require('./client/retry/service-config-builder')
 
 const valueOfString = (envName) => {
@@ -147,7 +145,6 @@ const init = (initOptions = {}) => {
     readConfigJson(initOptions))
 
   log.init(agentConfig.logLevel)
-  setLog(makeLogLevelLog(agentConfig.logLevel))
 
   Object.entries(REQUIRE_CONFIG).forEach(([propertyName, description]) => {
     if (agentConfig.enable && !agentConfig[propertyName]) {

lib/context/async-id.js

@@ -13,6 +13,11 @@ class AsyncId {
   static asyncIdGenerator = new SequenceGenerator(1)
   static nonAsyncId = new AsyncId(0, 0)
 
+  // DefaultAsyncIdGenerator.java: nextAsyncId()
+  static make() {
+    return new AsyncId(AsyncId.asyncIdGenerator.next, 0)
+  }
+
   constructor(asyncId, sequence) {
     this.asyncId = asyncId
     this.sequence = sequence
@@ -31,6 +36,11 @@ class AsyncId {
     return new AsyncId(asyncId, this.sequence + 1)
   }
 
+  // DefaultAsyncId.java: nextLocalAsyncId
+  nextLocalAsyncId2() {
+    return new AsyncId(this.asyncId, this.sequence + 1)
+  }
+
   sequenceNextLocalAsyncId() {
     if (this === AsyncId.nonAsyncId) {
       return this.nextLocalAsyncId() 

lib/context/disable-span-recorder.js

@@ -24,8 +24,6 @@ class DisableSpanRecorder {
     recordRemoteAddr() {}
 
     recordException() {}
-
-    recordSpanEvent() {}
 }
 
 module.exports = DisableSpanRecorder

lib/context/disable-trace.js

@@ -16,6 +16,10 @@ class DisableTrace {
         this.spanRecorder = new DisableSpanRecorder()
     }
 
+    getTraceRoot() {
+        return this.traceRoot
+    }
+
     traceBlockBegin() {
         return new DisableSpanEventRecorder()
     }

lib/context/remote-trace-root-builder.js

@@ -6,8 +6,8 @@
 
 'use strict'
 
-const TraceRootBuilder = require("./trace-root-builder")
-const TraceIdBuilder = require("./trace/trace-id-builder")
+const TraceRootBuilder = require('./trace-root-builder')
+const TraceIdBuilder = require('./trace/trace-id-builder')
 
 class RemoteTraceRoot {
     constructor(traceId, traceRoot) {
@@ -18,6 +18,10 @@ class RemoteTraceRoot {
     getTraceStartTime() {
         return this.traceRoot.getTraceStartTime()
     }
+
+    isSampled() {
+        return true
+    }
 }
 
 class RemoteTraceRootBuilder {

lib/context/sequence-generator.js

@@ -24,6 +24,13 @@ class SequenceGenerator {
     return this.next.toString()
   }
 
+  getAndIncrement() {
+    if (this.sequence > this.maxValue) {
+      this.sequence = this.initValue
+    }
+    return this.sequence++
+  }
+
   // for test
   reset() {
     this.sequence = this.initValue

lib/context/span-builder.js

@@ -9,17 +9,74 @@
 class Span {
     constructor(traceRoot) {
         this.traceRoot = traceRoot
-        this.startTime = traceRoot.getTraceStartTime()
     }
 }
 
 class SpanBuilder {
     constructor(traceRoot) {
         this.traceRoot = traceRoot
+        this.annotations = []
+        this.startTime = traceRoot.getTraceStartTime()
+    }
+
+    setApiId(apiId) {
+        this.apiId = apiId
+        return this
+    }
+
+    setRpc(rpc) {
+        this.rpc = rpc
+        return this
+    }
+
+    setEndPoint(endPoint) {
+        this.endPoint = endPoint
+        return this
+    }
+
+    setRemoteAddress(remoteAddress) {
+        this.remoteAddress = remoteAddress
+        return this
+    }
+
+    addAnnotation(annotation) {
+        this.annotations.push(annotation)
+        return this
+    }
+
+    setExceptionInfo(id, message) {
+        this.exceptionInfo = { id, message }
+        return this
+    }
+
+    setAcceptorHost(acceptorHost) {
+        this.acceptorHost = acceptorHost
+        return this
+    }
+
+    setParentApplicationName(parentApplicationName) {
+        this.parentApplicationName = parentApplicationName
+        return this
+    }
+
+    setParentApplicationType(parentApplicationType) {
+        this.parentApplicationType = parentApplicationType
+        return this
     }
 
     build() {
-        return new Span(this.traceRoot)
+        const span = new Span(this.traceRoot)
+        this.startTime = this.startTime || Date.now()
+        span.apiId = this.apiId
+        span.rpc = this.rpc
+        span.endPoint = this.endPoint
+        span.remoteAddress = this.remoteAddress
+        span.annotations = this.annotations
+        span.exceptionInfo = this.exceptionInfo
+        span.acceptorHost = this.acceptorHost
+        span.parentApplicationName = this.parentApplicationName
+        span.parentApplicationType = this.parentApplicationType
+        return span
     }
 }
 

lib/context/span-chunk-builder.js

@@ -5,12 +5,8 @@
  */
 
 'use strict'
-class SpanChunk {
-    constructor(traceRoot, spanEventList) {
-        this.traceRoot = traceRoot
-        this.spanEventList = spanEventList
-    }
-}
+
+const SpanChunk = require('./trace/span-chunk2')
 
 class SpanChunkBuilder {
     constructor(traceRoot) {

lib/context/span-id.js

@@ -8,6 +8,8 @@
 
 // SpanId.java in Java agent
 class SpanId {
+  static nullSpanId = -1
+  
   constructor () {
     this.MAX_NUM = Number.MAX_SAFE_INTEGER
   }

lib/context/span-recorder.js

@@ -77,12 +77,6 @@ class SpanRecorder {
       this.span.err = 1
     }
   }
-
-  recordSpanEvent(spanEvent) {
-    if (this.span && spanEvent) {
-      this.span.spanEventList.push(spanEvent)
-    }
-  }
 }
 
 module.exports = SpanRecorder