pinpoint-node-agent/lib/context/span-chunk.js

Pinpoint node agent provided by NAVER

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

'use strict'

const spanMessages = require('../data/v1/Span_pb')
const { convertTransactionId, convertLocalAsyncId } = require('../data/grpc-data-convertor')
const SpanEvents = require('./span-events')

class SpanChunk {
  constructor(traceId, agentInfo, asyncId, spanEventList) {
    if (!traceId || !agentInfo) {
      throw new Error('Can not initialize SpanChunk', traceId, agentInfo)
    }

    this.agentId = agentInfo.agentId // required, from config
    this.applicationName = agentInfo.applicationName // required, from config
    this.agentStartTime = agentInfo.agentStartTime // required, from config
    this.serviceType = agentInfo.serviceType // required
    this.spanId = traceId.spanId // required
    this.parentSpanId = traceId.parentSpanId
    this.transactionIdObject = traceId.transactionId
    this.spanEventList = spanEventList
    this.endPoint = null // host, domain
    this.applicationServiceType = agentInfo.serviceType // from config
    this.localAsyncId = asyncId
  }

  static getFactoryMethod(traceId, agentInfo) {
    return (spanEventList) => new SpanChunk(traceId, agentInfo, null, spanEventList)
  }

  static getAsyncFactoryMethod(traceId, agentInfo, asyncId) {
    return (spanEventList) => new SpanChunk(traceId, agentInfo, asyncId, spanEventList)
  }

  get spanMessage() {
    const pSpanMessage = new spanMessages.PSpanMessage()
    const pSpanChunk = new spanMessages.PSpanChunk()
    pSpanChunk.setVersion(1)

    const pTransactionId = convertTransactionId(this.transactionIdObject)
    pSpanChunk.setTransactionid(pTransactionId)

    pSpanChunk.setSpanid(this.spanId)
    pSpanChunk.setEndpoint(this.endPoint)
    pSpanChunk.setApplicationservicetype(this.applicationServiceType)
    pSpanChunk.setLocalasyncid(convertLocalAsyncId(this.localAsyncId))

    const keyTime = this.keyTime
    pSpanChunk.setKeytime(keyTime)

    const spanEvents = new SpanEvents(this.spanEventList)
    const pSpanEvents = spanEvents.getpSpanEvents(keyTime)
    if (pSpanEvents.length > 0) {
      pSpanEvents.forEach(pSpanEvent => pSpanChunk.addSpanevent(pSpanEvent))
    }

    pSpanMessage.setSpanchunk(pSpanChunk)
    return pSpanMessage
  }

  get keyTime() {
    const minStartTimeSpanEvent = this.spanEventList.reduce((minStartTimeSpanEvent, currentSpanEvent) => minStartTimeSpanEvent.startTime < currentSpanEvent.startTime ? minStartTimeSpanEvent : currentSpanEvent)
    if (!minStartTimeSpanEvent) {
      return 0
    }
    return minStartTimeSpanEvent.startTime
  }
}

module.exports = SpanChunk