zipkin2.internal.V1ThriftSpanReader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zipkin Show documentation
There is a newer version: 3.4.2
/*
 * Copyright The OpenZipkin Authors
 * SPDX-License-Identifier: Apache-2.0
 */
package zipkin2.internal;

import zipkin2.Endpoint;
import zipkin2.v1.V1Span;

import static zipkin2.internal.ThriftCodec.readListLength;
import static zipkin2.internal.ThriftCodec.skip;
import static zipkin2.internal.ThriftField.TYPE_I32;
import static zipkin2.internal.ThriftField.TYPE_I64;
import static zipkin2.internal.ThriftField.TYPE_STOP;
import static zipkin2.internal.ThriftField.TYPE_STRING;
import static zipkin2.internal.ThriftField.TYPE_STRUCT;
import static zipkin2.internal.V1ThriftSpanWriter.ANNOTATIONS;
import static zipkin2.internal.V1ThriftSpanWriter.BINARY_ANNOTATIONS;
import static zipkin2.internal.V1ThriftSpanWriter.DEBUG;
import static zipkin2.internal.V1ThriftSpanWriter.DURATION;
import static zipkin2.internal.V1ThriftSpanWriter.ID;
import static zipkin2.internal.V1ThriftSpanWriter.NAME;
import static zipkin2.internal.V1ThriftSpanWriter.PARENT_ID;
import static zipkin2.internal.V1ThriftSpanWriter.TIMESTAMP;
import static zipkin2.internal.V1ThriftSpanWriter.TRACE_ID;
import static zipkin2.internal.V1ThriftSpanWriter.TRACE_ID_HIGH;

public final class V1ThriftSpanReader {
  static final String ONE = Character.toString((char) 1);

  public static V1ThriftSpanReader create() {
    return new V1ThriftSpanReader();
  }

  V1Span.Builder builder = V1Span.newBuilder();

  public V1Span read(ReadBuffer buffer) {
    if (builder == null) {
      builder = V1Span.newBuilder();
    } else {
      builder.clear();
    }

    ThriftField thriftField;

    while (true) {
      thriftField = ThriftField.read(buffer);
      if (thriftField.type == TYPE_STOP) break;

      if (thriftField.isEqualTo(TRACE_ID_HIGH)) {
        builder.traceIdHigh(buffer.readLong());
      } else if (thriftField.isEqualTo(TRACE_ID)) {
        builder.traceId(buffer.readLong());
      } else if (thriftField.isEqualTo(NAME)) {
        builder.name(buffer.readUtf8(buffer.readInt()));
      } else if (thriftField.isEqualTo(ID)) {
        builder.id(buffer.readLong());
      } else if (thriftField.isEqualTo(PARENT_ID)) {
        builder.parentId(buffer.readLong());
      } else if (thriftField.isEqualTo(ANNOTATIONS)) {
        int length = readListLength(buffer);
        for (int i = 0; i < length; i++) {
          AnnotationReader.read(buffer, builder);
        }
      } else if (thriftField.isEqualTo(BINARY_ANNOTATIONS)) {
        int length = readListLength(buffer);
        for (int i = 0; i < length; i++) {
          BinaryAnnotationReader.read(buffer, builder);
        }
      } else if (thriftField.isEqualTo(DEBUG)) {
        builder.debug(buffer.readByte() == 1);
      } else if (thriftField.isEqualTo(TIMESTAMP)) {
        builder.timestamp(buffer.readLong());
      } else if (thriftField.isEqualTo(DURATION)) {
        builder.duration(buffer.readLong());
      } else {
        skip(buffer, thriftField.type);
      }
    }

    return builder.build();
  }

  static final class AnnotationReader {
    static final ThriftField TIMESTAMP = new ThriftField(TYPE_I64, 1);
    static final ThriftField VALUE = new ThriftField(TYPE_STRING, 2);
    static final ThriftField ENDPOINT = new ThriftField(TYPE_STRUCT, 3);

    static void read(ReadBuffer buffer, V1Span.Builder builder) {
      long timestamp = 0;
      String value = null;
      Endpoint endpoint = null;

      ThriftField thriftField;
      while (true) {
        thriftField = ThriftField.read(buffer);
        if (thriftField.type == TYPE_STOP) break;

        if (thriftField.isEqualTo(TIMESTAMP)) {
          timestamp = buffer.readLong();
        } else if (thriftField.isEqualTo(VALUE)) {
          value = buffer.readUtf8(buffer.readInt());
        } else if (thriftField.isEqualTo(ENDPOINT)) {
          endpoint = ThriftEndpointCodec.read(buffer);
        } else {
          skip(buffer, thriftField.type);
        }
      }

      if (timestamp == 0 || value == null) return;
      builder.addAnnotation(timestamp, value, endpoint);
    }
  }

  static final class BinaryAnnotationReader {
    static final ThriftField KEY = new ThriftField(TYPE_STRING, 1);
    static final ThriftField VALUE = new ThriftField(TYPE_STRING, 2);
    static final ThriftField TYPE = new ThriftField(TYPE_I32, 3);
    static final ThriftField ENDPOINT = new ThriftField(TYPE_STRUCT, 4);

    static void read(ReadBuffer buffer, V1Span.Builder builder) {
      String key = null;
      String value = null;
      Endpoint endpoint = null;
      boolean isBoolean = false;
      boolean isString = false;

      while (true) {
        ThriftField thriftField = ThriftField.read(buffer);
        if (thriftField.type == TYPE_STOP) break;
        if (thriftField.isEqualTo(KEY)) {
          key = buffer.readUtf8(buffer.readInt());
        } else if (thriftField.isEqualTo(VALUE)) {
          value = buffer.readUtf8(buffer.readInt());
        } else if (thriftField.isEqualTo(TYPE)) {
          switch (buffer.readInt()) {
            case 0:
              isBoolean = true;
              break;
            case 6:
              isString = true;
              break;
          }
        } else if (thriftField.isEqualTo(ENDPOINT)) {
          endpoint = ThriftEndpointCodec.read(buffer);
        } else {
          skip(buffer, thriftField.type);
        }
      }
      if (key == null || value == null) return;
      if (isString) {
        builder.addBinaryAnnotation(key, value, endpoint);
      } else if (isBoolean && ONE.equals(value) && endpoint != null) {
        if (key.equals("sa") || key.equals("ca") || key.equals("ma")) {
          builder.addBinaryAnnotation(key, endpoint);
        }
      }
    }
  }

  V1ThriftSpanReader() {
  }
}