Ejemplo n.º 1
0
  public long extractTimestampMillis(String topic, final byte[] bytes) throws IOException {
    if (timestampFieldPath != null) {
      com.google.protobuf.Message decodedMessage = protobufUtil.decodeMessage(topic, bytes);
      int i = 0;
      for (; i < timestampFieldPath.length - 1; ++i) {
        decodedMessage =
            (com.google.protobuf.Message)
                decodedMessage.getField(
                    decodedMessage.getDescriptorForType().findFieldByName(timestampFieldPath[i]));
      }
      Object timestampObject =
          decodedMessage.getField(
              decodedMessage.getDescriptorForType().findFieldByName(timestampFieldPath[i]));
      if (timestampObject instanceof com.google.protobuf.Timestamp) {
        return Timestamps.toMillis((com.google.protobuf.Timestamp) timestampObject);
      } else {
        return toMillis((Long) timestampObject);
      }
    } else {
      // Assume that the timestamp field is the first field, is required,
      // and is a uint64.

      CodedInputStream input = CodedInputStream.newInstance(bytes);
      // Don't really care about the tag, but need to read it to get, to
      // the payload.
      input.readTag();
      return toMillis(input.readUInt64());
    }
  }
Ejemplo n.º 2
0
  public ProtobufMessageParser(SecorConfig config) {
    super(config);

    protobufUtil = new ProtobufUtil(config);
    if (protobufUtil.isConfigured()) {
      String timestampFieldName = mConfig.getMessageTimestampName();
      String timestampFieldSeparator = mConfig.getMessageTimestampNameSeparator();
      if (timestampFieldSeparator == null || timestampFieldSeparator.isEmpty()) {
        timestampFieldSeparator = ".";
      }
      LOG.info(
          "Using protobuf timestamp field path: {} with separator: {}",
          timestampFieldName,
          timestampFieldSeparator);
      timestampFieldPath = timestampFieldName.split(Pattern.quote(timestampFieldSeparator));
    } else {
      LOG.info(
          "Protobuf message class is not configured, will assume that timestamp is the first uint64 field");
    }
  }