# app/initializers/graphql_hive.rb
GraphQLHive.configure do |config|
  config.token = '<YOUR_TOKEN>'
end

# schema.rb
class Schema < GraphQL::Schema
  query QueryType

  trace_with(GraphQLHive::Tracer)
end

# app/initializers/graphql_hive.rb
GraphQLHive.report_schema_to_hive(
  schema: Schema,
  options: {
    # Required
    author: ENV['GITHUB_USER'],
    commit: ENV['GITHUB_COMMIT'],
    # Optional
    service: ENV['GRAPHQL_HIVE_SERVICE_NAME'],
    url: ENV['GRAPHQL_HIVE_SERVICE_URL'],
    force: ENV['GRAPHQL_HIVE_FORCE_REPORT']
  }
)

# config/puma.rb
preload_app!

on_worker_boot do
  GraphQLHive.start
end

# config/puma.rb

on_worker_shutdown do
  GraphQLHive.stop
end

# app/initializers/graphql_hive.rb
GraphQLHive.configure do |config|
  # Token is the only required configuration value.
  config.token = 'YOUR-REGISTRY-TOKEN'
  
  # The following are optional configuration values.
  
  # Enable/disable Hive Client.
  config.enabled = true
  # Verbose logs.
  config.debug = false
  # A custom logger.
  config.logger = MyLogger.new
  # Endpoint and port of the Hive API. Change this if you are using a self-hosted Hive instance.
  config.endpoint = 'app.graphql-hive.com'
  config.port = 80
  # Number of operations sent to Hive in a batch (AFTER sampling).
  config.buffer_size = 50
  # Size of the queue used to send operations to the buffer before sampling.
  config.queue_size = 1000
  # Report usage to Hive.
  config.collect_usage = true
  # Usage sampling configurations.
  config.collect_usage_sampling = {
    # % of operations recorded.
    sample_rate: 0.5,
    # Custom sampler to assign custom sampling rates.
    sampler: proc { |context| context.operation_name.includes?('someQuery') 1 : 0.5 },
    # Sample every distinct operation at least once.
    at_least_once: true,
    # Assign custom keys to distinguish between distinct operations.
    key_generator: proc { |context| context.operation_name }
  }

  # Pass an optional proc to client_info to help identify the client (ex: Apollo web app) that performed the query.
  config.client_info = proc { |context|
    { name: context.client_name, version: context.client_version }
  }
end