require 'grpcx/server' # or just 'grpcx'

class MyServer < GRPC::RpcServer
  include Grpcx::Server
end

# proceed as with usual GRPC::RpcServer:
server = MyServer.new(...)
server.handle(MyService.new)
server.add_http2_port '127.0.0.1:8080', :this_port_is_insecure
server.run_till_terminated

require 'grpclb/server'
require 'grpcx/server'

class MyServer < Grpclb::Server
  include Grpcx::Server
end

...

module Datadog::Notifications::Plugins
  class GRPC < Base

    def initialize(opts={})
      super
      Datadog::Notifications.subscribe 'process_action.grpc' do |reporter, event|
        record(reporter, event)
      end
    end

    private

    def record(reporter, event)
      action = event.payload[:action]
      status = event.payload[:exception] ? 'error' : 'ok'
      tags = self.tags + ["rpc:#{action}", "status:#{status}"]

      reporter.batch do
        reporter.increment 'grpc.request', tags: tags
        reporter.timing 'grpc.request.time', event.duration, tags: tags
      end
    end

  end
end

Datadog::Notifications.configure do |c|
  c.use Datadog::Notifications::Plugins::GRPC
end if RUNNING_IN_PROD?

ActiveSupport::Notifications.subscribe('process_action.grpc') do |_name, _start, _finish, _id, payload|
  e = payload[:exception_object]
  next unless e

  Raven.capture_exception(e)
end

Raven.configure do |config|
  config.should_capture = proc {|e|
    !e.is_a?(GRPC::BadStatus) # skip GRPC::BadStatus, but report everything else
  }
end