cgroups-mgr.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_UTIL_CGROUPS_MGR_H
#define IMPALA_UTIL_CGROUPS_MGR_H

#include <string>
#include <boost/thread/mutex.hpp>
#include <boost/unordered_map.hpp>
#include "common/status.h"
#include "util/metrics.h"
#include "util/thread.h"

namespace impala {

//
//
//

//
class CgroupsMgr {
 public:
  CgroupsMgr(MetricGroup* metrics);

  Status Init(const std::string& cgroups_hierarchy_path,
      const std::string& staging_cgroup);

  std::string UniqueIdToCgroup(const std::string& unique_id) const;

  int32_t VirtualCoresToCpuShares(int16_t v_cpu_cores);

  Status RegisterFragment(const TUniqueId& fragment_instance_id,
      const std::string& cgroup, bool* is_first);

  Status UnregisterFragment(const TUniqueId& fragment_instance_id,
      const std::string& cgroup);

  Status CreateCgroup(const std::string& cgroup, bool if_not_exists) const;

  Status DropCgroup(const std::string& cgroup, bool if_exists) const;

  Status SetCpuShares(const std::string& cgroup, int32_t num_shares);

  Status AssignThreadToCgroup(const Thread& thread, const std::string& cgroup) const;

  Status RelocateThreads(const std::string& src_cgroup,
      const std::string& dst_cgroup) const;

 private:
  Status GetCgroupPaths(const std::string& cgroup,
      std::string* cgroup_path, std::string* tasks_path) const;

  IntCounter* active_cgroups_metric_;

  std::string cgroups_hierarchy_path_;

  std::string staging_cgroup_;

  boost::mutex active_cgroups_lock_;

  boost::unordered_map<std::string, int32_t> active_cgroups_;
};

}

#endif