fragment-exec-state.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_SERVICE_FRAGMENT_EXEC_STATE_H
#define IMPALA_SERVICE_FRAGMENT_EXEC_STATE_H

#include <boost/bind.hpp>
#include <boost/thread/mutex.hpp>

#include "common/status.h"
#include "runtime/client-cache.h"
#include "runtime/plan-fragment-executor.h"
#include "service/fragment-mgr.h"

namespace impala {

class FragmentMgr::FragmentExecState {
 public:
  FragmentExecState(const TPlanFragmentInstanceCtx& fragment_instance_ctx,
      ExecEnv* exec_env)
    : fragment_instance_ctx_(fragment_instance_ctx),
      executor_(exec_env, boost::bind<void>(
          boost::mem_fn(&FragmentMgr::FragmentExecState::ReportStatusCb),
              this, _1, _2, _3)),
      client_cache_(exec_env->impalad_client_cache()) {
  }

  ~FragmentExecState() { }

  Status Cancel();

  Status Prepare(const TExecPlanFragmentParams& exec_params);

  void Exec();

  const TUniqueId& query_id() const {
    return fragment_instance_ctx_.query_ctx.query_id;
  }

  const TUniqueId& fragment_instance_id() const {
    return fragment_instance_ctx_.fragment_instance_id;
  }

  const TNetworkAddress& coord_address() const {
    return fragment_instance_ctx_.query_ctx.coord_address;
  }

  void set_exec_thread(Thread* exec_thread) { exec_thread_.reset(exec_thread); }

 private:
  TPlanFragmentInstanceCtx fragment_instance_ctx_;
  PlanFragmentExecutor executor_;
  ImpalaInternalServiceClientCache* client_cache_;
  TExecPlanFragmentParams exec_params_;

  boost::scoped_ptr<Thread> exec_thread_;

  boost::mutex status_lock_;

  Status exec_status_;

  void ReportStatusCb(const Status& status, RuntimeProfile* profile, bool done);

  Status UpdateStatus(const Status& status);
};

}

#endif