Contents

require 'shellwords'
require 'telemetry/number_helper'

module Telemetry
  module Metrics
    module Parser
      module LineProtocol
        extend Telemetry::NumberHelper

        def parse(line, use_shellwords: true)
          if use_shellwords
            raw_tags, raw_fields, timestamp = Shellwords.split(line.strip)
          else
            raw_tags, raw_fields, timestamp = line.strip.split
          end

          {
            measurement: get_measurement(line),
            tags: split_string_to_hash(raw_tags),
            fields: split_string_to_hash(raw_fields),
            timestamp: convert_to_number(timestamp)
          }
        end
        module_function :parse

        def get_measurement(line)
          measurement = line.split(',').first
          return nil if measurement.include? '='

          measurement
        end
        module_function :get_measurement

        def split_string_to_hash(raw_string)
          results = {}
          return results if raw_string.nil?

          raw_string.split(',').each do |string|
            next unless string.include? '='

            k, v = string.split('=')
            results[k.to_sym] = convert_to_number(v)
          end

          results
        end
        module_function :split_string_to_hash

        def to_line_protocol(measurement:, fields:, tags: {}, timestamp: DateTime.now.strftime('%Q'))
          "#{measurement},#{hash_to_line(tags)} #{hash_to_line(fields)} #{timestamp}"
        end
        module_function :to_line_protocol

        def hash_to_line(hash)
          hash.map { |k, v| "#{k}=#{v}" }.join(',').strip.delete_suffix(',')
        end
        module_function :hash_to_line
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
telemetry-metrics-parser-0.1.1	lib/telemetry/metrics/parser/line_protocol.rb