pinpoint-node-agent/lib/context/trace/trace-root.js

Pinpoint APM agent for Node.js — distributed tracing, error analysis, and URI statistics

/**
 * Pinpoint Node.js Agent
 * Copyright 2020-present NAVER Corp.
 * Apache License v2.0
 */

'use strict'

class TraceRoot {
    constructor(agentId, traceStartTime, transactionId) {
        this.agentId = agentId
        this.traceStartTime = traceStartTime
        this.transactionId = transactionId
    }

    getAgentId() {
        return this.agentId
    }

    getTraceStartTime() {
        return this.traceStartTime
    }

    getTransactionId() {
        return this.transactionId
    }

    isSampled() {
        return false
    }

    getShared() {
        if (!this.shared) {
            this.shared = new Shared()
        }
        return this.shared
    }

    hasErrorCode() {
        return this.shared?.getErrorCode?.()
    }

    toString() {
        return `TraceRoot(agentId=${this.agentId}, traceStartTime=${this.traceStartTime}, transactionId=${this.transactionId})`
    }

    setEnricher(name, value) {
        this.getShared().setEnricher(name, value)
    }

    getEnricher(name) {
        return this.getShared().getEnricher(name)
    }
}

class Shared {
    maskErrorCode(errorCode) {
        this.errorCode = errorCode
    }

    getErrorCode() {
        return this.errorCode
    }

    setEnricher(name, value) {
        if (!this.enrichers) {
            this.enrichers = new Map()
        }
        this.enrichers.set(name, value)
    }

    getEnricher(name) {
        return this.enrichers?.get(name)
    }
}

module.exports = TraceRoot