src/agent/protocol/grpc/SegmentObjectAdapter.ts

/*!
 *
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

import { Log, RefType, SegmentObject, SegmentReference, SpanObject } from '@/proto/language-agent/Tracing_pb';
import config from '@/config/AgentConfig';
import { KeyStringValuePair } from '@/proto/common/Common_pb';
import Segment from '@/trace/context/Segment';

/**
 * An adapter that adapts {@link Segment} objects to gRPC object {@link SegmentObject}.
 */
export default class SegmentObjectAdapter extends SegmentObject {
  constructor(segment: Segment) {
    super();
    this
      .setService(config.serviceName)
      .setServiceinstance(config.serviceInstance)
      .setTraceid(segment.relatedTraces[0].toString())
      .setTracesegmentid(segment.segmentId.toString())
      .setSpansList(segment.spans.map(span => new SpanObject()
        .setSpanid(span.id)
        .setParentspanid(span.parentId)
        .setStarttime(span.startTime)
        .setEndtime(span.endTime)
        .setOperationname(span.operation)
        .setPeer(span.peer)
        .setSpantype(span.type)
        .setSpanlayer(span.layer)
        .setComponentid(span.component.id)
        .setIserror(span.errored)
        .setLogsList(span.logs.map(log => new Log()
          .setTime(log.timestamp)
          .setDataList(log.items.map(logItem => new KeyStringValuePair()
            .setKey(logItem.key)
            .setValue(logItem.val),
          )),
        ))
        .setTagsList(span.tags.map(tag => new KeyStringValuePair()
          .setKey(tag.key)
          .setValue(tag.val),
        ))
        .setRefsList(span.refs.map(ref => new SegmentReference()
          .setReftype(RefType.CROSSPROCESS)
          .setTraceid(ref.traceId.toString())
          .setParenttracesegmentid(ref.segmentId.toString())
          .setParentspanid(ref.spanId)
          .setParentservice(ref.service)
          .setParentserviceinstance(ref.serviceInstance)
          .setNetworkaddressusedatpeer(ref.clientAddress),
        )),
      ));
  }
}