def setup(self, *args, **kwargs):
"""
Loads information about a deployed vcluster from a file.
The file to read from must be specified in environment variable
VCLUSTER_INFO.
:param args: Not used
:param kwargs: Not used
"""
conf_file = os.environ.get("VCLUSTER_INFO")
if not conf_file:
raise Exception("Environment variable VCLUSTER_INFO " +
"not set to vcluster output configuration file")
self.vcluster = vcluster.vcluster_from_conf(conf_file)
self.component_name_map.update({
components.MesosMaster().name:
"mesos-master",
components.MesosAgent().name:
"mesos-slave",
components.Zookeeper().name:
"zookeeper",
components.HostMgr().name:
"hostmgr",
components.JobMgr().name:
"jobmgr",
components.ResMgr().name:
"resmgr",
components.BatchPlacementEngine().name:
"placement",
components.StatelessPlacementEngine().name:
"placement_stateless",
})
def __init__(self):
self.component_name_map = {
components.MesosMaster().name: "peloton-mesos-master",
components.MesosAgent().name: "peloton-mesos-agent",
components.Zookeeper().name: "peloton-zk",
components.Cassandra().name: "peloton-cassandra",
components.HostMgr().name: "peloton-hostmgr",
components.JobMgr().name: "peloton-jobmgr",
components.ResMgr().name: "peloton-resmgr",
components.BatchPlacementEngine().name: "peloton-placement",
components.StatelessPlacementEngine().name: "peloton-placement",
}