class GraphQL::Tracing::DetailedTrace::RedisBackend

# File lib/graphql/tracing/detailed_trace/redis_backend.rb, line 9
def initialize(redis:, limit: nil)
  @redis = redis
  @key = KEY_PREFIX + "traces"
  @remrangebyrank_limit = limit ? -limit - 1 : nil
end

# File lib/graphql/tracing/detailed_trace/redis_backend.rb, line 33
def delete_all_traces
  @redis.del(@key)
end

# File lib/graphql/tracing/detailed_trace/redis_backend.rb, line 28
def delete_trace(id)
  @redis.zremrangebyscore(@key, id, id)
  nil
end

# File lib/graphql/tracing/detailed_trace/redis_backend.rb, line 37
def find_trace(id)
  str_data = @redis.zrange(@key, id, id, byscore: true).first
  if str_data.nil?
    nil
  else
    entry_to_trace(id, str_data)
  end
end

# File lib/graphql/tracing/detailed_trace/redis_backend.rb, line 46
def save_trace(operation_name, duration_ms, begin_ms, trace_data)
  id = begin_ms
  data = JSON.dump({ "o" => operation_name, "d" => duration_ms, "b" => begin_ms, "t" => Base64.encode64(trace_data) })
  @redis.pipelined do |pipeline|
    pipeline.zadd(@key, id, data)
    if @remrangebyrank_limit
      pipeline.zremrangebyrank(@key, 0, @remrangebyrank_limit)
    end
  end
  id
end

# File lib/graphql/tracing/detailed_trace/redis_backend.rb, line 15
def traces(last:, before:)
  before = case before
  when Numeric
    "(#{before}"
  when nil
    "+inf"
  end
  str_pairs = @redis.zrange(@key, before, 0, byscore: true, rev: true, limit: [0, last || 100], withscores: true)
  str_pairs.map do |(str_data, score)|
    entry_to_trace(score, str_data)
  end
end