class SchedulerTask(object):
def __init__(self, slave_id, slave_ip):
self.task = mesos_pb2.TaskInfo()
self.slave_id = slave_id
def add_resources_configurations(self, id):
self.task = Task(id, "dscheduler", self.slave_id)
self.task.configure_command_protobuf("dscheduler")
self.task.configure_memory_resources()
self.task.configure_container_protobuf(expose_ports=True)
self.task.configure_port_resources()
self.task.configure_cpu_resources()
class WorkerTask(object):
def __init__(self, slave_id, slave_ip):
self.task = mesos_pb2.TaskInfo()
self.slave_id = slave_id
self.slave_ip = slave_ip
def add_resources_configurations(self, id):
self.task = Task(id, "dwoker", self.slave_id)
# TODO: add config for test environment vs production evs for worker uri
self.task.configure_command_protobuf("dworker 192.168.99.100:33001".format(self.slave_ip))
self.task.configure_container_protobuf()
self.task.configure_cpu_resources()
self.task.configure_memory_resources()