gql_metrics_tracer 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8a0d96d8b78c4b80c1625c6b5500cebac944464197884576612f00ba6bc584cf
-  data.tar.gz: cf69077f4fc23379bf50e058df072aab751d4e676ef9c28e664d4632b2bd820f
+  metadata.gz: 932d56d4a22e85406cba06faebc9e220ea9da7389049fca1266cc7796684a0ba
+  data.tar.gz: 91f73c97ec02642e48576da8f0c24522b137b4b83a64dc705ab339bb4dae6edc
 SHA512:
-  metadata.gz: a65dd23fd1f46112ea4f389e373326e7da910df2f024135f5c49220e8becb7f85b21a35758432e37dc05583afd6342827f5f358ce2364f665c5a4f3ece3ba1ad
-  data.tar.gz: 2cb1aa8955955c854c8b864fb9c9f7dae8aeaf4ea0396104c6f0bbb5a0547fc75b7753154e52f8828d3359acdc95e23cfb39f65dd55fd74177693c8cb84cd34c
+  metadata.gz: 16d5cf2b5e49c0c09a7fe5ad356d09432baf46400c4b48ff86e517a17121cfead10a088718a84696ff400d283730de27f8bd6f6d49f1111aeb1d2178fe9e123e
+  data.tar.gz: f47e07317b13d0130e298d41d4171990c8670af9d8b3090822e251460a938a092020caf822d82b41f57aa478a029effa74471f7b887d0bb12d1b9cad9d8b5e54

data/lib/gql_metrics_tracer.rb

@@ -1,194 +1,66 @@
-# frozen_string_literal: true
-
 require "typhoeus"
 require "json"
-
-class GQLMetricsTracer
-  def self.use(schema_definition, _options = {})
-    tracer = new(schema_definition)
-    schema_definition.instrument(:query, tracer)
-    schema_definition.tracer(tracer)
-  end
-
-  def initialize(schema_definition)
-    @schema_id = nil
-    @schema_definition = schema_definition
-
-    publish_schema!
-  end
-
-  def before_query(query)
-    query.context["gqlmetrics-trace"] = {
-      "starts-ns" => precise_time,
-      "starts-timestamp" => Time.current,
-      "traces" => [],
-    }
-  end
-
-  def after_query(query)
-    ended_at_ns = precise_time
-    ended_timestamp = Time.current
-    started_at = query.context.dig("gqlmetrics-trace", "starts-timestamp")
-    started_at_ns = query.context.dig("gqlmetrics-trace", "starts-ns")
-    traces = query.context.dig("gqlmetrics-trace", "traces")
-
-    send_trace!(
-      query: query.document.to_query_string,
-      duration_ns: ended_at_ns - started_at_ns,
-      started_at: started_at,
-      ended_at: ended_timestamp,
-      traces: traces,
-    )
-  end
-
-  def trace(key, data, &block)
-    if key == "execute_field"
-      field = data[:context].present? ? data[:context].field : data[:field]
-      path = data[:context].present? ? data[:context].path : data[:path]
-      parent_type = data[:context].present? ? data[:context].parent_type : data[:owner].graphql_name
-      context = data[:context] || data[:object].context
-
-      trace_field!(field: field, path: path, parent_type: parent_type, context: context, &block)
-    else
-      yield
+require "active_support/all"
+
+require "gql_metrics_tracer/field_tracer"
+require "gql_metrics_tracer/query_tracer"
+require "gql_metrics_tracer/api_client"
+
+module GQLMetrics
+  class Tracer
+    def self.use(schema_definition, _options = {})
+      tracer = new(schema_definition)
+      schema_definition.instrument(:query, tracer)
+      schema_definition.tracer(tracer)
     end
-  end
 
-  private
-
-  attr_reader :schema_id, :schema_definition
-
-  def send_trace!(query:, duration_ns:, started_at:, ended_at:, traces:)
-    return unless graph_token_configured?
-    return if @schema_id.nil?
+    def initialize(schema_definition)
+      @api_client = ApiClient.new schema_definition: schema_definition
+      Thread.new { @api_client.publish_schema! }
+    end
 
-    Thread.new do
-      response = Typhoeus.post(
-        "#{gqlmetrics_http_host}/api/traces",
-        headers: {
-          'Content-Type': "application/json",
-          'Authorization': "Bearer #{ENV['GQLMETRICS_GRAPH_ID']}",
-        },
-        timeout: 30,
-        body: JSON.dump(
-          {
-            schema_id: @schema_id,
-            query: query,
-            started_at_ms: timestamp_to_ms(started_at),
-            ended_at_ms: timestamp_to_ms(ended_at),
-            duration_ns: duration_ns,
-            traces: traces,
-          },
+    def before_query(query)
+      query.context["gqlmetrics-trace"] = {
+        "query-tracer" => QueryTracer.new(
+          schema_id: schema_id,
+          query: query.document.to_query_string,
+          operation_name: query.context[:operation_name],
         ),
-      )
-
-      trace_id = JSON.parse(response.body)["id"]
-
-      if trace_id.present?
-        Rails.logger.info "[GQLMetricsTracer] Sent trace #{trace_id}"
-      else
-        Rails.logger.warn "[GQLMetricsTracer] Failed to send the trace..."
-      end
+      }
     end
-  end
 
-  def trace_field!(field:, path:, parent_type:, context:)
-    query_starts_at = context["gqlmetrics-trace"]["starts-ns"]
+    def after_query(query)
+      query_tracer = query.context.dig("gqlmetrics-trace", "query-tracer")
+      query_tracer.end_trace!
 
-    sql_queries = []
-    ActiveSupport::Notifications.subscribe "sql.active_record" do |event|
-      unless /(pg_)|(max_identifier_length)/.match?(event.payload[:sql])
-        sql_queries.push(
-          {
-            sql: event.payload[:sql],
-            start_time: event.time,
-            end_time: event.end,
-            duration_ns: event.duration * 1_000_000,
-            allocations_count: event.allocations,
-          },
-        )
-      end
+      Thread.new { api_client.send_trace!(query_tracer) }
     end
 
-    starts_at = precise_time
-    yield
-  ensure
-    ends_at = precise_time
-    ActiveSupport::Notifications.unsubscribe("sql.active_record")
-
-    trace_node = {
-      "response_name" => field.name,
-      "original_field_name" => field.name,
-      "type" => field.type.to_type_signature,
-      "parent_type" => parent_type.to_s,
-      "start_time" => starts_at - query_starts_at,
-      "end_time" => ends_at - query_starts_at,
-      "sql_queries" => sql_queries,
-      "path" => path,
-    }
-
-    context["gqlmetrics-trace"]["traces"] << trace_node
-  end
-
-  # Introspect the schema and upload to the ingress server
-  def publish_schema!
-    return unless graph_token_configured?
-
-    Thread.new do
-      # Sleep for 50ms so that the definition has time enough to initialize
-      sleep(0.05)
-      schema = schema_definition.execute(GraphQL::Introspection::INTROSPECTION_QUERY)
-
-      payload = {
-        query: PUBLISH_SCHEMA_QUERY,
-        variables: {
-          schema: schema.to_json,
-        },
-      }
-
-      response = Typhoeus.post(
-        "#{gqlmetrics_http_host}/graphql",
-        headers: {
-          'Content-Type': "application/json",
-          'Authorization': "Bearer #{ENV['GQLMETRICS_GRAPH_ID']}",
-        },
-        timeout: 10,
-        body: payload.to_json,
-      )
-
-      json_response = JSON.parse(response.body)
-      schema_id = json_response.dig("data", "publishSchema", "id")
-
-      if schema_id.present?
-        @schema_id = schema_id
+    def trace(key, data, &block)
+      if key == "execute_field"
+        context = data[:context] || data[:object].context
+        query_tracer = context.dig("gqlmetrics-trace", "query-tracer")
+
+        field = !data[:context].nil? ? data[:context].field : data[:field]
+        path = !data[:context].nil? ? data[:context].path : data[:path]
+        parent_type = !data[:context].nil? ? data[:context].parent_type : data[:owner].graphql_name
+
+        query_tracer.trace_field!(
+          field: field,
+          path: path,
+          parent_type: parent_type,
+          context: context,
+          &block
+        )
       else
-        Rails.logger.warn "[GQLMetricsTracer] Failed to publish the latest schema..."
-        nil
+        yield
       end
     end
-  end
-
-  def graph_token_configured?
-    ENV["GQLMETRICS_GRAPH_ID"].present?
-  end
 
-  def gqlmetrics_http_host
-    ENV.fetch("GQLMETRICS_HTTP_HOST", "http://localhost:5000")
-  end
+    private
 
-  def precise_time
-    Process.clock_gettime(Process::CLOCK_MONOTONIC, :nanosecond)
-  end
+    attr_reader :api_client
 
-  def timestamp_to_ms(timestamp)
-    (timestamp.to_f * 1_000).to_i
+    delegate :schema_id, to: :api_client
   end
-
-  PUBLISH_SCHEMA_QUERY = <<~PUBLISH_SCHEMA_QUERY
-    mutation PublishSchema($schema: String!) {
-      publishSchema(schema: $schema) {
-        id
-      }
-    }
-  PUBLISH_SCHEMA_QUERY
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gql_metrics_tracer
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Dan Andreasson
@@ -26,6 +26,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: graphql
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
@@ -40,6 +68,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.9'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: timecop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Explore your GraphQL metrics
 email: dan@saits.se
 executables: []