blob: 338a7f77edbf039e53c1abf44a7cd4f9bd7ccb2b [file] [log] [blame]
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you 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
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.
#ifndef __HOOK_MANAGER_HPP__
#define __HOOK_MANAGER_HPP__
#include <string>
#include <mesos/mesos.hpp>
#include <mesos/hook.hpp>
#include <process/future.hpp>
#include <stout/try.hpp>
namespace mesos {
namespace internal {
class HookManager
static Try<Nothing> initialize(const std::string& hookList);
// Exposed just for testing so that we can unload a given
// hook and remove it from the list of available hooks.
static Try<Nothing> unload(const std::string& hookName);
static bool hooksAvailable();
static Labels masterLaunchTaskLabelDecorator(
const TaskInfo& taskInfo,
const FrameworkInfo& frameworkInfo,
const SlaveInfo& slaveInfo);
static Resources masterLaunchTaskResourceDecorator(
const TaskInfo& taskInfo,
const Resources& slaveResources);
static void masterSlaveLostHook(const SlaveInfo& slaveInfo);
static Labels slaveRunTaskLabelDecorator(
const TaskInfo& taskInfo,
const ExecutorInfo& executorInfo,
const FrameworkInfo& frameworkInfo,
const SlaveInfo& slaveInfo);
static Environment slaveExecutorEnvironmentDecorator(
ExecutorInfo executorInfo);
static process::Future<DockerTaskExecutorPrepareInfo>
const Option<TaskInfo>& taskInfo,
const ExecutorInfo& executorInfo,
const std::string& containerName,
const std::string& containerWorkDirectory,
const std::string& mappedSandboxDirectory,
const Option<std::map<std::string, std::string>>& env);
static void slavePostFetchHook(
const ContainerID& containerId,
const std::string& directory);
static void slaveRemoveExecutorHook(
const FrameworkInfo& frameworkInfo,
const ExecutorInfo& executorInfo);
static TaskStatus slaveTaskStatusDecorator(
const FrameworkID& frameworkId,
TaskStatus status);
static Resources slaveResourcesDecorator(
const SlaveInfo& slaveInfo);
static Attributes slaveAttributesDecorator(
const SlaveInfo& slaveInfo);
} // namespace internal {
} // namespace mesos {
#endif // __HOOK_MANAGER_HPP__