threading-benchmark.cc

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.

#include <stdlib.h>
#include <stdio.h>
#include <iostream>
#include <vector>
#include <sstream>
#include "runtime/string-value.h"
#include "util/benchmark.h"
#include "util/cpu-info.h"
#include "util/thrift-util.h"
#include "gen-cpp/Types_types.h"

#include <boost/shared_ptr.hpp>
#include <thrift/protocol/TBinaryProtocol.h>
#include <sstream>
#include <thrift/TApplicationException.h>
#include <thrift/transport/TBufferTransports.h>

using namespace impala;
using namespace std;
using namespace apache::thrift::protocol;
using namespace apache::thrift::transport;

struct TestData {
  uint8_t* msg;
  uint32_t length;
  boost::shared_ptr<TTransport> tmem_transport;
  TBinaryProtocolFactoryT<TMemoryBuffer> tproto_factory;
  boost::shared_ptr<apache::thrift::protocol::TProtocol> tproto;
};


//boost::shared_ptr<apache::thrift::protocol::TProtocol>

void TestDeserializeSharedState(int size, void* d) {
  THostPort deserialized_msg;
  TestData* data = reinterpret_cast<TestData*>(d);
  try {
    deserialized_msg.read(data->tproto.get());
  } catch (apache::thrift::protocol::TProtocolException& e) {
    std::stringstream msg;
    msg << "couldn't deserialize thrift msg:\n" << e.what();
    LOG(INFO) << msg;
  } catch (...) {
    // TODO: Find the right exception for 0 bytes
    //  LOG(INFO) << "Crap!";
  }
}

void TestDeserialize(int size, void* d) {
  THostPort hostport;
  TestData* data = reinterpret_cast<TestData*>(d);
  if (!DeserializeThriftMsg(data->msg, data->length, &hostport).ok()) { LOG(INFO) << "FAILED"; }
}

int main(int argc, char **argv) {
  CpuInfo::Init();

  TestData data;
  string s;
  THostPort hostport;
  hostport.ipaddress = "0.2.3.4";
  hostport.hostname = "localhost";
  hostport.port = 12345;
  SerializeThriftMsgToString(&hostport, &s);
  data.msg = (uint8_t*)const_cast<char *>(s.c_str());
  data.length = s.size();

  data.tmem_transport.reset(new apache::thrift::transport::TMemoryBuffer(
      const_cast<uint8_t*>(data.msg), data.length));

  data.tproto = data.tproto_factory.getProtocol(data.tmem_transport);

  Benchmark suite("deserialize");
  suite.AddBenchmark("deserialize", TestDeserialize, &data);
  suite.AddBenchmark("deserialize-shared-state", TestDeserializeSharedState, &data);
  cout << suite.Measure();

  return 0;
}