in-process-servers.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_TESTUTIL_IN_PROCESS_SERVERS_H
#define IMPALA_TESTUTIL_IN_PROCESS_SERVERS_H

#include <boost/scoped_ptr.hpp>
#include <boost/shared_ptr.hpp>

#include "common/status.h"
#include "util/thread.h"
#include "runtime/exec-env.h"

namespace impala {

class ImpalaServer;
class ThriftServer;
class Webserver;
class MetricGroup;
class Statestore;

//
class InProcessImpalaServer {
 public:
  InProcessImpalaServer(const std::string& hostname, int backend_port,
                        int subscriber_port, int webserver_port,
                        const std::string& statestore_host, int statestore_port);

  Status StartWithClientServers(int beeswax_port, int hs2_port, bool use_statestore);

  Status StartAsBackendOnly(bool use_statestore);

  Status Join();

  ImpalaServer* impala_server() { return impala_server_; }

  MetricGroup* metrics() { return exec_env_->metrics(); }

  void SetCatalogInitialized();

 private:
  const std::string hostname_;

  const uint32_t backend_port_;

  ImpalaServer* impala_server_;

  boost::scoped_ptr<ExecEnv> exec_env_;

  boost::scoped_ptr<ThriftServer> be_server_;

  boost::scoped_ptr<ThriftServer> hs2_server_;

  boost::scoped_ptr<ThriftServer> beeswax_server_;

};

class InProcessStatestore {
 public:
  InProcessStatestore(int statestore_port, int webserver_port);

  Status Start();

  uint32_t port() { return statestore_port_; }

 private:
  boost::scoped_ptr<Webserver> webserver_;

  boost::scoped_ptr<MetricGroup> metrics_;

  uint32_t statestore_port_;

  boost::scoped_ptr<Statestore> statestore_;

  boost::scoped_ptr<ThriftServer> statestore_server_;

  boost::scoped_ptr<Thread> statestore_main_loop_;
};

}

#endif