pinpoint-node-agent/lib/context/trace/trace-id-builder.js

Pinpoint node agent provided by NAVER

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

'use strict'

const spanId = require('../span-id')
const spanMessages = require('../../data/v1/Span_pb')

const delimiter = '^'
// DefaultTraceId.java
class TraceId {
    constructor(agentId, agentStartTime, transactionId, parentSpanId, spanId, flags) {
        this.agentId = agentId
        this.agentStartTime = agentStartTime
        this.transactionId = transactionId
        this.parentSpanId = parentSpanId
        this.spanId = spanId
        this.flags = flags
    }

    getAgentId() {
        return this.agentId
    }

    getAgentStartTime() {
        return this.agentStartTime
    }

    getTransactionId() {
        return this.transactionId
    }

    getParentSpanId() {
        return this.parentSpanId
    }

    getSpanId() {
        return this.spanId
    }

    getFlags() {
        return this.flags
    }

    getNextTraceId() {
        return new TraceId(this.agentId, this.agentStartTime, this.transactionId, this.spanId, spanId.nextSpanId(this.spanId, this.parentSpanId), this.flags)
    }

    isRoot() {
        return this.parentSpanId == spanId.nullSpanId()
    }

    toString() {
        return `TraceId(agentId=${this.agentId}, agentStartTime=${this.agentStartTime}, transactionId=${this.transactionId}, parentSpanId=${this.parentSpanId}
                    , spanId=${this.spanId}, flags=${this.flags})`
    }

    toStringDelimiterFormatted() {
        return [this.agentId, this.agentStartTime, this.transactionId].join(delimiter)
    }

    toProtocolBuffer() {
        const pTransactionId = new spanMessages.PTransactionId()
        pTransactionId.setAgentid(this.agentId)
        pTransactionId.setAgentstarttime(this.agentStartTime)
        pTransactionId.setSequence(this.transactionId)
        return pTransactionId
    }
}

class TraceIdBuilder {
    constructor(agentId, agentStartTime, transactionId) {
        this.agentId = agentId
        this.agentStartTime = agentStartTime
        this.transactionId = transactionId
        this.parentSpanId = spanId.nullSpanId()
        this.flags = '0'
    }

    static makeByHeader(traceIdHeader) {
        const headers = traceIdHeader.split(delimiter)
        if (headers.length !== 3) {
            return
        }
        return new TraceIdBuilder(headers[0], headers[1], headers[2])
    }

    setSpanId(spanId) {
        this.spanId = spanId
        return this
    }

    setParentSpanId(parentSpanId) {
        this.parentSpanId = parentSpanId
        return this
    }

    setFlags(flags) {
        this.flags = flags
        return this
    }

    build() {
        return new TraceId(this.agentId, this.agentStartTime
            , this.transactionId, this.parentSpanId, this.spanId ?? spanId.newSpanId(), this.flags)
    }
}

module.exports = TraceIdBuilder