pinpoint-apm#86 method descriptor Builder

feelform · Jun 9, 2021 · 288bfc6 · 288bfc6
1 parent 3faaf4f
commit 288bfc6
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 2 deletions.
diff --git a/lib/context/method-descriptor-builder.js b/lib/context/method-descriptor-builder.js
@@ -8,7 +8,33 @@
 
 const MethodDescriptor = require('./method-descriptor')
 
+const MethodType = require('../constant/method-type').MethodType
+
 class MethodDescriptorBuilder {
+    constructor(moduleName, objectPath, methodName) {
+        this.moduleName = moduleName
+        this.objectPath = objectPath
+        this.methodName = methodName
+        this.type = MethodType.DEFAULT
+        this.apiDescriptor = this.getDescriptor()
+        this.apiId = 0
+    }
+
+    setType(type) {
+        this.type = type
+        return this
+    }
+
+    setAPIDescriptor(apiDescriptor) {
+        this.apiDescriptor = apiDescriptor
+        return this
+    }
+
+    setAPIID(apiID) {
+        this.apiID = apiID
+        return this
+    }
+
     build() {
         return new MethodDescriptor()
     }

diff --git a/lib/context/method-descriptor.js b/lib/context/method-descriptor.js
@@ -10,13 +10,18 @@ const MethodType = require('../constant/method-type').MethodType
 
 // https://github.com/pinpoint-apm/pinpoint/blob/master/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/recorder/AbstractRecorder.java
 class MethodDescriptor {
-  constructor(moduleName, objectPath, methodName, type, apiDescriptor) {
+  constructor(moduleName, objectPath, methodName, type, apiDescriptor, apiId) {
     this.moduleName = moduleName
     this.objectPath = objectPath
     this.methodName = methodName
     this.type = type || MethodType.DEFAULT
     this.apiDescriptor = apiDescriptor || this.getDescriptor()
-    this.apiId = 0
+
+    if (typeof apiId === 'number') {
+      this.apiId = apiId
+    } else {
+      this.apiId = 0
+    }
   }
 
   static create(moduleName, objectPath, methodName) {