gem 'jcw'

require 'jcw'

::JCW::Wrapper.configure do |config|
  config.service_name = "Service name"
  config.connection = { protocol: :udp, host: "127.0.0.1", port: 6831 }
  config.enabled = true
  config.tags = {
    hostname: "custom-hostname",
    custom_tag: "custom-tag-value",
  }
  config.rack_ignore_path_patterns = ["/api/test", %r{/sidekiq}]
  config.logger = Logger.new($stdout)
end

# Set middleware for wrapping all requests
Rails.application.middleware.use(JCW::Rack::Tracer)

::JCW::Wrapper.configure do |config|
  config.service_name = "Service name"
  config.enabled = true
  config.subscribe_to = %w[process_action.action_controller start_processing.action_controller] # set ActiveSupport::Notifications namespaces
  config.connection = { protocol: :tcp, url: "http://localhost:14268/api/traces", headers: { key: "value" } }
  config.tags = {
    hostname: "custom-hostname",
    custom_tag: "custom-tag-value",
  }
  config.rack_ignore_path_patterns = ["/api/test", %r{/sidekiq}]
  config.logger = Logger.new($stdout)
end

# Set middleware for wrapping all requests
Rails.application.middleware.use(JCW::Rack::Tracer)

# If you need send all logs with spans set on_finish_span and extend JaegerLoggerExtension
# Not recommended for UDP sender, because default max packet size is 65,000 bytes.
Rails.application.config.tap do |config|
  config.middleware.use(
    JCW::Rack::Tracer,
    on_finish_span: lambda do |span|
      JCW::Logger.current.logs.each { |log| span.log_kv(**log) }
      JCW::Logger.current.clear # Do not forget to avoid memory leaks
    end,
  )

  config.logger.extend(JCW::LoggerExtension)
end

# Add JCW::Interceptors::Gruf::Client Interceptor to Gruf Client Initializer
options = {}
client_options = { timeout: 10, interceptors: [JCW::Interceptors::Gruf::Client.new] }

client = Gruf::Client.new(
  service: Test::Service, options: options, client_options: client_options
)

request_method = "some_method"
client.call(request_method)

# Add Server Interceptor
Rails.configuration.to_prepare do
  Gruf.configure do |config|
    config.interceptors.use(JCW::Interceptors::Gruf::Server)
  end
end  

# Configure
::JCW::Wrapper.configure do |config|
  config.service_name = "Service Name"
  config.connection = { protocol: :udp, host: "127.0.0.1", port: 6831 }
  config.enabled = true
  config.subscribe_to = [/.*/]
  config.grpc_ignore_methods = %w[grpc.ignore.method]
end