[#242] Extract Class refactoring from ActiveTrace to ActiveRequestRepository

Author: feelform

lib/context/trace-context.js

@@ -23,7 +23,7 @@ const AsyncSpanChunkBuilder = require('./trace/async-span-chunk-builder')
 const ChildTraceBuilder = require('./trace/child-trace-builder')
 const DisableChildTrace = require('./trace/disable-child-trace')
 const disableAsyncId = require('./trace/disable-async-id')
-const ActiveTraceRepository = require('../metric/active-trace-repository')
+const activeRequestRepository = require('../metric/active-request-repository')
 
 class TraceContext {
   constructor(agentInfo, dataSender, config) {
@@ -35,7 +35,6 @@ class TraceContext {
       this.enableSampling = config.sampling
     }
     this.traceSampler = new TraceSampler(agentInfo, config)
-    this.activeRequestRepository = new ActiveTraceRepository()
   }
 
   getAgentInfo() {
@@ -83,7 +82,7 @@ class TraceContext {
     }
     try {
       trace.close()
-      this.activeRequestRepository.remove(trace.getTraceRoot())
+      activeRequestRepository.remove(trace.getTraceRoot())
       // activeTrace.remove(trace)
     } catch (e) {
       log.error('Fail to complete trace object', e)
@@ -118,7 +117,7 @@ class TraceContext {
   }
 
   newLocalTrace(traceRoot) {
-    this.activeRequestRepository.register(traceRoot)
+    activeRequestRepository.register(traceRoot)
     return new DisableTrace(traceRoot)
   }
 
@@ -137,7 +136,7 @@ class TraceContext {
     const spanBuilder = new SpanBuilder(traceRoot)
     const spanChunkBuilder = new SpanChunkBuilder(traceRoot)
     const repository = new SpanRepository(spanChunkBuilder, this.dataSender, this.agentInfo)
-    this.activeRequestRepository.register(traceRoot)
+    activeRequestRepository.register(traceRoot)
     return new Trace2(spanBuilder, repository)
   }
 

lib/metric/active-trace-repository.js lib/metric/active-request-repository.jslib/metric/active-trace-repository.js renamed to lib/metric/active-request-repository.js

@@ -9,7 +9,7 @@
 const SimpleCache = require('../utils/simple-cache')
 
 // DefaultActiveTraceRepository.java
-class ActiveTraceRepository {
+class ActiveRequestRepository {
     constructor() {
         this.activeTraceCache = new SimpleCache()
     }
@@ -29,4 +29,4 @@ class ActiveTraceRepository {
     }
 }
 
-module.exports = ActiveTraceRepository
+module.exports = new ActiveRequestRepository()

test/stats/active-trace.test.js

@@ -6,59 +6,20 @@
 
 const test = require('tape')
 const axios = require('axios')
-const { fixture, util } = require('../test-helper')
-const activeTrace = require('../../lib/metric/active-trace')
 const agent = require('../support/agent-singleton-mock')
 const express = require('express')
 const http = require('http')
 const grpc = require('@grpc/grpc-js')
 const services = require('../../lib/data/v1/Service_grpc_pb')
 const spanMessages = require('../../lib/data/v1/Span_pb')
+const activeRequestRepository = require('../../lib/metric/active-request-repository')
 
 const TEST_ENV = {
   host: 'localhost',
   port: 5005,
 }
 const getServerUrl = (path) => `http://${TEST_ENV.host}:${TEST_ENV.port}${path}`
 
-test(`Should record active trace in multiple call`, function (t) {
-  agent.bindHttp()
-
-  const PATH = '/active-trace'
-  const LASTONE_PATH = '/active-trace/lastone'
-  const SHUTDOWN = '/shutdown'
-  const app = new express()
-
-  app.get(PATH, async (req, res) => {
-    await util.sleep(2000)
-    res.send('ok get')
-  })
-
-  app.get(LASTONE_PATH, async (req, res) => {
-    res.send('ok get')
-  })
-
-  const server = app.listen(TEST_ENV.port, async function () {
-    Promise.all([
-      axios.get(getServerUrl(PATH), { httpAgent: new http.Agent({ keepAlive: false }) }),
-      axios.get(getServerUrl(PATH), { httpAgent: new http.Agent({ keepAlive: false }) }),
-      axios.get(getServerUrl(LASTONE_PATH), { httpAgent: new http.Agent({ keepAlive: false }) }),
-    ]).then((result) => {
-      t.equal(activeTrace.getAllTraces().length, 0)
-      t.equal('' + agent.mockAgentStartTime, agent.agentInfo.startTimestamp, "startTimestamp equals")
-
-      server.close()
-      t.end()
-    }).catch(() => {
-      server.close()
-      t.end()
-    })
-  })
-
-  t.equal(agent.mockAgentId, fixture.config.agentId, "Agent ID equals")
-  t.equal(agent.agentInfo, agent.pinpointClient.agentInfo, "AgentInfo equals")
-})
-
 test(`Active trace should be recorded with HTTP call`, function (t) {
   const collectorServer = new grpc.Server()
   collectorServer.addService(services.MetadataService, {
@@ -73,7 +34,7 @@ test(`Active trace should be recorded with HTTP call`, function (t) {
     const app = new express()
     app.get('/active-trace', async (req, res) => {
       agent.callbackTraceClose((trace) => {
-        const actualCached = agent.getTraceContext().activeRequestRepository.activeTraceCache.get(trace.getTraceRoot().getTransactionId())
+        const actualCached = activeRequestRepository.activeTraceCache.get(trace.getTraceRoot().getTransactionId())
         t.equal(actualCached, trace.getTraceRoot(), 'active trace traceRoot is cached')
       })
       setTimeout(() => {
@@ -85,7 +46,7 @@ test(`Active trace should be recorded with HTTP call`, function (t) {
       const result = await axios.get(getServerUrl('/active-trace'), { httpAgent: new http.Agent({ keepAlive: false }) })
       t.equal(result.status, 200, 'status code is 200')
       server.close(() => {
-        const cacheSize = agent.getTraceContext().activeRequestRepository.activeTraceCache.cache.size
+        const cacheSize = activeRequestRepository.activeTraceCache.cache.size
         t.equal(cacheSize, 0, 'active trace cache is empty')
         t.end()
       })

test/support/agent-singleton-mock.js

@@ -23,6 +23,7 @@ const transactionIdGenerator = require('../../lib/context/sequence-generators').
 const closedTraceWrapped = Symbol('closedTraceWrapped')
 const stringMetaService = require('../../lib/context/string-meta-service')
 const apiMetaService = require('../../lib/context/api-meta-service')
+const activeRequestRepository = require('../../lib/metric/active-request-repository')
 
 let traces = []
 const resetTraces = () => {
@@ -111,7 +112,7 @@ class MockAgent extends Agent {
         if (sampler.getSamplingCountGenerator()) {
             sampler.getSamplingCountGenerator().reset()
         }
-        this.traceContext.activeRequestRepository.activeTraceCache.cache.clear()
+        activeRequestRepository.activeTraceCache.cache.clear()
         transactionIdGenerator.reset()
 
         httpShared.clearPathMatcher()