# encoding: utf-8

require 'ting_yun/agent/datastore/metric_helper'
require 'ting_yun/agent/method_tracer_helpers'


module TingYun
  module Agent
    module Datastore
      def self.wrap(product, operation, collection = nil, ip_address = nil, port = nil, dbname=nil,  callback = nil )
        return yield unless operation
        klass_name, *metrics = TingYun::Agent::Datastore::MetricHelper.metrics_for(product, operation, ip_address ,
                                                                                   port,
                                                                        dbname,collection )
        TingYun::Agent::MethodTracerHelpers.trace_execution_scoped(metrics, {}, nil, klass_name) do
          t0 = Time.now
          begin
            yield
          ensure
            elapsed_time = (Time.now - t0).to_f
            if callback
              callback.call(elapsed_time)
            end
            config = {
                :product => product,
                :operation => operation,
                :database => collection,
                :host => ip_address,
                :port => port,
                :type => product,
                :nosql => klass_name
            }
            ::TingYun::Agent::Collector::TransactionSampler.notice_nosql_statement(config,elapsed_time*1000, :nosql)
          end
        end
      end
    end
  end
end