thrift-thread.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_THRIFT_THREAD_H
#define IMPALA_RPC_THRIFT_THREAD_H

#define HAVE_PTHREAD_H

#include <thrift/concurrency/Thread.h>

#include "common/logging.h"
#include "util/thread.h"
#include "util/promise.h"

namespace impala {

class ThriftThreadFactory : public apache::thrift::concurrency::ThreadFactory {
 public:
  ThriftThreadFactory(const std::string& group, const std::string& prefix)
      : group_(group), prefix_(prefix), count_(0) { }

  virtual boost::shared_ptr<apache::thrift::concurrency::Thread> newThread(
      boost::shared_ptr<apache::thrift::concurrency::Runnable> runnable) const;

  virtual apache::thrift::concurrency::Thread::id_t getCurrentThreadId() const;

 private:
  std::string group_;

  std::string prefix_;

  mutable int64_t count_;
};

class ThriftThread : public apache::thrift::concurrency::Thread {
 public:
  ThriftThread(const std::string& group, const std::string& name,
      boost::shared_ptr<apache::thrift::concurrency::Runnable> runnable);

  virtual void start();

  virtual void join();

  virtual id_t getId();

  virtual ~ThriftThread() { }

 private:
  void RunRunnable(boost::shared_ptr<apache::thrift::concurrency::Runnable> runnable,
      Promise<apache::thrift::concurrency::Thread::id_t>* promise);

  boost::scoped_ptr<impala::Thread> impala_thread_;

  apache::thrift::concurrency::Thread::id_t tid_;

  std::string group_;

  std::string name_;
};

}

#endif