rpc-trace.h

Go to the documentation of this file.

// Copyright 2012 Cloudera Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef IMPALA_RPC_RPC_TRACE_H
#define IMPALA_RPC_RPC_TRACE_H

#include "util/collection-metrics.h"
#include "rpc/thrift-server.h"
#include "util/internal-queue.h"

#include <thrift/TProcessor.h>
#include <boost/thread/thread.hpp>
#include <boost/unordered_map.hpp>

#include <rapidjson/document.h>

namespace impala {

class Webserver;
class MetricGroup;

class RpcEventHandler : public apache::thrift::TProcessorEventHandler {
 public:
  RpcEventHandler(const std::string& server_name, MetricGroup* metrics);

  virtual void* getContext(const char* fn_name, void* server_context);

  virtual void postWrite(void* ctx, const char* fn_name, uint32_t bytes);

  void ToJson(rapidjson::Value* server, rapidjson::Document* document);

  void Reset(const std::string& method_name);

  void ResetAll();

  std::string server_name() const { return server_name_; }

 private:
  struct MethodDescriptor {
    std::string name;

    StatsMetric<double>* time_stats;

    AtomicInt<uint32_t> num_in_flight;
  };

  typedef boost::unordered_map<std::string, MethodDescriptor*> MethodMap;

  struct InvocationContext {
    const int64_t start_time_ms;

    const ThriftServer::ConnectionContext* cnxn_ctx;

    MethodDescriptor* method_descriptor;

    InvocationContext(int64_t start_time, const ThriftServer::ConnectionContext* cnxn_ctx,
        MethodDescriptor* descriptor)
        : start_time_ms(start_time), cnxn_ctx(cnxn_ctx), method_descriptor(descriptor) { }
  };

  boost::mutex method_map_lock_;

  MethodMap method_map_;

  std::string server_name_;

  MetricGroup* metrics_;

};

void InitRpcEventTracing(Webserver* webserver);

}

#endif